住户导入创建共用线程池和用户机构插入加锁

linwe

2024-06-19 6b113a88e7b7b625405cc0b1d46efa22d46c71a2

 src/main/java/org/springblade/modules/system/service/impl/UserServiceImpl.java

@@ -17,7 +17,6 @@
package org.springblade.modules.system.service.impl;


import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
@@ -25,9 +24,10 @@
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import lombok.AllArgsConstructor;
import org.apache.commons.lang3.StringUtils;
import org.apache.ibatis.annotations.Param;
import org.apache.logging.log4j.util.Strings;
import org.flowable.idm.engine.impl.persistence.entity.UserEntity;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springblade.common.cache.DictCache;
import org.springblade.common.cache.ParamCache;
import org.springblade.common.cache.SysCache;
@@ -35,6 +35,7 @@
import org.springblade.common.constant.CommonConstant;
import org.springblade.common.constant.TenantConstant;
import org.springblade.common.enums.DictEnum;
import org.springblade.common.utils.SpinLockUtil;
import org.springblade.core.log.exception.ServiceException;
import org.springblade.core.mp.base.BaseServiceImpl;
import org.springblade.core.mp.support.Condition;
@@ -46,12 +47,9 @@
import org.springblade.core.tool.support.Kv;
import org.springblade.core.tool.utils.*;
import org.springblade.modules.auth.enums.UserEnum;
import org.springblade.modules.community.entity.CommunityEntity;
import org.springblade.modules.community.service.ICommunityService;
import org.springblade.modules.grid.service.IGridmanService;
import org.springblade.modules.house.entity.HouseholdEntity;
import org.springblade.modules.house.service.IHouseholdService;
import org.springblade.modules.house.vo.HouseholdVO;
import org.springblade.modules.police.entity.PoliceAffairsGridEntity;
import org.springblade.modules.police.service.IPoliceAffairsGridService;
import org.springblade.modules.property.entity.PropertyCompanyEntity;
@@ -65,6 +63,7 @@
import org.springblade.modules.system.vo.UserDetailVO;
import org.springblade.modules.system.vo.UserVO;
import org.springblade.modules.system.wrapper.UserWrapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

@@ -80,6 +79,9 @@
@Service
@AllArgsConstructor
public class UserServiceImpl extends BaseServiceImpl<UserMapper, User> implements IUserService {

   private static final Logger logger = LoggerFactory.getLogger(UserServiceImpl.class);

   private static final String GUEST_NAME = "guest";

   private final IUserDeptService userDeptService;
@@ -88,6 +90,10 @@
   private final IDeptService deptService;
   private final BladeTenantProperties tenantProperties;
   private final IPoliceAffairsGridService policeAffairsGridService;

   @Autowired
   private SpinLockUtil spinLockUtil;


   @Override
   @Transactional(rollbackFor = Exception.class)
@@ -202,18 +208,34 @@
   }

   private boolean submitUserDept(User user) {
      List<Long> deptIdList = Func.toLongList(user.getDeptId());
      List<UserDept> userDeptList = new ArrayList<>();
      deptIdList.forEach(deptId -> {
         UserDept userDept = new UserDept();
         userDept.setUserId(user.getId());
         userDept.setDeptId(deptId);
         userDeptList.add(userDept);
      });
      userDeptService.remove(Wrappers.<UserDept>update().lambda().eq(UserDept::getUserId, user.getId()));
      boolean b = userDeptService.saveBatch(userDeptList);
      return b;
      try {
         // 加锁
         boolean isLock = spinLockUtil.tryLock(user.getId().toString());
         boolean result = false;
         List<Long> deptIdList = Func.toLongList(user.getDeptId());
         List<UserDept> userDeptList = new ArrayList<>();
         deptIdList.forEach(deptId -> {
            UserDept userDept = new UserDept();
            userDept.setUserId(user.getId());
            userDept.setDeptId(deptId);
            userDeptList.add(userDept);
         });
         // 是否加锁成功
         logger.info("是否加锁成功：" + isLock);
         if (isLock) {
            userDeptService.remove(Wrappers.<UserDept>update().lambda().eq(UserDept::getUserId, user.getId()));
            result = userDeptService.saveBatch(userDeptList);
         }
         // 释放锁
         spinLockUtil.unlock(user.getId().toString());
         return result;
      } catch (Exception e) {
         // 释放锁
         spinLockUtil.unlock(user.getId().toString());
         throw new RuntimeException(e);
      }
   }


   @Override
   public IPage<User> selectUserPage(IPage<User> page, User user, String deptId, String tenantId) {
@@ -279,7 +301,7 @@
    * @param tenantId
    * @param account
    * @param password
    * @param type 登录类型 2：pc 3：app
    * @param type     登录类型 2：pc 3：app
    * @return
    */
   @Override
@@ -343,7 +365,6 @@
   }

   /**
    *
    * @param user
    * @param type 登录类型 2：pc 3：app
    * @return
@@ -356,7 +377,7 @@
      userInfo.setUser(user);
      if (Func.isNotEmpty(user)) {
         // 判断登录类型设置对应的角色id
         user.setRoleId(Strings.join(roleService.getRoleIdStrByRoleIdAndType(user.getRoleId(),type), ','));
         user.setRoleId(Strings.join(roleService.getRoleIdStrByRoleIdAndType(user.getRoleId(), type), ','));
         // 设置角色别名
         List<String> roleAlias = roleService.getRoleAliases(user.getRoleId());
         userInfo.setRoles(roleAlias);
@@ -539,6 +560,28 @@
   }

   @Override
   public boolean registerUser(UserVO user) {
      if (StringUtils.isEmpty(user.getAccount())) {
         throw new ServiceException("账号不能为空!");
      }
      if (StringUtils.isEmpty(user.getPassword())) {
         throw new ServiceException("密码不能为空!");
      }
      if (!StringUtils.equals(user.getPassword(), user.getPassword2())) {
         throw new ServiceException("两次的密码不一致!");
      }
      long count = count(Wrappers.<User>lambdaQuery().eq(User::getAccount, user.getAccount()));
      if (count > 0) {
         throw new ServiceException("改账号已存在，请更换其他账号!");
      }

      user.setPassword(DigestUtil.encrypt(user.getPassword()));
      user.setRoleId("1717429059648606209");
      boolean oauthTemp = save(user);
      return (oauthTemp);
   }

   @Override
   public boolean updatePlatform(Long userId, Integer userType, String userExt) {
      if (userType.equals(UserEnum.WEB.getCategory())) {
         UserWeb userWeb = new UserWeb();
@@ -684,7 +727,6 @@
   }



   /**
    * 处理漏绑定的user_dept
    */
@@ -753,7 +795,7 @@
               userInfo.setRoleId(userInfo.getRoleId() + ",1727864411451359233");
            }
            // 设置性别
            setSex(userInfo,userExcel);
            setSex(userInfo, userExcel);
            // 更新机构
            setDeptId(userInfo, userExcel);
            // 更新
@@ -771,7 +813,7 @@
            user.setPhone(userExcel.getPhone());
            user.setAccount(userExcel.getPhone());
            // 设置性别
            setSex(user,userExcel);
            setSex(user, userExcel);
            // 更新机构
            setDeptId(user, userExcel);
            // 设置初始密码
@@ -786,14 +828,15 @@

   /**
    * 设置性别
    *
    * @param user
    * @param userExcel
    */
   private void setSex(User user, PoliceUserExcel userExcel) {
      if (userExcel.getSex().contains("男")){
      if (userExcel.getSex().contains("男")) {
         user.setSex(1);
      }
      if (userExcel.getSex().contains("女")){
      if (userExcel.getSex().contains("女")) {
         user.setSex(2);
      }
   }