grpc interceptor

pkg/common/mw/check.go

@@ -7,11 +7,12 @@ import (
 	"encoding/base64"
 	"errors"
 	"fmt"
-	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/config"
 	"math/rand"
 	"strings"
 	"sync"
 	"time"
+
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/config"
 )
 
 var (

pkg/common/mw/intercept_chain.go

@@ -0,0 +1,27 @@
+package mw
+
+import (
+	"context"
+
+	"google.golang.org/grpc"
+)
+
+func InterceptChain(intercepts ...grpc.UnaryServerInterceptor) grpc.UnaryServerInterceptor {
+	l := len(intercepts)
+	return func(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp interface{}, err error) {
+		chain := func(currentInter grpc.UnaryServerInterceptor, currentHandler grpc.UnaryHandler) grpc.UnaryHandler {
+			return func(currentCtx context.Context, currentReq interface{}) (interface{}, error) {
+				return currentInter(
+					currentCtx,
+					currentReq,
+					info,
+					currentHandler)
+			}
+		}
+		chainHandler := handler
+		for i := l - 1; i >= 0; i-- {
+			chainHandler = chain(intercepts[i], chainHandler)
+		}
+		return chainHandler(ctx, req)
+	}
+}

pkg/common/mw/rpc_server_interceptor.go

@@ -3,12 +3,13 @@ package mw
 import (
 	"context"
 	"fmt"
-	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/constant"
 	"math"
 	"runtime"
 	"runtime/debug"
 	"strings"
 
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/constant"
+
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/config"
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/log"
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/mw/specialerror"