From 59b9f2ff0f95184f0c006e316f9e61586e3a9789 Mon Sep 17 00:00:00 2001
From: 777 <123@qwe.com>
Date: Sun, 18 Jan 2026 13:24:21 +0800
Subject: [PATCH] 123

---
 .../com/ruoyi/cai/service/UserCodeGenService.java     |  2 ++
 .../cai/service/impl/UserCodeGenServiceImpl.java      | 11 +++++++++--
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/ruoyi-cai/src/main/java/com/ruoyi/cai/service/UserCodeGenService.java b/ruoyi-cai/src/main/java/com/ruoyi/cai/service/UserCodeGenService.java
index 5c3854a4..6c6a1726 100644
--- a/ruoyi-cai/src/main/java/com/ruoyi/cai/service/UserCodeGenService.java
+++ b/ruoyi-cai/src/main/java/com/ruoyi/cai/service/UserCodeGenService.java
@@ -12,4 +12,6 @@ import com.ruoyi.cai.domain.UserCodeGen;
 public interface UserCodeGenService extends IService<UserCodeGen> {
 
     String getCodeGen();
+
+    String getCodeGenRandom();
 }
diff --git a/ruoyi-cai/src/main/java/com/ruoyi/cai/service/impl/UserCodeGenServiceImpl.java b/ruoyi-cai/src/main/java/com/ruoyi/cai/service/impl/UserCodeGenServiceImpl.java
index 996f568a..075172fe 100644
--- a/ruoyi-cai/src/main/java/com/ruoyi/cai/service/impl/UserCodeGenServiceImpl.java
+++ b/ruoyi-cai/src/main/java/com/ruoyi/cai/service/impl/UserCodeGenServiceImpl.java
@@ -1,6 +1,5 @@
 package com.ruoyi.cai.service.impl;
 
-import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.ruoyi.cai.domain.UserCodeGen;
 import com.ruoyi.cai.mapper.UserCodeGenMapper;
@@ -27,8 +26,16 @@ public class UserCodeGenServiceImpl extends ServiceImpl<UserCodeGenMapper, UserC
 
     private final Random random = new Random();
 
+
     @Override
-    public String getCodeGen() {
+    public String getCodeGen(){
+        UserCodeGen gen = new UserCodeGen();
+        this.save(gen);
+        return gen.getId()+"";
+    }
+
+    @Override
+    public String getCodeGenRandom() {
         // 使用数据库唯一索引保证不重复 + 随机生成
         for (int i = 0; i < MAX_RETRY_COUNT; i++) {
             // 生成随机用户号