Golang——gRPC认证和拦截器

一. OpenSSL

        1.1 介绍

        OpenSSL是一个开放源代码的软件库包，用于支持网络通讯过程中的加密。这个库提供的功能包含了SSL和TLS协议的实现，并可用于生成密钥、证书、进行密码运算等。

        其组成主要包括一下三个组件：

  1. openssl：多用途的命令行工具

  2. libcrypto：加密算法库

  3. libssl：加密模块应用库，实现了ssl及tls

openssl可以实现秘钥证书管理、对称加密和非对称加密 。

        官网：[ Downloads ] - /source/index.html

        1.2 Windows安装方法

        OpenSSL官网没有提供windows版本的安装包，可以选择其它开源平台提供的工具。

Win32/Win64 OpenSSL Installer for Windows - Shining Light Productions

        以该工具为例：

        进入下载界面，选择下载的版本，下载完，之后安装即可。

        1.3 生成公钥和私钥

        openssl命令详解-CSDN博客

生成私钥：openssl genrsa -out rsa_private_key.pem 1024
生成公钥：openssl rsa -in rsa_private_key.pem -pubout -out rsa_public_key.pem

二. gRPC认证

        gRPC默认内置了两种认证方式：

  • SSL/TLS认证
  • 基于Token的认证

        同时，gRPC提供了接口用于扩展自定义认证方式。

        2.1 TLS认证

        2.1.1 什么是TLS认证

        TLS（Transport Layer Security，安全传输层)，TLS是建立在传输层TCP协议之上的协议，服务于应用层，它的前身是SSL（Secure Socket Layer，安全套接字层），它实现了将应用层的报文进行加密后再交由TCP进行传输的功能。

        2.1.2 TLS的作用

TLS协议主要解决如下三个网络安全问题。

  • 保密(message privacy)，保密通过加密encryption实现，所有信息都加密传输，第三方无法嗅探；
  • 完整性(message integrity)，通过MAC校验机制，一旦被篡改，通信双方会立刻发现；
  • 认证(mutual authentication)，双方认证,双方都可以配备证书，防止身份被冒充；

        2.1.3 TLS认证实例

  • 证书制作

        制作公钥：自签名公钥(x509)，   制作私钥。

#生成一个名为server_private.key的RSA私钥，使用SHA256算法和4096位密钥长度。然后使用该私钥生成一个有效期为36500天的自签名证书，并将其保存为名为server.pem的文件。同时在证书中添加subjectAltName扩展，指定DNS名称为www.wy.com。
openssl req -newkey rsa:4096 -nodes -sha256 -keyout server_private.key -x509 -days 36500 -out server.pem -addext "subjectAltName =DNS:www.wy.com"
  •  openssl req：生成自签名证书
  • -newkey rsa:4096 ：生成新的4096位rsa密钥对
  • -sha256：使用sha256加密
  • -keyout：指定生成的私钥文件
  • -x509：指输出证书
  • -days 36500：有效期 36500
  • -out：输出证书的文件名
  • -addext：添加扩展

        注意需要在证书中添加subjectAltName扩展，指定DNS名称。不然在客户端连接服务器时会报错，报错信息为：

        rpc error: code = Unavailable desc = connection error: desc = "transport: authentication handshake failed: tls: failed to verify certificate: x509: certificate relies on legacy Common Name field, use SANs instead"

        因为go1.15 版本开始废弃CommonName，因此推荐使用SAN证书。如果想兼容之前的方式，需要设置环境变量 GODEBUG为 x509ignoreCN=0。(创建 SSL/TLS 证书时，证书依赖于传统的 Common Name (CN) 字段，而没有使用现代标准所推荐的 Subject Alternative Names (SANs) 字段。现代的 TLS 客户端（比如最新版本的浏览器和安全工具）要求证书使用 SANs 字段来指定有效的主机名。)

        自定义信息： 

You are about to be asked to enter information that will be incorporated
into your certificate request.
What you are about to enter is what is called a Distinguished Name or a DN.
There are quite a few fields but you can leave some blank
For some fields there will be a default value,
If you enter '.', the field will be left blank.
-----
Country Name (2 letter code) [AU]:CN #国家
State or Province Name (full name) [Some-State]:SHANGHAI #省份
Locality Name (eg, city) []:SHANGHAI #城市
Organization Name (eg, company) [Internet Widgits Pty Ltd]:BF #公司
Organizational Unit Name (eg, section) []:Dev #部门
Common Name (e.g. server FQDN or YOUR name) []:www.wy.com #服务器名称
Email Address []:xxx@xxx.com #邮箱地址
  • 目录结构

  • 示例代码 

         服务端代码：

package mainimport ("context""fmt""net"hello "sample-app/grpc/proto""google.golang.org/grpc""google.golang.org/grpc/credentials" //引入gRPC认证包
)const (//服务器地址Addr = "127.0.0.1:8080"
)type helloService struct{}//定义hello 服务
var HelloService = helloService{}//实现proto hello service方法
func (h helloService) SayHello(c context.Context, req *hello.HelloRequest) (*hello.HelloResponse, error) {resp := new(hello.HelloResponse)resp.Message = fmt.Sprintf("Hello %s", req.Name)return resp, nil
}func main() {ls, err := net.Listen("tcp", Addr)if err != nil {fmt.Println(err)return}//TLS认证cert, err := credentials.NewServerTLSFromFile("..\\..\\key\\server.pem", "..\\..\\key\\server_private.key")if err != nil {fmt.Println(err)return}//新建一个grpc服务器，并开启TLS认证//上面监听并没有进行连接客户端server := grpc.NewServer(grpc.Creds(cert))//注册HelloServicehello.RegisterHelloServer(server, HelloService)fmt.Println("Listen on" + Addr + "with TLS")//这里面才会连接客户端，需要进行认证server.Serve(ls)
}

  • credentials.NewServerTLSFromFile：从输入证书文件和密钥文件为服务端构造TLS凭证

  • grpc.Creds：返回一个ServerOption，用于设置服务器连接的凭证。

        客户端代码：

package mainimport ("context""fmt"hello "sample-app/grpc/proto""google.golang.org/grpc""google.golang.org/grpc/credentials"
)const (、//gRPC服务器地址Addr = "127.0.0.1:8080"
)func main() {//TLS连接cert, err := credentials.NewClientTLSFromFile("..\\..\\key\\server.pem", "www.wy.com")if err != nil {fmt.Println("credentials fail ", err)return}//请求连接的时候 需要认证conn, err := grpc.Dial(Addr, grpc.WithTransportCredentials(cert))if err != nil {fmt.Println("Dial fail", err)return}defer conn.Close()c := hello.NewHelloClient(conn)req := new(hello.HelloRequest)req.Name = "gRPC"resp, err := c.SayHello(context.Background(), req)if err != nil {fmt.Println("say hello fail", err)return}fmt.Println(resp.Message)
}

  • credentials.NewClientTLSFromFile：从输入的证书文件中为客户端构造TLS凭证。

  • grpc.WithTransportCredentials：配置连接级别的安全凭证（例如，TLS/SSL），返回一个DialOption，用于连接服务器。

        proto文件：

syntax="proto3";
package hello;
option go_package="hello";service Hello
{rpc SayHello(HelloRequest)returns(HelloResponse){}; 
}message HelloRequest
{string name = 1;
}message HelloResponse
{string message = 1;
}

使用下面命令生成pb.go文件：

protoc --go_out=plugins=grpc:"生成pb.go文件地址" -I="proto文件地址" "proto文件地址\文件"

演示：

        实际TLS认证不是这样，客户端和服务器时分离的。客户端有证书(包含公钥)，服务端有证书和私钥。

        客户端发送请求给服务器请求连接，服务器将证书通过私钥加密后发送给客户端。客户端有证书，里面包含服务器私钥对应的公钥。使用公钥对数据进行解密，获得证书数据，与本地证书数据进行比较。

         2.2 Token认证

        继续扩展上面的代码，实现TLS+Token认证机制。

        2.2.1 什么是Token认证

        Token认证是一种基于Token的身份验证方法，用于在客户端和服务器之间进行身份验证。以下是Token认证的主要概念、流程以及优缺点：

  • 主要概念

    • Token的含义：Token（令牌）是服务端生成的一串字符串，作为客户端进行请求的一个标识。
    • Token的组成：一般包括用户身份标识（uid）、时间戳（time）和签名（sign）等元素。
    • Token的作用：Token主要用于身份验证、授权、会话管理和跨域资源共享（CORS）等方面。

  • 认证流程

    • 用户登录并获取Token：用户使用用户名和密码登录，成功后服务端生成Token并发送给客户端。
    • 客户端存储和使用Token：客户端将Token保存在本地（如cookie或localStorage），并在后续请求中携带该Token。
    • 服务端验证Token：服务端收到请求后，验证Token的合法性，若合法则处理请求并返回数据。

        2.2.2 示例代码

        根据上面的代码，实现TLS+Token认证机制。

  • 认证原理

        客户端发送请求，会将Token放到context.Context上下文中，服务器收到请求，从上下文中获取Token验证，然后进行下一步操作。

  • 目录结构

  • 客户端代码

        grpc/credential包内默认定义了PerRPCCredentials接口，是提供用于自定义接口，他的作用是将所需安全认证信息添加到每个RPC上下文中。其包含两个方法。

type PerRPCCredentials interface {//获取当前请求认证所需的元数据GetRequestMetadata(ctx context.Context, uri ...string) (map[string]string, error)//是否需要基于TLS认证进行安全传输RequireTransportSecurity() bool
}

package mainimport ("context""fmt"hello "sample-app/grpc/proto""google.golang.org/grpc""google.golang.org/grpc/credentials"
)const (Addr = "127.0.0.1:8080"//是否使用TLSOpenTLS = true
)// 自定义认证
type Token struct {Appid  stringAppkey string
}// 实现自定义认证方法
func (t Token) GetRequestMetadata(ctx context.Context, uri ...string) (map[string]string, error) {return map[string]string{"appid":  t.Appid,"appkey": t.Appkey,}, nil
}// 实现自定义认证是否开启TLS
func (t Token) RequireTransportSecurity() bool {return OpenTLS
}func main() {//TLS连接var opt []grpc.DialOptionif OpenTLS {cert, err := credentials.NewClientTLSFromFile("..\\..\\key\\server.pem", "www.wy.com")if err != nil {fmt.Println("credentials fail ", err)return}opt = append(opt, grpc.WithTransportCredentials(cert))} else {opt = append(opt, grpc.WithInsecure())}//使用自定义认证tk := Token{Appid:  "101010",Appkey: "i am a key",}opt = append(opt, grpc.WithPerRPCCredentials(&tk))conn, err := grpc.Dial(Addr, opt...)if err != nil {fmt.Println("Dial fail", err)return}defer conn.Close()//初始化服务器c := hello.NewHelloClient(conn)req := new(hello.HelloRequest)req.Name = "gRPC"resp, err := c.SayHello(context.Background(), req)if err != nil {fmt.Println("say hello fail", err)return}fmt.Println(resp.Message)
}

        定义一个结构Token，包含Token所需属性字段。实现PerRPCCredentials接口的两个方法。每次调用token信息会通过请求metadata传输到服务端。

        下面查看服务端如何获取metadata中信息。 

  • 服务端代码

        使用metadata.FromIncomingContext：从上下文中获取元数据。

package mainimport ("context""fmt""net"hello "sample-app/grpc/proto""google.golang.org/grpc""google.golang.org/grpc/codes""google.golang.org/grpc/credentials" //引入gRPC认证包"google.golang.org/grpc/metadata"
)var Addr = "127.0.0.1:8080"type helloService struct{}var HelloService = helloService{}func (h helloService) SayHello(c context.Context, req *hello.HelloRequest) (*hello.HelloResponse, error) {//认证md, ok := metadata.FromIncomingContext(c)if !ok {return nil, grpc.Errorf(codes.Unauthenticated, "无Token认证信息")}var appid stringvar appkey stringvals := md.Get("appid")if len(vals) != 0 {appid = vals[0]}val_key := md.Get("appkey")if len(val_key) != 0 {appkey = val_key[0]}//认证tokenif appid != "101010" || appkey != "i am a key" {return nil, grpc.Errorf(codes.Unauthenticated, "Token认证信息错误: Appid:%s, Appkey:%s", appid, appkey)}//fmt.Println("authenticated succ " + appid + "-" + appkey)resp := new(hello.HelloResponse)resp.Message = fmt.Sprintf("Hello %s \nToken info: Appid=%s, AppKey=%s", req.Name, appid, appkey)return resp, nil
}func main() {ls, err := net.Listen("tcp", Addr)if err != nil {fmt.Println(err)return}//TLS认证cert, err := credentials.NewServerTLSFromFile("..\\..\\key\\server.pem", "..\\..\\key\\server_private.key")if err != nil {fmt.Println(err)return}//新建一个grpc服务器，并开启TLS认证server := grpc.NewServer(grpc.Creds(cert))//注册HelloServicehello.RegisterHelloServer(server, HelloService)fmt.Println("Listen on " + Addr + " with TLS")server.Serve(ls)
}
  • 演示 

        成功： 

         失败：

        补充：

        google.golang.org/grpc/credentials/oauth 包已实现了用于 Google API 的 oauth 和 jwt 验证
的方法，使用方法可以参考 官方文档 。在实际应用中，我们可以根据自己的业务需求实现合适的验证方 式。

三. 拦截器

        grpc服务端和客户端都提供了interceptor功能，可以在请求前后处理一些通用逻辑，比如：记录日志，tracing，身份认证等。

        在上面自定义Token认证的示例中，认证信息是由每个服务中的方法处理并认证，如果有大量的接口，这种姿势就不优雅了，每一个接口实现都需要先处理认证信息。这个时候interceptor就可以用来解决这个问题，在请求被转到具体接口之前处理认证信息，一处认证到处无忧。在客户端，我们增加一个请求日志，记录请求相关的参数和耗时等。

        3.1 grpc的interceptor

        gRPC服务端和客户端均可实现各自的拦截器，根据rpc的两种请求方式可分为两种：

  • Unary Interceptor(一元拦截器)
  • Stream Interceptor(流式拦截器)

        rpc的两种请求方式：

  • 一元请求(Unary)：客户端发送一个请求给服务端，然后立即得到一个响应，但是服务器端并不能主动向客户端发送消息。
  • 流式请求：当数据量比较大，可能会对响应时间产生影响，可以使用流式来分批次来传输数据，不需要一次性就处理完数据。
    • 服务端流式请求(server streaming)：客户端发送一个请求给服务端，然后服务端就会向客户端逐渐返回一系列的消息。
    • 客户端流式请求(client streaming)：客户端就像一个流，连续发送多个消息给服务端，然后等待服务端的一个响应。
    • 双向流式请求（Bidirectional streaming）：客户端和服务端可以连续互发消息。

对应Protobuf文件service定义：

        下面是一个例子：

        一元请求：request和response不需要stream。

        服务端流式请求：request不需要stream，response需要stream

        客户端流式请求：request需要stream，response不需要stream。

        双向流式请求：request和response都需要stream。

syntax = "proto3";option go_package = "github.com/lixd/grpc-go-example/features/proto/echo";package echo;// Echo 服务，包含了4种类型API
service Echo {// UnaryAPIrpc UnaryEcho(EchoRequest) returns (EchoResponse) {}// SServerStreamingrpc ServerStreamingEcho(EchoRequest) returns (stream EchoResponse) {}// ClientStreamingErpc ClientStreamingEcho(stream EchoRequest) returns (EchoResponse) {}// BidirectionalStreamingrpc BidirectionalStreamingEcho(stream EchoRequest) returns (stream EchoResponse) {}
}message EchoRequest {string message = 1;
}message EchoResponse {string message = 1;
}

        3.2 一元拦截器

        对于一元服务器拦截器，只需要定义UnaryServerInterceptor方法即可。其中，handler(ctx, req)即调用rpc方法。

type UnaryServerInterceptor func(ctx context.Context,   //rpc上下文req interface{},       //rpc请求参数info *UnaryServerInfo, //rpc方法信息handler UnaryHandler,  //rpc方法本身，真正执行逻辑
)(interface{}, error){return handler(ctx, req)
}

        对于一元客户端拦截器，一样需要定义一个方法UnaryClientInterceptor，其中invoker()才真正请求rpc。

type UnaryClientInterceptor func(ctx context.Context,  //rpc上下文method string,        //调用方法名req,                  //rpc请求参数reply interface{},    //rpc响应结果cc *ClientConn,       //连接句柄invoker UnaryInvoker, //调用rpc本身方法   opts ...CallOption    //调用配置
) error {return invoker(ctx, method, req, reply, cc, opts...)
}

        一元拦截器的实现，根据handle和invoker的前后：调用前预处理，调用rpc方法，调用后处理。

  • 服务端代码
package mainimport ("context""fmt""net"hello "sample-app/grpc/proto""google.golang.org/grpc""google.golang.org/grpc/codes""google.golang.org/grpc/credentials" //引入gRPC认证包"google.golang.org/grpc/metadata"
)var Addr = "127.0.0.1:8080"type helloService struct{}var HelloService = helloService{}func (h helloService) SayHello(c context.Context, req *hello.HelloRequest) (*hello.HelloResponse, error) {//fmt.Println("authenticated succ " + appid + "-" + appkey)resp := new(hello.HelloResponse)resp.Message = fmt.Sprintf("Hello %s", req.Name)return resp, nil
}func main() {ls, err := net.Listen("tcp", Addr)if err != nil {fmt.Println(err)return}//TLS认证cert, err := credentials.NewServerTLSFromFile("..\\..\\key\\server.pem", "..\\..\\key\\server_private.key")if err != nil {fmt.Println(err)return}var opts []grpc.ServerOption//开启TLS认证opts = append(opts, grpc.Creds(cert))//注册拦截器opts = append(opts, grpc.UnaryInterceptor(UnaryServerInterceptor))//server := grpc.NewServer(grpc.Creds(cert))server := grpc.NewServer(opts...)//注册HelloServicehello.RegisterHelloServer(server, HelloService)fmt.Println("Listen on " + Addr + " with TLS " + "Interceptor")server.Serve(ls)
}func auth(ctx context.Context) error {md, ok := metadata.FromIncomingContext(ctx)if !ok {return grpc.Errorf(codes.Unauthenticated, "无Token认证信息")}var appid stringvar appkey stringvals := md.Get("appid")if len(vals) != 0 {appid = vals[0]}val_key := md.Get("appkey")if len(val_key) != 0 {appkey = val_key[0]}//认证tokenif appid != "101010" || appkey != "i am a key" {return grpc.Errorf(codes.Unauthenticated, "Token认证信息错误: Appid:%s, Appkey:%s", appid, appkey)}return nil
}func UnaryServerInterceptor(ctx context.Context, req interface{}, into *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp interface{}, err error) {err = auth(ctx)if err != nil {return nil, err}return handler(ctx, req)
}
  • 客户端代码
package mainimport ("context""fmt""log"hello "sample-app/grpc/proto""time""google.golang.org/grpc""google.golang.org/grpc/credentials"
)const (Addr = "127.0.0.1:8080"//是否使用TLSOpenTLS = true
)// 自定义认证
type Token struct {Appid  stringAppkey string
}// 实现自定义认证方法
func (t Token) GetRequestMetadata(ctx context.Context, uri ...string) (map[string]string, error) {return map[string]string{"appid":  t.Appid,"appkey": t.Appkey,}, nil
}// 实现自定义认证是否开启TLS
func (t Token) RequireTransportSecurity() bool {return OpenTLS
}func main() {//TLS连接var opt []grpc.DialOptionif OpenTLS {cert, err := credentials.NewClientTLSFromFile("..\\..\\key\\server.pem", "www.wy.com")if err != nil {fmt.Println("credentials fail ", err)return}opt = append(opt, grpc.WithTransportCredentials(cert))} else {opt = append(opt, grpc.WithInsecure())}//使用自定义认证tk := Token{Appid:  "101010",Appkey: "i am not a key",}opt = append(opt, grpc.WithPerRPCCredentials(&tk))//加入拦截器opt = append(opt, grpc.WithUnaryInterceptor(UnaryClientInterceptor))conn, err := grpc.Dial(Addr, opt...)if err != nil {fmt.Println("Dial fail", err)return}defer conn.Close()//初始化服务器c := hello.NewHelloClient(conn)req := new(hello.HelloRequest)req.Name = "gRPC"resp, err := c.SayHello(context.Background(), req)if err != nil {fmt.Println("say hello fail", err)return}fmt.Println(resp.Message)
}func UnaryClientInterceptor(ctx context.Context, method string, req, reply interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {stat_time := time.Now()err := invoker(ctx, method, req, reply, cc, opts...)log.Printf("method=%s, req=%v, reply=%v, duration=%s, error=%v\n", method, req, reply, time.Since(stat_time), err)return err
}
  • 运行结果

认证成功：

认证失败：

         3.3 流式拦截器

        流式拦截器的实现与一元拦截器一致，实现提供的方法即可，方法参数含义如下：

type StreamServerInterceptor func(srv interface{},        //rpc请求参数ss ServerStream,        //服务端stream对象info *StreamServerInfo, //rpc方法信息handler StreamHandler   //rpc方法本身，真正执行逻辑
) (err error) {return handler(src, ss)
}
type StreamClientInterceptor func(ctx context.Context,//rpc上下文desc *StreamDesc,   //流信息cc *ClientConn,     //连接句柄method string,      //调用方法名streamer Streamer,  //调用rpc方法本身opts ...CallOption  //调用配置	    
) (ClientStream, error) {//流操作预处理clientStream, err := streamer(ctx, desc, cc, method, opts...)//根据某些条件，通过clientStream拦截流操作return clientStream, err
}

        与其他拦截器不同，客户端流式拦截器的实现分为两个部分，流操作预处理和流操作拦截，其不能在事后进行rpc方法调用，只能通过ClientStream对象进行操作拦截。即需要进行rpc方法调用后，才能进行操作拦截。例如根据特定的metadata，调用ClientStream.CloseSend()终止流操作。

        下面实现了一个打印请求和响应日志的拦截器，只是函数签名变成了grpc.StreamServerInterceptor 和 grpc.StreamClientInterceptor 。

  • proto文件
syntax="proto3";
package hello;
option go_package="hello";service Hello
{rpc SayHello(HelloRequest)returns(HelloResponse){}; //双向流式请求rpc Streaming(stream StreamRequest) returns (s StreamResponse) {}
}message HelloRequest
{string name = 1;
}message HelloResponse
{string message = 1;
}message StreamRequest
{string input = 1;
}message StreamResponse
{string output = 1;
}

使用下面命令生成go语言文件：

protoc --go_out=plugins=grpc:.\grpc\proto -I=.\grpc\proto .\grpc\proto\hello.proto
  • 服务端代码

        服务端定义结构体，实现了proto文件HelloServer接口，即实现了SayHello和Streaming方法。

        实现了拦截器方法StreamServerInterceptor(函数名随便定义)。

        服务端的实现其实和一元拦截器的使用方式没什么太大区别，但是流的特性在于请求和响应不是一次性完成的，而是多次发送和接收数据。所以我们可能需要在发送和接收数据的过程中处理一些公共逻辑。这才是拦截器特别的地方。

        我们注意到handler方法调用的第二个参数grpc.ServerStream接口类型，这个类型包含了SendMsg和RecvMsg方法，所以我们可以使用一个自定义类型实现这个接口，并重写这两个方法，我们就可以实现打印日志的目的。

package mainimport ("context""fmt""io""log""net"hello "sample-app/grpc/proto""strconv""google.golang.org/grpc""google.golang.org/grpc/codes""google.golang.org/grpc/credentials" //引入gRPC认证包"google.golang.org/grpc/metadata"
)var Addr = "127.0.0.1:8080"type helloService struct{}var HelloService = helloService{}func (h helloService) SayHello(c context.Context, req *hello.HelloRequest) (*hello.HelloResponse, error) {resp := new(hello.HelloResponse)resp.Message = fmt.Sprintf("Hello %s", req.Name)return resp, nil
}func (h helloService) Streaming(stream hello.Hello_StreamingServer) error {for n := 0; ; {res, err := stream.Recv()if err == io.EOF {return nil}if err != nil {return err}v, _ := strconv.Atoi(res.Input)log.Printf("[server streaming] recv : %d\n", v)n += vlog.Printf("[server streaming] send : %d\n", n)stream.Send(&hello.StreamResponse{Output: strconv.Itoa(n)})}
}func main() {ls, err := net.Listen("tcp", Addr)if err != nil {fmt.Println(err)return}//TLS认证cert, err := credentials.NewServerTLSFromFile("..\\..\\key\\server.pem", "..\\..\\key\\server_private.key")if err != nil {fmt.Println(err)return}var opts []grpc.ServerOption//开启TLS认证opts = append(opts, grpc.Creds(cert))//注册拦截器opts = append(opts, grpc.StreamInterceptor(StreamServerInterceptor))//server := grpc.NewServer(grpc.Creds(cert))server := grpc.NewServer(opts...)//注册HelloServicehello.RegisterHelloServer(server, HelloService)fmt.Println("Listen on " + Addr + " with TLS " + "with StreamInterceptor")server.Serve(ls)
}// 里面定义grpc.ServerStream接口类型的属性
// 是为了重写SendMsg和RecvMsg方法
type serverStream struct {//需要使用匿名字段//内嵌方法重写grpc.ServerStream
}// 重写ServerStream的SendMsg方法
func (s serverStream) SendMsg(m interface{}) error {//发送数据前处理log.Printf("[server SendMsg]: send : %T\n", m)return s.ServerStream.SendMsg(m)
}// 重写ServerStream的RecvMsg方法
func (s serverStream) RecvMsg(m interface{}) error {//接收数据前处理log.Printf("[server Recv Stream]: recv : %T\n", m)return s.ServerStream.RecvMsg(m)
}// 流式拦截器
func StreamServerInterceptor(srv interface{}, ss grpc.ServerStream, info *grpc.StreamServerInfo, handler grpc.StreamHandler) error {//前置逻辑log.Printf("[StreamServerInterceptor] accept request : %s\n", info.FullMethod)arg := serverStream{ss}err := handler(srv, arg)return err
}
  • 客户端代码

        客户端代码和服务端代码类似，只是对应数据处理的接口变成了grpc.ClientStream。

package mainimport ("context""fmt""io""log"hello "sample-app/grpc/proto""strconv""google.golang.org/grpc""google.golang.org/grpc/credentials"
)const (Addr = "127.0.0.1:8080"//是否使用TLSOpenTLS = true
)func main() {//TLS连接var opt []grpc.DialOptionif OpenTLS {cert, err := credentials.NewClientTLSFromFile("..\\..\\key\\server.pem", "www.wy.com")if err != nil {fmt.Println("credentials fail ", err)return}opt = append(opt, grpc.WithTransportCredentials(cert))} else {opt = append(opt, grpc.WithInsecure())}//加入流式拦截器opt = append(opt, grpc.WithStreamInterceptor(StreamClientInterceptor))conn, err := grpc.Dial(Addr, opt...)if err != nil {fmt.Println("Dial fail", err)return}defer conn.Close()//初始化服务器c := hello.NewHelloClient(conn)//单项请求req := new(hello.HelloRequest)req.Name = "gRPC"resp, err := c.SayHello(context.Background(), req)if err != nil {fmt.Println("say hello fail", err)return}fmt.Println(resp.Message)//流式发送数据Streaming(c)
}type clientStream struct {grpc.ClientStream
}func (c clientStream) SendMsg(m interface{}) error {log.Printf("[client SendMsg] send : %T\n", m)return c.ClientStream.SendMsg(m)
}func (c clientStream) RecvMsg(m interface{}) error {log.Printf("[client RecvMsg] recv : %T\n", m)return c.ClientStream.RecvMsg(m)
}// 拦截器方法
func StreamClientInterceptor(ctx context.Context, desc *grpc.StreamDesc, cc *grpc.ClientConn, method string, streamer grpc.Streamer, opts ...grpc.CallOption) (grpc.ClientStream, error) {//前置逻辑log.Printf("[StreamClientInterceptor] send req : %s\n", method)//请求s, err := streamer(ctx, desc, cc, method, opts...)if err != nil {return nil, err}return clientStream{s}, nil
}// 流式发送
func Streaming(pb hello.HelloClient) error {stream, err := pb.Streaming(context.Background())if err != nil {return err}for n := 0; n < 5; n++ {log.Printf("[client Streaming] send : %d\n", n)err = stream.Send(&hello.StreamRequest{Input: strconv.Itoa(n)})if err != nil {return err}resp, err := stream.Recv()//发送完毕，退出if err == io.EOF {return nil}if err != nil {return err}log.Printf("[client Streaming] recv : %s\n", resp.Output)}//停止发送stream.CloseSend()return nil
}
  • 演示

        注意点：

  • server和client的recv和send互成一对，最后一次输出recv是结束消息，err==io.EOF
  • 在自定义的RecvMsg方法中，前置位置只能读取消息的类型，而无法读取消息的实际数据，因为这个时候收到的消息还没有解析，如果需要接收消息的实际数据，需要把自定义的处理逻辑放后面。
func (s serverStream) RecvMsg(m interface{}) error {err := s.ServerStream.RecvMsg(m)log.Printf("[server Recv Stream]: recv : %T\n", m)return err
}
        项目推荐：  go-grpc-middleware
        这个项目对interceptor 进行了封装，支持多个拦截器的链式组装，对于需要多种处理的地方使用起 来会更方便些。

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.mzph.cn/pingmian/29832.shtml

如若内容造成侵权/违法违规/事实不符，请联系多彩编程网进行投诉反馈email:809451989@qq.com，一经查实，立即删除！

相关文章

有效招聘营销策略的六个组成部分

有效招聘营销策略的六个组成部分

任何想吸引更多人购买其产品的公司都必须投资于市场营销。然而&#xff0c;当涉及到让更多的人了解公司的工作时&#xff0c;许多有效的营销活动可能不是招聘团队的首要考虑因素。为了超越招聘委员会上的“发布祈祷”策略&#xff0c;有必要包括有效招聘营销策略的所有组成部分…
阅读更多...
车联网车载设备

车联网车载设备

智能网联主要通过OBU&#xff08;On Board Unit,车载单元&#xff09;实现。OBU是一种安装在车辆上用于实现V2X通信的硬件设备&#xff0c;可实现和其他车辆OBU&#xff08;PC5&#xff09;、路侧RSU&#xff08;PC5&#xff09;、行人&#xff08;PC5&#xff09;和V2X平台&am…
阅读更多...
基于Redis实现共享session登录

基于Redis实现共享session登录

搭配食用&#xff1a;Redis&#xff08;基础篇&#xff09;-CSDN博客 项目实现前的 Mysql中的表&#xff1a; 表说明tb_user用户表tb_user_info用户详情表tb_shop商户信息表tb_shop_type商户类型表tb_blog用户日记表&#xff08;达人探店日记)tb_follow用户关注表tb_voucher优…
阅读更多...
if/case条件测试语句

if/case条件测试语句

一 条件测试 1.1返回码 $? $? 返回码 用来哦按段命令或者脚本是否执行成功 0 true为真就是成功成立 非0 false 失败或者异常 1.2 test 命令 可以进行条件测试 然后根据返回值来判断条件是否成立 -e &#xff1a;exist 测试目录或者目录是否存在 -d : director…
阅读更多...
【产品经理】订单处理6-审单方案

【产品经理】订单处理6-审单方案

电商系统中订单管理员会对特殊类型的订单进行审核&#xff0c;普通订单则自动审核&#xff0c;本节讲述自动审单方案、手动审单以及加急审单。 一、自动审单 自动审单方案可按照方案形式制定&#xff0c;可一次性制定多套审单方案。 1. 审单通过条件有 执行店铺&#xff…
阅读更多...
同三维T80006EHL-4K30CN 单路4K30 HDMI编码器（全国产化）

同三维T80006EHL-4K30CN 单路4K30 HDMI编码器（全国产化）

同三维T80006EHL-4K30CN 单路4K30 HDMI编码器 带1路HDMI环出和1路3.5音频输入&#xff0c;支持4K30&#xff0c;所有元器件全国产 一、 产品简介&#xff1a; T80006EHL-4K30CN 4K编码器&#xff08;采集盒&#xff09;是一款全国产化的专业4K HDMI音视频编码产品&#xff0c;…
阅读更多...
2024年先进机械电子、电气工程与自动化国际学术会议(ICAMEEA 2024)

2024年先进机械电子、电气工程与自动化国际学术会议(ICAMEEA 2024)

2024年先进机械电子、电气工程与自动化国际学术会议(ICAMEEA 2024) 2024 International Conference on Advanced Mechatronic, Electrical Engineering and Automation 会议地点&#xff1a;杭州&#xff0c;中国 网址&#xff1a;www.icameea.com 邮箱: icameeasub-conf.c…
阅读更多...
5.How Fast Should You Be When Learning?(你应该用多快的速度学习?)

5.How Fast Should You Be When Learning?(你应该用多快的速度学习?)

Normally when I talk about learing quickly, I’m using speed as a synonym for efficiency.Use more effective methods and you’ll learn more in less time.All else being equal, that means you’re learing faster. 通常我在谈到快速学习时&#xff0c;是把“速度&qu…
阅读更多...
【自记录】记一台i7四代老爷笔记本重装Win10 22H2

【自记录】记一台i7四代老爷笔记本重装Win10 22H2

1 背景 手上有一台退役的老爷笔记本&#xff0c;清华同方锋锐 U430。因为比较轻便&#xff0c;最近重新翻出来用于临时抓个包和简单的脚本语言&#xff08;比方说Python&#xff09;编辑工作。但是现在像VSCode的新版本都要求Win10以上&#xff0c;像Continue这类的AI插件都跑…
阅读更多...
ssh-add id_rsa_gitlab1 Error connecting to agent: No such file or directory

ssh-add id_rsa_gitlab1 Error connecting to agent: No such file or directory

ssh-add id_rsa_gitlab1 Error connecting to agent: No such file or directory 目录 ssh-add id_rsa_gitlab1 Error connecting to agent: No such file or directory1. 启动 SSH 代理2. 添加 SSH 密钥3. 使用 Git Bash 或其他终端4. 使用 Pageant&#xff08;适用于 PuTTY 用…
阅读更多...
基于DDD的编码实践

基于DDD的编码实践

分层设计 领域驱动设计&#xff08;Domain-driven design, DDD&#xff09; 作为一种复杂软件系统的应对方案&#xff0c;在设计和编码提供了一种新的解决方式&#xff0c;即领域驱动&#xff0c;要求程序员在设计和编码时从领域专家的角度出发来实现架构/代码&#xff0c;做到…
阅读更多...
工作随机：oracle集群下的vip intermediate，failed over处理

工作随机：oracle集群下的vip intermediate，failed over处理

文章目录 前言一、问题排查二、恢复db2使用1.确认db2 vip状态2.恢复db2 的vip3.检查监听&#xff1a; 前言 在对数据库进行巡检发现&#xff0c;集群中一个节点的备份没有执行&#xff0c;未生成当天的任何日志&#xff0c;查询/var/spool/oracle 信息发现提示&#xff1a;no …
阅读更多...
经验分享，在线文本比较工具

经验分享，在线文本比较工具

这里分享一个在线文本比较工具&#xff0c;打开网页即用&#xff0c;很方便 网址&#xff1a; https://www.jq22.com/textDifference 截图&#xff1a;
阅读更多...
Python语言在地球科学交叉领域中的实践技术融合应用

Python语言在地球科学交叉领域中的实践技术融合应用

Python是功能强大、免费、开源&#xff0c;实现面向对象的编程语言&#xff0c;Python能够运行在Linux、Windows、Macintosh、AIX操作系统上及不同平台&#xff08;x86和arm&#xff09;&#xff0c;Python简洁的语法和对动态输入的支持&#xff0c;再加上解释性语言的本质&…
阅读更多...
代码审计中XSS挖掘一些体会

代码审计中XSS挖掘一些体会

0x01 XSS的挖掘思路 1.1 反射型 直接搜索 echo print_r print之类的函数即可 也可以寻找$_GET变量来判断是否存在输出&#xff08;不过对于代码审计来说除非实在挖不出漏洞&#xff0c;否则没必要关注反射xss&#xff09; 1.2 dom型 和反射型差不多 需要看网站的前端javascr…
阅读更多...
openGauss 6.0一主二备高可用架构部署，可靠很行

openGauss 6.0一主二备高可用架构部署，可靠很行

作者&#xff1a;IT邦德 中国DBA联盟(ACDU)成员&#xff0c;10余年DBA工作经验&#xff0c; Oracle、PostgreSQL ACE CSDN博客专家及B站知名UP主&#xff0c;全网粉丝10万 擅长主流Oracle、MySQL、PG、高斯及Greenplum备份恢复&#xff0c; 安装迁移&#xff0c;性能优化、故障…
阅读更多...
MySQL JDBC驱动包引入有版本要求吗

MySQL JDBC驱动包引入有版本要求吗

提示&#xff1a;有关数据库的任何操作&#xff0c;请事先都做好备份&#xff0c;一定不会错的&#xff1b; 文章目录 前言一、com.mysql.jdbc.Driver和com.mysql.cj.jdbc.Driver如何选择&#xff1f;1、概念2、引入驱动3、总结 前言 新老项目的交替中&#xff0c;如果你使用的…
阅读更多...
【并发编程实战】内存模型--解决可见性和有序性的利器

【并发编程实战】内存模型--解决可见性和有序性的利器

一.前言 在前面讲了三个问题&#xff0c; 缓存导致的可见性问题&#xff0c;编译优化带来的有序性问题&#xff0c;线程切换带来的原子性问题。既然存在问题&#xff0c;那么总要有解决方案的&#xff0c;这一章里主要就是解决这三个问题的关键点--内存模型 二.内存模型 2.1 …
阅读更多...
hdfs高可用文件系统架构

hdfs高可用文件系统架构

1、整体架构 2、角色简介 2.1、namenode NameNode 是 HDFS 集群中的核心组件&#xff0c;负责管理文件系统的元数据、处理客户端请求、管理数据块、确保数据完整性和高可用性。由于其重要性&#xff0c;NameNode 的性能和可靠性直接影响整个 HDFS 集群的性能和可靠性。在生产…
阅读更多...
【漏洞复现】CRMEB开源电商系统 /api/products SQL注入漏洞(CVE-2024-36837)

【漏洞复现】CRMEB开源电商系统 /api/products SQL注入漏洞(CVE-2024-36837)

0x01 产品简介 CRMEB开源电商系统是一款由西安众邦网络科技有限公司打造的全栈式电商解决方案&#xff0c;旨在为开发者和商家提供高性能、智能化的电商平台服务。该系统集成了CRM(客户关系管理)、ERP(企业资源规划)和EB(电子商务)的功能&#xff0c;通过深度结合这些功能&…
阅读更多...
推荐文章
最新文章

Copyright @ 2022~2023