weaver-hrm-salary/src/com/engine/salary/service/auth/AuthMemberServiceImpl.java

package com.engine.salary.service.auth;

import cn.hutool.core.collection.CollUtil;
import com.engine.common.util.ServiceUtil;
import com.engine.core.impl.Service;
import com.engine.hrmelog.entity.dto.LoggerContext;
import com.engine.salary.config.SalaryElogConfig;
import com.engine.salary.constant.SalaryDefaultTenantConstant;
import com.engine.salary.entity.auth.param.AuthMemberSaveParam;
import com.engine.salary.entity.auth.po.AuthMemberPO;
import com.engine.salary.entity.auth.po.AuthRoleEmpPO;
import com.engine.salary.entity.auth.po.AuthRolePO;
import com.engine.salary.entity.taxagent.po.TaxAgentPO;
import com.engine.salary.enums.OperateTypeEnum;
import com.engine.salary.enums.auth.TargetTypeEnum;
import com.engine.salary.exception.SalaryRunTimeException;
import com.engine.salary.mapper.auth.AuthMemberMapper;
import com.engine.salary.mapper.auth.AuthRoleEmpMapper;
import com.engine.salary.mapper.auth.AuthRoleMapper;
import com.engine.salary.service.TaxAgentService;
import com.engine.salary.service.impl.TaxAgentServiceImpl;
import com.engine.salary.util.SalaryEntityUtil;
import com.engine.salary.util.SalaryI18nUtil;
import com.engine.salary.util.db.IdGenerator;
import com.engine.salary.util.db.MapperProxyFactory;
import com.google.common.collect.Lists;
import weaver.hrm.User;

import java.util.*;
import java.util.stream.Collectors;

public class AuthMemberServiceImpl extends Service implements AuthMemberService {


    private AuthMemberMapper getAuthMemberMapper() {
        return MapperProxyFactory.getProxy(AuthMemberMapper.class);
    }

    private AuthRoleMapper getAuthRoleMapper() {
        return MapperProxyFactory.getProxy(AuthRoleMapper.class);
    }

    private AuthRoleEmpMapper getAuthRoleEmpMapper() {
        return MapperProxyFactory.getProxy(AuthRoleEmpMapper.class);
    }


    private TaxAgentService getTaxAgentService(User user) {
        return ServiceUtil.getService(TaxAgentServiceImpl.class, user);
    }


    @Override
    public Long saveMember(AuthMemberSaveParam param) {

        AuthRolePO rolePO = getAuthRoleMapper().getById(param.getRoleId());
        if (rolePO == null) {
            throw new SalaryRunTimeException("角色不存在！");
        }

        Date now = new Date();
        AuthMemberPO po = AuthMemberPO.builder()
                .id(IdGenerator.generate())
                .roleId(param.getRoleId())
                .target(param.getTarget())
                .targetType(param.getTargetType())
                .targetName(param.getTargetName())
                .creator((long) user.getUID())
                .createTime(now)
                .updateTime(now)
                .deleteType(0)
                .tenantKey(SalaryDefaultTenantConstant.DEFAULT_TENANT_KEY)
                .build();
        getAuthMemberMapper().insertIgnoreNull(po);

        LoggerContext<TaxAgentPO> loggerContext = new LoggerContext<>();
        loggerContext.setUser(user);
        loggerContext.setTargetId(rolePO.getTaxAgentId() + "");
        loggerContext.setTargetName(rolePO.getName() + "成员：" + po.getId());
        loggerContext.setOperateType(OperateTypeEnum.ADD.getValue());
        loggerContext.setOperateTypeName(SalaryI18nUtil.getI18nLabel(0, "新增成员"));
        loggerContext.setOperatedesc(SalaryI18nUtil.getI18nLabel(0, "新增成员"));
        loggerContext.setNewValues(po);
        SalaryElogConfig.taxAgentLoggerTemplate.write(loggerContext);


        return po.getId();
    }

    @Override
    public void deleteMember(Long id) {
        AuthMemberPO po = getAuthMemberMapper().getById(id);
        if (po == null) {
            throw new SalaryRunTimeException("成员不存在！");
        }

        getAuthMemberMapper().deleteByIds(Collections.singleton(id));

        AuthRolePO rolePO = getAuthRoleMapper().getById(po.getRoleId());
        LoggerContext<TaxAgentPO> loggerContext = new LoggerContext<>();
        loggerContext.setUser(user);
        loggerContext.setTargetId(rolePO.getTaxAgentId() + "");
        loggerContext.setTargetName(rolePO.getName() + "成员：" + po.getId());
        loggerContext.setOperateType(OperateTypeEnum.DELETE.getValue());
        loggerContext.setOperateTypeName(SalaryI18nUtil.getI18nLabel(0, "删除成员"));
        loggerContext.setOperatedesc(SalaryI18nUtil.getI18nLabel(0, "删除成员"));
        loggerContext.setNewValues(po);
        SalaryElogConfig.taxAgentLoggerTemplate.write(loggerContext);
    }

    @Override
    public void syncMember(Long roleId) {
        AuthRolePO rolePO = getAuthRoleMapper().getById(roleId);
        if (rolePO == null) {
            throw new SalaryRunTimeException("角色不存在！");
        }

        //1、获取成员
        List<AuthMemberPO> members = getAuthMemberMapper().listSome(AuthMemberPO.builder().roleId(roleId).build());
        Map<Integer, Set<String>> targetTypeMap = SalaryEntityUtil.group2Map(members, AuthMemberPO::getTargetType, AuthMemberPO::getTarget);

        //2、获取人员
        Set<Long> empIds = new HashSet<>();
        for (Integer targetType : targetTypeMap.keySet()) {
            Set<String> targetIds = targetTypeMap.get(targetType);
            List<Long> ids = TargetTypeEnum.parseByValue(targetType).getEmpIds(targetIds);
            empIds.addAll(ids);
        }

        //3、更新成员关系
        getAuthRoleEmpMapper().deleteByRoleId(roleId);
        Date now = new Date();
        List<AuthRoleEmpPO> roleEmpPOS = empIds.stream()
                .map(empId -> AuthRoleEmpPO.builder()
                        .id(IdGenerator.generate())
                        .employeeId(empId)
                        .roleId(roleId)
                        .creator((long) user.getUID())
                        .createTime(now)
                        .updateTime(now)
                        .deleteType(0)
                        .tenantKey(SalaryDefaultTenantConstant.DEFAULT_TENANT_KEY)
                        .build())
                .collect(Collectors.toList());
        if (CollUtil.isNotEmpty(roleEmpPOS)) {
            List<List<AuthRoleEmpPO>> partition = Lists.partition(roleEmpPOS, 100);
            partition.forEach(list -> getAuthRoleEmpMapper().batchInsert(list));
        }
    }
}