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.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 org.springframework.stereotype.Service;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.transaction.annotation.Transactional;


@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;

    /**
     * @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);
        roleMapper.insert(role);
        //新建角色功能权限
        for (RoleFun roleFun : roleVO.getRoleFunList()) {
            roleFun.setRoleId(role.getRoleId());
            roleFun.setCpId(role.getCpId());
        }
        roleFunService.saveRoleFun(roleVO.getRoleFunList());
        //新建角色敏感信息权限
        for (RoleSensitive roleSensitive : roleVO.getRoleSensitiveList()) {
            roleSensitive.setRoleId(role.getRoleId());
            roleSensitive.setCpId(role.getCpId());
        }
        roleSensitiveService.saveRoleSensitive(roleVO.getRoleSensitiveList());
        return ResponseResultUtil.success();
    }

    /**
     * @desc : 编辑角色
     * @author : 于继渤
     * @date : 2023/1/5 9:39
     */
    @Transactional(
            rollbackFor = {Exception.class}
    )
    public ResponseResultVO<?> updateRole(RoleVo roleVO) {
        Role role = roleConvert.convertToPo(roleVO);
        roleMapper.updateById(role);

        //删除历史功能权限
        roleFunService.deleteRoleId(roleVO.getRoleId());
        //删除历史敏感信息权限
        roleSensitiveService.deleteRoleId(roleVO.getRoleId());

        //新建角色功能权限
        for (RoleFun roleFun : roleVO.getRoleFunList()) {
            roleFun.setRoleId(role.getRoleId());
            roleFun.setCpId(role.getCpId());
        }
        roleFunService.saveRoleFun(roleVO.getRoleFunList());
        //新建角色敏感信息权限
        for (RoleSensitive roleSensitive : roleVO.getRoleSensitiveList()) {
            roleSensitive.setRoleId(role.getRoleId());
            roleSensitive.setCpId(role.getCpId());
        }
        roleSensitiveService.saveRoleSensitive(roleVO.getRoleSensitiveList());
        return ResponseResultUtil.success();
    }

}