From a9de97ec2f33c540e4dce56af7a11ccd30a65f5a Mon Sep 17 00:00:00 2001
From: Fancy <Fancy.fx@outlook.com>
Date: 星期五, 13 十二月 2024 11:57:49 +0800
Subject: [PATCH] application

---
 pms-parent/pms-web-base/src/main/java/com/dy/pmsBase/user/UserSv.java |   20 ++++++++++++++++----
 1 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/pms-parent/pms-web-base/src/main/java/com/dy/pmsBase/user/UserSv.java b/pms-parent/pms-web-base/src/main/java/com/dy/pmsBase/user/UserSv.java
index 4b034da..630f72e 100644
--- a/pms-parent/pms-web-base/src/main/java/com/dy/pmsBase/user/UserSv.java
+++ b/pms-parent/pms-web-base/src/main/java/com/dy/pmsBase/user/UserSv.java
@@ -6,11 +6,13 @@
 import com.dy.pmsGlobal.daoBa.BaUserRoleMapper;
 import com.dy.pmsGlobal.pojoBa.BaUser;
 import com.dy.pmsGlobal.pojoBa.BaUserRole;
+import com.dy.pmsGlobal.util.QrCodeConstant;
 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;
 
@@ -26,7 +28,6 @@
     private BaUserMapper dao;
     @Autowired
     private BaUserRoleMapper urDao;
-    private static final String stationPrefix = "101" ;
 
     /**
      * 鐢熸垚鎸囧畾浣嶆暟闅忔満鏁板瓧瀵嗙爜
@@ -77,7 +78,7 @@
         rsVo.obj = this.dao.selectSome(params) ;
         rsVo.obj.parallelStream().forEach(item->{
             try {
-                byte[] codes = QrCodeUtil.genQrCode(stationPrefix+item.id);
+                byte[] codes = QrCodeUtil.genQrCode(QrCodeConstant.TypeWorker+item.id);
                 item.qrCode = "data:image/jpeg;base64," + Base64.encode(codes);
             } catch (IOException e) {
                 e.printStackTrace();
@@ -96,7 +97,12 @@
      */
     @Transactional
     public Long save(BaUser po){
-        this.dao.insertSelective(po) ;
+        try{
+            dao.insertSelective(po) ;
+        }catch(DuplicateKeyException e){
+            log.error(e.getMessage());
+            throw new RuntimeException("鎵嬫満鍙风爜閲嶅");
+        }
         Long id = po.getId();
         this.saveUserRoles(id, po.roleIds) ;
         return id ;
@@ -109,7 +115,13 @@
      */
     @Transactional
     public int update(BaUser po) {
-        int count = this.dao.updateByPrimaryKeySelective(po);
+        int count=0;
+        try{
+            count = this.dao.updateByPrimaryKeySelective(po);
+        }catch(DuplicateKeyException e){
+            log.error(e.getMessage());
+            throw new RuntimeException("鎵嬫満鍙风爜閲嶅");
+        }
         this.saveUserRoles(po.id, po.roleIds) ;
         return count ;
     }

--
Gitblit v1.8.0