账户权限管理

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
@@ -56,16 +62,13 @@ public class PrivilageUserServiceImpl extends ServiceImpl<PrivilageUserDOMapper,
 
     @Override
     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.setStaffCode("");
+        poiStoreStaff.setStoreCode(poiStore.getCode());
         poiStoreStaff.setType(type);
         poiStoreStaff.setCreateTime(new Date());
         poiStoreStaff.setUpdateTime(new Date());
@@ -75,8 +78,7 @@ public class PrivilageUserServiceImpl extends ServiceImpl<PrivilageUserDOMapper,
         poiStoreStaff.setDeleted(0);
         //插入
         poiStoreStaffDOMapper.insert(poiStoreStaff);
-            }
+
-        }
     }
 
     @Override