package com.ruoyi.web.controller.system;
|
|
import cn.afterturn.easypoi.excel.ExcelExportUtil;
|
import cn.afterturn.easypoi.excel.ExcelImportUtil;
|
import cn.afterturn.easypoi.excel.entity.ExportParams;
|
import cn.afterturn.easypoi.excel.entity.ImportParams;
|
import cn.hutool.json.JSONObject;
|
import com.ruoyi.common.annotation.Log;
|
import com.ruoyi.common.basic.PageInfo;
|
import com.ruoyi.common.core.controller.BaseController;
|
import com.ruoyi.common.core.domain.AjaxResult;
|
import com.ruoyi.common.core.domain.R;
|
import com.ruoyi.common.core.domain.entity.SysDept;
|
import com.ruoyi.common.core.domain.entity.SysRole;
|
import com.ruoyi.common.core.domain.entity.SysUser;
|
import com.ruoyi.common.core.domain.entity.TDept;
|
import com.ruoyi.common.core.exception.ServiceException;
|
import com.ruoyi.common.enums.BusinessType;
|
import com.ruoyi.common.utils.SecurityUtils;
|
import com.ruoyi.common.utils.StringUtils;
|
import com.ruoyi.common.utils.WebUtils;
|
import com.ruoyi.framework.web.service.TokenService;
|
import com.ruoyi.system.dto.SysUserUpdateStatusDTO;
|
import com.ruoyi.system.importExcel.EmployeeImportExcel;
|
import com.ruoyi.system.query.SysUserQuery;
|
import com.ruoyi.system.service.*;
|
import com.ruoyi.system.vo.system.SysUserVO;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import io.swagger.annotations.ApiParam;
|
import org.apache.poi.ss.usermodel.Workbook;
|
import org.springframework.beans.BeanUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.transaction.annotation.Propagation;
|
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.util.CollectionUtils;
|
import org.springframework.validation.annotation.Validated;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.servlet.ServletOutputStream;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.IOException;
|
import java.io.InputStream;
|
import java.net.URLEncoder;
|
import java.text.ParseException;
|
import java.text.SimpleDateFormat;
|
import java.time.LocalDateTime;
|
import java.util.*;
|
import java.util.stream.Collectors;
|
|
/**
|
* 用户信息
|
*
|
* @author ruoyi
|
*/
|
@Api(tags = "用户信息")
|
@RestController
|
@RequestMapping("/system/user")
|
public class SysUserController extends BaseController {
|
@Autowired
|
private ISysUserService userService;
|
|
@Autowired
|
private ISysRoleService roleService;
|
|
@Autowired
|
private ISysDeptService deptService;
|
@Autowired
|
private TDeptService tDeptService;
|
@Autowired
|
private TokenService tokenService;
|
|
|
/**
|
* 获取用户列表
|
*/
|
@ApiOperation(value = "获取用户列表")
|
@PostMapping("/list")
|
public AjaxResult<PageInfo<SysUserVO>> list(@RequestBody SysUserQuery query) {
|
PageInfo<SysUserVO> list = userService.pageList(query);
|
return AjaxResult.success(list);
|
}
|
|
/**
|
* 获取用户列表
|
*/
|
@ApiOperation(value = "根据部门id获取用户列表")
|
@GetMapping("/listByDeptId")
|
public AjaxResult<PageInfo<SysUserVO>> listByDeptId(@ApiParam(value = "部门id")Integer deptId,
|
@ApiParam(value = "分页参数，当前页码")
|
Integer pageNum,
|
@ApiParam(value = "分页参数，每页数量，默认为10") Integer pageSize ) {
|
PageInfo<SysUserVO> list = userService.listByDeptId(deptId,pageNum,pageSize);
|
return AjaxResult.success(list);
|
}
|
@ApiOperation(value = "获取用户列表-不分页")
|
@PostMapping("/listNotPage")
|
|
public AjaxResult listNotPage() {
|
List<SysUser> list = userService.selectList();
|
return AjaxResult.success(list);
|
}
|
|
|
/**
|
* 获取用户详情
|
*/
|
@ApiOperation(value = "获取用户详情")
|
@GetMapping("/getDetail")
|
public AjaxResult getDetail(@RequestParam Long userId) {
|
SysUser sysUser = userService.selectUserById(userId);
|
SysUserVO sysUserVO = new SysUserVO();
|
BeanUtils.copyProperties(sysUser, sysUserVO);
|
|
return AjaxResult.success(sysUser);
|
}
|
|
/**
|
* 获取用户详情
|
*/
|
@ApiOperation(value = "获取用户详情")
|
@GetMapping("/queryDetail")
|
public AjaxResult queryDetail() {
|
Long userId = tokenService.getLoginUser().getUserId();
|
SysUser sysUser = userService.selectUserById(userId);
|
SysUserVO sysUserVO = new SysUserVO();
|
BeanUtils.copyProperties(sysUser, sysUserVO);
|
|
return AjaxResult.success(sysUser);
|
}
|
|
|
/**
|
* 获取用户数量统计
|
*/
|
@ApiOperation(value = "获取用户数量统计")
|
@PostMapping("/getUserCount")
|
public AjaxResult getUserCount() {
|
Map<String, Integer> map = new HashMap<>();
|
|
Integer userCountSum = userService.selectCount(null);
|
Integer normalCount = userService.selectCount(0);// 正常
|
Integer stopCount = userService.selectCount(1);// 停用
|
|
map.put("all", userCountSum);
|
map.put("normal", normalCount);
|
map.put("stop", stopCount);
|
|
return AjaxResult.success(map);
|
}
|
|
/**
|
* 移除黑名单
|
*/
|
@GetMapping("/removeBlackList")
|
public AjaxResult removeBlackList(@RequestParam String ids) {
|
String[] split = ids.split(",");
|
List<Long> id = new ArrayList<>();
|
for (String s : split) {
|
id.add(Long.valueOf(s));
|
}
|
userService.updateUserIfBlack(id);
|
return AjaxResult.success();
|
}
|
|
/**
|
* 新增用户
|
*/
|
@ApiOperation(value = "新增用户管理")
|
@Log(title = "用户信息-新增用户", businessType = BusinessType.INSERT)
|
@PostMapping("/add")
|
public AjaxResult add(@Validated @RequestBody SysUser user) {
|
user.setUserName(user.getUserName());
|
if (!userService.checkUserNameUnique(user)) {
|
return error("新增用户'" + user.getUserName() + "'失败，登录账号已存在");
|
} else if (StringUtils.isNotEmpty(user.getPhonenumber()) && !userService.checkPhoneUnique(user)) {
|
return error("新增用户'" + user.getUserName() + "'失败，手机号码已存在");
|
}
|
user.setUserName(user.getPhonenumber());
|
user.setCreateBy(getUsername());
|
user.setPassword(SecurityUtils.encryptPassword("123456"));
|
userService.insertUser(user);
|
return AjaxResult.success();
|
}
|
|
@Log(title = "员工导入", businessType = BusinessType.INSERT)
|
@ApiOperation(value = "员工导入")
|
@Transactional(rollbackFor = Exception.class, propagation = Propagation.REQUIRES_NEW)
|
@PostMapping("/importUser")
|
public R<Object> importUser(@RequestPart("file") MultipartFile file) {
|
ImportParams params = new ImportParams();
|
params.setHeadRows(1); // 表头行数
|
InputStream inputStream = null;
|
List<EmployeeImportExcel> employeeExcelList;
|
|
try {
|
inputStream = file.getInputStream();
|
employeeExcelList = ExcelImportUtil.importExcel(inputStream,
|
EmployeeImportExcel.class, params);
|
} catch (Exception e) {
|
e.printStackTrace();
|
throw new ServiceException("员工导入失败: " + e.getMessage());
|
} finally {
|
if (inputStream != null) {
|
try {
|
inputStream.close();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
|
if (CollectionUtils.isEmpty(employeeExcelList)) {
|
throw new ServiceException("员工数据为空!");
|
}
|
|
int successCount = 0;
|
int failCount = 0;
|
List<String> errorMessages = new ArrayList<>();
|
|
for (EmployeeImportExcel employeeExcel : employeeExcelList) {
|
try {
|
// 数据验证
|
|
|
// 数据验证
|
if (StringUtils.isEmpty(employeeExcel.getNickName())) {
|
errorMessages.add("第" + (successCount + failCount + 1) + "行：姓名为空");
|
failCount++;
|
continue;
|
}
|
|
if (StringUtils.isEmpty(employeeExcel.getPhoneNumber())) {
|
errorMessages.add("第" + (successCount + failCount + 1) + "行：联系电话为空");
|
failCount++;
|
continue;
|
}
|
|
// 转换并保存用户
|
SysUser sysUser = convertToSysUser(employeeExcel);
|
if (StringUtils.isEmpty(employeeExcel.getDepartment())) {
|
errorMessages.add("第" + (successCount + failCount + 1) + "行：所在股室为空");
|
failCount++;
|
continue;
|
}
|
|
TDept dept = tDeptService.lambdaQuery().eq(TDept::getDeptName, employeeExcel.getDepartment())
|
.last("limit 1").one();
|
if (dept == null){
|
errorMessages.add("第" + (successCount + failCount + 1) + "行：所在股室不存在");
|
failCount++;
|
continue;
|
}
|
sysUser.setDeptId(dept.getId().toString());
|
userService.insertUser(sysUser);
|
successCount++;
|
|
} catch (Exception e) {
|
errorMessages.add("第" + (successCount + failCount + 1) + "行：" + e.getMessage());
|
failCount++;
|
e.printStackTrace();
|
}
|
}
|
|
JSONObject result = new JSONObject();
|
result.put("successCount", successCount);
|
result.put("failCount", failCount);
|
result.put("totalCount", successCount + failCount);
|
if (!errorMessages.isEmpty()) {
|
result.put("errorMessages", errorMessages);
|
return R.fail(result.get("errorMessages"));
|
}
|
|
return R.ok("导入完成", result.toString());
|
}
|
|
/**
|
* 解析日期字符串为Date对象
|
* @param dateStr 日期字符串，格式如：2022.09
|
* @return Date对象
|
*/
|
private Date parseDate(String dateStr) {
|
if (StringUtils.isEmpty(dateStr)) {
|
return null;
|
}
|
try {
|
// 支持 YYYY.MM 格式
|
SimpleDateFormat sdf = new SimpleDateFormat("yyyy/MM/dd");
|
return sdf.parse(dateStr);
|
} catch (ParseException e) {
|
try {
|
// 支持 YYYY-MM 格式
|
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM");
|
return sdf.parse(dateStr);
|
} catch (ParseException ex) {
|
try {
|
// 支持 YYYY-MM-DD 格式
|
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
|
return sdf.parse(dateStr);
|
} catch (ParseException exc) {
|
return null;
|
}
|
}
|
}
|
}
|
|
/**
|
* 将EmployeeImportExcel转换为SysUser
|
* @param employeeExcel 员工导入Excel数据
|
* @return SysUser对象
|
*/
|
private SysUser convertToSysUser(EmployeeImportExcel employeeExcel) {
|
SysUser sysUser = new SysUser();
|
sysUser.setPassword(SecurityUtils.encryptPassword("123456"));
|
|
// 基本信息
|
sysUser.setNickName(employeeExcel.getNickName());
|
sysUser.setUserName(employeeExcel.getPhoneNumber());
|
sysUser.setPhonenumber(employeeExcel.getPhoneNumber());
|
// 性别转换
|
if ("男".equals(employeeExcel.getSex())) {
|
sysUser.setSex("0");
|
} else if ("女".equals(employeeExcel.getSex())) {
|
sysUser.setSex("1");
|
} else {
|
sysUser.setSex("0");
|
}
|
|
// 身份证
|
sysUser.setIdCard(employeeExcel.getIdCard());
|
|
// 政治面貌
|
sysUser.setPoliticalOutlook(employeeExcel.getPoliticalOutlook());
|
|
// 时间字段
|
sysUser.setParticipationTime(parseDate(employeeExcel.getParticipationTime()));
|
sysUser.setCompanyTime(parseDate(employeeExcel.getCompanyTime()));
|
sysUser.setPositionTime(parseDate(employeeExcel.getPositionTime()));
|
// 学历信息
|
sysUser.setFirstDegree(employeeExcel.getFirstDegree());
|
sysUser.setFirstCollege(employeeExcel.getFirstCollege());
|
sysUser.setFirstMajor(employeeExcel.getFirstMajor());
|
sysUser.setHighestDegree(employeeExcel.getHighestDegree());
|
sysUser.setHighestCollege(employeeExcel.getHighestCollege());
|
sysUser.setHighestMajor(employeeExcel.getHighestMajor());
|
|
// 职务和部门
|
sysUser.setPosition(employeeExcel.getPosition());
|
sysUser.setOrganization(employeeExcel.getOrganization());
|
|
// 系统字段
|
sysUser.setStatus("0"); // 正常状态
|
sysUser.setDelFlag("0"); // 未删除
|
sysUser.setCreateBy(getUsername());
|
|
return sysUser;
|
}
|
|
@ApiOperation(value = "员工导入模板下载")
|
@GetMapping("/importDownload")
|
public void importDownload() {
|
List<EmployeeImportExcel> employeeImportExcels = new ArrayList<>();
|
EmployeeImportExcel employeeImportExcel = new EmployeeImportExcel();
|
employeeImportExcels.add(employeeImportExcel);
|
ExportParams exportParams = new ExportParams();
|
exportParams.setTitle("员工信息导入模板");
|
exportParams.setSheetName("员工信息");
|
Workbook workbook = ExcelExportUtil.exportExcel(exportParams, EmployeeImportExcel.class, employeeImportExcels);
|
HttpServletResponse response = WebUtils.response();
|
ServletOutputStream outputStream = null;
|
try {
|
String fileName = URLEncoder.encode("员工信息导入模板.xls", "utf-8");
|
response.setHeader("Content-Disposition", "attachment;filename=" + fileName);
|
response.setHeader("content-Type", "application/vnd.ms-excel");
|
response.setHeader("Pragma", "no-cache");
|
response.setHeader("Cache-Control", "no-cache");
|
outputStream = response.getOutputStream();
|
workbook.write(outputStream);
|
} catch (IOException e) {
|
e.printStackTrace();
|
System.out.println("员工导入模板下载失败!");
|
} finally {
|
if (outputStream != null) {
|
try {
|
outputStream.close();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
}
|
|
/**
|
* 修改用户
|
*/
|
// @PreAuthorize("@ss.hasPermi('system:user:edit')")
|
@ApiOperation(value = "修改用户管理")
|
@Log(title = "用户信息-修改用户", businessType = BusinessType.UPDATE)
|
@PostMapping("/edit")
|
public AjaxResult edit(@Validated @RequestBody SysUser user) {
|
user.setUserName(user.getPhonenumber());
|
// userService.checkUserAllowed(user);
|
// userService.checkUserDataScope(user.getUserId());
|
if (!userService.checkUserNameUnique(user)) {
|
return error("修改用户'" + user.getUserName() + "'失败，登录账号已存在");
|
} else if (StringUtils.isNotEmpty(user.getPhonenumber()) && !userService.checkPhoneUnique(user)) {
|
return error("修改用户'" + user.getUserName() + "'失败，手机号码已存在");
|
}
|
user.setUserName(user.getPhonenumber());
|
user.setUpdateBy(getUsername());
|
if (StringUtils.isNotEmpty(user.getPassword())) {
|
user.setPassword(SecurityUtils.encryptPassword(user.getPassword()));
|
}
|
return AjaxResult.success(userService.updateUser(user));
|
}
|
|
/**
|
* 删除用户
|
*/
|
// @PreAuthorize("@ss.hasPermi('system:user:remove')")
|
@ApiOperation(value = "批量删除用户")
|
@Log(title = "用户信息-批量删除用户", businessType = BusinessType.DELETE)
|
@DeleteMapping("/deleteById/{ids}")
|
public AjaxResult remove(@PathVariable String ids) {
|
String[] split = ids.split(",");
|
List<Long> userIds = new ArrayList<>();
|
for (String s : split) {
|
userIds.add(Long.valueOf(s));
|
}
|
if (userIds.contains(getUserId())) {
|
return error("当前用户不能删除");
|
}
|
return AjaxResult.success(userService.deleteUserByIds(userIds));
|
}
|
|
/**
|
* 重置密码
|
*/
|
// @PreAuthorize("@ss.hasPermi('system:user:resetPwd')")
|
@ApiOperation(value = "重置密码")
|
@Log(title = "用户信息-重置密码", businessType = BusinessType.UPDATE)
|
@PostMapping("/resetPwd")
|
public AjaxResult resetPwd(@RequestBody SysUser user) {
|
userService.checkUserAllowed(user);
|
// userService.checkUserDataScope(user.getUserId());
|
user.setPassword(SecurityUtils.encryptPassword(user.getPassword()));
|
user.setUpdateBy(getUsername());
|
return AjaxResult.success(userService.resetPwd(user));
|
}
|
|
/**
|
* 状态修改
|
*/
|
@ApiOperation(value = "状态修改/启用禁用")
|
@Log(title = "用户信息-状态修改", businessType = BusinessType.UPDATE)
|
@PutMapping("/changeStatus")
|
public AjaxResult changeStatus(@RequestBody SysUserUpdateStatusDTO dto) {
|
SysUser loginUser = tokenService.getLoginUser().getUser();
|
SysUser user = new SysUser();
|
user.setUserId(dto.getUserId());
|
user.setStatus(String.valueOf(dto.getStatus()));
|
// user.setRemark(dto.getRemark());
|
user.setUpdateBy(getUsername());
|
user.setDisableRemark(dto.getRemark());
|
user.setOperatingTime(LocalDateTime.now());
|
user.setOperatingPerson(loginUser.getNickName() + "(" + loginUser.getUserName() + ")");
|
return AjaxResult.success(userService.updateUserStatus(user));
|
}
|
|
/**
|
* 根据用户编号获取授权角色
|
*/
|
// @PreAuthorize("@ss.hasPermi('system:user:query')")
|
@GetMapping("/authRole/{userId}")
|
public AjaxResult authRole(@PathVariable("userId") Long userId) {
|
AjaxResult ajax = AjaxResult.success();
|
SysUser user = userService.selectUserById(userId);
|
List<SysRole> roles = roleService.selectRolesByUserId(userId);
|
ajax.put("user", user);
|
ajax.put("roles", SysUser.isAdmin(userId) ? roles : roles.stream().filter(r -> !r.isAdmin()).collect(Collectors.toList()));
|
return ajax;
|
}
|
|
/**
|
* 用户授权角色
|
*/
|
// @PreAuthorize("@ss.hasPermi('system:user:edit')")
|
@Log(title = "用户管理", businessType = BusinessType.GRANT)
|
@PutMapping("/authRole")
|
public AjaxResult insertAuthRole(Long userId, Long[] roleIds) {
|
userService.checkUserDataScope(userId);
|
userService.insertUserAuth(userId, roleIds);
|
return AjaxResult.success();
|
}
|
|
/**
|
* 获取部门树列表
|
*/
|
// @PreAuthorize("@ss.hasPermi('system:user:list')")
|
@GetMapping("/deptTree")
|
public AjaxResult deptTree(SysDept dept) {
|
return AjaxResult.success(deptService.selectDeptTreeList(dept));
|
}
|
}
|
