如何在 Go 中实现一个支持中间件嵌套调用的轻量级 RPC 框架

冬明君_2997

冬明君_2997

2026-05-13

366人浏览

原创

Go轻量RPC框架通过统一中间件签名func(context.Context, *rpc.Request, interface{}, func() error) error实现嵌套调用,服务端借ServerCodec注入钩子,客户端封装Do方法支持Before/After回调,全程透传context与error。

go 里实现支持中间件嵌套调用的轻量级 rpc 框架,核心不在于重写 net/rpc,而在于把中间件逻辑注入到请求生命周期的关键钩子点——服务端用 handler 包裹,客户端用 call 前后拦截,且中间件必须能链式传递上下文和错误。

用 func(context.Context, *rpc.Request, interface{}, func() error) error 统一中间件签名

中间件必须可嵌套,就不能依赖固定参数列表。统一用函数类型接收 context.Context、*rpc.Request、响应值指针、以及一个“继续向下执行”的 next 函数。这样每层中间件决定是否调用 next(),还能在前后做日志、鉴权、超时控制等。

常见错误是把中间件写成 func(*rpc.Request) error,导致无法透传 context 或提前终止调用链。

  • 所有中间件必须返回 error,以便上层统一处理失败(比如直接返回 rpc.ErrShutdown)
  • 响应值用 interface{} 指针接收,避免中间件内部对具体结构体强依赖
  • next() 不应暴露底层连接或 codec,只负责触发后续 handler 或实际 RPC 调用

服务端:用 rpc.Server.RegisterCodec + 自定义 ServerCodec 插入中间件链

标准 rpc.Server 不提供 handler 注册钩子,但允许替换 Codec。你可以包装原始 ServerCodec,在 ReadRequestHeader 后、ReadRequestBody 前启动中间件链,在 WriteResponse 前注入后置逻辑。

典型场景是统一加 trace ID、校验 token、限流——这些都得在反序列化请求体前完成,否则可能浪费解析开销。

  • 不要在 Register 阶段注册中间件,那只是绑定方法名,不参与执行流
  • 自定义 ServerCodec 必须完整代理原始 codec 的所有方法,漏掉 Close 会导致连接泄漏
  • 中间件链执行顺序 = 注册顺序,前置中间件先执行;若某层返回非 nil error,应跳过 next() 并直接 WriteResponse 错误

客户端:封装 rpc.Client.Call 为 Do 方法并支持 Before/After 回调

客户端没法改 rpc.Client 内部逻辑,所以最轻量做法是包装一层结构体,暴露 Do(ctx context.Context, serviceMethod string, args interface{}, reply interface{}) error,并在其中按序执行 Before 列表(如注入 header)、调用原 Call、再执行 After(如记录耗时)。

使用Go语言搭建家庭相册系统-相关课件
使用Go语言搭建家庭相册系统-相关课件

使用Go语言搭建家庭相册系统-相关课件

下载

容易踩的坑是把中间件写成同步阻塞式重试——这会卡死整个调用,正确做法是在 Before 中设置 ctx 超时,让 Call 自身承担重试语义(或由单独的 retry middleware 处理)。

  • Before 可修改 args(比如加签名字段),但不能改 reply 指针本身
  • After 接收 error 参数,用于判断是否成功,避免重复解包 response
  • 不要在中间件里直接操作 rpc.Client 的 conn 字段——它是未导出字段,且 Go 1.22+ 已标记为 internal

中间件嵌套时 context 传递必须显式透传,不能依赖闭包捕获

多个中间件嵌套时,如果用闭包捕获外层 ctx,会导致超时/取消信号无法向下传递。必须把更新后的 ctx 作为参数传给下一层 next,并在每层中间件开头用 ctx = ctx 显式接收。

例如鉴权中间件生成新 ctx 带 user info,日志中间件就得从这个 ctx 里取,而不是读原始传入的 ctx。

  • 每次调用 next(ctx, req, reply, ...) 前,确保 ctx 已被当前中间件增强(如 context.WithValue 或 context.WithTimeout)
  • 避免在中间件中用 context.Background() 创建子 context,会切断取消链
  • 调试时打印 ctx.Err() 是排查“调用没响应却也不报错”的最快方式

真正难的不是写通中间件链,而是保证每个环节的 error 被正确归因——是 codec 解析失败?还是中间件主动拒绝?还是下游服务无响应?这些 error 类型必须区分清楚,否则运维时根本没法定位是框架问题还是业务逻辑问题。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

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

下载

相关标签:

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

相关专题

更多
C语言变量命名
C语言变量命名

c语言变量名规则是:1、变量名以英文字母开头;2、变量名中的字母是区分大小写的;3、变量名不能是关键字;4、变量名中不能包含空格、标点符号和类型说明符。php中文网还提供c语言变量的相关下载、相关课程等内容,供大家免费下载使用。

2023.06.20

3469

3

c语言入门自学零基础
c语言入门自学零基础

C语言是当代人学习及生活中的必备基础知识,应用十分广泛,本专题为大家c语言入门自学零基础的相关文章,以及相关课程,感兴趣的朋友千万不要错过了。

2023.07.25

2408

9

c语言运算符的优先级顺序
c语言运算符的优先级顺序

c语言运算符的优先级顺序是括号运算符 > 一元运算符 > 算术运算符 > 移位运算符 > 关系运算符 > 位运算符 > 逻辑运算符 > 赋值运算符 > 逗号运算符。本专题为大家提供c语言运算符相关的各种文章、以及下载和课程。

2023.08.02

1340

5

c语言数据结构
c语言数据结构

数据结构是指将数据按照一定的方式组织和存储的方法。它是计算机科学中的重要概念,用来描述和解决实际问题中的数据组织和处理问题。数据结构可以分为线性结构和非线性结构。线性结构包括数组、链表、堆栈和队列等,而非线性结构包括树和图等。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.09

1278

4

c语言random函数用法
c语言random函数用法

c语言random函数用法:1、random.random,随机生成(0,1)之间的浮点数;2、random.randint,随机生成在范围之内的整数,两个参数分别表示上限和下限;3、random.randrange,在指定范围内,按指定基数递增的集合中获得一个随机数;4、random.choice,从序列中随机抽选一个数;5、random.shuffle,随机排序。

2023.09.05

1396

5

c语言const用法
c语言const用法

const是关键字,可以用于声明常量、函数参数中的const修饰符、const修饰函数返回值、const修饰指针。详细介绍:1、声明常量,const关键字可用于声明常量,常量的值在程序运行期间不可修改,常量可以是基本数据类型,如整数、浮点数、字符等,也可是自定义的数据类型;2、函数参数中的const修饰符,const关键字可用于函数的参数中,表示该参数在函数内部不可修改等等。

2023.09.20

2298

7

c语言get函数的用法
c语言get函数的用法

get函数是一个用于从输入流中获取字符的函数。可以从键盘、文件或其他输入设备中读取字符,并将其存储在指定的变量中。本文介绍了get函数的用法以及一些相关的注意事项。希望这篇文章能够帮助你更好地理解和使用get函数 。

2023.09.20

3740

8

c数组初始化的方法
c数组初始化的方法

c语言数组初始化的方法有直接赋值法、不完全初始化法、省略数组长度法和二维数组初始化法。详细介绍:1、直接赋值法,这种方法可以直接将数组的值进行初始化;2、不完全初始化法,。这种方法可以在一定程度上节省内存空间;3、省略数组长度法,这种方法可以让编译器自动计算数组的长度;4、二维数组初始化法等等。

2023.09.22

17455

6

c语言中null和NULL的区别
c语言中null和NULL的区别

c语言中null和NULL的区别是:null是C语言中的一个宏定义,通常用来表示一个空指针,可以用于初始化指针变量,或者在条件语句中判断指针是否为空;NULL是C语言中的一个预定义常量,通常用来表示一个空值,用于表示一个空的指针、空的指针数组或者空的结构体指针。

2023.09.22

549

3

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程