feat: invitation code

TensoRaws · Jul 16, 2024 · 18c4767 · 18c4767
1 parent cbbefbc
commit 18c4767
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/internal/service/user/invitation.go b/internal/service/user/invitation.go
@@ -15,6 +15,8 @@ type InvitationGenResponse struct {
 	InvitationCode string `json:"invitation_code"`
 }
 
+type InvitationMeResponse []cache.UserInvitation
+
 // InvitationGen 生成邀请码 (POST /invitation/gen)
 func InvitationGen(c *gin.Context) {
 	userID, _ := resp.GetUserIDFromGinContext(c)
@@ -50,6 +52,10 @@ func InvitationMe(c *gin.Context) {
 		return
 	}
 
-	resp.OKWithData(c, codeList)
+	if len(codeList) == 0 {
+		resp.OKWithData(c, InvitationMeResponse{})
+	} else {
+		resp.OKWithData(c, InvitationMeResponse(codeList))
+	}
 	log.Logger.Infof("User %d got invitation code list successfully!", userID)
 }