|
@@ -1,6 +1,7 @@
|
|
|
package com.taais.biz.controller;
|
|
|
|
|
|
import cn.dev33.satoken.annotation.SaCheckPermission;
|
|
|
+import cn.hutool.core.collection.CollectionUtil;
|
|
|
import com.taais.biz.domain.vo.BatchDataResult;
|
|
|
import cn.hutool.core.util.IdUtil;
|
|
|
import com.taais.biz.domain.Data;
|
|
@@ -93,9 +94,12 @@ public class DataController extends BaseController {
|
|
|
@SaCheckPermission("demo:data:export")
|
|
|
@Log(title = "数据管理", businessType = BusinessType.EXPORT)
|
|
|
@PostMapping("/export")
|
|
|
- public void export(HttpServletResponse response, DataBo dataBo) {
|
|
|
+ public void export(HttpServletResponse response, @RequestBody DataBo dataBo) {
|
|
|
// 查询数据列表
|
|
|
List<DataVo> list = dataService.selectList(dataBo);
|
|
|
+ if (CollectionUtil.isNotEmpty(dataBo.getSelectedIds())) {
|
|
|
+ list = list.stream().filter(item -> dataBo.getSelectedIds().contains(item.getId())).collect(Collectors.toList());
|
|
|
+ }
|
|
|
|
|
|
// 创建一个临时目录用于存放文件
|
|
|
String tempDirPath = System.getProperty("java.io.tmpdir") + "/export_" + System.currentTimeMillis();
|
|
@@ -144,7 +148,7 @@ public class DataController extends BaseController {
|
|
|
} catch (IOException e) {
|
|
|
throw new RuntimeException("生成Excel文件失败", e);
|
|
|
}
|
|
|
- // 将临时目录打包为ZIP文件
|
|
|
+ // 将临时目录打包为ZIP文件C:\Users\10418\AppData\Local\Temp\/export_1726842007326
|
|
|
File zipFile = new File(tempDir.getParent(), "export_" + System.currentTimeMillis() + ".zip");
|
|
|
try (ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(zipFile))) {
|
|
|
for (File file : Objects.requireNonNull(tempDir.listFiles())) {
|