thrift中实现的robin

最新推荐文章于 2024-07-12 09:42:16 发布

_狼_

最新推荐文章于 2024-07-12 09:42:16 发布

阅读量117

点赞数

本文链接：https://blog.csdn.net/nrs12345/article/details/86501434

版权

最近重温thrift源码时，发现很优雅的实现了lb, 用到了Iterator，在此记录下。

protected static class SelectorThreadLoadBalancer {
    private final Collection<? extends SelectorThread> threads;
    private Iterator<? extends SelectorThread> nextThreadIterator;

    public <T extends SelectorThread> SelectorThreadLoadBalancer(Collection<T> threads) {
      if (threads.isEmpty()) {
        throw new IllegalArgumentException("At least one selector thread is required");
      }
      this.threads = Collections.unmodifiableList(new ArrayList<T>(threads));
      nextThreadIterator = this.threads.iterator();
    }

    public SelectorThread nextThread() {
      // Choose a selector thread (round robin)
      if (!nextThreadIterator.hasNext()) {
        nextThreadIterator = threads.iterator();
      }
      return nextThreadIterator.next();
    }
  }