From 82de5a307466894bbb0258f8a63a26a7bb96d80d Mon Sep 17 00:00:00 2001
From: zhang <zc857179121@qq.com>
Date: 星期四, 16 十月 2025 09:55:35 +0800
Subject: [PATCH] 13

---
 src/main/java/com/zy/asrs/controller/StaDescController.java |  115 ++++++++++++++++++++++++++++++++++++++++++++++-----------
 1 files changed, 93 insertions(+), 22 deletions(-)

diff --git a/src/main/java/com/zy/asrs/controller/StaDescController.java b/src/main/java/com/zy/asrs/controller/StaDescController.java
index 5c32944..81bb038 100644
--- a/src/main/java/com/zy/asrs/controller/StaDescController.java
+++ b/src/main/java/com/zy/asrs/controller/StaDescController.java
@@ -5,14 +5,21 @@
 import com.baomidou.mybatisplus.mapper.EntityWrapper;
 import com.baomidou.mybatisplus.mapper.Wrapper;
 import com.baomidou.mybatisplus.plugins.Page;
-import com.zy.asrs.entity.StaDesc;
-import com.zy.asrs.service.StaDescService;
-import com.zy.common.web.BaseController;
 import com.core.annotations.ManagerAuth;
 import com.core.common.BaseRes;
 import com.core.common.Cools;
 import com.core.common.DateUtils;
 import com.core.common.R;
+import com.zy.asrs.entity.StaDesc;
+import com.zy.asrs.entity.param.StaDescInitParam;
+import com.zy.asrs.mapper.StaDescMapper;
+import com.zy.asrs.service.StaDescService;
+import com.zy.common.web.BaseController;
+import org.apache.ibatis.session.ExecutorType;
+import org.apache.ibatis.session.SqlSession;
+import org.apache.ibatis.session.SqlSessionFactory;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 
@@ -21,8 +28,62 @@
 @RestController
 public class StaDescController extends BaseController {
 
+    private static final Logger log = LoggerFactory.getLogger(StaDescController.class);
     @Autowired
     private StaDescService staDescService;
+    @Autowired
+    private SqlSessionFactory sqlSessionFactory;
+
+    @RequestMapping(value = "/staDesc/init/auth")
+    @ManagerAuth(memo = "鍒濆鍖栫珯鐐硅矾寰�")
+    public R init(StaDescInitParam param) {
+        SqlSession sqlSession = null;
+        try {
+            if (param.getTypeDesc() == 1) {
+                staDescService.delete(new EntityWrapper<>());
+            }
+            String[] startStaList = param.getStartStaList().split(";");
+            String[] endStaList = param.getEndStaList().split(";");
+            List<StaDesc> staDescList = new ArrayList<>();
+            sqlSession = sqlSessionFactory.openSession(ExecutorType.BATCH, false);
+            StaDescMapper sqlSessionMapper = sqlSession.getMapper(StaDescMapper.class);
+            for (String startSta : startStaList) {
+                for (String endSta : endStaList) {
+                    for (Integer type : param.getType()) {
+                        int sameRes = staDescService.selectCount(new EntityWrapper<StaDesc>()
+                                .eq("type_no", type)
+                                .eq("stn_no", Integer.parseInt(startSta))
+                                .eq("crn_no", param.getCrnNo())
+                                .eq("crn_stn", Integer.parseInt(endSta)));
+                        if (sameRes > 0) {
+                            continue;
+                        }
+                        StaDesc staDesc = new StaDesc();
+                        staDesc.setCrnNo(param.getCrnNo());
+                        staDesc.setTypeNo(type);
+                        staDesc.setStnNo(Integer.parseInt(startSta));
+                        staDesc.setCrnStn(Integer.parseInt(endSta));
+                        staDesc.setModiUser(getUserId());
+                        staDesc.setModiTime(new Date());
+                        staDesc.setAppeUser(getUserId());
+                        staDesc.setAppeTime(new Date());
+                        sqlSessionMapper.insert(staDesc);
+//                        staDescList.add(staDesc);
+                    }
+                }
+            }
+            sqlSession.flushStatements();
+            sqlSession.clearCache();
+
+//            staDescService.insertBatch(staDescList);
+        } catch (Exception e) {
+            return R.error(e.getMessage());
+        } finally {
+            sqlSession.close();
+        }
+
+        return R.ok("ok1").add("ok2");
+    }
 
     @RequestMapping(value = "/staDesc/{id}/auth")
     @ManagerAuth
@@ -32,22 +93,24 @@
 
     @RequestMapping(value = "/staDesc/list/auth")
     @ManagerAuth
-    public R list(@RequestParam(defaultValue = "1")Integer curr,
-                  @RequestParam(defaultValue = "10")Integer limit,
-                  @RequestParam(required = false)String orderByField,
-                  @RequestParam(required = false)String orderByType,
-                  @RequestParam Map<String, Object> param){
+    public R list(@RequestParam(defaultValue = "1") Integer curr,
+                  @RequestParam(defaultValue = "10") Integer limit,
+                  @RequestParam(required = false) String orderByField,
+                  @RequestParam(required = false) String orderByType,
+                  @RequestParam Map<String, Object> param) {
         excludeTrash(param);
         EntityWrapper<StaDesc> wrapper = new EntityWrapper<>();
         convert(param, wrapper);
-        if (!Cools.isEmpty(orderByField)){wrapper.orderBy(humpToLine(orderByField), "asc".equals(orderByType));}
+        if (!Cools.isEmpty(orderByField)) {
+            wrapper.orderBy(humpToLine(orderByField), "asc".equals(orderByType));
+        }
         return R.ok(staDescService.selectPage(new Page<>(curr, limit), wrapper));
     }
 
-    private <T> void convert(Map<String, Object> map, EntityWrapper<T> wrapper){
-        for (Map.Entry<String, Object> entry : map.entrySet()){
+    private <T> void convert(Map<String, Object> map, EntityWrapper<T> wrapper) {
+        for (Map.Entry<String, Object> entry : map.entrySet()) {
             String val = String.valueOf(entry.getValue());
-            if (val.contains(RANGE_TIME_LINK)){
+            if (val.contains(RANGE_TIME_LINK)) {
                 String[] dates = val.split(RANGE_TIME_LINK);
                 wrapper.ge(entry.getKey(), DateUtils.convert(dates[0]));
                 wrapper.le(entry.getKey(), DateUtils.convert(dates[1]));
@@ -60,6 +123,14 @@
     @RequestMapping(value = "/staDesc/add/auth")
     @ManagerAuth(memo = "绔欑偣璺緞娣诲姞")
     public R add(StaDesc staDesc) {
+        int sameRes = staDescService.selectCount(new EntityWrapper<StaDesc>()
+                .eq("type_no", staDesc.getTypeNo())
+                .eq("stn_no", staDesc.getStnNo())
+                .eq("crn_no", staDesc.getCrnNo())
+                .eq("crn_stn", staDesc.getCrnStn()));
+        if (sameRes > 0) {
+            return R.error("璺緞宸插瓨鍦�");
+        }
         staDesc.setModiUser(getUserId());
         staDesc.setModiTime(new Date());
         staDesc.setAppeUser(getUserId());
@@ -68,10 +139,10 @@
         return R.ok();
     }
 
-	@RequestMapping(value = "/staDesc/update/auth")
-	@ManagerAuth(memo = "绔欑偣璺緞淇敼")
-    public R update(StaDesc staDesc){
-        if (Cools.isEmpty(staDesc) || null==staDesc.getTypeId()){
+    @RequestMapping(value = "/staDesc/update/auth")
+    @ManagerAuth(memo = "绔欑偣璺緞淇敼")
+    public R update(StaDesc staDesc) {
+        if (Cools.isEmpty(staDesc) || null == staDesc.getTypeId()) {
             return R.error();
         }
         staDesc.setModiUser(getUserId());
@@ -82,12 +153,12 @@
 
     @RequestMapping(value = "/staDesc/delete/auth")
     @ManagerAuth(memo = "绔欑偣璺緞鍒犻櫎")
-    public R delete(@RequestParam String param){
+    public R delete(@RequestParam String param) {
         List<StaDesc> list = JSONArray.parseArray(param, StaDesc.class);
-        if (Cools.isEmpty(list)){
+        if (Cools.isEmpty(list)) {
             return R.error();
         }
-        for (StaDesc entity : list){
+        for (StaDesc entity : list) {
             staDescService.delete(new EntityWrapper<>(entity));
         }
         return R.ok();
@@ -95,7 +166,7 @@
 
     @RequestMapping(value = "/staDesc/export/auth")
     @ManagerAuth(memo = "绔欑偣璺緞瀵煎嚭")
-    public R export(@RequestBody JSONObject param){
+    public R export(@RequestBody JSONObject param) {
         List<String> fields = JSONObject.parseArray(param.getJSONArray("fields").toJSONString(), String.class);
         EntityWrapper<StaDesc> wrapper = new EntityWrapper<>();
         Map<String, Object> map = excludeTrash(param.getJSONObject("staDesc"));
@@ -111,7 +182,7 @@
         wrapper.like("crn_no", condition);
         Page<StaDesc> page = staDescService.selectPage(new Page<>(0, 10), wrapper);
         List<Map<String, Object>> result = new ArrayList<>();
-        for (StaDesc staDesc : page.getRecords()){
+        for (StaDesc staDesc : page.getRecords()) {
             Map<String, Object> map = new HashMap<>();
             map.put("id", staDesc.getCrnNo());
             map.put("value", staDesc.getCrnNo());
@@ -124,7 +195,7 @@
     @ManagerAuth
     public R query(@RequestBody JSONObject param) {
         Wrapper<StaDesc> wrapper = new EntityWrapper<StaDesc>().eq(humpToLine(String.valueOf(param.get("key"))), param.get("val"));
-        if (null != staDescService.selectOne(wrapper)){
+        if (null != staDescService.selectOne(wrapper)) {
             return R.parse(BaseRes.REPEAT).add(getComment(StaDesc.class, String.valueOf(param.get("key"))));
         }
         return R.ok();

--
Gitblit v1.9.1