diff --git a/src/main/java/com/controller/MessageInfoController.java b/src/main/java/com/controller/MessageInfoController.java new file mode 100644 index 0000000..49c7a51 --- /dev/null +++ b/src/main/java/com/controller/MessageInfoController.java @@ -0,0 +1,115 @@ +package com.example.controller; + +import cn.hutool.core.collection.CollUtil; +import cn.hutool.core.collection.CollectionUtil; +import cn.hutool.core.io.IoUtil; +import cn.hutool.core.util.ObjectUtil; +import cn.hutool.core.util.StrUtil; +import cn.hutool.poi.excel.ExcelUtil; +import cn.hutool.poi.excel.ExcelWriter; +import com.example.common.Result; +import com.example.common.ResultCode; +import com.example.entity.MessageInfo; +import com.example.service.*; +import com.example.vo.MessageInfoVo; + +import com.github.pagehelper.PageHelper; +import com.github.pagehelper.PageInfo; +import org.springframework.web.bind.annotation.*; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.ServletOutputStream; +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import java.io.IOException; +import java.util.LinkedHashMap; +import java.util.List; +import java.util.Map; +import java.util.stream.Collectors; + +@RestController +@RequestMapping(value = "/messageInfo") +public class MessageInfoController { + @Resource + private MessageInfoService messageInfoService; + + @PostMapping + public Result add(@RequestBody MessageInfoVo messageInfo) { + messageInfoService.add(messageInfo); + return Result.success(messageInfo); + } + + @DeleteMapping("/{id}") + public Result delete(@PathVariable Long id) { + messageInfoService.delete(id); + return Result.success(); + } + + @PutMapping + public Result update(@RequestBody MessageInfoVo messageInfo) { + messageInfoService.update(messageInfo); + return Result.success(); + } + + @GetMapping("/{id}") + public Result detail(@PathVariable Long id) { + MessageInfo messageInfo = messageInfoService.findById(id); + return Result.success(messageInfo); + } + + @GetMapping + public Result> all() { + return Result.success(messageInfoService.findAll()); + } + + @GetMapping("/page/{name}") + public Result> page(@PathVariable String name, + @RequestParam(defaultValue = "1") Integer pageNum, + @RequestParam(defaultValue = "5") Integer pageSize, + HttpServletRequest request) { + return Result.success(messageInfoService.findPage(name, pageNum, pageSize, request)); + } + + /** + * 批量通过excel添加信息 + * @param file excel文件 + * @throws IOException + */ + @PostMapping("/upload") + public Result upload(MultipartFile file) throws IOException { + + List infoList = ExcelUtil.getReader(file.getInputStream()).readAll(MessageInfo.class); + if (!CollectionUtil.isEmpty(infoList)) { + // 处理一下空数据 + List resultList = infoList.stream().filter(x -> ObjectUtil.isNotEmpty(x.getName())).collect(Collectors.toList()); + for (MessageInfo info : resultList) { + messageInfoService.add(info); + } + } + return Result.success(); + } + + @GetMapping("/getExcelModel") + public void getExcelModel(HttpServletResponse response) throws IOException { + // 1. 生成excel + Map row = new LinkedHashMap<>(); + row.put("name", "小李"); + row.put("content", "来了"); + row.put("time", "TIME"); + + List> list = CollUtil.newArrayList(row); + + // 2. 写excel + ExcelWriter writer = ExcelUtil.getWriter(true); + writer.write(list, true); + + response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=utf-8"); + response.setHeader("Content-Disposition","attachment;filename=messageInfoModel.xlsx"); + + ServletOutputStream out = response.getOutputStream(); + writer.flush(out, true); + writer.close(); + IoUtil.close(System.out); + } +}