From 5c7bbb84d67016e7e798135584b9a2fd5524492b Mon Sep 17 00:00:00 2001
From: whyclxw <810412026@qq.com>
Date: 星期一, 19 五月 2025 08:52:55 +0800
Subject: [PATCH] 添加电源要加入包机组

---
 src/main/java/com/whyc/service/PowerInfService.java |   37 ++++++++++++++++++++++++++++++++++++-
 1 files changed, 36 insertions(+), 1 deletions(-)

diff --git a/src/main/java/com/whyc/service/PowerInfService.java b/src/main/java/com/whyc/service/PowerInfService.java
index c56aac7..321d3ff 100644
--- a/src/main/java/com/whyc/service/PowerInfService.java
+++ b/src/main/java/com/whyc/service/PowerInfService.java
@@ -6,10 +6,14 @@
 import com.github.pagehelper.PageInfo;
 import com.whyc.dto.PowerDto;
 import com.whyc.dto.Response;
+import com.whyc.mapper.BaojigroupPowerMapper;
+import com.whyc.mapper.BaojigroupUsrMapper;
 import com.whyc.mapper.PowerInfMapper;
 import com.whyc.mapper.StationInfMapper;
 import com.whyc.pojo.db_station.PowerInf;
 import com.whyc.pojo.db_station.StationInf;
+import com.whyc.pojo.db_user.BaojigroupPower;
+import com.whyc.pojo.db_user.BaojigroupUsr;
 import com.whyc.pojo.db_user.User;
 import com.whyc.util.ActionUtil;
 import org.apache.commons.math3.analysis.function.Power;
@@ -17,6 +21,7 @@
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
+import java.util.ArrayList;
 import java.util.List;
 
 @Service
@@ -27,6 +32,11 @@
     @Autowired(required = false)
     private StationInfMapper sinfMapper;
 
+    @Autowired(required = false)
+    private BaojigroupUsrMapper bjUserMapper;
+
+    @Autowired(required = false)
+    private BaojigroupPowerMapper bjPowerMapper;
 
     /*娣诲姞鐢垫簮
     1.鍒ゆ柇褰撳墠鐢ㄦ埛鏄惁鏄寘缁勭敤鎴�
@@ -79,7 +89,32 @@
             powerId += 1;
         }
         addpinf.setPowerId(powerId);
-        mapper.insert(addpinf);
+        int flag=mapper.insert(addpinf);
+        if(flag>0){
+            //褰撳墠浜烘墍鍦ㄧ殑鍖呮満缁勬坊鍔犵數婧愭満鎴�
+            insertInbaoji(stationId,powerId);
+        }
+
+    }
+    //褰撳墠浜烘墍鍦ㄧ殑鍖呮満缁勬坊鍔犵數婧愭満鎴�
+    public void insertInbaoji(Integer stationId,Integer powerId){
+        //閲嶅惎绾跨▼骞跺皢閿佸姞鍒板綋鍓嶄汉鐨勫寘鏈虹粍
+        User uinf= ActionUtil.getUser();
+        QueryWrapper wrapper1=new QueryWrapper();
+        wrapper1.eq("uid",uinf.getId());
+        List<BaojigroupUsr> groupIdList=bjUserMapper.selectList(wrapper1);
+        List<BaojigroupPower> baojigroupList=new ArrayList();
+        if(groupIdList!=null&&groupIdList.size()>0){
+            for (BaojigroupUsr bjUser:groupIdList) {
+                BaojigroupPower bjPower=new BaojigroupPower();
+                bjPower.setStationId(stationId);
+                bjPower.setPowerId(powerId);
+                bjPower.setBaojiGroupId(bjUser.getBaojiGroupId());
+                baojigroupList.add(bjPower);
+            }
+        }
+        bjPowerMapper.insertBatchSomeColumn(baojigroupList);
+        //processSurveyService.setUpThreadRestart();
     }
     //鑾峰彇鐢垫簮鍝佺墝(涓嬫媺)
     public Response getCompanyByUid(Integer uid) {

--
Gitblit v1.9.1