登录失败修改状态

guyue

1 天以前 286cc1963d6d5a26f26c78d542974691b80a86f7

 src/main/java/com/linghu/controller/UserController.java

@@ -1,17 +1,32 @@
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.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.http.HttpHeaders;
import org.springframework.http.MediaType;
@@ -22,9 +37,8 @@

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.time.LocalDateTime;
import java.util.*;
import java.util.stream.Collectors;

@RestController
@@ -43,7 +57,7 @@
    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.error("该邮箱已存在");
            return ResponseResult.success("该邮箱已存在");
        }
        user.setStatus("正常");
        userService.save(user);
@@ -66,7 +80,7 @@
                        .ne(User::getUser_id, user.getUser_id())
                .eq(User::getUser_email, user.getUser_email()));
        if (list != null && list.size() > 0) {
            return ResponseResult.error("该邮箱已存在");
            return ResponseResult.success("该邮箱已存在");
        }
        userService.updateById(user);
        return ResponseResult.success();
@@ -104,15 +118,48 @@
    @GetMapping("/downloadTemplate")
    @ApiOperation("下载模板")
    public ResponseEntity<byte[]> downloadTemplate() throws IOException {
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        EasyExcel.write(out, UserExcel.class).sheet("账号模板").doWrite(new ArrayList<>());
        // 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=keyword_template.xlsx")
                .header(HttpHeaders.CONTENT_DISPOSITION, "attachment;filename=user_template.xlsx")
                .contentType(MediaType.APPLICATION_OCTET_STREAM)
                .body(out.toByteArray());
    }


    /**
     * 导入文件
@@ -123,97 +170,120 @@
    @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 {
            // 检查文件是否为空
            // 1. 检查文件是否为空
            if (file.isEmpty()) {
                return ResponseResult.error("上传文件不能为空");
            }

            // 读取Excel数据
            List<UserExcel> excelList = EasyExcel.read(file.getInputStream())
                    .head(UserExcel.class)
                    .sheet()
                    .doReadSync();
            // 2. 读取数据库中已有的用户和分组，用于重复校验
            // 2.1 已存在的用户（按用户名和邮箱分组）
            List<User> userList = userService.list();
            Map<String, User> userNameMap = userList.stream()
                    .collect(Collectors.toMap(
                            User::getUser_name,
                            user -> user, // 变量名修正为user（原excel变量名不合理）
                            (oldValue, newValue) -> oldValue // 重复用户名保留旧值
                    ));
            Map<String, User> userEmailMap = userList.stream()
                    .collect(Collectors.toMap(
                            User::getUser_email,
                            user -> user,
                            (oldValue, newValue) -> oldValue // 重复邮箱保留旧值
                    ));

            // 数据转换与验证
            // 2.2 已存在的分组（按分组名称映射）
            List<Sectionalization> sectionalizationList = sectionalizationService.list();
            Map<String, Sectionalization> sectionMap = sectionalizationList.stream()
                    .collect(Collectors.toMap(
                            Sectionalization::getSectionalization_name,
                            section -> section,
                            (oldValue, newValue) -> oldValue
                    ));

            // 3. 读取Excel数据（使用自定义监听器）
            UserExcelListener listener = new UserExcelListener();
            EasyExcel.read(file.getInputStream(), UserExcel.class, listener)
                    .sheet() // 读取第一个sheet
                    .doRead(); // 执行读取

            // 4. 校验Excel数据并转换为User对象
            List<ExcelDataWithRow<UserExcel>> excelDataList = listener.getDataList();
            List<User> insertUserList = new ArrayList<>();
            List<String> errorMessages = new ArrayList<>();
            for (UserExcel excel : excelList) {
                // 检查必要字段
                if (!StringUtils.hasText(excel.getUser_name())) {
                    errorMessages.add("账户名不能为空");
                    continue;
                }
                if (!StringUtils.hasText(excel.getUser_email())) {
                    errorMessages.add("邮箱不能为空");
                    continue;
                }
                if (!StringUtils.hasText(String.valueOf(excel.getPassword()))) {
                    errorMessages.add("密码不能为空");
                    continue;
            Set<String> emailSet = new HashSet<>(); // 用于校验导入数据中的邮箱重复

            for (ExcelDataWithRow<UserExcel> excelData : excelDataList) {
                int rowNum = excelData.getRowNumber();
                UserExcel excelRowData = excelData.getData();
                List<String> rowErrors = new ArrayList<>();

                // 4.1 校验必填字段
                if (!StringUtils.hasText(excelRowData.getUser_name())) {
                    rowErrors.add("账户名不能为空");
                } else if (userNameMap.containsKey(excelRowData.getUser_name())) {
                    rowErrors.add("账户名已存在（数据库中重复）");
                }

                if (!StringUtils.hasText(excel.getSectionalization_name())) {
                    errorMessages.add("分组不能为空");
                    continue;
                if (!StringUtils.hasText(excelRowData.getUser_email())) {
                    rowErrors.add("邮箱不能为空");
                } else {
                    if (userEmailMap.containsKey(excelRowData.getUser_email())) {
                        rowErrors.add("邮箱已存在（数据库中重复）");
                    }
                    if (emailSet.contains(excelRowData.getUser_email())) {
                        rowErrors.add("邮箱重复（导入数据中重复）");
                    }
                }
                if (nameMap.containsKey(excel.getUser_name())){
                    errorMessages.add("用户名称重复");
                    continue;

                if (!StringUtils.hasText(excelRowData.getPassword())) {
                    rowErrors.add("密码不能为空");
                }

                if (!StringUtils.hasText(excelRowData.getSectionalization_name())) {
                    rowErrors.add("账户分组不能为空");
                } else if (!sectionMap.containsKey(excelRowData.getSectionalization_name())) {
                    rowErrors.add("账户分组不存在（数据库中无此分组）");
                }

                // 4.2 处理校验结果
                if (!rowErrors.isEmpty()) {
                    // 记录错误信息
                    errorMessages.add(String.format("第%d行：%s", rowNum, String.join("；", rowErrors)));
                } else {
                    // 校验通过，转换为User对象
                    User user = new User();
                    user.setUser_name(excelRowData.getUser_name());
                    user.setUser_email(excelRowData.getUser_email());
                    user.setPassword(excelRowData.getPassword()); // 实际场景建议加密存储
                    user.setSectionalization_id(sectionMap.get(excelRowData.getSectionalization_name()).getSectionalization_id());
                    user.setStatus("正常");

                    insertUserList.add(user);
                    emailSet.add(excelRowData.getUser_email()); // 记录已添加的邮箱，用于去重
                }
            }
            // 处理错误

            // 5. 处理错误或执行批量插入
            if (!errorMessages.isEmpty()) {
                return ResponseResult.error("数据验证失败: " + String.join("; ", errorMessages));
                // 组装错误信息
                StringBuilder errorMsg = new StringBuilder();
                errorMsg.append("导入失败，共").append(errorMessages.size()).append("条数据有误：\n");
                errorMessages.forEach(msg -> errorMsg.append(msg).append("\n"));
                return ResponseResult.error( errorMsg.toString());
            } else {
                // 批量插入数据库
                userService.saveBatch(insertUserList);
                return ResponseResult.success("导入成功，共新增" + insertUserList.size() + "条用户数据");
            }
            // 开始添加
            for (UserExcel excel : excelList) {
                User user = new User();
                if (sectionalizationMap.containsKey(excel.getSectionalization_name())) {
                    user.setSectionalization_id(sectionalizationMap.get(excel.getSectionalization_name()).getSectionalization_id());
                }else {
                    //新增
                    Sectionalization sectionalization = new Sectionalization();
                    sectionalization.setSectionalization_name(excel.getSectionalization_name());
                    sectionalizationService.save(sectionalization);
                    user.setSectionalization_id(sectionalization.getSectionalization_id());
                }
                user.setUser_email(excel.getUser_email());
                user.setUser_name(excel.getUser_name());
                user.setPassword(excel.getPassword());
                user.setStatus("正常");
                userService.save(user);

            }
            // 返回信息
            return ResponseResult.success();
        }
        catch (Exception e) {
            // 记录详细异常信息

            return ResponseResult.error("文件解析失败：" + e.getMessage());
        } catch (Exception e) {
            // 捕获异常（如文件格式错误、IO异常等）
            return ResponseResult.error("导入失败：" + e.getMessage());
        }
    }




}