package com.mindskip.xzs.controller.admin; import com.github.pagehelper.PageInfo; import com.mindskip.xzs.base.BaseApiController; import com.mindskip.xzs.base.RestResponse; import com.mindskip.xzs.domain.*; import com.mindskip.xzs.domain.enums.UserStatusEnum; import com.mindskip.xzs.domain.other.KeyValue; import com.mindskip.xzs.domain.vo.UserVO; import com.mindskip.xzs.repository.UserDepartmentMapper; import com.mindskip.xzs.service.*; import com.mindskip.xzs.utility.DateTimeUtil; import com.mindskip.xzs.utility.PageInfoHelper; import com.mindskip.xzs.utility.convert.UserClassConvert; import com.mindskip.xzs.utility.excel.ExcelUtils; import com.mindskip.xzs.viewmodel.admin.user.*; import org.apache.commons.lang3.ObjectUtils; import org.apache.commons.lang3.StringUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; import javax.validation.Valid; import java.util.*; import java.util.stream.Collectors; @RestController("AdminUserController") @RequestMapping(value = "/api/admin/user") public class UserController extends BaseApiController { private final UserService userService; private final UserEventLogService userEventLogService; private final AuthenticationService authenticationService; private final DepartmentService departmentService; private final UserDepartmentMapper userDepartmentMapper; private final TagService tagService; @Autowired public UserController(UserService userService, UserEventLogService userEventLogService, AuthenticationService authenticationService, DepartmentService departmentService, UserDepartmentMapper userDepartmentMapper, TagService tagService) { this.userService = userService; this.userEventLogService = userEventLogService; this.authenticationService = authenticationService; this.departmentService = departmentService; this.userDepartmentMapper = userDepartmentMapper; this.tagService = tagService; } @RequestMapping(value = "/page/list", method = RequestMethod.POST) public RestResponse> pageList(@RequestBody UserPageRequestVM model) { model.setDepartmentId(ObjectUtils.isNotEmpty(model.getDepartmentId()) ? model.getDepartmentId() : getAdminDeptIds()); PageInfo page = userService.userPage(model); page.setList(page.getList().stream().map(e->{ List userDepartments = userDepartmentMapper.selectByUserId(e.getId()); List list = new ArrayList<>(); for (UserDepartment userDepartment : userDepartments) { Department byId = departmentService.getById(userDepartment.getDepartmentId()); if (ObjectUtils.isNotEmpty(byId)) { list.add(byId); } } e.setDeptNames(StringUtils.join(list.stream().map(Department::getName).collect(Collectors.toList()), ",")); e.setDeptIdList(list.stream().map(Department::getId).collect(Collectors.toList())); e.setTagNames(tagService.selectTagNamesByUserId(e.getId())); return e; }).collect(Collectors.toList())); return RestResponse.ok(page); } @RequestMapping(value = "/event/page/list", method = RequestMethod.POST) public RestResponse> eventPageList(@RequestBody UserEventPageRequestVM model) { PageInfo pageInfo = userEventLogService.page(model); PageInfo page = PageInfoHelper.copyMap(pageInfo, d -> { UserEventLogVM vm = modelMapper.map(d, UserEventLogVM.class); vm.setCreateTime(DateTimeUtil.dateFormat(d.getCreateTime())); return vm; }); return RestResponse.ok(page); } @RequestMapping(value = "/select/{id}", method = RequestMethod.POST) public RestResponse select(@PathVariable Integer id) { User user = userService.getUserById(id); UserResponseVM userVm = UserResponseVM.from(user); List userDepartments = userDepartmentMapper.selectByUserId(user.getId()); String deptIds = ""; for (UserDepartment userDepartment : userDepartments) { deptIds = deptIds + userDepartment.getDepartmentId().toString() + ","; } userVm.setDeptIds(deptIds.equals("") ? "" : deptIds.substring(0,deptIds.length()-1)); userVm.setTagIds(tagService.selectTagIdsByUserId(user.getId())); userVm.setPassword(authenticationService.pwdDecode(user.getPassword())); return RestResponse.ok(userVm); } @RequestMapping(value = "/current", method = RequestMethod.POST) public RestResponse current() { User user = getCurrentUser(); UserResponseVM userVm = UserResponseVM.from(user); return RestResponse.ok(userVm); } @RequestMapping(value = "/edit", method = RequestMethod.POST) public RestResponse edit(@RequestBody @Valid UserCreateVM model) { if (model.getId() == null) { //create User existUser = userService.getUserByUserName(model.getUserName()); if (null != existUser) { return new RestResponse<>(2, "用户已存在"); } if (StringUtils.isBlank(model.getPassword())) { return new RestResponse<>(3, "密码不能为空"); } } if (StringUtils.isBlank(model.getBirthDay())) { model.setBirthDay(null); } User user = modelMapper.map(model, User.class); if (model.getId() == null) { String encodePwd = authenticationService.pwdEncode(model.getPassword()); user.setPassword(encodePwd); user.setUserUuid(UUID.randomUUID().toString()); user.setCreateTime(new Date()); user.setLastActiveTime(new Date()); user.setDeleted(false); userService.insertByFilter(user); } else { if (!StringUtils.isBlank(model.getPassword())) { String encodePwd = authenticationService.pwdEncode(model.getPassword()); user.setPassword(encodePwd); } user.setModifyTime(new Date()); userService.updateByIdFilter(user); } Map oldDeptUser = userDepartmentMapper.selectByUserId(user.getId()).stream().collect(Collectors.toMap(UserDepartment::getDepartmentId, userDept -> userDept)); userDepartmentMapper.removeByUserId(user.getId()); for (Integer s : model.getDeptIds()) { UserDepartment userDepartment = new UserDepartment(); userDepartment.setUserId(user.getId()); userDepartment.setDepartmentId(s); if (Objects.nonNull(oldDeptUser.get(s))) { userDepartment.setDeptAdmin(oldDeptUser.get(s).getDeptAdmin()); } else { userDepartment.setDeptAdmin("0"); } userDepartmentMapper.insert(userDepartment); } if (ObjectUtils.isNotEmpty(model.getTagIds())) { tagService.removeUserTagByUserId(user.getId()); tagService.saveBatchUserTag(model.getTagIds().stream().map( tagId -> new UserTag() {{ setUserId(user.getId()); setTagId(tagId); }} ).collect(Collectors.toList())); } return RestResponse.ok(user); } @RequestMapping(value = "/update", method = RequestMethod.POST) public RestResponse update(@RequestBody @Valid UserUpdateVM model) { User user = userService.selectById(getCurrentUser().getId()); modelMapper.map(model, user); user.setModifyTime(new Date()); userService.updateByIdFilter(user); return RestResponse.ok(); } @RequestMapping(value = "/changeStatus/{id}", method = RequestMethod.POST) public RestResponse changeStatus(@PathVariable Integer id) { User user = userService.selectById(id); UserStatusEnum userStatusEnum = UserStatusEnum.fromCode(user.getStatus()); Integer newStatus = userStatusEnum == UserStatusEnum.Enable ? UserStatusEnum.Disable.getCode() : UserStatusEnum.Enable.getCode(); user.setStatus(newStatus); user.setModifyTime(new Date()); userService.updateByIdFilter(user); return RestResponse.ok(newStatus); } @RequestMapping(value = "/delete/{id}", method = RequestMethod.POST) public RestResponse delete(@PathVariable Integer id) { User user = userService.getUserById(id); user.setDeleted(true); userService.updateByIdFilter(user); return RestResponse.ok(); } @RequestMapping(value = "/selectByUserName", method = RequestMethod.POST) public RestResponse> selectByUserName(@RequestBody String userName) { List keyValues = userService.selectByUserName(userName); return RestResponse.ok(keyValues); } @PostMapping("/import") public RestResponse importUser(@RequestPart("file") MultipartFile file) throws Exception { List userVOS = ExcelUtils.readMultipartFile(file, UserVO.class) .stream().map(e -> { e.setUserLevel(departmentService.getName(e.getDepartment()).getId()); String encodePwd = authenticationService.pwdEncode(e.getPassword()); e.setPassword(encodePwd); e.setUserUuid(UUID.randomUUID().toString()); e.setCreateTime(new Date()); e.setLastActiveTime(new Date()); e.setDeleted(false); e.setAge(null); return e; }).collect(Collectors.toList()); List users = UserClassConvert.INSTANCE.UserVOListToUserList(userVOS); userService.insertUsers(users); return RestResponse.ok(); } @RequestMapping(value = "/conversion", method = RequestMethod.GET) public RestResponse conversion() { List users = userService.getUsers(); for (User user : users) { UserDepartment userDepartment = new UserDepartment(); userDepartment.setUserId(user.getId()); userDepartment.setDepartmentId(user.getUserLevel()); userDepartmentMapper.insert(userDepartment); } return RestResponse.ok(); } }