add Refresh

bff/internal/response/user.go

@@ -13,49 +13,66 @@ import (
 type ctxKey string
 
 const (
-	CtxUserId   ctxKey = "X-User-Id"
-	CtxUserName ctxKey = "X-User-Name"
+	CtxUserId      ctxKey = "X-User-Id"
+	CtxUserName    ctxKey = "X-User-Name"
+	CtxRefresh     ctxKey = "X-Refresh"
+	CtxServiceCode ctxKey = "X-Service-Code"
 )
 
 type UserInfo struct {
-	ID     int64
-	Name   string
-	RawUID string
-	Valid  bool
+	ID      int64
+	Name    string
+	RawUID  string
+	Refresh string
+	Valid   bool
 }
 
 func GetUserInfo(ctx context.Context) UserInfo {
-	rawUID, ok := ctx.Value(CtxUserId).(string)
-	if !ok || rawUID == "" {
-		return UserInfo{Valid: false}
-	}
+	rawUID, _ := ctx.Value(CtxUserId).(string)
 	name, _ := ctx.Value(CtxUserName).(string)
+	refresh, _ := ctx.Value(CtxRefresh).(string)
+
+	info := UserInfo{
+		RawUID:  rawUID,
+		Name:    name,
+		Refresh: refresh,
+	}
+	if rawUID == "" {
+		return info
+	}
 	uid, err := strconv.ParseInt(rawUID, 10, 64)
 	if err != nil {
-		return UserInfo{RawUID: rawUID, Name: name, Valid: false}
+		info.Valid = false
+		return info
 	}
-	return UserInfo{ID: uid, Name: name, RawUID: rawUID, Valid: true}
+	info.ID = uid
+	info.Valid = true
+	return info
 }
 
 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)
+		newCtx = context.WithValue(newCtx, CtxUserId, r.Header.Get("X-User-Id"))
+		newCtx = context.WithValue(newCtx, CtxUserName, r.Header.Get("X-User-Name"))
+		newCtx = context.WithValue(newCtx, CtxRefresh, r.Header.Get("X-Refresh"))
 		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)
+	md := metadata.New(map[string]string{})
+
 	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)
+		md.Set("x-user-id", user.RawUID)
+		md.Set("x-user-name", url.QueryEscape(user.Name))
 	}
+
+	if user.Refresh != "" {
+		md.Set("x-refresh", user.Refresh)
+	}
+
+	ctx = metadata.NewOutgoingContext(ctx, md)
 	return invoker(ctx, method, req, reply, cc, opts...)
 }