feat(invite): 新增用户邀请码创建与查询接口

src/main/java/com/yolo/keyborad/controller/UserController.java

@@ -8,9 +8,13 @@ import com.yolo.keyborad.model.dto.AppleLoginReq;
 import com.yolo.keyborad.model.dto.user.*;
 import com.yolo.keyborad.model.entity.KeyboardFeedback;
 import com.yolo.keyborad.model.entity.KeyboardUser;
+import com.yolo.keyborad.model.entity.KeyboardUserInviteCodes;
+import com.yolo.keyborad.model.vo.user.InviteCodeRespVO;
 import com.yolo.keyborad.model.vo.user.KeyboardUserInfoRespVO;
 import com.yolo.keyborad.model.vo.user.KeyboardUserRespVO;
 import com.yolo.keyborad.service.IAppleService;
+import com.yolo.keyborad.service.KeyboardFeedbackService;
+import com.yolo.keyborad.service.KeyboardUserInviteCodesService;
 import com.yolo.keyborad.service.UserService;
 import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.Parameter;
@@ -41,8 +45,10 @@ public class UserController {
     private UserService userService;
 
     @Resource
-    private com.yolo.keyborad.service.KeyboardFeedbackService feedbackService;
+    private KeyboardFeedbackService feedbackService;
 
+    @Resource
+    private KeyboardUserInviteCodesService inviteCodesService;
     /**
      * 苹果登录
      *
@@ -133,4 +139,15 @@ public class UserController {
         return ResultUtils.success(userService.bindInviteCode(bindInviteCodeDTO));
     }
 
+    @GetMapping("/inviteCode")
+    @Operation(summary = "查询邀请码", description = "查询用户自己的邀请码")
+    public BaseResponse<InviteCodeRespVO> getInviteCode() {
+        long userId = StpUtil.getLoginIdAsLong();
+        KeyboardUserInviteCodes inviteCode = inviteCodesService.getUserInviteCode(userId);
+        if (inviteCode == null) {
+            inviteCode = inviteCodesService.createInviteCode(userId);
+        }
+        InviteCodeRespVO respVO = BeanUtil.copyProperties(inviteCode, InviteCodeRespVO.class);
+        return ResultUtils.success(respVO);
+    }
 }