玩具系统 玩具系统 玩具系统

package com.rabbiter.pet.controller; import cn.hutool.core.collection.CollUtil; import cn.hutool.core.util.StrUtil; import cn.hutool.poi.excel.ExcelReader; import cn.hutool.poi.excel.ExcelUtil; import cn.hutool.poi.excel.ExcelWriter; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.rabbiter.pet.controller.dto.UserDTO; import com.rabbiter.pet.controller.dto.UserPasswordDTO; import com.rabbiter.pet.common.Constants; import com.rabbiter.pet.common.Result; import com.rabbiter.pet.config.interceptor.AuthAccess; import com.rabbiter.pet.entity.User; import com.rabbiter.pet.exception.ServiceException; import com.rabbiter.pet.service.IUserService; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; import javax.annotation.Resource; import javax.servlet.ServletOutputStream; import javax.servlet.http.HttpServletResponse; import java.io.InputStream; import java.net.URLEncoder; import java.util.List; /** * <p> * 前端控制器 * </p> * * @author * @since 2023-01-26 */ @RestController @RequestMapping("/user") public class UserController { @Resource private IUserService userService; @PostMapping("/login") public Result login(@RequestBody UserDTO userDTO) { String username = userDTO.getUsername(); String password = userDTO.getPassword(); if (StrUtil.isBlank(username) || StrUtil.isBlank(password)) { return Result.error(Constants.CODE_400, "参数错误"); } UserDTO dto = userService.login(userDTO); return Result.success(dto); } @PostMapping("/register") public Result register(@RequestBody UserDTO userDTO) { String username = userDTO.getUsername(); String password = userDTO.getPassword(); if (StrUtil.isBlank(username) || StrUtil.isBlank(password)) { return Result.error(Constants.CODE_400, "参数错误"); } userDTO.setNickname(userDTO.getUsername()); return Result.success(userService.register(userDTO)); } // 新增或者更新 @PostMapping public Result save(@RequestBody User user) { String username = user.getUsername(); if (StrUtil.isBlank(username)) { return Result.error(Constants.CODE_400, "参数错误"); } if (user.getId() != null) { user.setPassword(null); } else { user.setNickname(user.getUsername()); if (user.getPassword() == null) { user.setPassword("123"); } } return Result.success(userService.saveOrUpdate(user)); } /** * 修改密码 * * @param userPasswordDTO * @return */ @PostMapping("/password") public Result password(@RequestBody UserPasswordDTO userPasswordDTO) { userService.updatePassword(userPasswordDTO); return Result.success(); } @AuthAccess @PutMapping("/reset") public Result reset(@RequestBody UserPasswordDTO userPasswordDTO) { if (StrUtil.isBlank(userPasswordDTO.getUsername()) || StrUtil.isBlank(userPasswordDTO.getPhone())) { throw new ServiceException("-1", "参数异常"); } QueryWrapper<User> queryWrapper = new QueryWrapper<>(); queryWrapper.eq("username", userPasswordDTO.getUsername()); queryWrapper.eq("phone", userPasswordDTO.getPhone()); List<User> list = userService.list(queryWrapper); if (CollUtil.isNotEmpty(list)) { User user = list.get(0); user.setPassword("123"); userService.updateById(user); } return Result.success(); } @DeleteMapping("/{id}") public Result delete(@PathVariable Integer id) { return Result.success(userService.removeById(id)); } @PostMapping("/del/batch") public Result deleteBatch(@RequestBody List<Integer> ids) { return Result.success(userService.removeByIds(ids)); } @GetMapping public Result findAll() { return Result.success(userService.list()); } @GetMapping("/{id}") public Result findOne(@PathVariable Integer id) { return Result.success(userService.getById(id)); } @GetMapping("/username/{username}") public Result findByUsername(@PathVariable String username) { QueryWrapper<User> queryWrapper = new QueryWrapper<>(); queryWrapper.eq("username", username); return Result.success(userService.getOne(queryWrapper)); } @GetMapping("/page") public Result findPage(@RequestParam Integer pageNum, @RequestParam Integer pageSize, @RequestParam(defaultValue = "") String username, @RequestParam(defaultValue = "") String email, @RequestParam(defaultValue = "") String address) { QueryWrapper<User> queryWrapper = new QueryWrapper<>(); queryWrapper.orderByDesc("id"); if (!"".equals(username)) { queryWrapper.like("username", username); } if (!"".equals(email)) { queryWrapper.like("email", email); } if (!"".equals(address)) { queryWrapper.like("address", address); } return Result.success(userService.page(new Page<>(pageNum, pageSize), queryWrapper)); } /** * 导出接口 */ @GetMapping("/export") public void export(HttpServletResponse response) throws Exception { // 从数据库查询出所有的数据 List<User> list = userService.list(); // 通过工具类创建writer 写出到磁盘路径 // ExcelWriter writer = ExcelUtil.getWriter(filesUploadPath + "/用户信息.xlsx"); // 在内存操作，写出到浏览器 ExcelWriter writer = ExcelUtil.getWriter(true); //自定义标题别名 writer.addHeaderAlias("username", "用户名"); writer.addHeaderAlias("password", "密码"); writer.addHeaderAlias("nickname", "昵称"); writer.addHeaderAlias("email", "邮箱"); writer.addHeaderAlias("phone", "电话"); writer.addHeaderAlias("address", "地址"); writer.addHeaderAlias("createTime", "创建时间"); writer.addHeaderAlias("avatarUrl", "头像"); // 一次性写出list内的对象到excel，使用默认样式，强制输出标题 writer.write(list, true); // 设置浏览器响应的格式 response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=utf-8"); String fileName = URLEncoder.encode("用户信息", "UTF-8"); response.setHeader("Content-Disposition", "attachment;filename=" + fileName + ".xlsx"); ServletOutputStream out = response.getOutputStream(); writer.flush(out, true); out.close(); writer.close(); } /** * excel 导入 * * @param file * @throws Exception */ @PostMapping("/import") public Result imp(MultipartFile file) throws Exception { InputStream inputStream = file.getInputStream(); ExcelReader reader = ExcelUtil.getReader(inputStream); // 方式1：(推荐) 通过 javabean的方式读取Excel内的对象，但是要求表头必须是英文，跟javabean的属性要对应起来 // List<User> list = reader.readAll(User.class); // 方式2：忽略表头的中文，直接读取表的内容 List<List<Object>> list = reader.read(1); List<User> users = CollUtil.newArrayList(); for (List<Object> row : list) { User user = new User(); user.