0

0

如何在Golang中实现RPC调用限速_Golang RPC请求流量控制方法

P粉602998670

P粉602998670

发布时间:2026-01-12 11:46:02

|

746人浏览过

|

来源于php中文网

原创

Go net/rpc 本身不支持限速,需通过包装 net.Listener 实现连接频次限制,或包装 net.Conn 在 Read() 中按 RPC 帧限速;精确 QPS 限速应改用 gRPC + 拦截器。

如何在golang中实现rpc调用限速_golang rpc请求流量控制方法

Go net/rpc 本身不支持限速,必须自己加中间层

Go 标准库net/rpc 是一个轻量级、无状态的 RPC 框架,它不提供任何请求频率控制能力。限速逻辑不能写在服务端方法内部(那样会污染业务),也不能依赖 HTTP 中间件(因为 net/rpc 默认走 TCP,不经过 HTTP )。真正可行的方式是在连接建立后、请求分发前插入限速器。

net.Listener 包装器实现连接级限速

最干净的做法是包装 net.Listener,在 Accept() 返回连接前做速率判断。这样能控制新连接建立频率,适合防御突发连接洪峰。注意:这不是“每秒请求数”(QPS)限速,而是“每秒新建连接数”限制。

  • 适用于防止客户端疯狂重连或扫描类攻击
  • 使用 golang.org/x/time/rate.Limiter 配合 net.Listener 实现
  • 不要在 Accept() 中阻塞太久,建议设置超时或使用非阻塞 TryAccept 类逻辑
type RateLimitedListener struct {
    net.Listener
    limiter *rate.Limiter
}

func (l *RateLimitedListener) Accept() (net.Conn, error) {
    if !l.limiter.Allow() {
        return nil, errors.New("connection rejected: rate limit exceeded")
    }
    return l.Listener.Accept()
}

rpc.ServeConn 前拦截并限速单连接内请求

如果要控制单个 TCP 连接内的请求处理速度(比如防止一个恶意客户端复用连接狂发请求),就得在 rpc.ServeConn 调用前包装 net.Conn。核心是重写 Read() 方法,在每次读到完整 RPC 帧(含 header + body)后再触发一次限速检查。

  • net/rpc 使用自定义二进制协议,每个请求以 4 字节长度头开始,需先读头再读体
  • 直接包装 net.Conn 并在 Read() 中做 limiter.WaitN(ctx, n) 可控但复杂
  • 更稳妥的做法是改用 jsonrpc2gRPC 等带上下文和中间件能力的框架

实际项目中推荐用 gRPC + grpc-middleware 实现 QPS 限速

如果你的场景真需要精确到每秒 N 次方法调用的限速(比如 User.Create 接口限 100 QPS),net/rpc 不是合适选择。gRPC 天然支持拦截器,配合 go.uber.org/ratelimitgolang.org/x/time/rate 很容易实现。

ImgCreator AI
ImgCreator AI

一款AI图像生成工具,适合创建插图、动画和概念设计图像。

下载

立即学习go语言免费学习笔记(深入)”;

  • 限速粒度可设为 method、user-id、ip 等维度
  • 错误返回统一为 status.Errorf(codes.ResourceExhausted, "...")
  • 注意:gRPC 流式接口需单独处理,limiter.Allow() 要放在每次 Recv() 后而非连接建立时
func rateLimitInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
    if !limiter.Allow() {
        return nil, status.Errorf(codes.ResourceExhausted, "rate limit exceeded")
    }
    return handler(ctx, req)
}

真正难的不是写限速代码,而是确定限速维度和阈值——同一接口对内部系统和外部 API 的容忍度可能差十倍;而误把健康检查探针当成恶意请求限掉,这种事在灰度期特别常见。

相关专题

更多
golang如何定义变量
golang如何定义变量

golang定义变量的方法:1、声明变量并赋予初始值“var age int =值”;2、声明变量但不赋初始值“var age int”;3、使用短变量声明“age :=值”等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

178

2024.02.23

golang有哪些数据转换方法
golang有哪些数据转换方法

golang数据转换方法:1、类型转换操作符;2、类型断言;3、字符串和数字之间的转换;4、JSON序列化和反序列化;5、使用标准库进行数据转换;6、使用第三方库进行数据转换;7、自定义数据转换函数。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

226

2024.02.23

golang常用库有哪些
golang常用库有哪些

golang常用库有:1、标准库;2、字符串处理库;3、网络库;4、加密库;5、压缩库;6、xml和json解析库;7、日期和时间库;8、数据库操作库;9、文件操作库;10、图像处理库。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

337

2024.02.23

golang和python的区别是什么
golang和python的区别是什么

golang和python的区别是:1、golang是一种编译型语言,而python是一种解释型语言;2、golang天生支持并发编程,而python对并发与并行的支持相对较弱等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

208

2024.03.05

golang是免费的吗
golang是免费的吗

golang是免费的。golang是google开发的一种静态强类型、编译型、并发型,并具有垃圾回收功能的开源编程语言,采用bsd开源协议。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

388

2024.05.21

golang结构体相关大全
golang结构体相关大全

本专题整合了golang结构体相关大全,想了解更多内容,请阅读专题下面的文章。

194

2025.06.09

golang相关判断方法
golang相关判断方法

本专题整合了golang相关判断方法,想了解更详细的相关内容,请阅读下面的文章。

189

2025.06.10

golang数组使用方法
golang数组使用方法

本专题整合了golang数组用法,想了解更多的相关内容,请阅读专题下面的文章。

192

2025.06.17

Java 项目构建与依赖管理(Maven / Gradle)
Java 项目构建与依赖管理(Maven / Gradle)

本专题系统讲解 Java 项目构建与依赖管理的完整体系,重点覆盖 Maven 与 Gradle 的核心概念、项目生命周期、依赖冲突解决、多模块项目管理、构建加速与版本发布规范。通过真实项目结构示例,帮助学习者掌握 从零搭建、维护到发布 Java 工程的标准化流程,提升在实际团队开发中的工程能力与协作效率。

3

2026.01.12

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
WEB前端教程【HTML5+CSS3+JS】
WEB前端教程【HTML5+CSS3+JS】

共101课时 | 8.2万人学习

JS进阶与BootStrap学习
JS进阶与BootStrap学习

共39课时 | 3.1万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号