package com.moral.api.controller; import io.swagger.annotations.Api; import io.swagger.annotations.ApiImplicitParam; import io.swagger.annotations.ApiImplicitParams; import io.swagger.annotations.ApiOperation; import lombok.extern.slf4j.Slf4j; import java.util.Map; import javax.servlet.http.HttpServletRequest; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RequestMethod; import org.springframework.web.bind.annotation.RestController; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.moral.api.entity.User; import com.moral.api.service.UserService; import com.moral.constant.ResponseCodeEnum; import com.moral.constant.ResultMessage; import com.moral.util.PageResult; import com.moral.util.WebUtils; @Slf4j @Api(tags = {"用户"}) @RestController @RequestMapping("/user") public class UserController { @Autowired private UserService userService; @ApiOperation(value = "添加账户", notes = "添加账户") @ApiImplicitParams({ @ApiImplicitParam(name = "account", value = "账户,长度6-10", required = true, paramType = "query", dataType = "String"), @ApiImplicitParam(name = "password", value = "密码,长度6-20", required = true, paramType = "query", dataType = "String"), @ApiImplicitParam(name = "userName", value = "账户名称", required = false, paramType = "query", dataType = "String"), @ApiImplicitParam(name = "email", value = "邮箱,格式123456@qq.com", required = false, paramType = "query", dataType = "String"), @ApiImplicitParam(name = "mobile", value = "手机号,1开头11为数字", required = false, paramType = "query", dataType = "String"), @ApiImplicitParam(name = "wechat", value = "微信", required = false, paramType = "query", dataType = "String"), @ApiImplicitParam(name = "token", value = "token", required = true, paramType = "header", dataType = "String") }) @RequestMapping(value = "addUser", method = RequestMethod.POST) public ResultMessage addUser(User user, HttpServletRequest request) { if (user.getAccount().isEmpty() || user.getPassword().isEmpty()) { return ResultMessage.fail(ResponseCodeEnum.PARAMETERS_IS_MISSING.getCode(), ResponseCodeEnum.PARAMETERS_IS_MISSING.getMsg()); } String token = request.getHeader("token"); Map result = userService.addUser(user, token); if (!result.isEmpty()) { return ResultMessage.fail((int) result.get("code"), result.get("msg").toString()); } return ResultMessage.ok(); } @ApiOperation(value = "删除账户", notes = "删除账户") @ApiImplicitParams({ @ApiImplicitParam(name = "userId", value = "用户id", required = true, paramType = "query", dataType = "String"), @ApiImplicitParam(name = "token", value = "token", required = true, paramType = "header", dataType = "String") }) @RequestMapping(value = "deleteUser", method = RequestMethod.POST) public ResultMessage deleteUser(String userId) { if (userId == null) { return ResultMessage.fail(ResponseCodeEnum.PARAMETERS_IS_MISSING.getCode(), ResponseCodeEnum.PARAMETERS_IS_MISSING.getMsg()); } QueryWrapper queryWrapper = new QueryWrapper<>(); queryWrapper.eq("id", userId); User user = userService.getOne(queryWrapper); if (user == null) { return ResultMessage.fail(ResponseCodeEnum.ACCOUNT_NOT_EXIST.getCode(), ResponseCodeEnum.ACCOUNT_NOT_EXIST.getMsg()); } userService.deleteUser(user); return ResultMessage.ok(); } @ApiOperation(value = "修改账户信息", notes = "修改账户信息") @ApiImplicitParams({ @ApiImplicitParam(name = "account", value = "账户,长度6-10", required = true, paramType = "query", dataType = "String"), @ApiImplicitParam(name = "password", value = "密码,长度6-20", required = true, paramType = "query", dataType = "String"), @ApiImplicitParam(name = "userName", value = "账户名称", required = false, paramType = "query", dataType = "String"), @ApiImplicitParam(name = "email", value = "邮箱,格式123456@qq.com", required = false, paramType = "query", dataType = "String"), @ApiImplicitParam(name = "mobile", value = "手机号,1开头11位数字", required = false, paramType = "query", dataType = "String"), @ApiImplicitParam(name = "wechat", value = "微信", required = false, paramType = "query", dataType = "String"), @ApiImplicitParam(name = "token", value = "token", required = true, paramType = "header", dataType = "String") }) @RequestMapping(value = "updateUser", method = RequestMethod.POST) public ResultMessage updateUser(User user, HttpServletRequest request) { if (user.getAccount().isEmpty() || user.getPassword().isEmpty()) { return ResultMessage.fail(ResponseCodeEnum.PARAMETERS_IS_MISSING.getCode(), ResponseCodeEnum.PARAMETERS_IS_MISSING.getMsg()); } String token = request.getHeader("token"); Map result = userService.updateUser(user, token); if (!result.isEmpty()) { return ResultMessage.fail((int) result.get("code"), result.get("msg").toString()); } return ResultMessage.ok(); } @ApiOperation(value = "分页查询账户列表", notes = "分页查询账户列表") @ApiImplicitParams({ @ApiImplicitParam(name = "page", value = "当前页", required = false, paramType = "query", dataType = "String"), @ApiImplicitParam(name = "size", value = "每页条数", required = false, paramType = "query", dataType = "String"), @ApiImplicitParam(name = "order", value = "排序字段", required = false, paramType = "query", dataType = "String"), @ApiImplicitParam(name = "token", value = "token", required = true, paramType = "header", dataType = "String"), @ApiImplicitParam(name = "account", value = "账号模糊查询", required = false, paramType = "query", dataType = "String"), @ApiImplicitParam(name = "userName", value = "用户名模糊查询", required = false, paramType = "header", dataType = "String"), @ApiImplicitParam(name = "mobile", value = "手机号模糊查询", required = false, paramType = "query", dataType = "String"), @ApiImplicitParam(name = "email", value = "邮箱模糊查询", required = false, paramType = "header", dataType = "String"), @ApiImplicitParam(name = "wechat", value = "微信模糊查询", required = false, paramType = "header", dataType = "String"), }) @RequestMapping(value = "selectUsers", method = RequestMethod.POST) public ResultMessage selectUsers(HttpServletRequest request) { Map parameters = WebUtils.getParametersStartingWith(request, null); parameters.put("token", request.getHeader("token")); Page userPage = userService.selectUsers(parameters); PageResult pageResult = new PageResult<>( userPage.getTotal(), userPage.getPages(), userPage.getRecords() ); return ResultMessage.ok(pageResult); } }