iboss-server-mdm/src/main/java/com/dk/mdm/service/mst/RoleService.java

package com.dk.mdm.service.mst;

import com.dk.common.infrastructure.annotaiton.Pagination;
import com.dk.common.infrastructure.constant.Constant;
import com.dk.common.model.pojo.PageList;
import com.dk.common.response.ResponseResultUtil;
import com.dk.common.response.ResponseResultVO;
import com.dk.mdm.infrastructure.convert.mst.RoleConvert;
import com.dk.mdm.model.pojo.mst.Role;
import com.dk.mdm.mapper.mst.RoleMapper;
import com.dk.common.service.BaseService;
import com.dk.common.mapper.BaseMapper;
import com.dk.mdm.model.pojo.mst.RoleFun;
import com.dk.mdm.model.pojo.mst.RoleSensitive;
import com.dk.mdm.model.query.mst.RoleQuery;
import com.dk.mdm.model.response.mst.RoleResponse;
import com.dk.mdm.model.vo.mst.RoleVo;
import com.dk.mdm.service.common.CommonService;
import org.springframework.stereotype.Service;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.transaction.annotation.Transactional;

import java.util.Map;


@Service
@Transactional
public class RoleService extends BaseService<Role> {

    @Override
    public BaseMapper<Role> getRepository() {
        return roleMapper;
    }

    @Autowired
    private RoleMapper roleMapper;

    @Autowired
    private RoleFunService roleFunService;

    @Autowired
    private RoleSensitiveService roleSensitiveService;

    @Autowired
    private RoleConvert roleConvert;

    @Autowired
    private CommonService commonService;
    /**
     * @desc : 重写主键
     * @author : 周兴
     * @date : 2024/2/29 20:29
     */
    @Override
    public String getPrimaryKey() {
        return "role_id";
    }

    /**
     * @desc : 查询角色
     * @author : 于继渤
     * @date : 2023/1/5 9:39
     */
    @Pagination
    public ResponseResultVO<PageList<RoleResponse>> selectByCond(RoleQuery roleQuery) {
        return super.mergeListWithCount(roleQuery, roleMapper.selectByCond(roleQuery),
                roleMapper.countByCond(roleQuery));
    }

    /**
     * @desc : 新建角色
     * @author : 于继渤
     * @date : 2023/1/5 9:39
     */
    @Transactional(
            rollbackFor = {Exception.class}
    )
    public ResponseResultVO<?> saveRole(RoleVo roleVO) {
        Role role = roleConvert.convertToPo(roleVO);
        Integer displayNo = roleMapper.selectDisplayNo(new RoleQuery());
        role.setDisplayNo(displayNo);
        //设置编码
        Map<String, Object> uniqueNoteCode = commonService.getUniqueNoteCode(Constant.docNameConstant.ROLE.getName(), true);
        role.setRoleId(uniqueNoteCode.get("outId").toString());
        role.setRoleCode(uniqueNoteCode.get("outNote").toString());
        //新建
        roleMapper.insert(role);
        //设置角色权限明细
        setRoleItem(roleVO, role);

        return ResponseResultUtil.success();
    }

    private void setRoleItem(RoleVo roleVO, Role role) {
        if (roleVO.getRoleFunList() != null && roleVO.getRoleFunList().size() > 0) {
            //新建角色功能权限
            for (RoleFun roleFun : roleVO.getRoleFunList()) {
                roleFun.setRoleId(role.getRoleId());
                roleFun.setCpId(role.getCpId());
            }
            roleFunService.saveRoleFun(roleVO.getRoleFunList());
        }
        if (roleVO.getRoleSensitiveList() != null && roleVO.getRoleSensitiveList().size() > 0) {
            //新建角色敏感信息权限
            for (RoleSensitive roleSensitive : roleVO.getRoleSensitiveList()) {
                roleSensitive.setRoleId(role.getRoleId());
                roleSensitive.setCpId(role.getCpId());
            }
            roleSensitiveService.saveRoleSensitive(roleVO.getRoleSensitiveList());
        }
    }

    /**
     * @desc : 编辑角色
     * @author : 于继渤
     * @date : 2023/1/5 9:39
     */
    @Transactional(
            rollbackFor = {Exception.class}
    )
    public ResponseResultVO<?> updateRole(RoleVo roleVO) {
        Role role = roleConvert.convertToPo(roleVO);
        updateByUuid(role);
        if (roleVO.getRoleFunList() != null) {
            //删除历史功能权限
            roleFunService.deleteRoleId(roleVO.getRoleId());
        }
        if (roleVO.getRoleSensitiveList() != null) {
            //删除历史敏感信息权限
            roleSensitiveService.deleteRoleId(roleVO.getRoleId());
        }

        //设置角色权限明细
        setRoleItem(roleVO, role);
        return ResponseResultUtil.success();
    }



}