initial import

core/rpc/clientinterceptors/timeoutinterceptor.go

@@ -0,0 +1,25 @@
+package clientinterceptors
+
+import (
+	"context"
+	"time"
+
+	"zero/core/contextx"
+
+	"google.golang.org/grpc"
+)
+
+const defaultTimeout = time.Second * 2
+
+func TimeoutInterceptor(timeout time.Duration) grpc.UnaryClientInterceptor {
+	if timeout <= 0 {
+		timeout = defaultTimeout
+	}
+
+	return func(ctx context.Context, method string, req, reply interface{}, cc *grpc.ClientConn,
+		invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
+		ctx, cancel := contextx.ShrinkDeadline(ctx, timeout)
+		defer cancel()
+		return invoker(ctx, method, req, reply, cc, opts...)
+	}
+}