From bea818128ff533b3103856f95af38ac7bea77468 Mon Sep 17 00:00:00 2001
From: pxh4iabou <2044878308@qq.com>
Date: Sun, 30 Apr 2023 01:24:00 +0800
Subject: [PATCH] ADD file via upload

---
 .../com/controller/AdminInfoController.java   | 138 ++++++++++++++++++
 1 file changed, 138 insertions(+)
 create mode 100644 src/main/java/com/controller/AdminInfoController.java

diff --git a/src/main/java/com/controller/AdminInfoController.java b/src/main/java/com/controller/AdminInfoController.java
new file mode 100644
index 0000000..fedc87d
--- /dev/null
+++ b/src/main/java/com/controller/AdminInfoController.java
@@ -0,0 +1,138 @@
+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.AdminInfo;
+import com.example.service.AdminInfoService;
+import com.example.exception.CustomException;
+import com.example.common.ResultCode;
+import com.example.vo.AdminInfoVo;
+
+import com.github.pagehelper.PageHelper;
+import com.github.pagehelper.PageInfo;
+import com.example.service.*;
+import org.springframework.web.bind.annotation.*;
+import org.springframework.beans.factory.annotation.Value;
+import cn.hutool.core.util.StrUtil;
+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 = "/adminInfo")
+public class AdminInfoController {
+
+    @Resource
+    private AdminInfoService adminInfoService;
+
+    @PostMapping
+    public Result<AdminInfo> add(@RequestBody AdminInfoVo adminInfo) {
+        adminInfoService.add(adminInfo);
+        return Result.success(adminInfo);
+    }
+
+    @DeleteMapping("/{id}")
+    public Result delete(@PathVariable Long id) {
+        adminInfoService.delete(id);
+        return Result.success();
+    }
+
+    @PutMapping
+    public Result update(@RequestBody AdminInfoVo adminInfo) {
+        adminInfoService.update(adminInfo);
+        return Result.success();
+    }
+
+    @GetMapping("/{id}")
+    public Result<AdminInfo> detail(@PathVariable Long id) {
+        AdminInfo adminInfo = adminInfoService.findById(id);
+        return Result.success(adminInfo);
+    }
+
+    @GetMapping
+    public Result<List<AdminInfoVo>> all() {
+        return Result.success(adminInfoService.findAll());
+    }
+
+    @GetMapping("/page/{name}")
+    public Result<PageInfo<AdminInfoVo>> page(@PathVariable String name,
+                                                @RequestParam(defaultValue = "1") Integer pageNum,
+                                                @RequestParam(defaultValue = "5") Integer pageSize,
+                                                HttpServletRequest request) {
+        return Result.success(adminInfoService.findPage(name, pageNum, pageSize, request));
+    }
+
+    @PostMapping("/register")
+    public Result<AdminInfo> register(@RequestBody AdminInfo adminInfo) {
+        if (StrUtil.isBlank(adminInfo.getName()) || StrUtil.isBlank(adminInfo.getPassword())) {
+            throw new CustomException(ResultCode.PARAM_ERROR);
+        }
+        return Result.success(adminInfoService.add(adminInfo));
+    }
+
+    /**
+    * 批量通过excel添加信息
+    * @param file excel文件
+    * @throws IOException
+    */
+    @PostMapping("/upload")
+    public Result upload(MultipartFile file) throws IOException {
+
+        List<AdminInfo> infoList = ExcelUtil.getReader(file.getInputStream()).readAll(AdminInfo.class);
+        if (!CollectionUtil.isEmpty(infoList)) {
+            // 处理一下空数据
+            List<AdminInfo> resultList = infoList.stream().filter(x -> ObjectUtil.isNotEmpty(x.getName())).collect(Collectors.toList());
+            for (AdminInfo info : resultList) {
+                adminInfoService.add(info);
+            }
+        }
+        return Result.success();
+    }
+
+    @GetMapping("/getExcelModel")
+    public void getExcelModel(HttpServletResponse response) throws IOException {
+        // 1. 生成excel
+        Map<String, Object> row = new LinkedHashMap<>();
+		row.put("name", "admin");
+		row.put("password", "123456");
+		row.put("nickName", "管理员");
+		row.put("sex", "男");
+		row.put("age", 22);
+		row.put("birthday", "TIME");
+		row.put("phone", "18843232356");
+		row.put("address", "上海市");
+		row.put("code", "111");
+		row.put("email", "aa@163.com");
+		row.put("cardId", "342425199001116372");
+		row.put("level", 1);
+
+        List<Map<String, Object>> 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=adminInfoModel.xlsx");
+
+        ServletOutputStream out = response.getOutputStream();
+        writer.flush(out, true);
+        writer.close();
+        IoUtil.close(System.out);
+    }
+}