grpc-middleware实现grpc调用重试

proto

syntax = "proto3";
option go_package = ".;proto";
package proto;
service Greeter {
  rpc SayHello (HelloRequest) returns (HelloResponse);
}

message HelloRequest {
  string name = 1;
}

message HelloResponse {
  string message = 1;
}

生成对应的go代码

protoc -I . demo.proto --go_out=plugins=grpc:.

server

package main

import (
	"ShopBefore/rpc/grpc_retry/proto"
	"context"
	"google.golang.org/grpc"
	"log"
	"net"
	"time"
)

type Server struct{
    }

func (s Server) SayHello(ctx context.Context, request *proto.HelloRequest) (*proto.HelloResponse, error) {
    
	log.Println("收到请求.....")
	// 设置一秒超时,再接口这里休眠两秒
	time.Sleep(2 * time.Second)
	return &proto.HelloResponse{
    Message: "Hello" + request.Name}, nil
}

func main() {
    
	server := grpc.NewServer()
	proto.RegisterGreeterServer(server, &Server{
    })
	listen, err := net.Listen("tcp", ":8081")
	if err != nil {
    
		panic(err)
	}
	err = server.Serve(listen)
	if err != nil {
    
		panic(err)
	}
}

client

package main

import (
	"ShopBefore/rpc/grpc_retry/proto"
	"context"
	"fmt"
	"github.com/grpc-ecosystem/go-grpc-middleware/retry"
	"google.golang.org/grpc"
	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/credentials/insecure"
	"time"
)

func main() {
    
	retryOpts := []grpc_retry.CallOption{
    
		// 最大重试次数
		grpc_retry.WithMax(3),
		// 超时时间
		grpc_retry.WithPerRetryTimeout(1 * time.Second),
		// 只有返回对应的code才会执行重试
		grpc_retry.WithCodes(codes.Unknown, codes.DeadlineExceeded, codes.Unavailable),
	}
	conn, err := grpc.Dial("127.0.0.1:8081",
		grpc.WithTransportCredentials(insecure.NewCredentials()),
		grpc.WithUnaryInterceptor(grpc_retry.UnaryClientInterceptor(retryOpts...)))
	defer func(conn *grpc.ClientConn) {
    
		err := conn.Close()
		if err != nil {
    
			panic(err)
		}
	}(conn)
	if err != nil {
    
		panic(err)
	}
	client := proto.NewGreeterClient(conn)
	hello, err := client.SayHello(context.Background(), &proto.HelloRequest{
    Name: "番茄炒蛋"})
	if err != nil {
    
		panic(err)
	}
	fmt.Println(hello)
}

测试

server

client

client发送一次请求,由于每次都会超时,进入重试,server确实收到了三次请求.最后client报超时的错也是正常的