ajax 9 місяців тому
батько
коміт
558bdf473f

+ 6 - 2
taais-modules/taais-biz/src/main/java/com/taais/biz/controller/DataController.java

@@ -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())) {

+ 2 - 0
taais-modules/taais-biz/src/main/java/com/taais/biz/domain/bo/DataBo.java

@@ -7,6 +7,7 @@ import jakarta.validation.constraints.NotBlank;
 import jakarta.validation.constraints.NotNull;
 
 import java.util.Date;
+import java.util.List;
 
 /**
  * 数据管理业务对象 data
@@ -105,5 +106,6 @@ public class DataBo {
      */
     private Boolean labeled;
 
+    private List<Long> selectedIds;
 
 }

+ 2 - 0
taais-modules/taais-biz/src/main/java/com/taais/biz/service/impl/DataServiceImpl.java

@@ -111,6 +111,8 @@ public class DataServiceImpl extends BaseServiceImpl<DataMapper, Data> implement
         queryWrapper.and(DATA.LABELED.eq
             (dataBo.getLabeled()));
 
+
+
         return queryWrapper;
     }