Java NIO流量整形核心是写就绪时按令牌桶节奏放行字节：维护独立令牌桶，周期补令牌，预占不足则注册OP_WRITE等待；Netty的ChannelTrafficShapingHandler已封装该逻辑并支持动态调整。
Java NIO 中实现流量整形（Traffic Shaping），核心是**在数据写入通道前控制发送速率**，避免突发流量压垮下游或触发限流。NIO 本身不提供内置的流量整形组件，需结合 SelectionKey.OP_WRITE 事件、时间轮/滑动窗口逻辑、以及缓冲区调度来手动实现。关键不是“拦截字节”，而是“按节奏放行字节”。
基于写就绪事件 + 令牌桶的写控速
这是最常用且贴近网络栈实际的方式：不阻塞线程，利用 Selector 对 OP_WRITE 的通知机制，配合令牌桶动态决定每次能写多少。
- 为每个 Channel 维护一个独立的令牌桶（如
AtomicLong tokens），按目标速率（如 1MB/s）周期性补充令牌（例如每 10ms 补 10KB） - 当有数据要发送时，先尝试“预占”所需令牌（如待写 8KB → 需 8 个 1KB 令牌），若不足则取消本次写操作，注册
OP_WRITE并等待下次就绪 - 就绪后，从 ByteBuffer 中读取最多「当前可用令牌数 × 单位大小」的字节写出；写完立即更新剩余令牌，并取消
OP_WRITE（除非缓冲区仍有未写完数据） - 注意：必须使用
SocketChannel.write(ByteBuffer)的返回值判断实际写出量，不能假设全写完；未写完的数据要保留在 buffer 中并保持OP_WRITE注册状态
使用 Netty 的 ChannelTrafficShapingHandler
如果你用 Netty（底层基于 NIO），直接复用其成熟实现是最稳妥的选择——它已封装了全局/单连接速率限制、平滑突发处理、延迟补偿等细节。
Alibabacloud Sdk Client Initialization For Java
下载
在 Java 中初始化和管理阿里云 SDK客户端。包括单例模式、线程安全、endpoint 与 region 配置、VPC 终端节点、同步与异步等。
- 服务端对每个连接限速：new
ChannelTrafficShapingHandler(1024 * 1024, 1024 * 1024, 1000)（写速率、读速率、检查周期 ms） - 支持动态调整：
trafficShapingHandler.configure(512 * 1024) - 底层仍依赖
OP_WRITE调度，但自动管理写队列、唤醒时机和异常回退，比手写更健壮
避免常见陷阱
手写整形容易踩坑，重点注意三点：
立即学习“Java免费学习笔记（深入）”；
- 不要在 OP_WRITE 就绪时无脑循环 write()：NIO 的 write 是非阻塞的，一次调用可能只写部分数据，但反复调用直到 buffer 清空会吃光 CPU；应每次只写一帧或按令牌限额写，再让出控制权
-
时间精度不能依赖 System.currentTimeMillis()：高频率补令牌时，用
System.nanoTime()计算间隔，避免毫秒级时钟跳跃导致令牌误补 - 关闭连接时必须清理整形状态：清除令牌桶、取消写兴趣集、释放关联 buffer，否则内存泄漏+无效调度
轻量替代：固定窗口计数器（适合粗粒度限速）
如果只要求“每秒最多发 X 字节”，不需要平滑，可用更简单的方案：
- 每个 Channel 持有一个
long windowStartMs和AtomicLong bytesInWindow - 每次准备写前，检查是否跨窗：
if (now - windowStartMs >= 1000) { reset(); } - 若
bytesInWindow.addAndGet(needBytes) 则允许写，否则拒绝或排队 - 优点：无定时任务、无锁竞争少；缺点：存在临界时刻的两倍突发（如窗口末尾写 1B，新窗口立刻写满）
