grpc-middleware实现grpc调用重试

最新推荐文章于 2024-05-29 09:40:55 发布

.番茄炒蛋

最新推荐文章于 2024-05-29 09:40:55 发布

阅读量374

点赞数 2

分类专栏： Go学习 grpc入门和进阶文章标签： golang

本文链接：https://blog.csdn.net/qq_43135259/article/details/125901761

版权

Go学习同时被 2 个专栏收录

58 篇文章 6 订阅

订阅专栏

grpc入门和进阶

14 篇文章 6 订阅

订阅专栏

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