From 406cc9a54e82c89a75bdd4c680cdc0c96f45a857 Mon Sep 17 00:00:00 2001
From: Fancy <Fancy.fx@outlook.com>
Date: 星期一, 24 六月 2024 13:14:12 +0800
Subject: [PATCH] update login check API

---
 pms-parent/pms-web-platform/src/main/java/com/dy/pmsPlatform/station/StationSv.java |   64 +++++++++++++++++++++++++++----
 1 files changed, 55 insertions(+), 9 deletions(-)

diff --git a/pms-parent/pms-web-platform/src/main/java/com/dy/pmsPlatform/station/StationSv.java b/pms-parent/pms-web-platform/src/main/java/com/dy/pmsPlatform/station/StationSv.java
index 0389ee9..3890a5a 100644
--- a/pms-parent/pms-web-platform/src/main/java/com/dy/pmsPlatform/station/StationSv.java
+++ b/pms-parent/pms-web-platform/src/main/java/com/dy/pmsPlatform/station/StationSv.java
@@ -1,14 +1,21 @@
 package com.dy.pmsPlatform.station;
 
+import cn.hutool.core.codec.Base64;
+import com.alibaba.excel.util.StringUtils;
 import com.dy.common.webUtil.QueryResultVo;
 import com.dy.pmsGlobal.daoPlt.PltStationMapper;
+import com.dy.pmsGlobal.pojoPlt.PltProduct;
 import com.dy.pmsGlobal.pojoPlt.PltStation;
+import com.dy.pmsGlobal.util.QrCodeUtil;
+import com.google.zxing.WriterException;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.dubbo.common.utils.PojoUtils;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.dao.DuplicateKeyException;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
+import java.io.IOException;
 import java.util.List;
 import java.util.Map;
 
@@ -17,6 +24,9 @@
 public class StationSv {
 
     private PltStationMapper dao;
+    private static final String stationPrefix = "103" ;
+    private static final String DEFAULT_CODE = "0001";
+    private static final String CODE_FORMAT = "%04d";
 
     @Autowired
     public void setDao(PltStationMapper dao) {
@@ -24,12 +34,41 @@
     }
 
     @Transactional
-    public int save(PltStation station) throws Exception {
-        if(dao.selectByCodeId(station.id,station.code)>0){
-            throw new Exception("宸ョ珯缂栧彿宸插瓨鍦�");
-        }
-        int count = dao.insertSelective(station);
+    public int save(PltStation station) throws RuntimeException {
+
+//        if(dao.selectByCodeId(station.id,station.code)>0){
+//            throw new RuntimeException("宸ョ珯缂栧彿宸插瓨鍦�");
+//        }
+        int count = 0;
+        int flag=0;
+        do {
+            station.code = getNextCode();
+            try {
+                count = dao.insertSelective(station);
+                return count;
+            } catch (DuplicateKeyException e) {
+                // 濡傛灉鍑虹幇鍞竴绾︽潫杩濆弽寮傚父锛屽皾璇曡幏鍙栨柊鐨刢ode骞堕噸璇�
+                log.warn("鎻掑叆宸ョ珯鏃堕亣鍒板敮涓�绾︽潫寮傚父锛屽皾璇曡幏鍙栨柊鐨刢ode", e);
+            }
+            flag++;
+        } while (flag<5);
+
         return count;
+    }
+
+    private String getNextCode(){
+        log.info("鑾峰彇宸ョ珯缂栧彿");
+        String maxCode = dao.selectMaxCode();
+        if (StringUtils.isBlank(maxCode)) {
+            return DEFAULT_CODE;
+        } else {
+            int nextCode = Integer.parseInt(maxCode) + 1;
+            // 妫�鏌ユ孩鍑�
+            if (nextCode > 9999) {
+                throw new RuntimeException("宸ョ珯缂栧彿婧㈠嚭");
+            }
+            return String.format(CODE_FORMAT, nextCode);
+        }
     }
 
     /**
@@ -45,10 +84,10 @@
 
 
     @Transactional
-    public int update(PltStation station) throws Exception {
-        if(dao.selectByCodeId(station.id,station.code)>0){
-            throw new Exception("宸ョ珯缂栧彿宸插瓨鍦�");
-        }
+    public int update(PltStation station) throws RuntimeException {
+//        if(dao.selectByCodeId(station.id,station.code)>0){
+//            throw new RuntimeException("宸ョ珯缂栧彿宸插瓨鍦�");
+//        }
         int count = dao.updateByPrimaryKeySelective(station);
         return count;
     }
@@ -73,6 +112,9 @@
 
         //鏌ヨ绗﹀悎鏉′欢鐨勮褰�
         rsVo.obj = this.dao.selectSome(params) ;
+        rsVo.obj.parallelStream().forEach(item->{
+            item.qrCode = QrCodeUtil.genQrCodeString(stationPrefix+item.id);
+        });
         return rsVo ;
     }
 
@@ -83,5 +125,9 @@
         station.setDisabled(disabled);
         return dao.updateByPrimaryKeySelective(station);
     }
+
+    public List<PltStation> selectAll() {
+        return dao.selectAll();
+    }
 }
 

--
Gitblit v1.8.0