Java 怎么在 NIO 网络编程中实现流量整形功能

云枫大大_1275

云枫大大_1275

2026-09-08

863人浏览

原创

java nio流量整形核心是写就绪时按令牌桶节奏放行字节:维护独立令牌桶,周期补令牌,预占不足则注册op_write等待;netty的channeltrafficshapinghandler已封装该逻辑并支持动态调整。

java 怎么在 nio 网络编程中实现流量整形功能

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
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 调度,但自动管理写队列、唤醒时机和异常回退,比手写更健壮

避免常见陷阱

手写整形容易踩坑,重点注意三点:

  • 不要在 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,新窗口立刻写满)

Java免费学习笔记:立即使用
解锁 Java 大师之旅:从入门到精通的终极指南

相关文章

编程速学教程(入门课程)
编程速学教程(入门课程)

编程怎么学习?编程怎么入门?编程在哪学?编程怎么学才快?不用担心,这里为大家提供了编程速学教程(入门课程),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

java 网络编程

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
java
java

Java是一个通用术语,用于表示Java软件及其组件,包括“Java运行时环境 (JRE)”、“Java虚拟机 (JVM)”以及“插件”。php中文网还为大家带了Java相关下载资源、相关课程以及相关文章等内容,供大家免费下载使用。

2023.06.15

9317

6

java正则表达式语法
java正则表达式语法

java正则表达式语法是一种模式匹配工具,它非常有用,可以在处理文本和字符串时快速地查找、替换、验证和提取特定的模式和数据。本专题提供java正则表达式语法的相关文章、下载和专题,供大家免费下载体验。

2023.07.05

6482

9

java自学难吗
java自学难吗

Java自学并不难。Java语言相对于其他一些编程语言而言,有着较为简洁和易读的语法,本专题为大家提供java自学难吗相关的文章,大家可以免费体验。

2023.07.31

5772

8

java配置jdk环境变量
java配置jdk环境变量

Java是一种广泛使用的高级编程语言,用于开发各种类型的应用程序。为了能够在计算机上正确运行和编译Java代码,需要正确配置Java Development Kit(JDK)环境变量。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.08.01

1024

3

java保留两位小数
java保留两位小数

Java是一种广泛应用于编程领域的高级编程语言。在Java中,保留两位小数是指在进行数值计算或输出时,限制小数部分只有两位有效数字,并将多余的位数进行四舍五入或截取。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.08.02

848

3

java基本数据类型
java基本数据类型

java基本数据类型有:1、byte;2、short;3、int;4、long;5、float;6、double;7、char;8、boolean。本专题为大家提供java基本数据类型的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

1216

5

java有什么用
java有什么用

java可以开发应用程序、移动应用、Web应用、企业级应用、嵌入式系统等方面。本专题为大家提供java有什么用的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

2449

5

java在线网站
java在线网站

Java在线网站是指提供Java编程学习、实践和交流平台的网络服务。近年来,随着Java语言在软件开发领域的广泛应用,越来越多的人对Java编程感兴趣,并希望能够通过在线网站来学习和提高自己的Java编程技能。php中文网给大家带来了相关的视频、教程以及文章,欢迎大家前来学习阅读和下载。

2023.08.03

19791

3

配置java环境变量
配置java环境变量

配置Java环境变量是为了让操作系统能够识别和使用Java的相关命令和功能。本专题为大家提供配置java环境变量相关文章,帮助大家解决问题。

2023.08.03

1115

8

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
dev.java 官方:Learn Java
dev.java 官方:Learn Java

共0课时 | 0人学习

Java JDBC数据库连接官方教程
Java JDBC数据库连接官方教程

共0课时 | 0人学习