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报超时的错也是正常的