123

ace-common/src/main/java/com/github/wxiaoqi/security/common/util/UUIDUtils.java

@@ -14,6 +14,13 @@ public class UUIDUtils {
             "W", "X", "Y", "Z" };
 
 
+    public static String[] chars1 = new String[] { "a", "b", "c", "d", "e", "f",
+            "g", "h", "i", "j", "k", "l", "m", "n", "p", "q", "r", "s",
+            "t", "u", "v", "w", "x", "y", "z", "1", "2", "3", "4", "5",
+            "6", "7", "8", "9", "A", "B", "C", "D", "E", "F", "G", "H", "I",
+            "J", "K", "L", "M", "N",  "P", "Q", "R", "S", "T", "U", "V",
+            "W", "X", "Y", "Z" };
+
     public static String generateShortUuid() {
         StringBuffer shortBuffer = new StringBuffer();
         String uuid = UUID.randomUUID().toString().replace("-", "");
@@ -25,4 +32,16 @@ public class UUIDUtils {
         return shortBuffer.toString();
 
     }
+    public static String genCodes(int length) {
+        StringBuffer shortBuffer = new StringBuffer();
+        String uuid = UUID.randomUUID().toString().replace("-", "");
+        for (int i = 0; i < length; i++) {
+            String str = uuid.substring(i * 4, i * 4 + 4);
+            int x = Integer.parseInt(str, 16);
+            shortBuffer.append(chars1[x % 0x3E]);
+        }
+        return shortBuffer.toString();
+    }
+
+
 }

ace-modules/ace-admin/src/main/java/com/github/wxiaoqi/security/admin/rpc/service/AppPermissionService.java

@@ -15,6 +15,7 @@ import com.github.wxiaoqi.security.api.vo.user.AppUserInfo;
 import com.github.wxiaoqi.security.common.msg.BaseResponse;
 import com.github.wxiaoqi.security.common.msg.ObjectRestResponse;
 import com.github.wxiaoqi.security.common.util.EmojiFilter;
+import com.github.wxiaoqi.security.common.util.UUIDUtils;
 import com.github.wxiaoqi.security.common.util.VerificationUtils;
 import com.github.wxiaoqi.security.common.util.process.ResultCode;
 import com.github.wxiaoqi.security.common.util.process.SystemConfig;
@@ -248,6 +249,8 @@ public class AppPermissionService {
             rsUserDetail.setIsdel(0);
             rsUserDetail.setCrtHost(getIp());
             setCreateIPInfo(rsUserDetail);
+            //生成邀请码
+            rsUserDetail.setCode(UUIDUtils.genCodes(6));
             appUserDetailBiz.insertSelective(rsUserDetail);
             log.error("注册：新增用户详情： " + userid);
              //自动登录获取优惠卷
@@ -754,6 +757,8 @@ public class AppPermissionService {
             rsUserDetail.setCreatetime(now);
             rsUserDetail.setUpdatetime(now);
             rsUserDetail.setIsdel(0);
+            //生成邀请码
+            rsUserDetail.setCode(UUIDUtils.genCodes(6));
             //设置来源
             rsUserDetail.setChannel(UserSourceEnum.APPLET.getCode());
             rsUserDetail.setCrtHost(getIp());