From 5d8f655a5b3625833cddd9d9f81c29a28d7a0fda Mon Sep 17 00:00:00 2001 From: Bunny <1319900154@qq.com> Date: Wed, 27 Nov 2024 23:14:16 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E5=AF=BC=E5=87=BA=E8=B4=A6=E5=8D=95?= =?UTF-8?q?=E4=BD=BF=E7=94=A8Excel=E5=AF=BC=E5=85=A5=E8=B4=A6=E5=8D=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../bill/{ => excel}/BillExportDto.java | 2 +- .../bill/excel/BillImportUserDto.java | 42 +++++++++ .../financial/bill/user/BillAddUserDto.java | 6 -- .../controller/financial/BillController.java | 10 +- .../controller/system/FilesController.java | 7 ++ .../bunny/services/excel/BillAddUserDAO.java | 69 +++++++++++++- .../services/excel/BillAddUserListener.java | 86 +++++++++++++++--- .../bunny/services/factory/BillFactory.java | 2 +- .../service/financial/BillService.java | 10 +- .../financial/impl/BillServiceImpl.java | 34 ++++++- .../services/service/system/FilesService.java | 6 ++ .../service/system/impl/FilesServiceImpl.java | 38 ++++++++ .../resources/static/bill-add-template.xlsx | Bin 0 -> 10910 bytes .../financial/impl/BillServiceImplTest.java | 66 +++++++++++++- .../src/test/java/excel/fill/FillTest.java | 30 ------ 15 files changed, 347 insertions(+), 61 deletions(-) rename dao/src/main/java/cn/bunny/dao/dto/financial/bill/{ => excel}/BillExportDto.java (95%) create mode 100644 dao/src/main/java/cn/bunny/dao/dto/financial/bill/excel/BillImportUserDto.java create mode 100644 service/src/main/resources/static/bill-add-template.xlsx delete mode 100644 service/src/test/java/excel/fill/FillTest.java diff --git a/dao/src/main/java/cn/bunny/dao/dto/financial/bill/BillExportDto.java b/dao/src/main/java/cn/bunny/dao/dto/financial/bill/excel/BillExportDto.java similarity index 95% rename from dao/src/main/java/cn/bunny/dao/dto/financial/bill/BillExportDto.java rename to dao/src/main/java/cn/bunny/dao/dto/financial/bill/excel/BillExportDto.java index 27fafe7..29af416 100644 --- a/dao/src/main/java/cn/bunny/dao/dto/financial/bill/BillExportDto.java +++ b/dao/src/main/java/cn/bunny/dao/dto/financial/bill/excel/BillExportDto.java @@ -1,4 +1,4 @@ -package cn.bunny.dao.dto.financial.bill; +package cn.bunny.dao.dto.financial.bill.excel; import com.fasterxml.jackson.annotation.JsonFormat; import io.swagger.v3.oas.annotations.media.Schema; diff --git a/dao/src/main/java/cn/bunny/dao/dto/financial/bill/excel/BillImportUserDto.java b/dao/src/main/java/cn/bunny/dao/dto/financial/bill/excel/BillImportUserDto.java new file mode 100644 index 0000000..4bbb304 --- /dev/null +++ b/dao/src/main/java/cn/bunny/dao/dto/financial/bill/excel/BillImportUserDto.java @@ -0,0 +1,42 @@ +package cn.bunny.dao.dto.financial.bill.excel; + +import com.alibaba.excel.annotation.ExcelProperty; +import com.fasterxml.jackson.annotation.JsonFormat; +import io.swagger.v3.oas.annotations.media.Schema; +import lombok.AllArgsConstructor; +import lombok.Builder; +import lombok.Data; +import lombok.NoArgsConstructor; + +import java.math.BigDecimal; +import java.time.LocalDateTime; + +@Data +@AllArgsConstructor +@NoArgsConstructor +@Builder +@Schema(name = "BillAddDto对象", title = "账单信息添加内容", description = "账单信息添加内容") +public class BillImportUserDto { + + @Schema(name = "username", title = "类型：1 - 收入，-1 - 支出") + @ExcelProperty(index = 0) + private String type; + + @Schema(name = "amount", title = "金额") + @ExcelProperty("金额") + private BigDecimal amount; + + @Schema(name = "description", title = "描述") + @ExcelProperty("描述") + private String description; + + @Schema(name = "transactionDate", title = "交易日期") + @JsonFormat(locale = "zh", timezone = "GMT+8", pattern = "yyyy-MM-dd HH:mm:ss") + @ExcelProperty("交易日期") + private LocalDateTime transactionDate; + + @Schema(name = "categoryId", title = "类别") + @ExcelProperty("类别") + private String categoryName; + +} \ No newline at end of file diff --git a/dao/src/main/java/cn/bunny/dao/dto/financial/bill/user/BillAddUserDto.java b/dao/src/main/java/cn/bunny/dao/dto/financial/bill/user/BillAddUserDto.java index fd191bd..f289df7 100644 --- a/dao/src/main/java/cn/bunny/dao/dto/financial/bill/user/BillAddUserDto.java +++ b/dao/src/main/java/cn/bunny/dao/dto/financial/bill/user/BillAddUserDto.java @@ -1,6 +1,5 @@ package cn.bunny.dao.dto.financial.bill.user; -import com.alibaba.excel.annotation.ExcelProperty; import com.fasterxml.jackson.annotation.JsonFormat; import io.swagger.v3.oas.annotations.media.Schema; import jakarta.validation.constraints.Max; @@ -25,28 +24,23 @@ public class BillAddUserDto { @NotNull(message = "类型不能为空") @Min(value = -1, message = "类型格式不正确") @Max(value = 1, message = "类型格式不正确") - @ExcelProperty(index = 0) private Byte type; @Schema(name = "amount", title = "金额") @NotNull(message = "金额不能为空") @Min(value = 0, message = "金额格式不正确") - @ExcelProperty("金额") private BigDecimal amount; @Schema(name = "description", title = "描述") - @ExcelProperty("描述") private String description; @Schema(name = "transactionDate", title = "交易日期") @NotNull(message = "交易日期不能为空") @JsonFormat(locale = "zh", timezone = "GMT+8", pattern = "yyyy-MM-dd HH:mm:ss") - @ExcelProperty("交易日期") private LocalDateTime transactionDate; @Schema(name = "categoryId", title = "类别id") @NotNull(message = "类别id不能为空") - @ExcelProperty("类别id") private Long categoryId; } \ No newline at end of file diff --git a/service/src/main/java/cn/bunny/services/controller/financial/BillController.java b/service/src/main/java/cn/bunny/services/controller/financial/BillController.java index f83e2b5..45cb978 100644 --- a/service/src/main/java/cn/bunny/services/controller/financial/BillController.java +++ b/service/src/main/java/cn/bunny/services/controller/financial/BillController.java @@ -1,10 +1,10 @@ package cn.bunny.services.controller.financial; import cn.bunny.dao.dto.financial.bill.BillDto; -import cn.bunny.dao.dto.financial.bill.BillExportDto; import cn.bunny.dao.dto.financial.bill.ExpendWithIncomeDto; import cn.bunny.dao.dto.financial.bill.admin.BillAddDto; import cn.bunny.dao.dto.financial.bill.admin.BillUpdateDto; +import cn.bunny.dao.dto.financial.bill.excel.BillExportDto; import cn.bunny.dao.dto.financial.bill.user.BillAddUserDto; import cn.bunny.dao.dto.financial.bill.user.BillUpdateUserDto; import cn.bunny.dao.entity.financial.Bill; @@ -23,6 +23,7 @@ import jakarta.servlet.http.HttpServletResponse; import jakarta.validation.Valid; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; +import org.springframework.web.multipart.MultipartFile; import reactor.core.publisher.Mono; import java.util.List; @@ -76,6 +77,13 @@ public class BillController { return Mono.just(Result.success(vo)); } + @Operation(summary = "使用Excel导入用户账单", description = "使用Excel导入用户账单") + @PostMapping("noManage/importBill") + public Mono>> importBill(MultipartFile file) { + billService.importBill(file); + return Mono.just(Result.success(ResultCodeEnum.ADD_SUCCESS)); + } + @Operation(summary = "导出用户账单信息", description = "导出用户账单信息") @PostMapping("noManage/exportBill") public void exportBill(@Valid @RequestBody BillExportDto dto, HttpServletResponse response) { diff --git a/service/src/main/java/cn/bunny/services/controller/system/FilesController.java b/service/src/main/java/cn/bunny/services/controller/system/FilesController.java index 422c311..1dcc710 100644 --- a/service/src/main/java/cn/bunny/services/controller/system/FilesController.java +++ b/service/src/main/java/cn/bunny/services/controller/system/FilesController.java @@ -16,6 +16,7 @@ import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import io.swagger.v3.oas.annotations.Operation; import io.swagger.v3.oas.annotations.Parameter; import io.swagger.v3.oas.annotations.tags.Tag; +import jakarta.servlet.http.HttpServletResponse; import jakarta.validation.Valid; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.http.ResponseEntity; @@ -83,6 +84,12 @@ public class FilesController { return filesService.downloadFilesByFilepath(filepath); } + @Operation(summary = "获取添加账单模板文件", description = "获取添加账单模板文件") + @GetMapping("noManage/getAddBillTemplate") + public void getAddBillTemplate(HttpServletResponse response) { + filesService.getAddBillTemplate(response); + } + // // 无法做权限校验 // @Operation(summary = "根据文件名访问resource下图片文件", description = "根据文件名访问resource下文件") // @GetMapping("noAuth/getResourceImagesByFilename/{filename}") diff --git a/service/src/main/java/cn/bunny/services/excel/BillAddUserDAO.java b/service/src/main/java/cn/bunny/services/excel/BillAddUserDAO.java index fe99a69..ddc9e72 100644 --- a/service/src/main/java/cn/bunny/services/excel/BillAddUserDAO.java +++ b/service/src/main/java/cn/bunny/services/excel/BillAddUserDAO.java @@ -1,11 +1,74 @@ package cn.bunny.services.excel; -import cn.bunny.dao.dto.financial.bill.user.BillAddUserDto; +import cn.bunny.common.service.context.BaseContext; +import cn.bunny.dao.dto.financial.bill.excel.BillImportUserDto; +import cn.bunny.dao.entity.financial.Bill; +import cn.bunny.dao.entity.financial.Category; +import cn.bunny.services.mapper.financial.CategoryMapper; +import cn.bunny.services.service.financial.BillService; +import cn.hutool.core.text.StrBuilder; +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; +import com.baomidou.mybatisplus.core.toolkit.StringUtils; +import com.baomidou.mybatisplus.core.toolkit.Wrappers; +import org.springframework.beans.BeanUtils; +import java.math.BigDecimal; +import java.time.LocalDateTime; import java.util.List; +import java.util.Map; +import java.util.concurrent.atomic.AtomicReference; +import java.util.stream.Collectors; public class BillAddUserDAO { - public void save(List list) { - // 如果是mybatis,尽量别直接调用多次insert,自己写一个mapper里面新增一个方法batchInsert,所有数据一次性插入 + + public void save(List cachedDataList, BillService billService, CategoryMapper categoryMapper, StrBuilder messageContent, AtomicReference index) { + Long userId = BaseContext.getUserId(); + + // 查询数据 + LambdaQueryWrapper queryWrapper = Wrappers.lambdaQuery().eq(Category::getUserId, userId).or().eq(Category::getIsBuiltin, true); + Map categoryMap = categoryMapper.selectList(queryWrapper).stream().collect(Collectors.toMap(Category::getCategoryName, Category::getId)); + + List billList = cachedDataList.stream() + // 判断是否有值 + .filter(billImportUserDto -> { + index.updateAndGet(i -> i == null ? 1 : i + 1); + + String billImportUserDtoType = billImportUserDto.getType(); + BigDecimal amount = billImportUserDto.getAmount(); + LocalDateTime transactionDate = billImportUserDto.getTransactionDate(); + String categoryName = billImportUserDto.getCategoryName(); + if (StringUtils.isEmpty(billImportUserDtoType) + || amount == null + || StringUtils.isEmpty(categoryName) + || transactionDate == null) { + + String string = "序号[" + index.get() + "]数据为空--> " + "类型：" + billImportUserDtoType + ", 金额：" + amount + ", 日期：" + transactionDate + ", 类别：" + categoryName + "
"; + messageContent.append(string); + return false; + } + return true; + }) + .map(billImportUserDto -> { + String billImportUserDtoType = billImportUserDto.getType(); + String categoryName = billImportUserDto.getCategoryName(); + + Bill bill = new Bill(); + BeanUtils.copyProperties(billImportUserDto, bill); + + // 设置用户id + bill.setUserId(userId); + + // 判断添加的类型是否正确 + byte type = billImportUserDtoType.equals("收入") ? Byte.parseByte("1") : Byte.parseByte("-1"); + bill.setType(type); + + // 添加的分类查找分类id + Long categoryId = categoryMap.get(categoryName); + bill.setCategoryId(categoryId); + + return bill; + }).toList(); + + billService.saveBatch(billList); } } \ No newline at end of file diff --git a/service/src/main/java/cn/bunny/services/excel/BillAddUserListener.java b/service/src/main/java/cn/bunny/services/excel/BillAddUserListener.java index 5a6e64d..0e237a2 100644 --- a/service/src/main/java/cn/bunny/services/excel/BillAddUserListener.java +++ b/service/src/main/java/cn/bunny/services/excel/BillAddUserListener.java @@ -1,27 +1,65 @@ package cn.bunny.services.excel; -import cn.bunny.dao.dto.financial.bill.user.BillAddUserDto; +import cn.bunny.common.service.context.BaseContext; +import cn.bunny.dao.dto.financial.bill.excel.BillImportUserDto; +import cn.bunny.dao.entity.system.Message; +import cn.bunny.dao.entity.system.MessageReceived; +import cn.bunny.services.mapper.financial.CategoryMapper; +import cn.bunny.services.mapper.message.MessageMapper; +import cn.bunny.services.mapper.message.MessageReceivedMapper; +import cn.bunny.services.service.financial.BillService; +import cn.hutool.core.text.StrBuilder; import com.alibaba.excel.context.AnalysisContext; +import com.alibaba.excel.exception.ExcelDataConvertException; import com.alibaba.excel.read.listener.ReadListener; import com.alibaba.excel.util.ListUtils; -import com.alibaba.fastjson2.JSON; +import com.alibaba.excel.util.StringUtils; import lombok.extern.slf4j.Slf4j; +import java.util.Base64; import java.util.List; +import java.util.concurrent.atomic.AtomicReference; +/** + * 无法被Spring管理必须使用构造函数注入方式 + */ @Slf4j -public class BillAddUserListener implements ReadListener { +public class BillAddUserListener implements ReadListener { + // 缓存数据 private static final int BATCH_COUNT = 100; + // 需要存入数据库的数据 private final BillAddUserDAO billAddUserDAO; - private List cachedDataList = ListUtils.newArrayListWithExpectedSize(BATCH_COUNT); + // 查询用户类别 + private final CategoryMapper categoryMapper; + // 用户消息 + private final MessageMapper messageMapper; + // 用户接受消息 + private final MessageReceivedMapper messageReceivedMapper; + // 存入数据库信息 + private final BillService billService; + private final StrBuilder messageContent = new StrBuilder(); + // 设置索引查看是第几个数据异常 + AtomicReference index = new AtomicReference<>(1); + private List cachedDataList = ListUtils.newArrayListWithExpectedSize(BATCH_COUNT); - public BillAddUserListener() { + public BillAddUserListener(CategoryMapper categoryMapper, MessageMapper messageMapper, MessageReceivedMapper messageReceivedMapper, BillService billService) { billAddUserDAO = new BillAddUserDAO(); + this.categoryMapper = categoryMapper; + this.billService = billService; + this.messageMapper = messageMapper; + this.messageReceivedMapper = messageReceivedMapper; } - public BillAddUserListener(BillAddUserDAO billAddUserDAO) { - this.billAddUserDAO = billAddUserDAO; + @Override + public void onException(Exception exception, AnalysisContext context) { + if (exception instanceof ExcelDataConvertException excelDataConvertException) { + int row = excelDataConvertException.getRowIndex() + 1; + int column = excelDataConvertException.getColumnIndex() + 1; + log.warn("第{}行，第{}列解析异常，请正确填写", row, column); + + messageContent.append("第").append(String.valueOf(row)).append("行，第").append(String.valueOf(column)).append("列解析异常，请正确填写
"); + } } /** @@ -30,8 +68,7 @@ public class BillAddUserListener implements ReadListener { * @param data one row value. Is is same as {@link AnalysisContext#readRowHolder()} */ @Override - public void invoke(BillAddUserDto data, AnalysisContext context) { - log.info("解析到一条数据:{}", JSON.toJSONString(data)); + public void invoke(BillImportUserDto data, AnalysisContext context) { cachedDataList.add(data); // 达到BATCH_COUNT了，需要去存储一次数据库，防止数据几万条数据在内存，容易OOM if (cachedDataList.size() >= BATCH_COUNT) { @@ -46,15 +83,40 @@ public class BillAddUserListener implements ReadListener { */ @Override public void doAfterAllAnalysed(AnalysisContext context) { + // 这里也要保存数据，确保最后遗留的数据也存储到数据库 saveData(); + if (StringUtils.isBlank(messageContent.toString())) return; + + // 获取当前用户id同时也是接受人的id + Long userId = BaseContext.getUserId(); + + // 将消息中的内容进行Base64编码，要求就是这样 + String encoded = Base64.getEncoder().encodeToString(messageContent.toString().getBytes()); + + // 如果有消息 + Message message = new Message(); + message.setTitle("导入账单时出错"); + message.setSendUserId(1L); + message.setMessageType("1851507850609356802"); + message.setSummary("导入账单时出错"); + message.setContent(encoded); + message.setEditorType("rich"); + message.setLevel("warning"); + message.setExtra("导入错误"); + messageMapper.insert(message); + + // 从之前保存的消息中获取消息id，保存到消息接收表中 + MessageReceived messageReceived = new MessageReceived(); + messageReceived.setMessageId(message.getId()); + messageReceived.setReceivedUserId(userId); + messageReceived.setStatus(false); + messageReceivedMapper.insert(messageReceived); } /** * 加上存储数据库 */ private void saveData() { - log.info("{}条数据，开始存储数据库！", cachedDataList.size()); - billAddUserDAO.save(cachedDataList); - log.info("存储数据库成功！"); + billAddUserDAO.save(cachedDataList, billService, categoryMapper, messageContent, index); } } diff --git a/service/src/main/java/cn/bunny/services/factory/BillFactory.java b/service/src/main/java/cn/bunny/services/factory/BillFactory.java index 7f9a0b4..97205cc 100644 --- a/service/src/main/java/cn/bunny/services/factory/BillFactory.java +++ b/service/src/main/java/cn/bunny/services/factory/BillFactory.java @@ -1,8 +1,8 @@ package cn.bunny.services.factory; import cn.bunny.common.service.exception.BunnyException; -import cn.bunny.dao.dto.financial.bill.BillExportDto; import cn.bunny.dao.dto.financial.bill.ExpendWithIncomeDto; +import cn.bunny.dao.dto.financial.bill.excel.BillExportDto; import cn.bunny.dao.excel.BillUserExportExcel; import cn.bunny.dao.pojo.result.ResultCodeEnum; import cn.bunny.dao.vo.financial.user.expendAndIncome.ExpendWithIncome; diff --git a/service/src/main/java/cn/bunny/services/service/financial/BillService.java b/service/src/main/java/cn/bunny/services/service/financial/BillService.java index 2758c89..af57259 100644 --- a/service/src/main/java/cn/bunny/services/service/financial/BillService.java +++ b/service/src/main/java/cn/bunny/services/service/financial/BillService.java @@ -1,10 +1,10 @@ package cn.bunny.services.service.financial; import cn.bunny.dao.dto.financial.bill.BillDto; -import cn.bunny.dao.dto.financial.bill.BillExportDto; import cn.bunny.dao.dto.financial.bill.ExpendWithIncomeDto; import cn.bunny.dao.dto.financial.bill.admin.BillAddDto; import cn.bunny.dao.dto.financial.bill.admin.BillUpdateDto; +import cn.bunny.dao.dto.financial.bill.excel.BillExportDto; import cn.bunny.dao.dto.financial.bill.user.BillAddUserDto; import cn.bunny.dao.dto.financial.bill.user.BillUpdateUserDto; import cn.bunny.dao.entity.financial.Bill; @@ -15,6 +15,7 @@ import cn.bunny.dao.vo.financial.user.expendAndIncome.ExpendWithIncomeListVo; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.IService; import jakarta.servlet.http.HttpServletResponse; +import org.springframework.web.multipart.MultipartFile; import java.util.List; @@ -107,4 +108,11 @@ public interface BillService extends IService { * @param response 响应体 */ void exportBillByAdmin(BillExportDto dto, HttpServletResponse response); + + /** + * 使用Excel导入用户账单 + * + * @param file 文件 + */ + void importBill(MultipartFile file); } diff --git a/service/src/main/java/cn/bunny/services/service/financial/impl/BillServiceImpl.java b/service/src/main/java/cn/bunny/services/service/financial/impl/BillServiceImpl.java index 47782fd..47d5816 100644 --- a/service/src/main/java/cn/bunny/services/service/financial/impl/BillServiceImpl.java +++ b/service/src/main/java/cn/bunny/services/service/financial/impl/BillServiceImpl.java @@ -3,10 +3,11 @@ package cn.bunny.services.service.financial.impl; import cn.bunny.common.service.context.BaseContext; import cn.bunny.common.service.exception.BunnyException; import cn.bunny.dao.dto.financial.bill.BillDto; -import cn.bunny.dao.dto.financial.bill.BillExportDto; import cn.bunny.dao.dto.financial.bill.ExpendWithIncomeDto; import cn.bunny.dao.dto.financial.bill.admin.BillAddDto; import cn.bunny.dao.dto.financial.bill.admin.BillUpdateDto; +import cn.bunny.dao.dto.financial.bill.excel.BillExportDto; +import cn.bunny.dao.dto.financial.bill.excel.BillImportUserDto; import cn.bunny.dao.dto.financial.bill.user.BillAddUserDto; import cn.bunny.dao.dto.financial.bill.user.BillUpdateUserDto; import cn.bunny.dao.entity.financial.Bill; @@ -17,10 +18,15 @@ import cn.bunny.dao.vo.financial.user.BillUserVo; import cn.bunny.dao.vo.financial.user.expendAndIncome.CategoryAmount; import cn.bunny.dao.vo.financial.user.expendAndIncome.ExpendWithIncome; import cn.bunny.dao.vo.financial.user.expendAndIncome.ExpendWithIncomeListVo; +import cn.bunny.services.excel.BillAddUserListener; import cn.bunny.services.factory.BillFactory; import cn.bunny.services.factory.HomeFactory; import cn.bunny.services.mapper.financial.BillMapper; +import cn.bunny.services.mapper.financial.CategoryMapper; +import cn.bunny.services.mapper.message.MessageMapper; +import cn.bunny.services.mapper.message.MessageReceivedMapper; import cn.bunny.services.service.financial.BillService; +import com.alibaba.excel.EasyExcel; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; @@ -29,7 +35,9 @@ import jakarta.servlet.http.HttpServletResponse; import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; +import org.springframework.web.multipart.MultipartFile; +import java.io.IOException; import java.time.LocalDate; import java.util.List; import java.util.stream.Collectors; @@ -47,9 +55,17 @@ public class BillServiceImpl extends ServiceImpl implements Bi @Autowired private HomeFactory homeFactory; + @Autowired private BillFactory billFactory; + @Autowired + private CategoryMapper categoryMapper; + @Autowired + private MessageMapper messageMapper; + @Autowired + private MessageReceivedMapper messageReceivedMapper; + /** * * 账单信息 服务实现类 * @@ -195,7 +211,7 @@ public class BillServiceImpl extends ServiceImpl implements Bi // 分类列表 List categoryAmountList = expendWithIncomeList.stream() - .filter(expendWithIncome -> dto.getType() != null && expendWithIncome.getType().equals(dto.getType())) + .filter(expendWithIncome -> dto.getType() != null && expendWithIncome.getType().equals(dto.getType()) && expendWithIncome.getCategoryName() != null) .collect(Collectors.groupingBy(ExpendWithIncome::getCategoryName, Collectors.summingDouble(expend -> expend.getAmount().doubleValue())) ).entrySet() @@ -237,6 +253,20 @@ public class BillServiceImpl extends ServiceImpl implements Bi billFactory.exportBill(dto, response); } + /** + * 使用Excel导入用户账单 + * + * @param file 文件 + */ + @Override + public void importBill(MultipartFile file) { + try { + EasyExcel.read(file.getInputStream(), BillImportUserDto.class, new BillAddUserListener(categoryMapper, messageMapper, messageReceivedMapper, this)).sheet().doRead(); + } catch (IOException e) { + throw new BunnyException(ResultCodeEnum.UPDATE_ERROR); + } + } + /** * 删除|批量删除账单信息 * diff --git a/service/src/main/java/cn/bunny/services/service/system/FilesService.java b/service/src/main/java/cn/bunny/services/service/system/FilesService.java index 809628b..bb1ee39 100644 --- a/service/src/main/java/cn/bunny/services/service/system/FilesService.java +++ b/service/src/main/java/cn/bunny/services/service/system/FilesService.java @@ -10,6 +10,7 @@ import cn.bunny.dao.vo.system.files.FileInfoVo; import cn.bunny.dao.vo.system.files.FilesVo; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.IService; +import jakarta.servlet.http.HttpServletResponse; import jakarta.validation.Valid; import org.springframework.http.ResponseEntity; @@ -84,4 +85,9 @@ public interface FilesService extends IService { * @return 媒体文件类型列表 */ Set getAllMediaTypes(); + + /** + * 获取添加账单模板文件 + */ + void getAddBillTemplate(HttpServletResponse response); } diff --git a/service/src/main/java/cn/bunny/services/service/system/impl/FilesServiceImpl.java b/service/src/main/java/cn/bunny/services/service/system/impl/FilesServiceImpl.java index c55491b..3b675aa 100644 --- a/service/src/main/java/cn/bunny/services/service/system/impl/FilesServiceImpl.java +++ b/service/src/main/java/cn/bunny/services/service/system/impl/FilesServiceImpl.java @@ -9,18 +9,25 @@ import cn.bunny.dao.dto.system.files.FileUploadDto; import cn.bunny.dao.dto.system.files.FilesAddDto; import cn.bunny.dao.dto.system.files.FilesDto; import cn.bunny.dao.dto.system.files.FilesUpdateDto; +import cn.bunny.dao.entity.financial.Category; import cn.bunny.dao.entity.system.Files; import cn.bunny.dao.pojo.common.MinioFilePath; import cn.bunny.dao.pojo.result.PageResult; import cn.bunny.dao.pojo.result.ResultCodeEnum; import cn.bunny.dao.vo.system.files.FileInfoVo; import cn.bunny.dao.vo.system.files.FilesVo; +import cn.bunny.services.mapper.financial.CategoryMapper; import cn.bunny.services.mapper.system.FilesMapper; import cn.bunny.services.service.system.FilesService; +import com.alibaba.excel.EasyExcel; +import com.alibaba.excel.ExcelWriter; +import com.alibaba.excel.write.metadata.WriteSheet; +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import jakarta.servlet.http.HttpServletResponse; import jakarta.validation.Valid; import lombok.SneakyThrows; import org.springframework.beans.BeanUtils; @@ -38,6 +45,7 @@ import java.io.IOException; import java.lang.reflect.Field; import java.util.HashSet; import java.util.List; +import java.util.Objects; import java.util.Set; /** @@ -63,6 +71,8 @@ public class FilesServiceImpl extends ServiceImpl implements @Autowired private FilesMapper filesMapper; + @Autowired + private CategoryMapper categoryMapper; /** * * 系统文件表 服务实现类 @@ -289,4 +299,32 @@ public class FilesServiceImpl extends ServiceImpl implements return Set.of(); } } + + /** + * 获取添加账单模板文件 + */ + @Override + public void getAddBillTemplate(HttpServletResponse response) { + // 设置数据库查询时间 + Long userId = BaseContext.getUserId(); + + // 查询数据 + LambdaQueryWrapper queryWrapper = Wrappers.lambdaQuery().eq(Category::getUserId, userId).or().eq(Category::getIsBuiltin, true); + List categoryList = categoryMapper.selectList(queryWrapper); + + String filenameTemplate = Objects.requireNonNull(getClass().getResource("/static/bill-add-template.xlsx")).getFile(); + if (filenameTemplate == null) throw new BunnyException(ResultCodeEnum.MISSING_TEMPLATE_FILES); + + try (ExcelWriter excelWriter = EasyExcel.write(response.getOutputStream()).withTemplate(filenameTemplate).build()) { + // 填充数据，类型数据要填充的在第二个sheet + WriteSheet writeSheet = EasyExcel.writerSheet(1).build(); + excelWriter.fill(categoryList, writeSheet); + + response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"); + response.setCharacterEncoding("utf-8"); + response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + "bill-add-template.xlsx"); + } catch (Exception e) { + throw new RuntimeException(e); + } + } } diff --git a/service/src/main/resources/static/bill-add-template.xlsx b/service/src/main/resources/static/bill-add-template.xlsx new file mode 100644 index 0000000000000000000000000000000000000000..6f70f3480edf6984f5737ab430c96df0967e93bb GIT binary patch literal 10910 zcmeHN^;=un)(-A21wxAj_m)C|;_mM5?k>fPOK~W!#oe{I6n7e&;uIl7XS}H1ONbJ0BnMlWn(A+fB+5vzyTmaYYExeI-A%! z>nVHKn>gt(x!YKi=E6eLWCNffpa0+be|!SH3FEfiEU5j*@t5c-6-vN078#uPJ&(j3 z`PB%6L_!o(pm4On#w?)z$3pZV{;vfJz`Hvi2P2GugML%`Q1do=ku8GD<#M>>+$Y>&G_1H!krmtX(0scH!ic1G;MJFOnrzZR4IOK*7 zXI4m1Ci5BCk6AOFvwQ>APre#12on`zSBTYB?jdlr+4>X{fpPq-&mb0Ek}oY4xvnw^ zO)dK~rhr8mR8h5vW`rnV;|}jyR)9ibfv>@bOK*v5;pi?t2h3IO4|^Uyv~eS`ffP0j zYX!c(*gV_XrHHptbM!5#_(oO+HX)=tb!f9(GQ<^N+O z{^hTi#mUI^u%HGXi{FO~Ud=9lz!a2n6A*7DQ}*$bSi-1}$^!wHJE@5=m5Bmj#C+R* z?uWt4Z=?1H$u2k8$|A6FxhWdl%7c?{9b6FTs2r0-9m>{wv0P^_W-n94Bt2+c+hgcT zn~HLzhE^#=r%#2eF~^zIh>&m!h=cKYQvEgiWi(a{uPdNt1(m*+2Uj(6Wben1r}@q& z74D(&g>p(CPNm`vIvJWRR(KCulU>{psVJFqT2vWkI&zVF=owpoJ`+ys#Ch;$luaL0 zBxl3EemO2SK$(5vt69f(`X$4omjkJ%bns*-AcCZ51>&dv6C_!20p8Rw002D#0Du8G z8Fy=DH#%F$kB{4BAWu0}X+IMt z6$w<-+4@iPed2{q{n8rl{>Ngv5>5P8LF(|FLqQsWG^&^|=PqOP%ejf? zl=hSg94=dm8ma)+O*oE5G#W=qqAYEc;=wXNqgvBJ8SXv#wGqk3>}4crZmq<0y5g;+ zL#Vn&s`gF3OPjs&tolz zY{TvYZIS8?&PbK5+~$tQd*F48n`XCXXb@K9L`lf_%#5KHp=FHI40-*X*TxdF>H5{& zGBD+C2Ml(jLvU8nLe>X)_RpzpnTdiob=eT96r3@}if9VsW~f#Xy9B9eg369lcw<$p zx_q?H4*mNxIz6<(d{&y^c9(@09?#MlF9MAZCp2^%V~0!va%~Z1mm7LW5%rs_Ba`30 zYM#}uT(#NZW%0>?FoyYKF;>@h5nQg! zacJS&vBOi8iBWAFN1Bj@qa4|n);!*Ax8vh_oxa|_bwsl5-}kez_*;=E;WFq3T+ys? zQSApR^y$wO(h3n2ob2I49u>w(f-3lqoR_Qk5SM%wkhH}eS&lCI^G+S+x94;jPM_ZB zrhN?u1HaUaguUDep8z)P$S&Tem}44kb|82Qd||YzeAr)Oh9KrYF(p}CXmAU8M+|~1 zA^;*31XF+dp1&gN-+l-R5*$L1`Tu*h#f@6_vY-YXzwZc~Yl@ zn~ccL*V!K}!ZQ>w-L{E(eNcyP>}*!qbfRK9e1H?DgeXH{DDm0Rj5Prk_qzBR&~P4^ zIok2-Jp(Wr!n_W|YM@V2&9F-MuSN`CHd=Lmn8g}J8BpfsDtM*EOzBihhZ^&4|F)W* zZItiQQT1sb<)17^uhwrf9}>)vp+a&Bh)Mj5jh)O*Oq`vVe_mMspyl*<9V;*kZtyX} z1BT#Lwe|dvCeEjXdXsiw9=HPU<_-R22=&m7{$c0wphKLXeh`tCSc7|(Q}n&%W7TxH z^=A(Y8If900c>PdJzHt`qq|q?lEMZI$7{7YevzxSy`GKjX>Y6LuZ;kI(yr_91~O+# zBi*{}W;tsFcr~H}HjL|i)e$#zcb?EXLywPRDce&MFgeciEY{QFP;#p0MAC2giD_O` z`K%6)kyvPI2Tm`Q-`SFv9br}{NCrOu3WOtmxR{h}O1*z~ToZs&37I@99K2MX#Bnohzk z6hZFtq`9n!16BUa*~%hWD9-1{{$Wb6H8`ga_u{^UsQ06>L_{m4C(E{-0owIYa#FG1 zlnxm*^UAK|*5Ev`ov1y&A(vtXP)uTvw5L7&bKLZv6MKpk|V$ULens#^8qf>5-AM+ieaJwYDy)06SI0CFG;Cfr16 z5GTJ;uB|b4kqid5;pap~S$VI!=(-}Z*7dXN5iueo=`m_M2cL37_H6zMuw1X&e87V8 zrVxj>c0T4OaYxBO z#~U$YIeI_8{O6XhEMqqC?&6MYbRzk z3#bCe#Z9->*w)hSK+~OOAZP$ThC*vNjaoFZ*0EI%H)@1E$tL{)mkJ4d8pqPA!>T2{ zFeaKL*GEMgZV$vpSxWBrbLvlybfM!JJ^iRG`RYiT_S?Y`z)kwl#buF=8)gwzTIP~L zD=~0AyHraw`(Sx(1?kzx(ifljxyRCuzA}34p**sV7-b(_CTulkyu`I-4iU2s_6C9| zDM9eEGfD43vW?WEo%$XoHWYjKH~|HtmV(`^P_e~~F#a2<=J?rm@?R4Lm226ya90lR z1P~g*N!D7K%xR4+4Uq`CVlNgtY_18s`t{v0Bx>1)kj)BH-?3qlyWtjuQi0GSR3$0K zkntl_dlldJu||a|ek0DMp$ZCRQ*c(P$J3I?5%lA+-$;*j|FF41?moJ@BxH(JeD5}P z*)bt2h(aUtF;0dZ-k(4FNk}w*W)>e0$^G_m@8Ev+SCD<8+T1$u{#< z^bz0ve?un{dfBYP91ALj5(4rMEDY6yj%Q1v+VVkV+ef5`_H9`d5^Zqbfff+@@f7sVv zDP1bZB{(ObdGIVF@~i2&{$9enob?k-zbB)CvZh(vkVJG8!tVct*_gFcESSN^ z6r>n=fN<-5ob?GJf<^j%)|v3mn{ z33a%0ZN_4Yai-8Xs2QhlPxE>$)Ob2TSs*D!y&0iQ`yU3#?u;NO{PbmuT%94LR9`(`goPwuK^`>Z{aQ9c0ZZm_aFNvo7)c_xH9R#>P+-6%>bJmYcYb`f#(9wtq_s%BerKOa67n=|(7rS}3+nbl}&R9-48(lCjS(8Jf1MJD3Xp~_&#mn+igC`QC9ok?0JqMk=499qW- zNIn;^v`Lgs#P!?gfpq@EmH4+kx3}J1`cWNOzAk8z>ji_i-DnsjcE0%$Lx4rr2$& zbA*NRwtwcXx;eIfOh zo4dv=U7kCp<8$)NgX`o@pNAhYe0n=C`ZeL#`>w|I9v|=fFjns`zPyUKvJLL%@)pJ( z$pBw3H4k7De0t|}&2x9J_eE&)J*J2x#@Wp5Yh6ELyDmUu zNdfAf9eC96Cn>C7jZ$9&8jbZdgT%f#+aABw+Hdkg%vijQmbV1CS84lIVW)ke>4zU} z#VvpybqIe?kyWh}$(ZLQ9*Q+@^m2&6?A7{PkSqq4mAI&}HLjwCvpDwPmtj#%MVsF3 zYyWAZ8TNf%-dC|FnxvLweBCpMn^2m_2@BfM@`4wBBxG1%@yxQCuxOw?RBEQtaC#T6 zlWlEte3yHD{wlOh^{0oF3T&KIBuC3Lk065S*mtf(eKZ>CobhaP)p{pNGz#CAOuFIu zOt*pL$7C2Z%c(~Qu)w_zZN#I%vi*c%=vcX=RLW$8l+@$R@G(qPI+zl>P!GW=slF@jAaF(`i_q<#yqoEfzTy5Dat@cj>qeporp3i!$YIw}v#U0AUeSbWr^qOz= zz7W*0d^PCW%&S&( z;U!fEh@3ZU5*q*Fb^S&9S@%UcH%h-jdF97hDgwQea#z$Z-%@1Gcr44QuJW4Q?Lot? z%nNlrkAmNJt2B;U*^g#rT`MjJte~eZZN$wpQb9!c*rP5K#_KMinZb%65pZ`d<#MG` z{n!Q-F27AYp88A9s)_w3|E%K&1K@j_%;C3N1Av3W;cwa&g>G>1`*?tZ5Er_EjooW+0;VSK0ryL(YcVpdfZr)ul!0n zwC4NiZc9~IpFI{}8M%&7FK_Kr8M0lQwD-xmLSSmc27f$z)JZUjb(j=;I|frBOsy6e zgqY5m0D7*5?yH2xrb3k4G`mlVNW$36J~%ueiXLZ6ML2HiPJUiz#rSgShLOg*0&niq z@cnYvR2(DOGWF^C65+IoD~3QA+~`NtwFLV&KZ>^|)3M+5EFo%W_&%d56m-215>^rN z;LeR8z5ijQ%2F-~Z@T@aSbF^UZ37cCgWBNTf^D6bLBaAa zI=xTld`VqimD*m2LM`8+m_R9y_db^y@}V0rmIPfhu?k-VkMA7xpxQchesEE-de{K( zb7fH-=^wlEL!lSCyVU*r(?6!_Avh%z1F!&qJr)3f_NUWwa`vz`arz^Q>s23(2D4*! zJlo|)0|9NU$I4cS>Xq;z|wEz$ACiC%(W)RwU9rblbz6t$Zo$R48NRGVirLE zKAkVInk%GOvpZU$c83EC-GXE-{HYwTCoaZU=P zNR@_1iBkRws&qL4o;k-??a)};L})$aHhKB{8#k7V5+2pUaU+P7j(IN4va#qSaik$n=>3ok_QN-UtH=891bU#VNySRzUuTk*{Wrp zSzCme2gbBXPP!*J&Fw~fcEOujck1sn3%fOrEK=E2xasB;!!?C2px8=45rZcD9Gq!T z^x3qpmCx7bYMZ22^$Vl5_%y6DjYBGlXD>p@vu(B%TljN> z-m-S>U4CumRO)I1<>Ck1{)lCrxkRuzbcWc$gqj=2vtCvTI3!N7 z_=_2}iWP))dlo37H4(>{Y8%Vn9|TOoB#+Lx3p+6R2Ly~f+pqj4b)Dfro5LS1MkwGk zhN;q*n^Z-)KH+yp=DG#F@1=uULi2)Ze{vxqc34yjDN3xzf)?=EYKf6DGT!Y>ArUb? zQ&)SZZ3m1DYb}B#< zLVQculH1JNv}eJT_yXu;=N#0bwPVa?k}lM((fV39UjlL})f?Bcx|DGybJJUR*v~cX z1$6n-83XF`m!h4#unnFK+KPG$yQ^9r4Zs2HQtmN!yVQco2`10fU zqJN{6tzKloTfU3Q_?5=?*;X-Yw^IkMo2M! z*m~Q3rWJ%~LUmaYmv(EMbU%pCdG%I?!Ac=HJ5?4hrKsC$haO}S3v(=jV>{zxqhXhj zwOb^S@QaI4Y8cnWVruwcc?i7Q`8deDBT3F@n>&d24qBa`Ze8ErkLW(4o%BT`UVg3=mQs{ZECMo$RYk`HpENASg8X! zW@AXMgazpy7~2`iJKEVhF&o)An*8BhAkFLlH4Pv~=N(rq+rxq@;s$yo`;qxZ7A&6& zN(8t-BZZM-rkQj$U{&KR2G?bEjg3Y2_CHOmewO(90l9bAtup5Lo#R`rh;9$~0!Vcz z)ScInzb05W@UXIKzmrJ|ZGjsCS7R7{~tX>KBo zXHd3?3_F|YB2LdFNm*1aR^N+f661{Y=Hvkrg(X2r$N|Q=G&9sye9Y7MB0n7eGhr|^ ztau*lDdOWPk-XrOM~(`-Y;2m03HSF6^u=@+!(-ipHjX)$@3VRA!J~2~53^e`TC<}j zwe;{bQJfJe(3X;kKJ6WLtGW$YB&;u_)6nph)970dMbPi9<6W5sZjJ7>>IWMH2H=hs zq!s4%E0?0im}@vCVX~yP`J5jlL<_MvqeM4Dwoe1^WcqDMU9lc1;f<0pIo=H2vwT!H zd_#H!Ws!DwqL=^AHaX=Jp_LD@NqPwMG5>0l2KM&mzPZlU0Ny+ zeOb&Zlw+4sqne>Z?rK#x3}%FHzDsSo!nH1Sy!Gg>2^vlu_C$c=<~KC^aY|7g&C4$g zLrZ~@jI2r#Ot~X5KT*U+u^6ie8T-j+J8G(=%UfSMIDjYqA(ENpxIoc_Jsi(MRMX|K zxbgHcN!QJL&(aGKmZ9$qvuqN1FZoCP(A4DhjVmpC~;F z{mZ1p8YI|G=1>&*<%zlBUXVz{7kswc6w0&A{s#R;t;FmeTJ_Rg1$3lRD_|UN1DQcW zBZr>PZejO=bl42D}8ku)K)!s|U0<&gdJ3Ae3AQKLMvMm%eBgBFI^9C;)(u z{8RY9w48s}^LxwjmnL~g6Y{?`Ab;2LdwKhp7CP*I)bg{~{kxXm3xU71h~oX!@_SkE zcLl#EWxo{M0e>p^m7@JF`g?TqOY}SGpQ69VI=^f9yW{>P4*)2WL)`iAZv1!gzjN$g g#dkq}5&swC%1gmPY~+u@HWWZVBvR<6{^QmE0c~joUjP6A literal 0 HcmV?d00001 diff --git a/service/src/test/java/cn/bunny/services/service/financial/impl/BillServiceImplTest.java b/service/src/test/java/cn/bunny/services/service/financial/impl/BillServiceImplTest.java index 4eba565..f4f6c8f 100644 --- a/service/src/test/java/cn/bunny/services/service/financial/impl/BillServiceImplTest.java +++ b/service/src/test/java/cn/bunny/services/service/financial/impl/BillServiceImplTest.java @@ -2,13 +2,20 @@ package cn.bunny.services.service.financial.impl; import cn.bunny.common.service.exception.BunnyException; import cn.bunny.dao.dto.financial.bill.ExpendWithIncomeDto; +import cn.bunny.dao.entity.financial.Category; +import cn.bunny.dao.entity.system.Message; +import cn.bunny.dao.entity.system.MessageReceived; import cn.bunny.dao.excel.BillUserExportExcel; import cn.bunny.dao.pojo.result.ResultCodeEnum; import cn.bunny.dao.vo.financial.user.expendAndIncome.ExpendWithIncome; import cn.bunny.services.mapper.financial.BillMapper; +import cn.bunny.services.mapper.financial.CategoryMapper; +import cn.bunny.services.mapper.message.MessageMapper; +import cn.bunny.services.mapper.message.MessageReceivedMapper; import com.alibaba.excel.EasyExcel; import com.alibaba.excel.ExcelWriter; import com.alibaba.excel.write.metadata.WriteSheet; +import com.baomidou.mybatisplus.core.toolkit.Wrappers; import org.junit.jupiter.api.Test; import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; @@ -16,16 +23,19 @@ import org.springframework.boot.test.context.SpringBootTest; import java.math.BigDecimal; import java.time.LocalDate; -import java.util.HashMap; -import java.util.List; -import java.util.Map; -import java.util.Objects; +import java.util.*; import java.util.concurrent.atomic.AtomicReference; @SpringBootTest class BillServiceImplTest { @Autowired private BillMapper billMapper; + @Autowired + private CategoryMapper categoryMapper; + @Autowired + private MessageMapper messageMapper; + @Autowired + private MessageReceivedMapper messageReceivedMapper; @Test void exportBill() { @@ -81,4 +91,52 @@ class BillServiceImplTest { throw new RuntimeException(e); } } + + @Test + void downloadAddBillTemplate() { + // 设置数据库查询时间 + Long userId = 1849444494908125181L; + + // 查询数据 + List categoryList = categoryMapper.selectList(Wrappers.lambdaQuery().eq(Category::getUserId, userId)); + + String filenameTemplate = Objects.requireNonNull(getClass().getResource("/static/bill-add-template.xlsx")).getFile(); + if (filenameTemplate == null) throw new BunnyException(ResultCodeEnum.MISSING_TEMPLATE_FILES); + + try (ExcelWriter excelWriter = EasyExcel.write("F:\\数据库备份\\bill-add-template.xlsx").withTemplate(filenameTemplate).build()) { + // 填充数据 + WriteSheet writeSheet = EasyExcel.writerSheet(1).build(); + excelWriter.fill(categoryList, writeSheet); + } catch (Exception e) { + throw new RuntimeException(e); + } + } + + @Test + void addBillByExcel() { + String encoded = Base64.getEncoder().encodeToString("messageContent.toString()".getBytes()); + + // 如果有消息 + Message message = new Message(); + message.setTitle("导入账单时出错"); + message.setSendUserId(1L); + message.setMessageType("1851507850609356802"); + message.setSummary("导入账单时出错"); + message.setContent(encoded); + message.setEditorType("rich"); + message.setLevel("warning"); + message.setExtra("导入错误"); + messageMapper.insert(message); + + // 从之前保存的消息中获取消息id，保存到消息接收表中 + // Long userId = BaseContext.getUserId(); + Long userId = 1849444494908125181L; + MessageReceived messageReceived = new MessageReceived(); + messageReceived.setMessageId(message.getId()); + messageReceived.setReceivedUserId(userId); + messageReceived.setStatus(false); + messageReceivedMapper.insert(messageReceived); + } + + } \ No newline at end of file diff --git a/service/src/test/java/excel/fill/FillTest.java b/service/src/test/java/excel/fill/FillTest.java deleted file mode 100644 index d3aab4b..0000000 --- a/service/src/test/java/excel/fill/FillTest.java +++ /dev/null @@ -1,30 +0,0 @@ -package excel.fill; - -import org.junit.jupiter.api.Test; -import org.springframework.boot.test.context.SpringBootTest; - -@SpringBootTest -public class FillTest { - - @Test - void fillTest1() { - // // 模板注意 用{} 来表示你要用的变量 如果本来就有"{","}" 特殊字符 用"\{","\}"代替 - // String filenameTemplate = "F:\\数据库备份\\bill-template.xlsx"; - // - // // 设置数据库查询时间 - // BillExportDto billExportDto = new BillExportDto(); - // billExportDto.setUserId(1849444494908125181L); - // billExportDto.setStartDate(LocalDate.of(2024, 11, 1)); - // billExportDto.setEndDate(LocalDate.of(2024, 11, 30)); - // - // EasyExcel.write("F:\\数据库备份\\" + billExportDto.getStartDate() + "~" + billExportDto.getEndDate() + ".xlsx").withTemplate(filenameTemplate).sheet().doFill(fillData); - - // // 方案2 根据Map填充 - // fileName = TestFileUtil.getPath() + "simpleFill" + System.currentTimeMillis() + ".xlsx"; - // // 这里 会填充到第一个sheet， 然后文件流会自动关闭 - // Map map = MapUtils.newHashMap(); - // map.put("name", "张三"); - // map.put("number", 5.2); - // EasyExcel.write(fileName).withTemplate(templateFileName).sheet().doFill(map); - } -}