From 04bd9283875970be1d51ad0e75303a753826f55d Mon Sep 17 00:00:00 2001
From: yangzhe123 <2824096059@qq.com>
Date: Wed, 3 Sep 2025 14:11:20 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A4=A7=E5=9D=9D-=E6=96=B0=E5=A2=9E=E6=96=87?=
 =?UTF-8?q?=E4=BB=B6=E8=8E=B7=E5=8F=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../project/hsz/controller/AttDamBaseController.java | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/gunshi/project/hsz/controller/AttDamBaseController.java b/src/main/java/com/gunshi/project/hsz/controller/AttDamBaseController.java
index d0e5233..595bbd4 100644
--- a/src/main/java/com/gunshi/project/hsz/controller/AttDamBaseController.java
+++ b/src/main/java/com/gunshi/project/hsz/controller/AttDamBaseController.java
@@ -92,13 +92,21 @@ public class AttDamBaseController extends AbstractCommonFileController{
     @Operation(summary = "列表")
     @PostMapping("/list")
     public R<List<AttDamBase>> list() {
-        return R.ok(service.lambdaQuery().list());
+        List<AttDamBase> list = service.lambdaQuery().list();
+        for (AttDamBase attDamBase : list) {
+            attDamBase.setFiles(fileService.getFiles(getGroupId(), attDamBase.getDamCode()));
+        }
+        return R.ok(list);
     }
 
     @Operation(summary = "获取主副坝,拦洪坝信息")
     @GetMapping("/info")
     public R<List<AttDamBase>> info(@Schema(name = "isMain",description = "是否主坝（0否 1是 2拦洪坝）") @RequestParam(name = "isMain") Integer isMain) {
-        return R.ok(service.lambdaQuery().eq(AttDamBase::getIsMain,isMain).list());
+        List<AttDamBase> list = service.lambdaQuery().eq(AttDamBase::getIsMain, isMain).list();
+        for (AttDamBase attDamBase : list) {
+            attDamBase.setFiles(fileService.getFiles(getGroupId(), attDamBase.getDamCode()));
+        }
+        return R.ok(list);
     }
 
     @Operation(summary = "分页")