go-zero/grpc的rpc服务间传递额外数据

go-zero/grpc的rpc服务间传递额外数据

2024/02/18

客户端:

go 复制代码
md := metadata.New(map[string]string{"signature": "0123456789", "timestamp": "2021-07-01 00:00:00"})
ctx := metadata.NewOutgoingContext(ctx, md)
addResp, err := adderClient.Add(ctx, addReq) // rpc 调用

初始化 md 也可如下方式:

go 复制代码
md := metadata.Pairs(
	"signature", "0123456789",
	"timestamp", "2021-07-01 00:00:00",
)
ctx := metadata.NewOutgoingContext(ctx, md)
addResp, err := adderClient.Add(ctx, addReq) // rpc 调用

追加新的如下:

go 复制代码
ctx = metadata.AppendToOutgoingContext(ctx, "noncestr", "abc")

也可使用 md 的 Set 和 Append 方法追加:

go 复制代码
md.Set("extra", "extra-value")
md.Append("append", "append-value")

服务端:

go 复制代码
md, _ := metadata.FromIncomingContext(ctx)
或直接:
vals := metadata.ValueFromIncomingContext(ctx, "signature")

注意 key 都会被转为小写,即使客户端为大写:

go 复制代码
// Keys beginning with "grpc-" are reserved for grpc-internal use only and may
// result in errors if set in metadata.
func New(m map[string]string) MD {
	md := make(MD, len(m))
	for k, val := range m {
		key := strings.ToLower(k)
		md[key] = append(md[key], val)
	}
	return md
}

func Pairs(kv ...string) MD {
	if len(kv)%2 == 1 {
		panic(fmt.Sprintf("metadata: Pairs got the odd number of input pairs for metadata: %d", len(kv)))
	}
	md := make(MD, len(kv)/2)
	for i := 0; i < len(kv); i += 2 {
		key := strings.ToLower(kv[i])
		md[key] = append(md[key], kv[i+1])
	}
	return md
}

func (md MD) Set(k string, vals ...string) {
	if len(vals) == 0 {
		return
	}
	k = strings.ToLower(k)
	md[k] = vals
}

而且 key 只能由 数字、字母和三个特殊字符"-_."组成,大写字母会自动被转为小写字母。

相关推荐
徐小黑ACG1 小时前
GO语言 使用protobuf
开发语言·后端·golang·protobuf
能来帮帮蒟蒻吗12 小时前
GO语言学习(16)Gin后端框架
开发语言·笔记·学习·golang·gin
JavaPub-rodert12 小时前
一道go面试题
开发语言·后端·golang
6<713 小时前
【go】静态类型与动态类型
开发语言·后端·golang
weixin_4209476416 小时前
windows golang,consul,grpc学习
windows·golang·consul
Json201131516 小时前
Gin、Echo 和 Beego三个 Go 语言 Web 框架的核心区别及各自的优缺点分析,结合其设计目标、功能特性与适用场景
前端·golang·gin·beego
二狗哈21 小时前
go游戏后端开发21:处理nats消息
开发语言·游戏·golang
能来帮帮蒟蒻吗1 天前
Go语言学习(15)结构体标签与反射机制
开发语言·笔记·学习·golang
returnShitBoy1 天前
Go语言中的defer关键字有什么作用?
开发语言·后端·golang
SoFlu软件机器人1 天前
Go/Rust 疯狂蚕食 Java 市场?老牌语言的 AI 化自救之路
java·golang·rust