From be921e144f1b4d1e5b2134d744d1c878bd63b3c7 Mon Sep 17 00:00:00 2001
From: ziin <hiziying@hotmail.com>
Date: Mon, 29 Dec 2025 15:04:30 +0800
Subject: [PATCH] =?UTF-8?q?feat(invite):=20=E6=94=AF=E6=8C=81=E7=A7=9F?=
 =?UTF-8?q?=E6=88=B7=E4=B8=8E=E7=94=A8=E6=88=B7=E4=B8=A4=E7=A7=8D=E9=82=80?=
 =?UTF-8?q?=E8=AF=B7=E7=A0=81=E7=B1=BB=E5=9E=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../model/entity/KeyboardUserInviteCodes.java | 14 +++++++++++
 .../KeyboardUserInviteCodesServiceImpl.java   |  1 +
 .../service/impl/UserServiceImpl.java         | 24 +++++++++++++++----
 3 files changed, 35 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/yolo/keyborad/model/entity/KeyboardUserInviteCodes.java b/src/main/java/com/yolo/keyborad/model/entity/KeyboardUserInviteCodes.java
index 2f444ab..53fbb7e 100644
--- a/src/main/java/com/yolo/keyborad/model/entity/KeyboardUserInviteCodes.java
+++ b/src/main/java/com/yolo/keyborad/model/entity/KeyboardUserInviteCodes.java
@@ -75,4 +75,18 @@ public class KeyboardUserInviteCodes {
     @TableField(value = "used_count")
     @Schema(description="邀请码已使用次数")
     private Integer usedCount;
+
+    /**
+     * 邀请码类型：USER=普通用户邀请码，TENANT=租户邀请码
+     */
+    @TableField(value = "invite_type")
+    @Schema(description="邀请码类型：USER=普通用户邀请码，TENANT=租户邀请码")
+    private String inviteType;
+
+    /**
+     * 邀请码所属租户ID（当inviteType=TENANT时使用）
+     */
+    @TableField(value = "tenant_id")
+    @Schema(description="邀请码所属租户ID（当inviteType=TENANT时使用）")
+    private Long tenantId;
 }
\ No newline at end of file
diff --git a/src/main/java/com/yolo/keyborad/service/impl/KeyboardUserInviteCodesServiceImpl.java b/src/main/java/com/yolo/keyborad/service/impl/KeyboardUserInviteCodesServiceImpl.java
index 37e612d..8e7a577 100644
--- a/src/main/java/com/yolo/keyborad/service/impl/KeyboardUserInviteCodesServiceImpl.java
+++ b/src/main/java/com/yolo/keyborad/service/impl/KeyboardUserInviteCodesServiceImpl.java
@@ -77,6 +77,7 @@ public class KeyboardUserInviteCodesServiceImpl extends ServiceImpl<KeyboardUser
         inviteCode.setExpiresAt(null); // 永久有效
         inviteCode.setMaxUses(null); // 不限次数
         inviteCode.setUsedCount(0); // 初始使用次数为0
+        inviteCode.setInviteType("USER"); // 默认为普通用户邀请码
 
         // 保存到数据库
         this.save(inviteCode);
diff --git a/src/main/java/com/yolo/keyborad/service/impl/UserServiceImpl.java b/src/main/java/com/yolo/keyborad/service/impl/UserServiceImpl.java
index cf40d0d..aaa8228 100644
--- a/src/main/java/com/yolo/keyborad/service/impl/UserServiceImpl.java
+++ b/src/main/java/com/yolo/keyborad/service/impl/UserServiceImpl.java
@@ -270,14 +270,22 @@ public class UserServiceImpl extends ServiceImpl<KeyboardUserMapper, KeyboardUse
                     userInvite.setBoundAt(new Date());
                     userInvite.setBindIp(request.getRemoteAddr());
                     userInvite.setBindUserAgent(request.getHeader("User-Agent"));
+                    // 记录邀请码类型快照（用户/租户）
+                    userInvite.setInviteType(inviteCode.getInviteType());
+                    userInvite.setInviteCode(inviteCode.getCode());
+                    // 如果是租户邀请码，记录租户ID
+                    if ("TENANT".equals(inviteCode.getInviteType()) && inviteCode.getTenantId() != null) {
+                        userInvite.setProfitTenantId(inviteCode.getTenantId());
+                        userInvite.setInviterTenantId(inviteCode.getTenantId());
+                    }
                     userInvitesService.save(userInvite);
 
                     // 更新邀请码使用次数
                     inviteCode.setUsedCount(inviteCode.getUsedCount() + 1);
                     inviteCodesService.updateById(inviteCode);
 
-                    log.info("User bound to invite code, userId={}, inviteCodeId={}, inviterUserId={}",
-                            keyboardUser.getId(), inviteCode.getId(), inviteCode.getOwnerUserId());
+                    log.info("User bound to invite code, userId={}, inviteCodeId={}, inviterUserId={}, inviteType={}",
+                            keyboardUser.getId(), inviteCode.getId(), inviteCode.getOwnerUserId(), inviteCode.getInviteType());
                 } catch (BusinessException e) {
                     // 邀请码验证失败，记录日志但不影响注册流程
                     log.warn("Failed to bind invite code for user {}: {}", keyboardUser.getId(), e.getMessage());
@@ -379,14 +387,22 @@ public class UserServiceImpl extends ServiceImpl<KeyboardUserMapper, KeyboardUse
         userInvite.setBoundAt(new Date());
         userInvite.setBindIp(request.getRemoteAddr());
         userInvite.setBindUserAgent(request.getHeader("User-Agent"));
+        // 记录邀请码类型快照（用户/租户）
+        userInvite.setInviteType(inviteCode.getInviteType());
+        userInvite.setInviteCode(inviteCode.getCode());
+        // 如果是租户邀请码，记录租户ID
+        if ("TENANT".equals(inviteCode.getInviteType()) && inviteCode.getTenantId() != null) {
+            userInvite.setProfitTenantId(inviteCode.getTenantId());
+            userInvite.setInviterTenantId(inviteCode.getTenantId());
+        }
         userInvitesService.save(userInvite);
 
         // 更新邀请码使用次数
         inviteCode.setUsedCount(inviteCode.getUsedCount() + 1);
         inviteCodesService.updateById(inviteCode);
 
-        log.info("User bound invite code, userId={}, inviteCodeId={}, inviterUserId={}",
-                userId, inviteCode.getId(), inviteCode.getOwnerUserId());
+        log.info("User bound invite code, userId={}, inviteCodeId={}, inviterUserId={}, inviteType={}",
+                userId, inviteCode.getId(), inviteCode.getOwnerUserId(), inviteCode.getInviteType());
 
         return true;
     }