netty为啥可靠

NIO闲聊

自从JAVA1.4推出NIO起,JAVA网络编程进入了一个全新的时代,传统网络IO(OIO)是傻等式的,一旦IO操作发起,那么用户线程就陷入很傻很天真的等待中,直到IO操作结束或者发生了断连,而NIO则要聪明许多是事件触发式的,只有当前有IO事件发生了,才会通知用户线程执行IO操作,当前操作结束之后不会阻塞等待可以执行其他的业务操作等待下一次事件,就好比上银行取钱,一种方式排队傻等直到排到你,一种是登记排号,登记完之后该干嘛干嘛去,等轮到你的时候业务员电话通知你去办理,脑子正常的都会喜欢后面那种方式。相比OIO的线程和连接的N对N,NIO只需少数几个线程处理N个连接,这就是著名的多路复用模型,对于Netty来讲,它并不是标准的多路复用,Netty的IO事件和非IO事件由同一个线程池进行调度,通过一个参数来控制IO事件和非IO事件的执行时间,这种方式下控制复杂一些,但是CPU使用率更高,因为会有很少的上下文切换(Context Switch)。

但是天下没有免费的午餐,NIO带来了性能方面的优势,但是相比OIO,在使用上复杂度飙升。比如我想执行一次写操作,OIO几行代码就搞定了:

[java]  view plain  copy
  1. OutputStream os = s.getOutputStream();  
  2. String str = "test";  
  3. os.write(str.getBytes());  
而只用NIO则需要一坨代码:

[java]  view plain  copy
  1. while (running) {  
  2.     SocketChannel socketChanel = serverSocketChanel.accept();  
  3.     socketChanel.configureBlocking(false);  
  4.     socketChanel.register(selector, SelectionKey.OP_READ  
  5.             | SelectionKey.OP_WRITE);  
  6.     int count = selector.select();  
  7.     if (count > 0) {  
  8.         Set<SelectionKey> keys = selector.selectedKeys();  
  9.         Iterator<SelectionKey> iter = keys.iterator();  
  10.         while (iter.hasNext()) {  
  11.             SelectionKey key = iter.next();  
  12.             if (key.isWritable()) {  
  13.                 ByteBuffer buff = ByteBuffer.allocate(1024);  
  14.                 buff.put("<html>test</html>".getBytes());  
  15.                 System.out.println("writable");  
  16.                 buff.flip();  
  17.                 socketChanel.write(buff);  
  18.                 buff.clear();  
  19.             }  
  20.             iter.remove();  
  21.         }  
  22.     }  
  23. }  

这也可以理解,还是拿前面取钱例子,第二种方式下,用户是爽了,但是银行需要搞一套排号 的系统,还要比较负责有耐心的业务员,总之银行很麻烦。除此之外NIO还带来了更多可靠性问题。Netty牛的地方就是屏蔽NIO编程的复杂性,简化编程模型,让开发者聚焦业务逻辑,而且针对NIO中的一些可靠性问题就行了处理。下面对Netty对几个可靠性问题处理进行学习。

连接超时处理

在OIO中,连接超时处理非常简单,只需调用一个setConnectTimeout方法设置连接超时时间即可,但是JDK的NIO API中并没有提供设置超时时间的方法,显然无论从服务器资源还是用户体验的角度,连接必须要用超时时间,一方面服务器的句柄资源是有限的,既然我不能为你服务那么请你尽早放手吧,长时间无法连接上服务器时非常有必要释放句柄资源,另一方面在用户发起操作时如果系统长时间不给响应显然不是一种好体验,这时候也非常有必要在超时之后提示用户当前无法连接上服务器。Netty对外提供了设置连接超时时间的API,通过ChannleOption.CONNECT_TIMEOUT_MILLIS设置超时时间timetou,Netty的处理是在调用建连之后马上启动一个延时任务,该任务在timeout时间过后执行,任务中执行关闭操作,连接建立成功之后取消这个任务,处理代码在AbstractNioUnsafe类的connect方法中:

[java]  view plain  copy
  1. @Override  
  2. public void connect(  
  3.         final SocketAddress remoteAddress, final SocketAddress localAddress, final ChannelPromise promise) {  
  4.     if (!promise.setUncancellable() || !ensureOpen(promise)) {  
  5.         return;  
  6.     }  
  7.   
  8.     try {  
  9.         if (connectPromise != null) {  
  10.             throw new IllegalStateException("connection attempt already made");  
  11.         }  
  12.   
  13.         boolean wasActive = isActive();  
  14.         if (doConnect(remoteAddress, localAddress)) {  
  15.             fulfillConnectPromise(promise, wasActive);  
  16.         } else {  
  17.             connectPromise = promise;  
  18.             requestedRemoteAddress = remoteAddress;  
  19.   
  20.             // Schedule connect timeout.  
  21.             int connectTimeoutMillis = config().getConnectTimeoutMillis();  
  22.             if (connectTimeoutMillis > 0) {  
  23.                 connectTimeoutFuture = eventLoop().schedule(new OneTimeTask() {  
  24.                     @Override  
  25.                     public void run() {  
  26.                         ChannelPromise connectPromise = AbstractNioChannel.this.connectPromise;  
  27.                         ConnectTimeoutException cause =  
  28.                                 new ConnectTimeoutException("connection timed out: " + remoteAddress);  
  29.                         if (connectPromise != null && connectPromise.tryFailure(cause)) {  
  30.                             close(voidPromise());  
  31.                         }  
  32.                     }  
  33.                 }, connectTimeoutMillis, TimeUnit.MILLISECONDS);  
  34.             }  
  35.   
  36.             promise.addListener(new ChannelFutureListener() {  
  37.                 @Override  
  38.                 public void operationComplete(ChannelFuture future) throws Exception {  
  39.                     if (future.isCancelled()) {  
  40.                         if (connectTimeoutFuture != null) {  
  41.                             connectTimeoutFuture.cancel(false);  
  42.                         }  
  43.                         connectPromise = null;  
  44.                         close(voidPromise());  
  45.                     }  
  46.                 }  
  47.             });  
  48.         }  
  49.     } catch (Throwable t) {  
  50.         if (t instanceof ConnectException) {  
  51.             Throwable newT = new ConnectException(t.getMessage() + ": " + remoteAddress);  
  52.             newT.setStackTrace(t.getStackTrace());  
  53.             t = newT;  
  54.         }  
  55.         promise.tryFailure(t);  
  56.         closeIfClosed();  
  57.     }  
  58. }  

分包传输

在TCP协议中,分包传输是非常,一份信息可能分几次达到目的地,在OIO中这是没有问题的,因为OIO是傻等式的,不读到完整的信息它是不会罢手的,但是NIO就不同了,它是基于事件的,只有有数据来了它才会去读取,那么问题来了,在读取到数据之后对数据进行业务解析时该如何处理?比如说我想解析一个整形数,但是当前只读取到了两个字节的数据,还有两个字节的数据在后面的传输包中,由于NIO的非阻塞性,业务数据的解析时机成了一个大问题,因为可能无法一次取到完整的数据。

基于上面这个问题,Netty框架设计了一个ReplayingDecoder来解决这种场景中的问题,ReplayingDecoder的核心原理是,当ReplayingDecoder在进行数据解析时,如果发现当前ByteBuf中所有可读数据并不完整,比如我想解析出一个整型数,但是ByteBuf中数据小于4个字节,那么此时会抛出一个Signal类型的Error,抛Error的操作在ReplayingDecoderBuffer中进行,一个ByteBuf的装饰器。在ReplayingDecoder会捕捉一个Error,捕捉到Signal之后会把ByteBuf中的读指针还原到之前的断点处(checkpoint,默认是ByteBuf的其实读位置),然后结束这次解析操作,等待下一次IO读事件。如果只是简单的整形数解析问题不大,但是如果数据解析逻辑复杂是,这种处理方式存在一个问题,在网络条件比较糟糕时,解析逻辑会反复执行多次,如果解析过程是一个耗CPU的操作,那么这对CPU是个大负担。可以通过ReplayingDecoder中的断点和状态机来解决这个问题,使用者可以在ReplayingDecoder中保存之前的解析结果、状态和读指针断点,举个例子,我要解析8个字节的数据,把前后四个字节都解析成整形数,并且把这两个数据相加当做解析结果,代码如下:

[java]  view plain  copy
  1. public class TowIntegerReplayingDecoder extends ReplayingDecoder<Integer> {  
  2.   
  3.     private static final int PARSE_1 = 1;  
  4.     private static final int PARSE_2 = 2;  
  5.     private int number1;  
  6.     private int number2;  
  7.   
  8.     @Override  
  9.     protected void decode(ChannelHandlerContext ctx, ByteBuf in,  
  10.             List<Object> out) throws Exception {  
  11.         switch (state()) {  
  12.         case PARSE_1:  
  13.             number1 = in.readInt();  
  14.             checkpoint(PARSE_2);  
  15.             break;  
  16.         case PARSE_2:  
  17.             number2 = in.readInt();  
  18.             checkpoint(PARSE_1);  
  19.             out.add(number1 + number2);  
  20.             break;  
  21.         default:  
  22.             break;  
  23.         }  
  24.   
  25.     }  
  26.   
  27. }  

在代码中,把解析分成两个阶段,当一个阶段解析完成之后,记录第一个阶段的解析结果,并且更新解析状态和读指针,这样如果由于数据不完整导致第二阶段的解析无法完成,下次IO事件触发时,该解析器会直接进入第二阶段的解析,而不会重复第一阶段的解析,这样会减少重复解析大概率。基于这种设计,ReplayingDecoder必须是Channel独有的,它的实例不能被共享,没有Channel实例必须有个单独的ReplayingDecoder解析器实例,而且不能添加Sharable注解,因为它是用状态了,如果在多个Channel中共享了,那么状态就乱套了。

Selector空轮询处理

在NIO中通过Selector的轮询当前是否有IO事件,根据JDK NIO api描述,Selector的select方法会一直阻塞,直到IO事件达到或超时,但是在Linux平台上这里有时会出现问题,在某些场景下select方法会直接返回,即使没有超时并且也没有IO事件到达,这就是著名的epoll bug,这是一个比较严重的bug,它会导致线程陷入死循环,会让CPU飙到100%,极大地影响系统的可靠性,到目前为止,JDK都没有完全解决这个问题。

但是Netty有效的规避了这个问题,经过实践证明,epoll bug已Netty框架解决,Netty的处理方式是这样的:

记录select空转的次数,定义一个阀值,这个阀值默认是512,可以在应用层通过设置系统属性io.netty.selectorAutoRebuildThreshold传入,当空转的次数超过了这个阀值,重新构建新Selector,将老Selector上注册的Channel转移到新建的Selector上,关闭老Selector,用新的Selector代替老Selector,详细实现可以查看NioEventLoop中的selector和rebuildSelector方法:

[java]  view plain  copy
  1. for (;;) {  
  2.     long timeoutMillis = (selectDeadLineNanos - currentTimeNanos + 500000L) / 1000000L;  
  3.     if (timeoutMillis <= 0) {  
  4.         if (selectCnt == 0) {  
  5.             selector.selectNow();  
  6.             selectCnt = 1;  
  7.         }  
  8.         break;  
  9.     }  
  10.   
  11.     int selectedKeys = selector.select(timeoutMillis);  
  12.     selectCnt ++;  
  13.   
  14.     if (selectedKeys != 0 || oldWakenUp || wakenUp.get() || hasTasks()) {  
  15.         // Selected something,  
  16.         // waken up by user, or  
  17.         // the task queue has a pending task.  
  18.         break;  
  19.     }  
  20.     if (selectedKeys == 0 && Thread.interrupted()) {  
  21.         // Thread was interrupted so reset selected keys and break so we not run into a busy loop.  
  22.         // As this is most likely a bug in the handler of the user or it's client library we will  
  23.         // also log it.  
  24.         //  
  25.         // See https://github.com/netty/netty/issues/2426  
  26.         if (logger.isDebugEnabled()) {  
  27.             logger.debug("Selector.select() returned prematurely because " +  
  28.                     "Thread.currentThread().interrupt() was called. Use " +  
  29.                     "NioEventLoop.shutdownGracefully() to shutdown the NioEventLoop.");  
  30.         }  
  31.         selectCnt = 1;  
  32.         break;  
  33.     }  
  34.     if (SELECTOR_AUTO_REBUILD_THRESHOLD > 0 &&  
  35.             selectCnt >= SELECTOR_AUTO_REBUILD_THRESHOLD) {  
  36.         // The selector returned prematurely many times in a row.  
  37.         // Rebuild the selector to work around the problem.  
  38.         logger.warn(  
  39.                 "Selector.select() returned prematurely {} times in a row; rebuilding selector.",  
  40.                 selectCnt);  
  41.   
  42.         rebuildSelector();  
  43.         selector = this.selector;  
  44.   
  45.         // Select again to populate selectedKeys.  
  46.         selector.selectNow();  
  47.         selectCnt = 1;  
  48.         break;  
  49.     }  
  50.   
  51.     currentTimeNanos = System.nanoTime();  
  52. }  
[java]  view plain  copy
  1. public void rebuildSelector() {  
  2.     if (!inEventLoop()) {  
  3.         execute(new Runnable() {  
  4.             @Override  
  5.             public void run() {  
  6.                 rebuildSelector();  
  7.             }  
  8.         });  
  9.         return;  
  10.     }  
  11.   
  12.     final Selector oldSelector = selector;  
  13.     final Selector newSelector;  
  14.   
  15.     if (oldSelector == null) {  
  16.         return;  
  17.     }  
  18.   
  19.     try {  
  20.         newSelector = openSelector();  
  21.     } catch (Exception e) {  
  22.         logger.warn("Failed to create a new Selector.", e);  
  23.         return;  
  24.     }  
  25.   
  26.     // Register all channels to the new Selector.  
  27.     int nChannels = 0;  
  28.     for (;;) {  
  29.         try {  
  30.             for (SelectionKey key: oldSelector.keys()) {  
  31.                 Object a = key.attachment();  
  32.                 try {  
  33.                     if (!key.isValid() || key.channel().keyFor(newSelector) != null) {  
  34.                         continue;  
  35.                     }  
  36.   
  37.                     int interestOps = key.interestOps();  
  38.                     key.cancel();  
  39.                     key.channel().register(newSelector, interestOps, a);  
  40.                     nChannels ++;  
  41.                 } catch (Exception e) {  
  42.                     logger.warn("Failed to re-register a Channel to the new Selector.", e);  
  43.                     if (a instanceof AbstractNioChannel) {  
  44.                         AbstractNioChannel ch = (AbstractNioChannel) a;  
  45.                         ch.unsafe().close(ch.unsafe().voidPromise());  
  46.                     } else {  
  47.                         @SuppressWarnings("unchecked")  
  48.                         NioTask<SelectableChannel> task = (NioTask<SelectableChannel>) a;  
  49.                         invokeChannelUnregistered(task, key, e);  
  50.                     }  
  51.                 }  
  52.             }  
  53.         } catch (ConcurrentModificationException e) {  
  54.             // Probably due to concurrent modification of the key set.  
  55.             continue;  
  56.         }  
  57.   
  58.         break;  
  59.     }  
  60.   
  61.     selector = newSelector;  
  62.   
  63.     try {  
  64.         // time to close the old selector as everything else is registered to the new one  
  65.         oldSelector.close();  
  66.     } catch (Throwable t) {  
  67.         if (logger.isWarnEnabled()) {  
  68.             logger.warn("Failed to close the old Selector.", t);  
  69.         }  
  70.     }  
  71.   
  72.     logger.info("Migrated " + nChannels + " channel(s) to the new Selector.");  
  73. }  

防止线程跑飞

线程是多路复用器的核心,所有IO事件执行的载体,一旦线程出现异常线程跑飞(run方法执行结束),那么可能会导致整个多路复用器不可用,导致挂载在多路复用器上的连接不可用,进而大量的业务请求失败。由于Netty中的同时处理IO事件和非IO事件逻辑,所以线程不仅仅要处理IO异常,业务测触发的异常也需要被正确的处理,一旦处理不当,会导致线程跑飞。Netty的处理是在run方法中catch所有的Throwable即所有的Exception和Error,不做任何处理,休眠1s继续执行循环,休眠1s的目的是为了防止捕获异常之后继续执行再次进入该异常形成死循环。实现代码在NioEventLoop的run方法中:

[java]  view plain  copy
  1. @Override  
  2. protected void run() {  
  3.     for (;;) {  
  4.         oldWakenUp = wakenUp.getAndSet(false);  
  5.         try {  
  6.             ...  
  7.         } catch (Throwable t) {  
  8.             logger.warn("Unexpected exception in the selector loop.", t);  
  9.   
  10.             // Prevent possible consecutive immediate failures that lead to  
  11.             // excessive CPU consumption.  
  12.             try {  
  13.                 Thread.sleep(1000);  
  14.             } catch (InterruptedException e) {  
  15.                 // Ignore.  
  16.             }  
  17.         }  
  18.     }  
  19. }  

内存保护

ByteBuf内存泄露保护

为了提升内存的利用率,Netty提供了内存池和对象池,内存泄露保护主要是针对Netty中的内存池的,Netty要求在使用完内存池中的内存之后要显示的归还,以免内存中的对象存在额外的引用造成内存泄露,Netty提供了SimpleChannelInboundHandler,该处理器会自动释放内存,使用者可以直接继承该处理器,它的channelRead方法的finally块中调用了释放内存的方法,另外内存泄露监控处理可以参考ResourceLeakDetector类中的代码:

[java]  view plain  copy
  1. @Override  
  2. public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {  
  3.     boolean release = true;  
  4.     try {  
  5.         if (acceptInboundMessage(msg)) {  
  6.             @SuppressWarnings("unchecked")  
  7.             I imsg = (I) msg;  
  8.             channelRead0(ctx, imsg);  
  9.         } else {  
  10.             release = false;  
  11.             ctx.fireChannelRead(msg);  
  12.         }  
  13.     } finally {  
  14.         if (autoRelease && release) {  
  15.             ReferenceCountUtil.release(msg);  
  16.         }  
  17.     }  
  18. }  

ByteBuf的内存溢出保护

为了防止一些超长的恶意流量耗尽服务器内存压垮服务器,有必要会缓存区设置上限,Netty做了如下处理:

  • 在内存分配的时候指定缓冲区长度上限(io.netty.buffer.ByteBufAllocator.buffer(int, int))。
  • 在对缓冲区进行写入操作的时候,如果缓冲区容量不足需要扩展,首先对最大容量进行判断,如果扩展后的容量超过上限,则拒绝扩展(io.netty.buffer.ByteBuf.ensureWritable(int)方法中处理)。
  • 在解码的时候,对消息长度进行判断,如果超过最大容量上限,则抛出解码异常,拒绝分配内存(io.netty.handler.codec.DelimiterBasedFrameDecoder.decode(ChannelHandlerContext, ByteBuf)方法中处理,在fail方法中抛出TooLongFrameException异常)。

连接中断处理

在客户端和服务端建立起连接之后,如果连接发生了意外中断,Netty也会及时释放连接句柄资源(因为TCP是全双工协议,通信双方都需要关闭和释放Socket句柄才不会发生句柄的泄漏,如不经过特殊处理是会发生句柄泄露的),原理如下:
在读取数据时会调用io.netty.buffer.AbstractByteBuf.writeBytes(ScatteringByteChannel, int),然后调用io.netty.buffer.ByteBuf.setBytes(int, ScatteringByteChannel, int),setBytes方法调用nio.channel.read,如果当前连接已经意外中断,会收到JDK NIO层抛出的ClosedChannelException异常,setBytes方法捕获该异常之后直接返回-1,
在NioByteUnsafe.read方法中,发现当前读取到的字节长度为-1,即调用io.netty.channel.nio.AbstractNioByteChannel.NioByteUnsafe.closeOnRead(ChannelPipeline)方法,然后调用io.netty.channel.AbstractChannel.AbstractUnsafe.close(ChannelPromise)关闭连接释放句柄资源。参考相关的代码:

[java]  view plain  copy
  1. //NioByteUnsafe.read方法  
  2. public void read() {  
  3.     ...  
  4.     boolean close = false;  
  5.     try {  
  6.         ...  
  7.         do {  
  8.             ...  
  9.             int localReadAmount = doReadBytes(byteBuf);  
  10.             if (localReadAmount <= 0) {  
  11.                 ...  
  12.                 close = localReadAmount < 0;  
  13.                 break;  
  14.             }  
  15.             ...  
  16.         } while (...);  
  17.   
  18.         ...  
  19.   
  20.         if (close) {  
  21.             closeOnRead(pipeline);  
  22.             close = false;  
  23.         }  
  24.     } catch (Throwable t) {  
  25.         ...  
  26.     } finally {  
  27.         ...  
  28.     }  
  29. }  
  30.   
  31. //NioSocketChannel.doReadBytes方法  
  32. protected int doReadBytes(ByteBuf byteBuf) throws Exception {  
  33.     return byteBuf.writeBytes(javaChannel(), byteBuf.writableBytes());  
  34. }  
  35.   
  36. //AbstractByteBuf.writeBytes方法  
  37. public int writeBytes(ScatteringByteChannel in, int length) throws IOException {  
  38.     ensureWritable(length);  
  39.     int writtenBytes = setBytes(writerIndex, in, length);  
  40.     if (writtenBytes > 0) {  
  41.         writerIndex += writtenBytes;  
  42.     }  
  43.     return writtenBytes;  
  44. }  
  45.   
  46. //UnpooledHeapByteBuf.setBytes方法  
  47. public int setBytes(int index, ScatteringByteChannel in, int length) throws IOException {  
  48.     ensureAccessible();  
  49.     try {  
  50.         return in.read((ByteBuffer) internalNioBuffer().clear().position(index).limit(index + length));  
  51.     } catch (ClosedChannelException e) {  
  52.         return -1;  
  53.     }  
  54. }  

流量整形

一般大型的系统都包含多个模块,在部署时不同的模块可能部署在不同的机器上,比如我司的项目,至少5个部件起,少了都不好意思拿出去见人。这种情况下系统运行时会涉及到大量的上下游部件的通信,但是由于不同服务器无论是从硬件配置,还是系统模块的业务特性都会存在差异,这就导致到服务器的处理能力,以及不同时间段服务器的负载都是有差异的,这就可能会导致问题:上下游消息的传递速度和下游部件的消息处理速度失去平衡,下游部件接收到的消息量远远超过了它的处理能力,导致大量的业务无法被及时的处理,甚至可能导致下游服务器被压垮。

在Netty框架中提供了流量整形处理机制来应付这种场景,通过控制服务器单位时间内发送/接收消息的字节数来使上下游服务器处理相对平衡的状态。Netty中的流量整形包含了两种:一种是针对单个连接的流量整形,另一种是针对全局即所有连接的流量整形。这两种方式的流量整形原理是类似的,只是流量整形器的作用域不同,一个是全局的,一个是连接建立后创建,连接关闭后被回收。GlobalTrafficShapingHandler处理全局流量整形,ChannelTrafficShapingHandler处理单链路流量整形,流量整形处理有三个重要的参数:

  • writeLimit:每秒最多可以写多个字节的数据。
  • readLimit:每秒最多可以读多少个字节的数据。
  • checkInterval:流量检查的间隔时间,默认1s。

以读操作为例,流量整形的工作过程大致如下:

  • 启动一个定时任务,每隔checkInterval毫秒执行一次,在任务中清除累加的读写字节数还原成0,更新上次流量整形检查时间。
  • 执行读操作,触发channelRead方法,记录当前已读取的字节数并且和上次流量整形检查之后的所有读操作读取的字节数进行累加。
  • 根据时间间隔和已读取的流量数计算当前流量判断当前读取操作是否已导致每秒读取的字节数超过了阀值readLimit,计算公式是:(bytes * 1000 / limit - interval) / 10 * 10,其中,bytes是上次流量整形检查之后的所有读操作累计读取的字节数,limit 就是readLimit,interval是当前时间距上次检查经过的时间毫秒数,如果该公式计算出来的值大于固定的阀值10,那么说明流量数已经超标,那么把该读操作放到延时任务中处理,延时的毫秒数就是上面那个公式计算出来的值。

下面是相关的代码:

[java]  view plain  copy
  1. //AbstractTrafficShapingHandler.channelRead方法  
  2. public void channelRead(final ChannelHandlerContext ctx, final Object msg) throws Exception {  
  3.     long size = calculateSize(msg);  
  4.     long curtime = System.currentTimeMillis();  
  5.   
  6.     if (trafficCounter != null) {  
  7.         //增加字节累计数  
  8.         trafficCounter.bytesRecvFlowControl(size);  
  9.         if (readLimit == 0) {  
  10.             // no action  
  11.             ctx.fireChannelRead(msg);  
  12.   
  13.             return;  
  14.         }  
  15.   
  16.         // compute the number of ms to wait before reopening the channel  
  17.         long wait = getTimeToWait(readLimit,  
  18.                 trafficCounter.currentReadBytes(),  
  19.                 trafficCounter.lastTime(), curtime);  
  20.         if (wait >= MINIMAL_WAIT) { // At least 10ms seems a minimal  
  21.             // time in order to  
  22.             // try to limit the traffic  
  23.             if (!isSuspended(ctx)) {  
  24.                 ctx.attr(READ_SUSPENDED).set(true);  
  25.   
  26.                 // Create a Runnable to reactive the read if needed. If one was create before it will just be  
  27.                 // reused to limit object creation  
  28.                 Attribute<Runnable> attr  = ctx.attr(REOPEN_TASK);  
  29.                 Runnable reopenTask = attr.get();  
  30.                 if (reopenTask == null) {  
  31.                     reopenTask = new ReopenReadTimerTask(ctx);  
  32.                     attr.set(reopenTask);  
  33.                 }  
  34.                 ctx.executor().schedule(reopenTask, wait,  
  35.                         TimeUnit.MILLISECONDS);  
  36.             } else {  
  37.                 // Create a Runnable to update the next handler in the chain. If one was create before it will  
  38.                 // just be reused to limit object creation  
  39.                 Runnable bufferUpdateTask = new Runnable() {  
  40.                     @Override  
  41.                     public void run() {  
  42.                         ctx.fireChannelRead(msg);  
  43.                     }  
  44.                 };  
  45.                 ctx.executor().schedule(bufferUpdateTask, wait, TimeUnit.MILLISECONDS);  
  46.                 return;  
  47.             }  
  48.         }  
  49.     }  
  50.     ctx.fireChannelRead(msg);  
  51. }  
  52.   
  53. //AbstractTrafficShapingHandler.getTimeToWait方法  
  54. private static long getTimeToWait(long limit, long bytes, long lastTime, long curtime) {  
  55.     long interval = curtime - lastTime;  
  56.     if (interval <= 0) {  
  57.         // Time is too short, so just lets continue  
  58.         return 0;  
  59.     }  
  60.     return (bytes * 1000 / limit - interval) / 10 * 10;  
  61. }  
  62.   
  63. private static class TrafficMonitoringTask implements Runnable {  
  64.     ...  
  65.     @Override  
  66.     public void run() {  
  67.         if (!counter.monitorActive.get()) {  
  68.             return;  
  69.         }  
  70.         long endTime = System.currentTimeMillis();  
  71.         //还原累计字节数,lastTime等变量  
  72.         counter.resetAccounting(endTime);  
  73.         if (trafficShapingHandler1 != null) {  
  74.             trafficShapingHandler1.doAccounting(counter);  
  75.         }  
  76.         counter.scheduledFuture = counter.executor.schedule(this, counter.checkInterval.get(),  
  77.                                                             TimeUnit.MILLISECONDS);  
  78.     }  
  79. }  






评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值