epmet-cloud-lingshan/epmet-user/epmet-user-server/src/main/java/com/epmet/controller/StaffRoleController.java

package com.epmet.controller;

import com.epmet.commons.tools.utils.Result;
import com.epmet.commons.tools.validator.ValidatorUtils;
import com.epmet.dto.GovStaffRoleDTO;
import com.epmet.dto.form.StaffRoleFormDTO;
import com.epmet.entity.GovStaffRoleEntity;
import com.epmet.service.GovStaffRoleService;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.ArrayList;
import java.util.List;

/**
 * @return
 * 工作人员相关api
 * @Author wxz
 * @Description
 * @Date 2020/4/22 22:44
 **/
@RestController
@RequestMapping("staffrole")
public class StaffRoleController {

    @Autowired
    private GovStaffRoleService govStaffRoleService;

    /**
     * 查询工作人员具有的角色列表
     * @return
     */
    @PostMapping("staffroles")
    public Result<List<GovStaffRoleDTO>> getRolesOfStaff(@RequestBody StaffRoleFormDTO staffRoleFormDTO) {
        ValidatorUtils.validateEntity(staffRoleFormDTO, StaffRoleFormDTO.GetRolesOfStaff.class);
        String staffId = staffRoleFormDTO.getStaffId();
        String orgId = staffRoleFormDTO.getOrgId();
        List<GovStaffRoleEntity> staffRoleEntities = govStaffRoleService.listRolesByStaffId(staffId, orgId);
        List<GovStaffRoleDTO> staffRoleDTOS = new ArrayList<>();
        staffRoleEntities.forEach(role -> {
            GovStaffRoleDTO dto = new GovStaffRoleDTO();
            BeanUtils.copyProperties(role, dto);
            staffRoleDTOS.add(dto);
        });
        return new Result<List<GovStaffRoleDTO>>().ok(staffRoleDTOS);
    }

    /**
     * 查询拥有指定角色的用户列表
     * @param staffRoleFormDTO
     * @return
     */
    @PostMapping("staffsinrole")
    public Result<List<GovStaffRoleDTO>> getStaffsInRole(@RequestBody StaffRoleFormDTO staffRoleFormDTO) {
        ValidatorUtils.validateEntity(staffRoleFormDTO, StaffRoleFormDTO.GetStaffsInRole.class);
        String roleKey = staffRoleFormDTO.getRoleKey();
        String orgId = staffRoleFormDTO.getOrgId();
        List<GovStaffRoleDTO> roleDTOS = govStaffRoleService.listStaffsInRole(roleKey, orgId);
        return new Result<List<GovStaffRoleDTO>>().ok(roleDTOS);
    }

}