From 4b17c39b3fcff630de72b8e9638438f900ea213e Mon Sep 17 00:00:00 2001
From: Caps <452713115@qq.com>
Date: Fri, 17 Apr 2020 10:15:44 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B4=A6=E6=88=B7=E6=9D=83=E9=99=90=E7=AE=A1?=
 =?UTF-8?q?=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../impl/PrivilageUserServiceImpl.java        | 48 ++++++++++---------
 1 file changed, 25 insertions(+), 23 deletions(-)

diff --git a/src/main/java/com/kiisoo/ic/system/service/impl/PrivilageUserServiceImpl.java b/src/main/java/com/kiisoo/ic/system/service/impl/PrivilageUserServiceImpl.java
index 370bf64..ee32d2a 100644
--- a/src/main/java/com/kiisoo/ic/system/service/impl/PrivilageUserServiceImpl.java
+++ b/src/main/java/com/kiisoo/ic/system/service/impl/PrivilageUserServiceImpl.java
@@ -3,7 +3,9 @@ package com.kiisoo.ic.system.service.impl;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.kiisoo.ic.constants.Constants;
+import com.kiisoo.ic.store.entity.PoiStore;
 import com.kiisoo.ic.store.entity.PoiStoreStaff;
+import com.kiisoo.ic.store.mapper.PoiStoreDOMapper;
 import com.kiisoo.ic.store.mapper.PoiStoreStaffDOMapper;
 import com.kiisoo.ic.system.entity.PrivilageUserDO;
 import com.kiisoo.ic.system.enums.UserEnum;
@@ -16,6 +18,7 @@ import java.util.Date;
 
 /**
  * 用户表  服务实现类
+ *
  * @author jinchaofan
  * @since 2020-02-19
  */
@@ -32,10 +35,13 @@ public class PrivilageUserServiceImpl extends ServiceImpl<PrivilageUserDOMapper,
      */
     private final PoiStoreStaffDOMapper poiStoreStaffDOMapper;
 
+    private final PoiStoreDOMapper poiStoreDOMapper;
+
     @Autowired
-    public PrivilageUserServiceImpl(PrivilageUserDOMapper privilageUserDOMapper, PoiStoreStaffDOMapper poiStoreStaffDOMapper) {
+    public PrivilageUserServiceImpl(PrivilageUserDOMapper privilageUserDOMapper, PoiStoreStaffDOMapper poiStoreStaffDOMapper,PoiStoreDOMapper poiStoreDOMapper) {
         this.privilageUserDOMapper = privilageUserDOMapper;
         this.poiStoreStaffDOMapper = poiStoreStaffDOMapper;
+        this.poiStoreDOMapper = poiStoreDOMapper;
     }
 
     @Override
@@ -55,39 +61,35 @@ public class PrivilageUserServiceImpl extends ServiceImpl<PrivilageUserDOMapper,
     }
 
     @Override
-    public void saveUserShopRelation(Long userId,String staffCode, Long shopId,Long type) {
+    public void saveUserShopRelation(Long userId, String staffCode, Long shopId, Long type) {
+        PoiStore poiStore = poiStoreDOMapper.selectById(shopId);
         PoiStoreStaff poiStoreStaff = new PoiStoreStaff();
         poiStoreStaff.setUserId(userId);
         poiStoreStaff.setUpdateBy(Constants.SYS_OPERATION);
-        QueryWrapper<PoiStoreStaff> wrapper = new QueryWrapper<>();
-        wrapper.eq("staff_code",staffCode);
-        if(staffCode != null && !staffCode.equals("")) {
-            //如果有就修改如果没有就插入
-            int update = poiStoreStaffDOMapper.update(poiStoreStaff, wrapper);
-            if (update == 0) {
-                poiStoreStaff.setStoreId(shopId);
-                poiStoreStaff.setType(type);
-                poiStoreStaff.setCreateTime(new Date());
-                poiStoreStaff.setUpdateTime(new Date());
-                poiStoreStaff.setStatus(1L);
-                poiStoreStaff.setUpdateBy(Constants.SYS_OPERATION);
-                poiStoreStaff.setCreateBy(Constants.SYS_OPERATION);
-                poiStoreStaff.setDeleted(0);
-                //插入
-                poiStoreStaffDOMapper.insert(poiStoreStaff);
-            }
-        }
+        poiStoreStaff.setStoreId(shopId);
+        poiStoreStaff.setStaffCode("");
+        poiStoreStaff.setStoreCode(poiStore.getCode());
+        poiStoreStaff.setType(type);
+        poiStoreStaff.setCreateTime(new Date());
+        poiStoreStaff.setUpdateTime(new Date());
+        poiStoreStaff.setStatus(1L);
+        poiStoreStaff.setUpdateBy(Constants.SYS_OPERATION);
+        poiStoreStaff.setCreateBy(Constants.SYS_OPERATION);
+        poiStoreStaff.setDeleted(0);
+        //插入
+        poiStoreStaffDOMapper.insert(poiStoreStaff);
+
     }
 
     @Override
     public void removeUserShopRelation(Long userId) {
         //并不是删除了店铺只是删除了店铺中的user_id
         QueryWrapper<PoiStoreStaff> wrapper = new QueryWrapper<>();
-        wrapper.eq("user_id",userId);
-        if(null != userId) {
+        wrapper.eq("user_id", userId);
+        if (null != userId) {
             PoiStoreStaff poiStoreStaff = new PoiStoreStaff();
             poiStoreStaff.setUserId(0L);
-            poiStoreStaffDOMapper.update(poiStoreStaff,wrapper);
+            poiStoreStaffDOMapper.update(poiStoreStaff, wrapper);
         }
     }