fuliqi
2024-10-17 8546b3d285af4235a0ef615a0c6e89486ae2c806
src/main/java/com/ycl/jxkg/service/impl/UserServiceImpl.java
@@ -1,7 +1,9 @@
package com.ycl.jxkg.service.impl;
import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.ExcelWriter;
import com.alibaba.excel.read.listener.PageReadListener;
import com.alibaba.excel.write.metadata.WriteSheet;
import com.baomidou.mybatisplus.extension.conditions.update.LambdaUpdateChainWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.github.pagehelper.PageHelper;
@@ -31,6 +33,7 @@
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.multipart.MultipartFile;
import javax.servlet.http.HttpServletResponse;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.*;
@@ -223,31 +226,49 @@
    @Override
    @SneakyThrows
    public void importTemplate(HttpServletResponse response) {
        List<UserExcelVo> list = new ArrayList<>();
        UserExcelVo userExcelVo = new UserExcelVo();
        userExcelVo.setUserName("zhangSan");
        userExcelVo.setRealName("张三");
        userExcelVo.setSex(1);
        userExcelVo.setRole(1);
        userExcelVo.setPhone("12345678901");
        userExcelVo.setAge(24);
        userExcelVo.setBirthDay("2000-01-01");
        list.add(userExcelVo);
        ExcelWriter excelWriter = EasyExcel.write(response.getOutputStream(), UserExcelVo.class).build();
        WriteSheet writeSheet = EasyExcel.writerSheet(0, "用户信息").build();
        excelWriter.write(list, writeSheet);
        excelWriter.finish();
    }
    @Override
    @SneakyThrows
    @Transactional
    public String importExcel(MultipartFile file) {
        List<User> userList = new ArrayList<>();
        List<String> errorList = new ArrayList<>();
        EasyExcel.read(file.getInputStream(), UserExcelVo.class, new PageReadListener<UserExcelVo>(dataList -> {
            dataList.forEach(item -> {
                User user = new User();
                user.setUserName(item.getUserName());
                user.setPassword(authenticationService.pwdEncode("!Password123456789"));
                user.setRealName(item.getRealName());
                user.setSex(item.getSex());
                user.setPhone(item.getPhone());
                try {
                    user.setBirthDay(new SimpleDateFormat("yyyy-MM-dd").parse(item.getBirthDay()));
                } catch (ParseException e) {
                    errorList.add(item.getRealName());
                }
                user.setAge(item.getAge());
                user.setRole(item.getRole());
                user.setStatus(UserStatusEnum.Enable.getCode());
                userList.add(user);
            });
        }));
        EasyExcel.read(file.getInputStream(), UserExcelVo.class, new PageReadListener<UserExcelVo>(dataList -> dataList.forEach(item -> {
            User user = new User();
            user.setUserName(item.getUserName());
            user.setPassword(authenticationService.pwdEncode("!Password123456789"));
            user.setRealName(item.getRealName());
            user.setSex(item.getSex());
            user.setPhone(item.getPhone());
            try {
                user.setBirthDay(new SimpleDateFormat("yyyy-MM-dd").parse(item.getBirthDay()));
            } catch (ParseException e) {
                errorList.add(item.getRealName());
            }
            user.setAge(item.getAge());
            user.setRole(item.getRole());
            user.setStatus(UserStatusEnum.Enable.getCode());
            user.setLastUpdatePasswordTime(new Date());
            userList.add(user);
        }))).sheet().doRead();
        if (!errorList.isEmpty()) {
            return "以下数据日期格式错误:\\n" + String.join(",", errorList);
            return "以下数据日期格式错误:" + String.join(",", errorList);
        } else {
            saveBatch(userList);
        }