From dc1c8c974d9469591ead4dd1d774bbfb97a33331 Mon Sep 17 00:00:00 2001
From: cjl <276999030@qq.com>
Date: Wed, 27 Sep 2023 17:51:23 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/wb' into wb

---
 screen-api/src/main/java/com/moral/api/service/impl/AllocationExtensionServiceImpl.java |   11 ++++++++---
 1 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/screen-api/src/main/java/com/moral/api/service/impl/AllocationExtensionServiceImpl.java b/screen-api/src/main/java/com/moral/api/service/impl/AllocationExtensionServiceImpl.java
index a2259b6..86b05aa 100644
--- a/screen-api/src/main/java/com/moral/api/service/impl/AllocationExtensionServiceImpl.java
+++ b/screen-api/src/main/java/com/moral/api/service/impl/AllocationExtensionServiceImpl.java
@@ -16,8 +16,11 @@
 import com.moral.api.service.AllocationExtensionService;
 import com.moral.api.service.AllocationService;
 import com.moral.api.service.FileTableService;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
+
+import java.util.List;
 
 /**
  * <p>
@@ -30,11 +33,11 @@
 @Service
 public class AllocationExtensionServiceImpl extends ServiceImpl<AllocationExtensionMapper, AllocationExtension> implements AllocationExtensionService {
 
-    private final AllocationService allocationService;
+    @Autowired
+    private AllocationService allocationService;
     private final FileTableService fileTableService;
 
-    public AllocationExtensionServiceImpl(AllocationService allocationService, FileTableService fileTableService) {
-        this.allocationService = allocationService;
+    public AllocationExtensionServiceImpl( FileTableService fileTableService) {
         this.fileTableService = fileTableService;
     }
 
@@ -63,6 +66,8 @@
 
     @Override
     public Page<AllocationExtensionPageExt> extPage(AllocationExtensionPageCond allocationExtensionPageCond) {
+        List<Integer> unitList = allocationService.unitResult();
+        allocationExtensionPageCond.setUnitList(unitList);
         return this.baseMapper.extPage(allocationExtensionPageCond.getPage().convertPage(), allocationExtensionPageCond);
     }
 

--
Gitblit v1.8.0