wangqifan преди 3 години
родител
ревизия
4a63fdcf0d
променени са 1 файла, в които са добавени 13 реда и са изтрити 11 реда
  1. 13 11
      src/main/java/com/izouma/nineth/service/UserService.java

+ 13 - 11
src/main/java/com/izouma/nineth/service/UserService.java

@@ -275,17 +275,19 @@ public class UserService {
             invite = inviteRepo.findFirstByCode(inviteCode).orElse(null);
         }
 
-        if (inviteType.equals(InviteType.AUCTION)) {
-            User inviteUser = userRepo.findById(invitor).orElseThrow(new BusinessException("暂无用户"));
-            invite = inviteRepo.findFirstByCode(String.valueOf(invitor)).orElse(null);
-            if (invite == null) {
-                Invite newOne = new Invite();
-                newOne.setInviteNum(0);
-                newOne.setPhone(inviteUser.getPhone());
-                newOne.setCode(String.valueOf(invitor));
-                newOne.setName(inviteUser.getNickname());
-                newOne.setInviteType(InviteType.AUCTION);
-                invite = inviteRepo.save(newOne);
+        if (inviteType != null) {
+            if (inviteType.equals(InviteType.AUCTION)) {
+                User inviteUser = userRepo.findById(invitor).orElseThrow(new BusinessException("暂无用户"));
+                invite = inviteRepo.findFirstByCode(String.valueOf(invitor)).orElse(null);
+                if (invite == null) {
+                    Invite newOne = new Invite();
+                    newOne.setInviteNum(0);
+                    newOne.setPhone(inviteUser.getPhone());
+                    newOne.setCode(String.valueOf(invitor));
+                    newOne.setName(inviteUser.getNickname());
+                    newOne.setInviteType(InviteType.AUCTION);
+                    invite = inviteRepo.save(newOne);
+                }
             }
         }
         smsService.verify(phone, code);