fanzherong_v 3 ヶ月 前
コミット
ec8af96874

+ 17 - 5
snowy-plugin/snowy-plugin-biz/src/main/java/vip/xiaonuo/biz/modular/user/service/impl/BizUserServiceImpl.java

@@ -105,10 +105,7 @@ import vip.xiaonuo.sys.api.SysUserApi;
 
 import java.io.*;
 import java.math.BigDecimal;
-import java.util.Comparator;
-import java.util.List;
-import java.util.Map;
-import java.util.Set;
+import java.util.*;
 import java.util.stream.Collectors;
 
 /**
@@ -962,7 +959,8 @@ public class BizUserServiceImpl extends ServiceImpl<BizUserMapper, BizUser> impl
         bizUser.setUserStatus("ENABLE");
         bizUser.setOpenId(wxUserRegisterParam.getOpenId());
         //生成推荐码
-        bizUser.setUserReferralCode(DateUtil.format(DateUtil.date(), DatePattern.PURE_DATETIME_PATTERN) + RandomUtil.randomNumbers(4));
+        //bizUser.setUserReferralCode(DateUtil.format(DateUtil.date(), DatePattern.PURE_DATETIME_PATTERN) + RandomUtil.randomNumbers(4));
+        bizUser.setUserReferralCode(generateReferralCode());
         bizUser.setName(wxUserRegisterParam.getName());
         this.save(bizUser);
         sysRoleApi.grantUserRole(bizUser.getId(), roleId);
@@ -1145,4 +1143,18 @@ public class BizUserServiceImpl extends ServiceImpl<BizUserMapper, BizUser> impl
             return false;
         }
     }
+
+
+    private static final String CHARACTERS = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
+    private static final int CODE_LENGTH = 10;
+
+    public static String generateReferralCode() {
+        StringBuilder sb = new StringBuilder();
+        Random random = new Random();
+        for (int i = 0; i < CODE_LENGTH; i++) {
+            int index = random.nextInt(CHARACTERS.length());
+            sb.append(CHARACTERS.charAt(index));
+        }
+        return sb.toString();
+    }
 }