From 580d70b15ae47bc180a0b579af8c47c506eefac6 Mon Sep 17 00:00:00 2001
From: huliguo <2023611923@qq.com>
Date: 星期五, 11 七月 2025 20:13:38 +0800
Subject: [PATCH] 修改
---
src/main/java/com/linghu/controller/UserController.java | 264 +++++++++++++++++++++++++++++++++++++++++++++++++---
1 files changed, 246 insertions(+), 18 deletions(-)
diff --git a/src/main/java/com/linghu/controller/UserController.java b/src/main/java/com/linghu/controller/UserController.java
index a24f880..adb5599 100644
--- a/src/main/java/com/linghu/controller/UserController.java
+++ b/src/main/java/com/linghu/controller/UserController.java
@@ -1,16 +1,48 @@
package com.linghu.controller;
+import com.alibaba.excel.EasyExcel;
+import com.alibaba.excel.ExcelWriter;
+import com.alibaba.excel.write.handler.SheetWriteHandler;
+import com.alibaba.excel.write.metadata.WriteSheet;
+import com.alibaba.excel.write.metadata.holder.WriteSheetHolder;
+import com.alibaba.excel.write.metadata.holder.WriteWorkbookHolder;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.linghu.listener.PlatformExcelListener;
+import com.linghu.listener.UserExcelListener;
import com.linghu.model.common.ResponseResult;
-import com.linghu.model.entity.Type;
+import com.linghu.model.entity.Platform;
+import com.linghu.model.entity.Sectionalization;
import com.linghu.model.entity.User;
+import com.linghu.model.excel.ExcelDataWithRow;
+import com.linghu.model.excel.PlatformExcel;
+import com.linghu.model.excel.UserExcel;
+import com.linghu.model.vo.UserPageVO;
+import com.linghu.service.SectionalizationService;
import com.linghu.service.UserService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
+import org.apache.poi.ss.usermodel.DataValidation;
+import org.apache.poi.ss.usermodel.DataValidationConstraint;
+import org.apache.poi.ss.usermodel.DataValidationHelper;
+import org.apache.poi.ss.usermodel.Sheet;
+import org.apache.poi.ss.util.CellRangeAddressList;
import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.PostMapping;
-import org.springframework.web.bind.annotation.RequestBody;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.MediaType;
+import org.springframework.http.ResponseEntity;
+import org.springframework.util.StringUtils;
+import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.time.LocalDateTime;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+import java.util.Objects;
+import java.util.stream.Collectors;
@RestController
@RequestMapping("/user")
@@ -19,34 +51,230 @@
@Autowired
private UserService userService;
-/*
- @PostMapping
- @ApiOperation(value = "添加类型")
- public ResponseResult<User> add(@RequestBody User user) {
- boolean success = typeService.save(type);
- if (success) {
- return ResponseResult.success(type);
- }
- return ResponseResult.error("添加类型失败");
- }
-*/
-
-
/**
* 新增用户
*/
+ @PostMapping
+ @ApiOperation(value = "添加")
+ public ResponseResult add(@RequestBody User user) {
+ List<User> list = userService.list(new LambdaQueryWrapper<User>().eq(User::getUser_email, user.getUser_email()));
+ if (list != null && list.size() > 0) {
+ return ResponseResult.success("该邮箱已存在");
+ }
+ user.setStatus("正常");
+ userService.save(user);
+ return ResponseResult.success();
+
+ }
+
+
/**
* 修改用户
*/
+ @PutMapping
+ @ApiOperation(value = "修改")
+ public ResponseResult edit(@RequestBody User user) {
+ User user1 = userService.getById(user.getUser_id());
+ if (user1 == null) {
+ return ResponseResult.error("该账户不存在");
+ }
+ List<User> list = userService.list(new LambdaQueryWrapper<User>()
+ .ne(User::getUser_id, user.getUser_id())
+ .eq(User::getUser_email, user.getUser_email()));
+ if (list != null && list.size() > 0) {
+ return ResponseResult.success("该邮箱已存在");
+ }
+ userService.updateById(user);
+ return ResponseResult.success();
+ }
/**
* 删除用户
*/
+ @DeleteMapping("/{user_id}")
+ @ApiOperation(value = "删除")
+ public ResponseResult delete(@PathVariable("user_id") Integer user_id) {
+ userService.removeById(user_id);
+ return ResponseResult.success();
+
+ }
/**
* 分页查询
*/
+ @GetMapping
+ @ApiOperation(value = "分页")
+ public ResponseResult<Page<UserPageVO>> page(@RequestParam(value = "pageSize", required = false, defaultValue = "10")Integer pageSize,
+ @RequestParam(value = "pageNum", required = false,defaultValue = "1")Integer pageNum,
+ @RequestParam(value = "sectionalization_id",required = false)Integer sectionalization_id,
+ @RequestParam(value = "status" ,required = false)String status) {
+ Page<UserPageVO> page = new Page<>(pageNum, pageSize);
+ return ResponseResult.success( userService.getPage(page,sectionalization_id,status));
+ }
+
+ /**
+ * 下载模板
+ */
+
+ @GetMapping("/downloadTemplate")
+ @ApiOperation("下载模板")
+ public ResponseEntity<byte[]> downloadTemplate() throws IOException {
+ // 1. 查询所有分组名称
+ List<Sectionalization> sections = sectionalizationService.list(null);
+ List<String> sectionNames = sections.stream()
+ .map(Sectionalization::getSectionalization_name)
+ .filter(Objects::nonNull)
+ .collect(Collectors.toList());
+
+ // 2. 创建Excel写出器
+ ByteArrayOutputStream out = new ByteArrayOutputStream();
+
+ // 3. 使用自定义的SheetWriteHandler来添加数据验证
+ ExcelWriter excelWriter = EasyExcel.write(out, UserExcel.class)
+ .registerWriteHandler(new SheetWriteHandler() {
+ @Override
+ public void afterSheetCreate(WriteWorkbookHolder writeWorkbookHolder,
+ WriteSheetHolder writeSheetHolder) {
+ Sheet sheet = writeSheetHolder.getSheet();
+
+ // 添加数据验证（下拉框）
+ DataValidationHelper helper = sheet.getDataValidationHelper();
+ CellRangeAddressList rangeList = new CellRangeAddressList(
+ 1, 65535, 3, 3); // D列（第4列）
+
+ DataValidationConstraint constraint = helper.createExplicitListConstraint(
+ sectionNames.toArray(new String[0]));
+ DataValidation validation = helper.createValidation(constraint, rangeList);
+ sheet.addValidationData(validation);
+ }
+ })
+ .build();
+
+ // 4. 写入空数据
+ WriteSheet writeSheet = EasyExcel.writerSheet("账号模板").build();
+ excelWriter.write(new ArrayList<>(), writeSheet);
+ excelWriter.finish();
+
+ // 5. 构建响应
+ return ResponseEntity.ok()
+ .header(HttpHeaders.CONTENT_DISPOSITION, "attachment;filename=user_template.xlsx")
+ .contentType(MediaType.APPLICATION_OCTET_STREAM)
+ .body(out.toByteArray());
+ }
+
+ /**
+ * 导入文件
+ */
+ // 导入文件
+ @Autowired
+ private SectionalizationService sectionalizationService;
+ @PostMapping("/importUserExcel")
+ @ApiOperation("导入用户数据")
+ public ResponseResult importUserExcel(@RequestParam("file") MultipartFile file) {
+ //查找出用户列表
+ List<User> userList = userService.list();
+ //名称
+ Map<String, User> nameMap = userList.stream()
+ .collect(Collectors.toMap(
+ User::getUser_name, // Key映射
+ excel -> excel, // Value映射
+ (oldValue, newValue) -> oldValue // 键冲突处理（保留旧值）
+ ));
+
+ //查找出分组列表
+ List<Sectionalization> list = sectionalizationService.list();
+ //分组
+ Map<String, Sectionalization> sectionalizationMap = list.stream()
+ .collect(Collectors.toMap(
+ Sectionalization::getSectionalization_name, // Key映射
+ excel -> excel, // Value映射
+ (oldValue, newValue) -> oldValue // 键冲突处理（保留旧值）
+ ));
+
+ try {
+ // 检查文件是否为空
+ if (file.isEmpty()) {
+ return ResponseResult.error("上传文件不能为空");
+ }
+ // 使用自定义监听器读取数据（包含行号）
+ UserExcelListener listener = new UserExcelListener();
+
+ // 读取Excel数据
+ EasyExcel.read(file.getInputStream())
+ .head(UserExcel.class)
+ .sheet()
+ .doReadSync();
+
+ // 数据转换与验证
+ List<ExcelDataWithRow<UserExcel>> excelList = listener.getDataList();
+ List<User> insertUserList = new ArrayList<>();
+ List<String> errorMessages = new ArrayList<>();
+ List<String> emailList = new ArrayList<>();
+ for (ExcelDataWithRow<UserExcel> excelData : excelList) {
+ int rowNum = excelData.getRowNumber();
+ UserExcel excel = excelData.getData();
+ List<String> rowErrors = new ArrayList<>();
+ // 检查必要字段
+ if (!StringUtils.hasText(excel.getUser_name())) {
+ rowErrors.add("用户名称不能为空");
+ }
+ if (!StringUtils.hasText(excel.getUser_email())) {
+ rowErrors.add("邮箱不能为空");
+ }else{
+ if (emailList.contains(excel.getUser_email())) {
+ rowErrors.add("导入数据中邮箱重复");
+ }
+ if (nameMap.containsKey(excel.getUser_email())){
+ rowErrors.add("邮箱与数据库中重复");
+ }
+ }
+ if (!StringUtils.hasText(String.valueOf(excel.getPassword()))) {
+ rowErrors.add("密码不能为空");
+ }
+
+ if (!StringUtils.hasText(excel.getSectionalization_name())) {
+ rowErrors.add("分组不能为空");
+ }else if (!sectionalizationMap.containsKey(excel.getSectionalization_name())){
+ rowErrors.add("分组不存在");
+ }
+
+ if (!rowErrors.isEmpty()) {
+ errorMessages.add(String.format("第%d行错误: %s", rowNum, String.join("；", rowErrors)));
+ }else {
+ // 构建Platform对象...
+ User user = new User();
+ user.setUser_name(excel.getUser_name());
+ user.setUser_email(excel.getUser_email());
+ user.setPassword(excel.getPassword());
+ user.setSectionalization_id(sectionalizationMap.get(excel.getSectionalization_name()).getSectionalization_id());
+ user.setStatus("正常");
+ insertUserList.add(user);
+ emailList.add(excel.getUser_email());
+ }
+ }
+ // 处理错误
+ if (!errorMessages.isEmpty()) {
+ // 构建清晰的错误信息
+ StringBuilder errorMsg = new StringBuilder();
+ errorMsg.append("导入失败，共").append(errorMessages.size()).append("条数据存在错误：\n");
+ errorMessages.forEach(msg -> errorMsg.append(msg).append("\n"));
+
+ return ResponseResult.error(400, errorMsg.toString());
+ }
+ // 开始添加
+ userService.saveBatch(insertUserList);
+ // 返回信息
+ return ResponseResult.success();
+ }
+ catch (Exception e) {
+ // 记录详细异常信息
+
+ return ResponseResult.error("文件解析失败：" + e.getMessage());
+ }
+ }
+
+
}
--
Gitblit v1.7.1