上一篇【第36篇】Netty时间轮高级应用——10亿级定时任务的工程实践下一篇【第38篇】Netty SSL TLS安全传输——HTTPS背后的Netty实现一、为什么需要流量整形某支付系统上游每天凌晨批量同步数据带宽被占满影响正常交易。流量整形平滑流量避免突发压垮下游。二、原理令牌桶算法令牌桶容量10 每秒产生5个令牌 时刻0: [T][T][T][T][T][_][_][_][_][_] (5个令牌) 发送3个包 → [T][T][_][_][_][_][_][_][_][_] (剩2个令牌) 时刻1: [T][T][T][T][T][T][T][_][_][_] (补充5个共7个) 发送6个包 → [T][_][_][_][_][_][_][_][_][_] (剩1个令牌) 发送4个包 → [T][_][_][_][_][_][_][_][_][_] ↑ 只有1个令牌只发送1个3个等待三、GlobalTrafficShapingHandler// 全局流量整形所有Channel共享带宽pipeline.addLast(newGlobalTrafficShapingHandler(eventExecutor,1024*1024,// writeLimit: 1MB/s 写速率1024*1024,// readLimit: 1MB/s 读速率1000// checkInterval: 1秒检查一次));// 源码核心AbstractTrafficShapingHandler// 计算等待时间延迟发送longwaittrafficCounter.readTimeToWait(size,limit,interval);if(wait0){ctx.executor().schedule(()-{ctx.write(msg,promise);},wait,TimeUnit.MILLISECONDS);}四、ChannelTrafficShapingHandler// 每个Channel独立限速pipeline.addLast(newChannelTrafficShapingHandler(512*1024,// 每个连接512KB/s0// 不限读));五、实战防DDoS连接限流publicclassAntiDdosHandlerextendsChannelInboundHandlerAdapter{privatefinalGlobalTrafficShapingHandlertrafficHandler;privatefinalAtomicIntegerconnectionsnewAtomicInteger();privatestaticfinalintMAX_CONNECTIONS10000;publicAntiDdosHandler(EventExecutorexecutor){trafficHandlernewGlobalTrafficShapingHandler(executor,1024*1024,0);}OverridepublicvoidchannelActive(ChannelHandlerContextctx){if(connections.incrementAndGet()MAX_CONNECTIONS){System.out.println(连接数超限拒绝新连接);ctx.close();}else{ctx.pipeline().addFirst(trafficHandler);ctx.fireChannelActive();}}OverridepublicvoidchannelInactive(ChannelHandlerContextctx){connections.decrementAndGet();ctx.fireChannelInactive();}}六、对比总结Handler范围配置GlobalTrafficShapingHandler所有Channel共享writeLimit1MB/sChannelTrafficShapingHandler单个Channel独立writeLimit512KB/sGlobalChannelTrafficShapingHandler全局单Channel双重控制灵活组合上一篇【第36篇】Netty时间轮高级应用——10亿级定时任务的工程实践下一篇【第38篇】Netty SSL TLS安全传输——HTTPS背后的Netty实现