Add grpc retry (#1160)

* Add grpc retry * Update grpc retry * Add tests * Fix a bug * Add api && some tests * Add comment * Add double check * Add server retry quota * Update optimize code * Fix bug * Update optimize code * Update optimize code * Fix bug
2021-10-27 19:46:07 +08:00
parent 496a2f341e
commit 462ddbb145
17 changed files with 544 additions and 10 deletions
--- a/zrpc/internal/server.go
+++ b/zrpc/internal/server.go
@@ -21,16 +21,18 @@ type (
 	baseRpcServer struct {
 		address            string
 		metrics            *stat.Metrics
+		maxRetries         int
 		options            []grpc.ServerOption
 		streamInterceptors []grpc.StreamServerInterceptor
 		unaryInterceptors  []grpc.UnaryServerInterceptor
 	}
 )

-func newBaseRpcServer(address string, metrics *stat.Metrics) *baseRpcServer {
+func newBaseRpcServer(address string, rpcServerOpts *rpcServerOptions) *baseRpcServer {
 	return &baseRpcServer{
-		address: address,
-		metrics: metrics,
+		address:    address,
+		metrics:    rpcServerOpts.metrics,
+		maxRetries: rpcServerOpts.MaxRetries,
 	}
 }