【rust-grpc-proxy】自定义metadata

前言

没有废话

原理

grpc可以通过metadata(元数据)实现类似http header的效果。
grpc底层使用http2通讯，而metadata通常会放在http2的header中传输，这也是本文支持的方式。
trailer 方式也能传输metadata，但我不常用，且处理较为麻烦，如无必要，则不支持。(绝对不承认自己懒)

实践

目标服务

首先修改一下我们之前的echo服务，增加一个拦截器，会将我们设置的metadata回显回来。

func metadataInterceptor() grpc.UnaryServerInterceptor {
	return func(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp interface{}, err error) {
		md, ok := metadata.FromIncomingContext(ctx)
		header := metadata.New(map[string]string{})
		if ok {
			for k, v := range md {
				if k == "content-type" || k == "content-length" {
					continue
				}
				header.Set(k, v[0])
			}
		}
		resp, err = handler(ctx, req)
		if err != nil {
			return nil, err
		}
		err = grpc.SetHeader(ctx, header)
		return resp, err
	}
}

将上面的代码加入到任何带有反射的grpc服务中，或者用项目中的测试程序目录是：example/metadata/server 环境需要mac才行

cd example/metadata

//会启动一个目标服务，并监听1234端口
./example.sh server

生成配置

运行命令生成配置

//example/metadata
./example.sh config 

可以看到配置中的内容

[metadata_filters]
prefix = ["echo-","greet-"]
match = ["use-name","use-id"]

• prefix 前缀匹配，配置为空则默认["md-"]
• match 完全匹配

运行程序

用如下命令运行代理

//mac系统 dir:example/metadata
./rust-grpc-proxy run -c config.toml

//或者在根目录运行
cargo run -- run -c ./example/metadata/config.toml

测试请求

curl --location --request GET 'http://127.0.0.1:6789/api/v1/echo/hello/get?query=666' \
--header 'echo-hello: world' \
--header 'md-hello: md-world' \
--header 'use-name: mynameiswd'

可以在response 的header看到 回显内容。表明测试成功

尾语

目前基础功能已经完成的差不了。根据计划，之后会完善使用体验上的内容。