Sentinel源码解析之StatisticSlot

淩亂°似流年 2022-10-16 12:28 212阅读 0赞

> 原创不易，转载请注明出处

### 文章目录 ###

*   *   *  前言
         *  1.entry方法源码解析
         *  2.exit 方法
         *  总结

--------------------

### 前言 ###

我们之前介绍完了slot里面的NodeSelectorSlot与ClusterBuilderSlot，按理说应该按照执行顺序往下介绍，但是下面这个LogSlot我们这里先不介绍，介绍一下它后面的StatisticSlot，这个slot从名字上也能看出来，就是统计slot，统计一些信息的，这个比较简单。

### 1.entry方法源码解析 ###

先来看下没有异常的代码

// Do some checking.
    // 先进入下面的slot里面进行流控/降级的判断
    fireEntry(context, resourceWrapper, node, count, prioritized, args);
    // 能够走到这里，说明通过了流控/降级等处理，然后下面就进行一些统计信息的更新
    // Request passed, add thread count and pass count.
    // 当前node的并发线程数+1
    node.increaseThreadNum();
    // 通过请求数+n（一般这里就是1）
    node.addPassRequest(count);
    // 如果有上游服务统计资源的话，相应的增加上游资源的统计
    if (context.getCurEntry().getOriginNode() != null) { 
        // Add count for origin node.
        context.getCurEntry().getOriginNode().increaseThreadNum();
        context.getCurEntry().getOriginNode().addPassRequest(count);
    }
    // 如果这个资源类型是in ，增加整个集群node的统计信息
    if (resourceWrapper.getType() == EntryType.IN) { 
        // Add count for global inbound entry node for global statistics.
        Constants.ENTRY_NODE.increaseThreadNum();
        Constants.ENTRY_NODE.addPassRequest(count);
    }
    // 如果这里注册了监听者的话，进行回调
    // Handle pass event with registered entry callback handlers.
    for (ProcessorSlotEntryCallback<DefaultNode> handler : StatisticSlotCallbackRegistry.getEntryCallbacks()) { 
        handler.onPass(context, resourceWrapper, node, count, args);
    }

你会发现，它与平常的slot有点不一样，别的slot都是执行一些业务逻辑，最后再调用fireEntry方法进入下一个slot进行处理，它是上来就执行了fireEntry方法让下面的slot进行处理。这样做的原因主要是由它功能决定的，它主要就是干统计的活，必须得后面得限流/降级slot执行完，才能够拿到能不能通过的结果，可以代码里面看到执行完fireEntry就说明通过了，这个时候就对一些node，进行某些指标的增加，increaseThreadNum 主要是这个资源并发线程数+1操作。addPassRequest就是通过请求+n操作。  
如果限流/降级这些slot没有通过的话，就会抛出异常，这个时候就不会再走fireEntry 下面一堆代码，而是走catch里面的

catch (PriorityWaitException ex) { 
    
                // 线程数量
                node.increaseThreadNum();
                if (context.getCurEntry().getOriginNode() != null) { 
                    // Add count for origin node.
                    context.getCurEntry().getOriginNode().increaseThreadNum();
                }
    
                if (resourceWrapper.getType() == EntryType.IN) { 
                    // Add count for global inbound entry node for global statistics.
                    Constants.ENTRY_NODE.increaseThreadNum();
                }
                // 回调
                // Handle pass event with registered entry callback handlers.
                for (ProcessorSlotEntryCallback<DefaultNode> handler : StatisticSlotCallbackRegistry.getEntryCallbacks()) { 
                    handler.onPass(context, resourceWrapper, node, count, args);
                }
            } catch (BlockException e) { 
                // Blocked, set block exception to current entry.
                //没有通过。将异常信息保存到当前的entry里面
                context.getCurEntry().setError(e);
                // 增加拒绝数量
                // Add block count.
                node.increaseBlockQps(count);
                if (context.getCurEntry().getOriginNode() != null) { 
                    context.getCurEntry().getOriginNode().increaseBlockQps(count);
                }
                // 增加集群node的拒绝数量
                if (resourceWrapper.getType() == EntryType.IN) { 
                    // Add count for global inbound entry node for global statistics.
                    Constants.ENTRY_NODE.increaseBlockQps(count);
                }
                // 处理回调
                // Handle block event with registered entry callback handlers.
                for (ProcessorSlotEntryCallback<DefaultNode> handler : StatisticSlotCallbackRegistry.getEntryCallbacks()) { 
                    handler.onBlocked(e, context, resourceWrapper, node, count, args);
                }
    
                throw e;
            } catch (Throwable e) { 
                // Unexpected error, set error to current entry.
                context.getCurEntry().setError(e);
                // 异常数量
                // This should not happen.
                node.increaseExceptionQps(count);
                if (context.getCurEntry().getOriginNode() != null) { 
                    context.getCurEntry().getOriginNode().increaseExceptionQps(count);
                }
                // 异常数量
                if (resourceWrapper.getType() == EntryType.IN) { 
                    Constants.ENTRY_NODE.increaseExceptionQps(count);
                }
                throw e;
            }

可以看到不同的异常有不同的统计方式PriorityWaitException 这个异常就是优先级等待异常，但是它并没有增加失败/拒绝通过的请求数，只是增加了并发线程数，这个其实意思就是让他等待，但是还是占用线程的。  
BlockException 这个就是拒绝异常，一般就是限流这块抛出来的  
Throwable 其他异常，这块统计异常请求数量。

### 2.exit 方法 ###

这个exit方法的调用时机其实是 entry执行完，执行你业务代码，不管你业务代码成功与否，都会调用exit方法。  
我们看下exit会进行哪些统计

if (context.getCurEntry().getError() == null) { 
          // Calculate response time (max RT is TIME_DROP_VALVE).
          long rt = TimeUtil.currentTimeMillis() - context.getCurEntry().getCreateTime();
          if (rt > Constants.TIME_DROP_VALVE) { 
              rt = Constants.TIME_DROP_VALVE;
          }
          // 记录rt（响应时间）
          // Record response time and success count.
          node.addRtAndSuccess(rt, count);
          if (context.getCurEntry().getOriginNode() != null) { 
              context.getCurEntry().getOriginNode().addRtAndSuccess(rt, count);
          }
          // 减少并发线程数
          node.decreaseThreadNum();
          // 减少并发线程数
          if (context.getCurEntry().getOriginNode() != null) { 
              context.getCurEntry().getOriginNode().decreaseThreadNum();
          }
          // 集群node添加rt与减少并发线程数
          if (resourceWrapper.getType() == EntryType.IN) { 
              Constants.ENTRY_NODE.addRtAndSuccess(rt, count);
              Constants.ENTRY_NODE.decreaseThreadNum();
          }
      } else { 
          // Error may happen.
      }
      // 回调
      // Handle exit event with registered exit callback handlers.
      Collection<ProcessorSlotExitCallback> exitCallbacks = StatisticSlotCallbackRegistry.getExitCallbacks();
      for (ProcessorSlotExitCallback handler : exitCallbacks) { 
          handler.onExit(context, resourceWrapper, count, args);
      }
    
      fireExit(context, resourceWrapper, count);

如果entry没抛出异常，会记录下rt，也就是业务代码执行的响应时间，然后将资源对应的并发线程数-1操作。最后调用fireExit 方法进入到下一个slot里面执行exit方法。

### 总结 ###

我们从源码中可以了解到StatisticSlot 其实就是个指标统计的slot，但是node背后一些指标的加减是个很有意思的过程，背后就是滑动窗口在支撑着，可以看下《[Sentinel源码解析之滑动窗口][Sentinel]》一文

[Sentinel]: https://inetyoung.blog.csdn.net/article/details/111828235