提交

jiangping

2024-07-12 2ce20c4dd9df60435afe5e6820103a37cf7f684a

 server/service/src/main/java/com/doumee/service/business/impl/UsersServiceImpl.java

@@ -1,8 +1,21 @@
package com.doumee.service.business.impl;

import cn.binarywang.wx.miniapp.bean.WxMaJscode2SessionResult;
import com.doumee.biz.system.SystemDictDataBiz;
import com.doumee.biz.zbom.model.IamUserRoleModel;
import com.doumee.biz.zbom.model.IamUserTypeModel;
import com.doumee.config.Jwt.JwtPayLoad;
import com.doumee.config.Jwt.JwtTokenUtil;
import com.doumee.core.annotation.excel.ExcelImporter;
import com.doumee.core.constants.ResponseStatus;
import com.doumee.core.exception.BusinessException;
import com.doumee.core.model.LoginUserInfo;
import com.doumee.core.model.PageData;
import com.doumee.core.model.PageWrap;
import com.doumee.core.utils.Constants;
import com.doumee.core.utils.Utils;
import com.doumee.dao.admin.request.ShopImport;
import com.doumee.dao.admin.request.UserImport;
import com.doumee.dao.business.ShopMapper;
import com.doumee.core.wx.WxMiniConfig;
import com.doumee.dao.business.UsersMapper;
@@ -23,14 +36,21 @@
import org.apache.commons.lang3.StringUtils;
import me.chanjar.weixin.common.error.WxErrorException;
import org.apache.commons.lang3.StringUtils;
import org.apache.shiro.SecurityUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;
import org.springframework.web.multipart.MultipartFile;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Objects;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

/**
 * 员工信息表Service实现
@@ -40,6 +60,10 @@
@Service
public class UsersServiceImpl implements UsersService {


    ExecutorService executor = Executors.newFixedThreadPool(1);
    @Autowired
    private RedisTemplate<String, Object> redisTemplate;
    @Autowired
    private UsersMapper usersMapper;
    @Autowired
@@ -278,7 +302,7 @@
        if(Objects.isNull(users)){
            throw new BusinessException(ResponseStatus.DATA_EMPTY.getCode(),"未查询到账户信息");
        }
        if(!Constants.equalsInteger(users.getIsdeleted(),Constants.ZERO)){
        if(!Constants.equalsInteger(users.getIsdeleted(), Constants.ZERO)){
            throw new BusinessException(ResponseStatus.NOT_ALLOWED.getCode(),"当前账户已删除");
        }
        if(!StringUtils.equals(users.getStatus(),Constants.ZERO+"")){
@@ -349,4 +373,214 @@
            e.printStackTrace();
        }
    }

    @Override
    @Transactional(rollbackFor = {BusinessException.class,Exception.class})
    public String importBatch(MultipartFile file){
        Boolean importing = (Boolean) redisTemplate.opsForValue().get(Constants.RedisKeys.IMPORTING_USERS);
        if(importing!=null && importing){
            throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(),"对不起，已存在导入或者上下级重置任务正在执行中，请稍后再试！");
        }
        redisTemplate.opsForValue().set(Constants.RedisKeys.IMPORTING_USERS,true);
        try {
            LoginUserInfo loginUserInfo = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
            ExcelImporter ie = null;
            List<UserImport> dataList =null;
            try {
                ie = new ExcelImporter(file,0,0);
                dataList = ie.getDataList(UserImport.class,null);
            }  catch (Exception e) {
                e.printStackTrace();
            }
            if(dataList == null || dataList.size() ==0){
                throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(),"对不起，录入数据为空！");
            }
            Date date =new Date();
            List<UserImport> finalDataList = dataList;
            Callable<String> task = () -> {
                dealUserDataBiz(finalDataList,date,loginUserInfo);
                return "异步任务完成";
            };
            executor.submit(task);

            return "导入成功";
        }catch (BusinessException e){
            throw e;
        }catch (Exception e){
            throw  new BusinessException(ResponseStatus.SERVER_ERROR.getCode(),"信息导入失败，请稍后重试");
        }
    }

    private void dealUserDataBiz(List<UserImport> dataList, Date date, LoginUserInfo loginUserInfo) {
        try {
            List<Users> list =  usersMapper.selectJoinList(Users.class,new MPJLambdaWrapper<Users>()
                    .selectAll(Users.class)
            );
            List<Users> insertList = new ArrayList<>();
            List<Users> updateList = new ArrayList<>();
            for(int i=0;i<dataList.size();i++){
                UserImport model = dataList.get(i);
                if(StringUtils.isBlank(model.getName())
                        &&StringUtils.isBlank(model.getCode()) ){
                    continue;
                }
                Users user = checkModelParam(model,insertList,updateList,i,loginUserInfo,list,date);
            }
            if(insertList .size() >0){
                int temp = 0;
                while(temp < insertList.size()){
                    int index;
                    if(temp + 500 <= insertList.size()){
                        index = temp+500;
                    }else{
                        index = insertList.size();
                    }
                    usersMapper.insert(insertList.subList(temp,index));
                    temp = index;
                }
            }
            if(updateList .size() >0){
                for(Users s : updateList){
                    //遍历更新数据
                    usersMapper.updateById(s);
                }
            }
        }catch (BusinessException e){
            throw e;
        }catch (Exception e){
            throw  new BusinessException(ResponseStatus.SERVER_ERROR.getCode(),"信息导入失败，请稍后重试");
        }finally {
            executor.shutdown();
            redisTemplate.delete(Constants.RedisKeys.IMPORTING_USERS);
        }
    }
    private Users getUserByUserID(String userId, List<Users> list) {
        if(list ==null || list.size() ==0){
            return null;
        }
        for(Users s :list){
            if(StringUtils.equals(s.getIamId(),userId)){
                return s;
            }
        }
        return null;
    }

    private Users checkModelParam(UserImport model, List<Users> insertList, List<Users> updateList, int index, LoginUserInfo loginUserInfo, List<Users> list, Date date) {

            if(StringUtils.isBlank(model.getUserId())){
                throw  new BusinessException(ResponseStatus.BAD_REQUEST.getCode(),"对不起，第"+(index+3)+"行【编码】信息不完整，请检查表格内容！");
            }
            if(StringUtils.isBlank(model.getName())){
                throw  new BusinessException(ResponseStatus.BAD_REQUEST.getCode(),"对不起，第"+(index+3)+"行【姓名】信息不完整，请检查表格内容！");
            }
            if(StringUtils.isBlank(model.getAccountNo())){
                throw  new BusinessException(ResponseStatus.BAD_REQUEST.getCode(),"对不起，第"+(index+3)+"行【登录账号】信息不完整，请检查表格内容！");
            }
            if(StringUtils.isBlank(model.getPhone())){
                throw  new BusinessException(ResponseStatus.BAD_REQUEST.getCode(),"对不起，第"+(index+3)+"行【手机号】信息不完整，请检查表格内容！");
            }
            Users users = getUserByUserID(model.getUserId(),list);
            if(users == null){
                users = new Users();
                users.setCreator(loginUserInfo.getId());
                users.setCreateDate(date);
                initUsersInfo(users,model,date,loginUserInfo);
                insertList.add(users);
            }else {
                initUsersInfo(users,model,date,loginUserInfo);
                updateList.add(users);
            }
            return  users;
        }

    private void initUsersInfo(Users users, UserImport model,Date date,LoginUserInfo loginUserInfo ){
        users.setEditor(loginUserInfo.getId());
        users.setEditDate(date);
        users.setName(model.getName());
        users.setIamUsername(model.getAccountNo());
        users.setPhone(model.getPhone());
        users.setStatus(model.getStatus());
        users.setEmail(model.getEmail());
        users.setIamId(model.getUserId());
        users.setCode(model.getCode());
        users.setSex(model.getSex());
        users.setStatus(model.getStatus());
        users.setIsdeleted(Constants.ZERO);
        users.setRoleIds("");
        users.setRoleNames("");
        users.setRoleCodes("");
        users.setUserTypes("");
        users.setIamOrgId("");
        getDepartidByParam(users,model.getOrgIds());
        setRolesInfo(users,model);
        getUserTypesByParam(users,model.getTypeCodes());
    }

    private String getUserTypesByParam(Users users,String typeCodes) {
        if( typeCodes !=null){
            String[] codes =typeCodes.split(",");
            for(String t : codes){
                if(StringUtils.isBlank(users.getUserTypes())){
                    users.setUserTypes("");
                }
                users.setUserTypes(users.getUserTypes()+"["+t+"];");
            }
        }
        return  users.getUserTypes();
    }

    private void setRolesInfo(Users users, UserImport model) {
        if( model.getRoleCodes() !=null){
            String[] codes = model.getRoleCodes().split(",");
            for(String t : codes){
                if(StringUtils.isBlank(users.getRoleCodes())){
                    users.setRoleCodes("");
                }
                users.setRoleCodes(users.getRoleCodes()+"["+t+"];");
            }
        }
        if( model.getRoleIds() !=null){
            String[] codes = model.getRoleIds().split(",");
            for(String t : codes){
                if(StringUtils.isBlank(users.getRoleIds())){
                    users.setRoleIds("");
                }
                users.setRoleIds(users.getRoleIds()+"["+t+"];");
            }
        }
        if( model.getRoleNames() !=null){
            String[] codes = model.getRoleNames().split(",");
            for(String t : codes){
                if(StringUtils.isBlank(users.getRoleNames())){
                    users.setRoleNames("");
                }
                users.setRoleNames(users.getRoleNames()+"["+t+"];");
            }
        }
    }

    private Shop getDepartidByParam(Users users , String orgIds) {
        if( orgIds ==null){
          return null;
        }
        String[] orgArray = orgIds.split(",");
        if( orgArray !=null && orgArray.length>0){
            for(String t : orgArray){
                Shop shop = shopMapper.selectOne(new QueryWrapper<Shop>().lambda()
                        .eq(Shop::getOrgId,t)
                        .last("limit 1"));
                if(shop!=null) {
                    users.setIamOrgId(t);
                    users.setDepartmentId(shop.getId());
                    return  shop;
                }
            }
        }
        users.setIamOrgId(orgIds+"--尚未同步到组织");

        return null;
    }

}