邀请

@@ -22,6 +22,7 @@ import org.springframework.util.StringUtils;
 
 import java.util.List;
 import java.util.Map;
+import java.util.Optional;
 import java.util.stream.Collectors;
 
 @Service
@@ -231,7 +232,7 @@ public class InviteServiceImpl implements IInviteService {
          */
         Integer bClassInviteCode = null;
         if (!CollectionUtils.isEmpty(inviters)){
-            bClassInviteCode = getInviteCode(inviters, InviterType.STORED_SELLER);
+            bClassInviteCode = getInviteCode(inviters, InviterType.CLASS_B_AGENT);
         }
 
         boolean storedSeller = storedSellerService.isStoredSeller(uid);
@@ -270,9 +271,10 @@ public class InviteServiceImpl implements IInviteService {
 
 
     private Integer getInviteCode(List<Inviter> inviters, InviterType inviterType){
-        return inviters.stream()
+        Optional<Inviter> inviter =  inviters.stream()
                 .filter(x -> x.getType().intValue() == inviterType.getType() && x.getStatus().intValue() == InviterStatus.VALID.getCode())
-                .findAny().get().getInviteCode();
+                .findAny();
+        return inviter.isPresent() ? inviter.get().getInviteCode() : null;
     }