From e34e91568095f88b3731046d18d776961556e089 Mon Sep 17 00:00:00 2001
From: java110 <928255095@qq.com>
Date: 星期五, 11 十二月 2020 09:17:32 +0800
Subject: [PATCH] 优化员工认证功能

---
 service-user/src/main/java/com/java110/user/bmo/staffAppAuth/impl/UpdateStaffAppAuthBMOImpl.java |   15 ++++++++++++++-
 1 files changed, 14 insertions(+), 1 deletions(-)

diff --git a/service-user/src/main/java/com/java110/user/bmo/staffAppAuth/impl/UpdateStaffAppAuthBMOImpl.java b/service-user/src/main/java/com/java110/user/bmo/staffAppAuth/impl/UpdateStaffAppAuthBMOImpl.java
index 26b72a9..047da78 100644
--- a/service-user/src/main/java/com/java110/user/bmo/staffAppAuth/impl/UpdateStaffAppAuthBMOImpl.java
+++ b/service-user/src/main/java/com/java110/user/bmo/staffAppAuth/impl/UpdateStaffAppAuthBMOImpl.java
@@ -1,6 +1,8 @@
 package com.java110.user.bmo.staffAppAuth.impl;
 
 import com.java110.core.annotation.Java110Transactional;
+import com.java110.core.factory.GenerateCodeFactory;
+import com.java110.dto.staffAppAuth.StaffAppAuthDto;
 import com.java110.intf.user.IStaffAppAuthInnerServiceSMO;
 import com.java110.po.staffAppAuth.StaffAppAuthPo;
 import com.java110.user.bmo.staffAppAuth.IUpdateStaffAppAuthBMO;
@@ -22,7 +24,18 @@
     @Java110Transactional
     public ResponseEntity<String> update(StaffAppAuthPo staffAppAuthPo) {
 
-        int flag = staffAppAuthInnerServiceSMOImpl.updateStaffAppAuth(staffAppAuthPo);
+        StaffAppAuthDto staffAppAuthDto = new StaffAppAuthDto();
+        staffAppAuthDto.setAppType(staffAppAuthPo.getAppType());
+        staffAppAuthDto.setStaffId(staffAppAuthPo.getStaffId());
+        staffAppAuthDto.setStoreId(staffAppAuthPo.getStoreId());
+        int count = staffAppAuthInnerServiceSMOImpl.queryStaffAppAuthsCount(staffAppAuthDto);
+        int flag = 0;
+        if (count > 0) {
+            flag = staffAppAuthInnerServiceSMOImpl.updateStaffAppAuth(staffAppAuthPo);
+        } else {
+            staffAppAuthPo.setAuId(GenerateCodeFactory.getGeneratorId(GenerateCodeFactory.CODE_PREFIX_auId));
+            flag = staffAppAuthInnerServiceSMOImpl.saveStaffAppAuth(staffAppAuthPo);
+        }
 
         if (flag > 0) {
             return ResultVo.createResponseEntity(ResultVo.CODE_OK, "淇濆瓨鎴愬姛");

--
Gitblit v1.8.0