123

ace-modules/ace-admin/src/main/java/com/github/wxiaoqi/security/admin/biz/AppUserRelationBiz.java

@@ -62,7 +62,9 @@ public class AppUserRelationBiz extends BaseBiz<AppUserRelationMapper, AppUserRe
     private  AppStaffUserBiz staffUserBiz;
 
     private static final HashMap<Integer, Boolean> map =  new HashMap<>();
+    private static final HashMap<Integer, Boolean> parentMap =  new HashMap<>();
     private static final Object obj = new Object();
+    private static final Object parentObj = new Object();
 
     private boolean lock(Integer userId) {
         synchronized (obj) {
@@ -73,6 +75,21 @@ public class AppUserRelationBiz extends BaseBiz<AppUserRelationMapper, AppUserRe
         }
         return false;
     }
+    private boolean parentLock(Integer userId) {
+        synchronized (parentObj) {
+            if (parentMap.get(userId) == null) {
+                parentMap.put(userId, true);
+                return true;
+            }
+        }
+        return false;
+    }
+
+    private void unParentLock(Integer userId) {
+        synchronized (parentObj) {
+            parentMap.remove(userId);
+        }
+    }
 
     private void unLock(Integer userId) {
         synchronized (obj) {
@@ -108,10 +125,7 @@ public class AppUserRelationBiz extends BaseBiz<AppUserRelationMapper, AppUserRe
                 AppUserRelation relation = getMyBiz().getRelationByUserId(parentId);
                 Long time = System.currentTimeMillis();
                 if (relation == null) {
-                    relation = new AppUserRelation();
-                    relation.setUserId(parentId);
-                    relation.setBindType(type);
-                    insertSelective(relation);
+                    insertLock(parentId, type);
                 }
                 Long bindTime = time - validTime;
                 //判断用户是否有有效的下线
@@ -124,12 +138,10 @@ public class AppUserRelationBiz extends BaseBiz<AppUserRelationMapper, AppUserRe
                         relation.setBindType(type);
                         relation.setBindTime(time);
                         insertSelective(relation);
-                        if (relation != null) {
-                            AppStaffUserDTO staffUserDTO = new AppStaffUserDTO();
-                            staffUserDTO.setSuId(relation.getParentId());
-                            staffUserDTO.setUid(relation.getUserId());
-                            staffUserBiz.addAppStaffUser(staffUserDTO);
-                        }
+                        AppStaffUserDTO staffUserDTO = new AppStaffUserDTO();
+                        staffUserDTO.setSuId(relation.getParentId());
+                        staffUserDTO.setUid(relation.getUserId());
+                        staffUserBiz.addAppStaffUser(staffUserDTO);
                     } else {
                         //判断用户是否有有效的上线
                         log.info("----userId===" + userId + "----bindTime====" + bindTime + "----relation.getBindTime()===" + relation.getBindTime());
@@ -150,6 +162,23 @@ public class AppUserRelationBiz extends BaseBiz<AppUserRelationMapper, AppUserRe
 
     }
 
+    private void insertLock(Integer userId, Integer type) {
+        try {
+            if (parentLock(userId)) {
+                AppUserRelation relation = getMyBiz().getRelationByUserId(userId);
+                if (relation == null) {
+                    relation = new AppUserRelation();
+                    relation.setUserId(userId);
+                    relation.setBindType(type);
+                    insertSelective(relation);
+                }
+            }
+        }catch (Exception e){
+        }finally {
+            unParentLock(userId);
+        }
+    }
+
     //首页关系绑定
     public ObjectRestResponse appBindRelation(Integer userId, String code) {
         Integer parentId = 0;