u

code/allapp/allapp-go/biz/handler/user/user_service.go

@@ -15,40 +15,37 @@ import (
 	"github.com/cloudwego/hertz/pkg/protocol/consts"
 )
 
-// ListUserSpaces .
-// @router /user/spaces [GET]
-func ListUserSpaces(ctx context.Context, c *app.RequestContext) {
+// GetUserSpace .
+// @router /user/space [GET]
+func GetUserSpace(ctx context.Context, c *app.RequestContext) {
 	var err error
-	var req user.ListUserSpacesParams
+	var req user.GetUserSpaceParams
 	err = c.BindAndValidate(&req)
 	if err != nil {
 		c.String(consts.StatusBadRequest, err.Error())
 		return
 	}
 
-	resp := new(user.ListUserSpacesResult)
+	resp := new(user.UserSpace)
 
 	userID := middleware.GetUserID(ctx)
 
-	// 查询
-	spaces, err := db.Queries.ListUserSpaces(ctx, userID)
+	space, err := db.Queries.GetSpaceByUserId(ctx, userID)
+
 	if err != nil {
+		if errorx.IsNotFound(err) {
+			response.Success(c).Data(resp).Send()
+			return
+		}
+
 		errorx.AddError(c, err)
 		return
 	}
 
-	userSpaces := make([]*user.UserSpace, 0, len(spaces))
-
-	for _, space := range spaces {
-		userSpaces = append(userSpaces, &user.UserSpace{
-			Id:         space.ID,
-			Name:       space.Name.String,
-			InviteCode: space.InviteCode,
-			Role:       space.Role,
-		})
-	}
-
-	resp.Spaces = userSpaces
+	resp.Id = space.ID
+	resp.Name = space.Name.String
+	resp.InviteCode = space.InviteCode
+	resp.Role = space.Role
 
 	response.Success(c).Data(resp).Send()
 }