完善用户注册邀请逻辑

src/main/java/com/xs/core/service/Impl/user/UserServiceImpl.java

@@ -27,6 +27,7 @@ import java.net.URLDecoder;
 import java.nio.charset.StandardCharsets;
 import java.time.LocalDateTime;
 import java.util.List;
+import java.util.Objects;
 import java.util.concurrent.CompletableFuture;
 
 /**
@@ -113,7 +114,7 @@ public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements IU
             LambdaQueryWrapper<User> queryWrapper = new LambdaQueryWrapper<>();
             queryWrapper.eq(User::getInviteCode, loginReq.getShare_code());
             inviteUser = getOne(queryWrapper);
-            if (ObjUtil.isNotNull(inviteUser)) {
+            if (ObjUtil.isNotNull(inviteUser)&& !Objects.equals(inviteUser.getId(), user.getId())) {
                 user.setReferrerId(String.valueOf(inviteUser.getId()));
                 user.setPassengerFlowWay(2);
                 user.setId(IdWorker.getId());

src/main/java/com/xs/core/utils/InviteCodeGenerator.java

@@ -35,31 +35,25 @@ public class InviteCodeGenerator {
         }
 
         //邀请码+随机码组成一个字符
-        String tempCode = code.toString() + randomCode.toString();
+        String tempCode = code + randomCode.toString();
         byte[] digest = DigestUtil.md5(tempCode);
         String sMD5EncryptResult = DatatypeConverter.printHexBinary(digest).toUpperCase();
-        String[] resUrl = new String[4];
-        //得到 4组短链接字符串
-        for (int i = 0; i < 4; i++) {
-            // 把加密字符按照 8 位一组 16 进制与 0x3FFFFFFF 进行位与运算
-            String sTempSubString = sMD5EncryptResult.substring(i * 8, i * 8 + 8);
-            // 这里需要使用 long 型来转换，因为 Inteper .parseInt() 只能处理 31 位 , 首位为符号位 , 如果不用 long ，则会越界
-            long lHexLong = 0x3FFFFFFF & Long.parseLong(sTempSubString, 16);
-            String outChars = "";
-            //循环获得每组6位的字符串
-            for (int j = 0; j < 6; j++) {
-                // 把得到的值与 0x0000003D 进行位与运算，取得字符数组 chars 索引(具体需要看chars数组的长度   以防下标溢出，注意起点为0)
-                long index = 0x0000003D & lHexLong;
-                // 把取得的字符相加
-                outChars += CHARACTERS[(int) index];
-                // 每次循环按位右移 5 位
-                lHexLong = lHexLong >> 5;
-            }
-            // 把字符串存入对应索引的输出数组
-            resUrl[i] = outChars;
-        }
 
-        return resUrl[0];
+        // 把加密字符按照 8 位一组 16 进制与 0x3FFFFFFF 进行位与运算
+        String sTempSubString = sMD5EncryptResult.substring(0, 8);
+        // 这里需要使用 long 型来转换，因为 Inteper .parseInt() 只能处理 31 位 , 首位为符号位 , 如果不用 long ，则会越界
+        long lHexLong = 0x3FFFFFFF & Long.parseLong(sTempSubString, 16);
+        StringBuilder outChars = new StringBuilder();
+        //循环获得每组6位的字符串
+        for (int j = 0; j < 6; j++) {
+            // 把得到的值与 0x0000003D 进行位与运算，取得字符数组 chars 索引(具体需要看chars数组的长度   以防下标溢出，注意起点为0)
+            long index = 0x0000003D & lHexLong;
+            // 把取得的字符相加
+            outChars.append(CHARACTERS[(int) index]);
+            // 每次循环按位右移 5 位
+            lHexLong = lHexLong >> 5;
+        }
+        return outChars.toString();
     }
 
     public static void main(String[] args) {