Jelajahi Sumber

推荐人 bug 修改

785834757 2 tahun lalu
induk
melakukan
124974d3fa

+ 5 - 2
src/main/java/com/ydtech/modules/esm/service/impl/EsmUserReferrerServiceImpl.java

@@ -63,13 +63,16 @@ public class EsmUserReferrerServiceImpl extends ServiceImpl<EsmUserReferrerMappe
             return null;
         }
         List<EsmUserReferrer> esmUserReferrers =
-                esmUserReferrerService.list(new LambdaQueryWrapper<EsmUserReferrer>().eq(EsmUserReferrer::getAffiliation,
-                        esmUserReferrer.getAffiliation()));
+                esmUserReferrerService.list(new LambdaQueryWrapper<EsmUserReferrer>()
+                        .eq(EsmUserReferrer::getAffiliation, esmUserReferrer.getAffiliation())
+                        .ne(EsmUserReferrer::getId, esmUserReferrer.getId()));
         findParentsRecursive(esmUserReferrer.getReferrerId(), esmUserReferrers, referrers);
         return referrers;
     }
 
     private static void findParentsRecursive(String userId, List<EsmUserReferrer> esmUserReferrers, List<EsmUserReferrer> referrers) {
+        if(esmUserReferrers.isEmpty())
+            return;
         EsmUserReferrer collect = esmUserReferrers.stream().filter(item -> item.getId().equals(userId)).findFirst().get();
         if (collect != null) {
             if (collect.getId().equals(collect.getReferrerId())) {