首页 > 代码库 > Flume-NG源码阅读之SinkGroups和SinkRunner

Flume-NG源码阅读之SinkGroups和SinkRunner

  在AbstractConfigurationProvider类中loadSinks方法会调用loadSinkGroups方法将所有的sink和sinkgroup放到了Map<String, SinkRunner> sinkRunnerMap之中。

  SinkRunner可能对应一个sink也可能对应一个sinkgroup。因为如果配置文件中有sinkgroup则这个sinkgroup对应的sink会组成一个group然后封装为一个sinkRunner,然后不在sinkgroup中的sink会自己成为一个sinkRunner。每个SinkRunner的构造方法的参数是一个SinkProcessor是用来处理多个sink的。

  一、如果一个SinkRunner对应一个sink。SinkProcessor pr = new DefaultSinkProcessor()是默认的SinkProcessor。loadSinkGroups方法中的相关代码如下:

1           SinkProcessor pr = new DefaultSinkProcessor();
2           List<Sink> sinkMap = new ArrayList<Sink>();
3           sinkMap.add(entry.getValue());
4           pr.setSinks(sinkMap);
5           Configurables.configure(pr, new Context());
6           sinkRunnerMap.put(entry.getKey(),new SinkRunner(pr));    

  DefaultSinkProcessor.configure(Context context)是空方法,所有这句Configurables.configure(pr, new Context())没啥作用,重要的是DefaultSinkProcessor.process()方法,就一句代码就是return sink.process()直接调用sink的process。setSinks方法只会设置一个sink。DefaultSinkProcessor的start()方法也会直接调用sink.start()来启动sink。

  二、如果一个SinkRunner对应多个sink。则会构造一个SinkGroup group = new SinkGroup(groupSinks)然后获取SinkProcessor:group.getProcessor()。loadSinkGroups方法中的相关代码如下:

1           SinkGroup group = new SinkGroup(groupSinks);
2           Configurables.configure(group, groupConf);
3           sinkRunnerMap.put(comp.getComponentName(), new SinkRunner(group.getProcessor()));   

  Configurables.configure(group, groupConf)会调用SinkGroupConfiguration.configure(Context context),该方法会获取配置文件中关于"processor."的属性通过getKnownSinkProcessor方法获取SinkProcessorType(是FailoverSinkProcessor或者是LoadBalancingSinkProcessor),并执行该SinkProcessor.configure(processorContext)进行实例化和配置。

  1、如果SinkProcessor是LoadBalancingSinkProcessor,这是负载均衡的processor,会将channel中的发送到指定的所有sink。通过配置选择器selector来选择何种方式的负载均衡,1.3有两种:ROUND_ROBIN,轮询,就是轮流向channel发送数据;RANDOM,随机选择channel发送数据。只有这个SinkProcessor有选择器。

  (1)configure(Context context)方法。先获取选择器selector的类型,默认是ROUND_ROBIN,轮询;获取backoff(是否使用推迟算法,就是sink.process出问题后对这个sink设置惩罚时间,在此期间不再认为其可活动)的boolean值(默认false就是不启用);根据类型构造相应的选择器对象RoundRobinSinkSelector(实际上会构造一个RoundRobinOrderSelector)或者RandomOrderSinkSelector(实际上会构造一个RandomOrderSelector);然后实例化并设置sinks;最后对selector执行其configure(context)方法进行初始化。

  A、RoundRobinSinkSelector的实际操作者是RoundRobinOrderSelector extends OrderSelector,它实现了createIterator()方法,该方法用来选出所有的sink及其可活动sinkl的索引封装成一个SpecificOrderIterator<T>(indexOrder, getObjects())并返回,可以通过SpecificOrderIterator.hasNext()方法判断是否还有sink,用next()方法获取下一个sink。这样可以按照索引递增的顺序依次获取sink进行操作。SpecificOrderIterator主要是将两个:一个是索引数组,一个是sink列表。createIterator()方法代码如下:

 1 @Override
 2   public Iterator<T> createIterator() {
 3     List<Integer> activeIndices = getIndexList();  //会获取最新的活动的sink的索引列表
 4     int size = activeIndices.size();
 5     // possible that the size has shrunk so gotta adjust nextHead for that
 6     if (nextHead >= size) {    //可能会出现sink的总数调整,所以总得getIndexList()并调整nextHead
 7       nextHead = 0;
 8     }
 9     int begin = nextHead++;    //注意++在后面说明是先赋值,在自加
10     if (nextHead == activeIndices.size()) {
11       nextHead = 0;
12     }
13 
14     int[] indexOrder = new int[size];
15 
16     for (int i = 0; i < size; i++) {
17       indexOrder[i] = activeIndices.get((begin + i) % size);
18     }
19 
20     return new SpecificOrderIterator<T>(indexOrder, getObjects());    //组成两个数组,大小都一样
21   }

  createIterator()方法中总会调用getIndexList()方法,因为可能有sink中断,或者sinkgroup再调整等情况,使得sinkgroup中实际活动的sink数产生变化。nextHead始终指向下一个可活动的sink索引。indexOrder是新的活动sink的索引数组;getObjects()则返回所有sink的List,通过索引即可即可获取此List中对应的sink。

  B、RandomOrderSinkSelector的实际操作者是RandomOrderSelector extends OrderSelector,它实现了createIterator()方法:

 1 public synchronized Iterator<T> createIterator() {
 2     List<Integer> indexList = getIndexList();
 3 
 4     int size = indexList.size();
 5     int[] indexOrder = new int[size];
 6     //indexList由于remove操作会动态变化,所以一直使用indexList.size()会获得实际大小
 7     while (indexList.size() != 1) {
 8       int pick = random.nextInt(indexList.size());
 9       indexOrder[indexList.size() - 1] = indexList.remove(pick);    //取出pick位置的索引,这句总是使得indexOrder从后向前插入数据
10     }
11 
12     indexOrder[0] = indexList.get(0);    //将最后一个索引放入indexOrder
13 
14     return new SpecificOrderIterator<T>(indexOrder, getObjects());
15   }

  这个方法最终只是将"可活动"的sink的顺序按随机的方式打乱了而已。注意的一个是总是调用indexList.size()动态获取最新的大小;一个是indexOrder[indexList.size() - 1]始终是从后向前插入数据;一个是indexOrder[0] = indexList.get(0)将最后一个插入保证完整性。

  A和B都是OrderSelector抽象类的子类,都只实现了createIterator()方法,对于getIndexList()和sink.process()方法出现错误的时的selector.informSinkFailed(sink)都是一样的这两个方法决定了出现问题的sink的推迟时间,如果要修改推迟时间可以重写这两个方法。当sink.process运行出问题时informSinkFailed会更新对应sink的FailureState(就三个数,sequentialFails记录出错次数、restoreTime记录出错后惩罚恢复时间(在此期间不再认为是可活动的sink,通过getIndexList()来过滤)、lastFail记录上一次出错时间,三个初始化都是0),maxTimeout默认是3000。看informFailure代码:

public void informFailure(T failedObject) {
    if (!shouldBackOff) {        //不允许推迟
      return;
    }
    FailureState state = stateMap.get(failedObject);
    long now = System.currentTimeMillis();    //获取现在系统时间
    long delta = now - state.lastFail;        //获取和上次失败时间之间的时间间隔
    long lastBackoffLength = Math.min(maxTimeout, 1000 * (1 << state.sequentialFails));    //获取上一次要推迟的时间增量
    long allowableDiff = lastBackoffLength + CONSIDER_SEQUENTIAL_RANGE;        //CONSIDER_SEQUENTIAL_RANGE=2000
    if (allowableDiff > delta) {
      if (state.sequentialFails < EXP_BACKOFF_COUNTER_LIMIT) {        //说明是连续失败
        state.sequentialFails++;
      }
    } else {            //说明期间曾重新正确process,重新计数
      state.sequentialFails = 1;
    }
    state.lastFail = now;
    state.restoreTime = now + Math.min(maxTimeout, 1000 * (1 << state.sequentialFails));
  }

  这个informFailure方法有需要说明的地方:1、如何判断是连续失败?关键在于CONSIDER_SEQUENTIAL_RANGE这个变量,等于2000,因为首先在推迟时间增量是1000的倍数,而且在推迟时间内是“不被认可”的,是不被认为是可活动的,所以超过推迟时间后自然会被重新认为是活动的,if (allowableDiff > delta)这句代码写的不够友好,比较费解,它等价于if ((now-state.restoreTime) < CONSIDER_SEQUENTIAL_RANGE),如果再次失败且比推迟时间等于1000说明是当重新认为是活动的第一次执行就失败,明显是连续失败,所以在失败次数不超过EXP_BACKOFF_COUNTER_LIMIT(等于16)时就增加state.sequentialFails,一旦超过16就不再增加就是16;当((now-state.restoreTime) >= CONSIDER_SEQUENTIAL_RANGE)成立时说明成功process至少一次但这次失败,需要重置state.sequentialFails为1。

  getIndexList()方法用来过滤“不被认可”的sink的索引。代码如下:

 1 protected List<Integer> getIndexList() {
 2     long now = System.currentTimeMillis();
 3 
 4     List<Integer> indexList = new ArrayList<Integer>();
 5 
 6     int i = 0;
 7     for (T obj : stateMap.keySet()) {        //是sink的集合
 8       if (!isShouldBackOff() || stateMap.get(obj).restoreTime < now) {    //
 9         indexList.add(i);        //将索引存储
10       }
11       i++;
12     }
13     return indexList;
14   }

  如果shouldBackOff=true则会返回的列表将是所有的sink的索引。stateMap.get(obj).restoreTime < now这句会过滤掉当前还处在惩罚时间内不被认可的sink的索引。

  (2)start()方法会先启动AbstractSinkProcessor.start()方法将所有的sink启动(start()),然后启动选择器 selector.start()。

  (3)process()方法遍历sinkIterator一次获取可活动的sink,执行sink.process()方法,如果有异常就跳出循环并执行失败处理informSinkFailed。

  2、如果SinkProcessor是FailoverSinkProcessor,这是容错的processor,一旦有一个sink中断可以使用其他的代替。

  (1)setSinks(List<Sink> sinks)方法会将sinks列表中的所有sink,先调用父类的setSinks方法为的是可以执行父类的start和stop方法(子类中没有实现这俩方法),然后放入Map<String, Sink> sinks中。

  (2)configure(Context context)方法,会先获取中断时间的上限maxPenalty,然后将所有的sink及其对应的优先级放入liveSinks(这是一个TreeMap,默认根据键值的自然顺序排序存储),最后activeSink = liveSinks.get(liveSinks.lastKey())获取优先级最高的sink作为活动sink。failedSinks = new PriorityQueue<FailedSink>()是一个保存中断sink的一个优先级队列。

  (3)process()方法。循环执行如果failedSinks不为空并且记录惩罚时间小于当前系统时间,则取出failedSinks的head然后尝试执行getSink().process()如果能获取到Rady状态说明这个节点又重新建立了链接,则将其加入liveSinks,并重新获取优先级最高的sink作为activeSink,如果获取的是backOff状态则重新将其加入failedSinks,返回状态,如果出现异常则cur.incFails()重新记录惩罚时间并加入failedSinks,惩罚时间会动态变化,会根据失败的次数增加(会和设置的比较取较大者)。如果failedSinks为空或者当前系统小于惩罚时间则使用当前活动的sink:activeSink.process()。

  注:惩罚时间是动态变化的,会随着链接失败的次数而变化,失败次数越多到下次使用它的间隔越长。

  返回顶端sinkRunnerMap会在Application.startAllComponents方法中调用,放放到LifecycleSupervisor.supervise方法中去执行,最终会执行SinkRunner.start()方法来启动组件。

 1 public void start() {
 2     SinkProcessor policy = getPolicy();
 3 
 4     policy.start();
 5 
 6     runner = new PollingRunner();
 7 
 8     runner.policy = policy;
 9     runner.counterGroup = counterGroup;
10     runner.shouldStop = new AtomicBoolean();
11 
12     runnerThread = new Thread(runner);
13     runnerThread.setName("SinkRunner-PollingRunner-" +
14         policy.getClass().getSimpleName());
15     runnerThread.start();
16 
17     lifecycleState = LifecycleState.START;
18   }

  上述代码中的policy其实就是SinkProcessor,可能是LoadBalancingSinkProcessor、FailoverSinkProcessor、DefaultSinkProcessor三者中其中之一。policy.start()会启动SinkProcessor。然后会启动一个线程PollingRunner,该线程会始终执行policy.process()方法根据返回的状态做一些统计。这就是我们自定义也要实现process方法的所在,及其需要返回Status的原因。

  至此SinkGroup的介绍完结。