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