调整middleware

internal/server/middleware/middleware.go

@@ -2,7 +2,6 @@ package middleware
 
 import (
 	"context"
-	"encoding/json"
 	"net/url"
 	"strconv"
 
@@ -30,20 +29,13 @@ func GetAccountID(c context.Context) int64 {
 	return 0
 }
 
-func GetAuthToken(c context.Context) string {
-	token := c.Value("token")
-	if _, ok := token.(string); ok {
-		return token.(string)
+func GetMain(c context.Context) bool {
+	main := c.Value("main")
+	if _, ok := main.(string); ok {
+		_main, _ := strconv.ParseBool(main.(string))
+		return _main
 	}
-	return ""
-}
-
-func GetJwtClaims(c context.Context) json.RawMessage {
-	claim := c.Value("claim")
-	if _, ok := claim.(json.RawMessage); ok {
-		return claim.(json.RawMessage)
-	}
-	return []byte{}
+	return false
 }
 
 func GetPageOffset(c context.Context) (string, int64) {
@@ -73,16 +65,10 @@ func GrpcValue(handler middleware.Handler) middleware.Handler {
 				ctx = context2.AppendToContext(ctx, "accountId", accountID[0])
 			}
 
-			claimMap := md.Get("jwt_claims")
-			if len(claimMap) > 0 {
-				_claimMap := json.RawMessage(claimMap[0])
-				ctx = context2.AppendToContext(ctx, "jwt_claims", _claimMap)
+			main := md.Get("main")
+			if len(main) > 0 {
+				ctx = context2.AppendToContext(ctx, "main", main[0])
 			}
-			token := md.Get("auth_token")
-			if len(token) > 0 {
-				ctx = context2.AppendToContext(ctx, "auth_token", token[0])
-			}
-
 			//nextId := md.Get("nextId")
 			//if len(nextId) > 0 {
 			//	ctx = context2.AppendToContext(ctx, "nextId", nextId[0])