grpc拦截器+metadata进行接口统一校验

最新推荐文章于 2025-02-08 00:00:00 发布

原创

最新推荐文章于 2025-02-08 00:00:00 发布

· 330 阅读

3 ·

版权

文章标签：

#go #grpc #拦截器

本文详细介绍了如何在gRPC中使用自定义拦截器和内置metadata进行接口调用的统一校验，以及两种方法：一是自定义拦截器配合客户端元数据；二是通过实现credentials.PerRPCCredentials接口。同时展示了server端如何对接口进行拦截并进行验证。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

grpc拦截器+metadata进行接口统一校验

hello.proto

syntax = "proto3";

option go_package ="./;proto";

service Server {
   
  rpc SayHello(HelloReq) returns (HelloRes);
}

message HelloReq{
   
  string name = 1;
}

message HelloRes{
   
  string msg = 1;
}

proto生成go文件命令

protoc --go_out=. --go-grpc_out=. hello.proto

client.go

方法一：自定义拦截器，实现统一参数校验

package main

import (
	"Go_Bible/grpc_token_auth_test/proto"
	"context"
	"fmt"
	"google.golang.org/grpc"
	"google.golang.org/grpc/metadata"
)

func main() {
   
	/*自定义拦截器,搭配metadata元数据，进行登录验证*/
	authInterceptor := func(ctx context.Context, method string, req, reply any, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
   
		//fmt.Println("客户端封装metadata，封装auth")
		// 手动创建元数据
		md := metadata.New(map[string]string{
   
			"appid":      "10010",
			"access_key": "1234",
		})
		// 创建出站请求的上下文
		ctx = metadata.NewOutgoingContext(ctx, md)
		err := invoker(ctx, method, req, reply, cc, opts...)
		if err != nil {
   
			fmt.Println("客户端接口调用错误-->" + err.Error())
		}
		return err
	}
	// 放入拦截器
	op

最低0.47元/天解锁文章