rename rpcx to zrpc

zrpc/internal/clientinterceptors/durationinterceptor.go

@@ -0,0 +1,30 @@
+package clientinterceptors
+
+import (
+	"context"
+	"path"
+	"time"
+
+	"github.com/tal-tech/go-zero/core/logx"
+	"github.com/tal-tech/go-zero/core/timex"
+	"google.golang.org/grpc"
+)
+
+const slowThreshold = time.Millisecond * 500
+
+func DurationInterceptor(ctx context.Context, method string, req, reply interface{},
+	cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
+	serverName := path.Join(cc.Target(), method)
+	start := timex.Now()
+	err := invoker(ctx, method, req, reply, cc, opts...)
+	if err != nil {
+		logx.WithDuration(timex.Since(start)).Infof("fail - %s - %v - %s", serverName, req, err.Error())
+	} else {
+		elapsed := timex.Since(start)
+		if elapsed > slowThreshold {
+			logx.WithDuration(elapsed).Slowf("[RPC] ok - slowcall - %s - %v - %v", serverName, req, reply)
+		}
+	}
+
+	return err
+}