使用context.Context管理RPC请求的超时、取消和元数据传递，gRPC原生支持上下文，而net/rpc需封装模拟，推荐gRPC以实现更完整的上下文控制。
在Golang中处理RPC请求上下文，核心是使用context.Context来传递请求范围的值、控制超时和取消信号。尤其是在gRPC或标准库net/rpc中，合理使用上下文能提升服务的可控性和可观测性。
理解上下文在RPC中的作用
Context 是Go中用于跨API边界传递截止时间、取消信号和请求范围值的标准机制。在RPC调用中，客户端发起请求时创建上下文，服务端通过该上下文获取调用元信息，并响应取消或超时。
常见用途包括：
- 设置请求超时，避免长时间阻塞
- 传递认证token、trace ID等元数据
- 支持优雅取消正在进行的远程调用
在gRPC中使用上下文
gRPC原生支持context.Context作为每个方法的第一个参数。
立即学习“go语言免费学习笔记（深入）”；
示例：客户端设置超时ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) defer cancel()resp, err := client.GetUser(ctx, &pb.UserID{Id: 123}) if err != nil { log.Fatal(err) }
示例：服务端读取上下文值
func (s *UserService) GetUser(ctx context.Context, req *pb.UserID) (*pb.User, error) {
// 检查是否已被取消
select {
case <-ctx.Done():
return nil, ctx.Err()
default:
}
// 获取传入的元数据（如token）
if md, ok := metadata.FromIncomingContext(ctx); ok {
if tokens := md["authorization"]; len(tokens) > 0 {
// 验证token
}
}
// 实际业务逻辑
user := &pb.User{Name: "Alice"}
return user, nil}
在标准库 net/rpc 中集成上下文
Go的net/rpc包本身不支持上下文，但可以通过封装参数来模拟。
一种常见做法是将context.Context信息编码进请求结构体中。
示例：为RPC方法添加上下文支持type Request struct {
Context map[string]string // 模拟传递trace_id, timeout等
Data interface{}
}
type Response struct {
Result interface{}
Error string
}
func (t Arith) Multiply(req Request, resp Response) error {
// 模拟从req.Context恢复上下文
traceID := req.Context["traceid"]
timeoutStr := req.Context["timeout"]
timeout, := time.ParseDuration(timeoutStr)
ctx, cancel := context.WithTimeout(context.Background(), timeout)
defer cancel()
// 使用ctx进行数据库查询或其他IO操作
select {
case <-time.After(2 * time.Second):
resp.Result = 42
case <-ctx.Done():
resp.Error = ctx.Err().Error()
return ctx.Err()
}
return nil}
注意：net/rpc限制较多，推荐使用gRPC替代以获得完整的上下文支持。
传递自定义元数据
使用context.WithValue可安全传递请求本地数据，但应限于请求元信息，而非核心参数。
示例：在中间件中注入trace IDctx := context.WithValue(context.Background(), "trace_id", "abc-123")
// 在handler中读取
if traceID, ok := ctx.Value("trace_id").(string); ok {
log.Printf("handling request %s", traceID)
}
建议使用自定义类型键避免冲突：
type ctxKey string
const TraceIDKey ctxKey = "trace_id"
ctx := context.WithValue(parent, TraceIDKey, "abc-123")
// ...
traceID := ctx.Value(TraceIDKey).(string)
基本上就这些。关键是把上下文当作请求生命周期的管理工具，结合超时、取消和元数据传递，让RPC调用更健壮。
