Grpc middleware implements grpc call retry

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;
}

Generate corresponding go Code

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(" Receive a request .....")
	//  Set one second timeout , Sleep here for two seconds 
	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{
    
		//  max retries 
		grpc_retry.WithMax(3),
		//  Timeout time 
		grpc_retry.WithPerRetryTimeout(1 * time.Second),
		//  Only return the corresponding code Will retry 
		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: " fried eggs with tomatoes "})
	if err != nil {
    
		panic(err)
	}
	fmt.Println(hello)
}

test

server

client

client Send a request , Because it will time out every time , Enter and try again ,server I did receive three requests . Last client It is also normal to make a timeout error