用户信息传到service

bff/internal/response/user.go

@@ -0,0 +1,61 @@
+package response
+
+import (
+	"context"
+	"net/http"
+	"net/url"
+	"strconv"
+
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/metadata"
+)
+
+type ctxKey string
+
+const (
+	CtxUserId   ctxKey = "X-User-Id"
+	CtxUserName ctxKey = "X-User-Name"
+)
+
+type UserInfo struct {
+	ID     int64
+	Name   string
+	RawUID string
+	Valid  bool
+}
+
+func GetUserInfo(ctx context.Context) UserInfo {
+	rawUID, ok := ctx.Value(CtxUserId).(string)
+	if !ok || rawUID == "" {
+		return UserInfo{Valid: false}
+	}
+	name, _ := ctx.Value(CtxUserName).(string)
+	uid, err := strconv.ParseInt(rawUID, 10, 64)
+	if err != nil {
+		return UserInfo{RawUID: rawUID, Name: name, Valid: false}
+	}
+	return UserInfo{ID: uid, Name: name, RawUID: rawUID, Valid: true}
+}
+
+func UserReadMiddleware(next http.HandlerFunc) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		uid := r.Header.Get("X-User-Id")
+		uname := r.Header.Get("X-User-Name")
+		newCtx := r.Context()
+		newCtx = context.WithValue(newCtx, CtxUserId, uid)
+		newCtx = context.WithValue(newCtx, CtxUserName, uname)
+		next(w, r.WithContext(newCtx))
+	}
+}
+
+func UserClientInterceptor(ctx context.Context, method string, req, reply any, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
+	user := GetUserInfo(ctx)
+	if user.Valid {
+		md := metadata.New(map[string]string{
+			"x-user-id":   user.RawUID,
+			"x-user-name": url.QueryEscape(user.Name),
+		})
+		ctx = metadata.NewOutgoingContext(ctx, md)
+	}
+	return invoker(ctx, method, req, reply, cc, opts...)
+}