结构修改

helper/auth.go → middleware/auth.go

-package helper
+package middleware
 
 import (
 	"context"
@@ -9,7 +9,8 @@ import (
 	"strconv"
 	"time"
 
-	"golang.org/x/time/rate"
+	"im-microservice/helper"
+
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/status"
@@ -25,7 +26,7 @@ func permissionError(msg string) error {
 
 func checksum(req map[string]string) error {
 
-	err, AppSecret := GetSecretByKey(req["Appkey"])
+	err, AppSecret := helper.GetSecretByKey(req["Appkey"])
 	if err != nil {
 		return status.Error(codes.NotFound, "appkey不存在")
 	}
@@ -53,8 +54,7 @@ func Auth(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo,
 	handler grpc.UnaryHandler) (interface{}, error) {
 
 	if info.FullMethod != health_path {
-		limiter := rate.NewLimiter(rate.Every(100*time.Millisecond), 1)
-		if limiter.Allow() {
+
 		request_map := make(map[string]string)
 		s := fmt.Sprintf("%v", req)
 
@@ -87,9 +87,6 @@ func Auth(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo,
 		if err := checksum(request_map); err != nil {
 			return nil, err
 		}
-		} else {
-			return nil, status.Error(codes.Unauthenticated, "请求超过限制,请稍后再试")
-		}
 
 	}
 	return handler(ctx, req)

middleware/interceptor.go

+package middleware
+
+import (
+	"context"
+	"time"
+
+	"golang.org/x/time/rate"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+)
+
+func Interceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo,
+	handler grpc.UnaryHandler) (interface{}, error) {
+
+	if info.FullMethod != health_path {
+		limiter := rate.NewLimiter(rate.Every(100*time.Millisecond), 1)
+		if limiter.Allow() {
+			return handler(ctx, req)
+		} else {
+			return nil, status.Error(codes.Unauthenticated, "请求超过限制,请稍后再试")
+		}
+	}
+
+}