From a072af5084816dd411f206477638b47c3dcd2ba9 Mon Sep 17 00:00:00 2001
From: LegnaYet <1023868505@qq.com>
Date: Sun, 26 Apr 2020 18:47:48 +0800
Subject: [PATCH] =?UTF-8?q?=E9=97=A8=E5=BA=97=E5=8F=B7=E7=AE=A1=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../kiisoo/ic/store/service/PoiStoreStaffService.java | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/kiisoo/ic/store/service/PoiStoreStaffService.java b/src/main/java/com/kiisoo/ic/store/service/PoiStoreStaffService.java
index e818996..b214434 100644
--- a/src/main/java/com/kiisoo/ic/store/service/PoiStoreStaffService.java
+++ b/src/main/java/com/kiisoo/ic/store/service/PoiStoreStaffService.java
@@ -45,6 +45,7 @@ import java.util.*;
 import java.util.concurrent.atomic.AtomicInteger;
 import java.util.stream.Collectors;
 
+import static com.kiisoo.ic.constants.Constants.SELLER_TYPE;
 import static com.kiisoo.ic.store.constant.Constants.*;
 
 /**
@@ -107,7 +108,7 @@ public class PoiStoreStaffService {
         records.forEach(poiStoreStaff -> {
             PoiStoreStaffVO poiStoreStaffVO = new PoiStoreStaffVO();
             BeanUtils.copyProperties(poiStoreStaff, poiStoreStaffVO);
-            poiStoreStaffVO.setRole(Constants.SELLER_TYPE.get(poiStoreStaff.getType()));
+            poiStoreStaffVO.setRole(SELLER_TYPE.get(poiStoreStaff.getType()));
             //拿到用户信息
             PrivilageUserBean privilageUserBean = privilageUserDOMapper.selectUserInfoAndRoleInfo(poiStoreStaff.getUserId());
             if(null != privilageUserBean){
@@ -177,7 +178,7 @@ public class PoiStoreStaffService {
             poiStoreStaffs.forEach(poiStoreStaff -> {
                 PoiStoreStaffVO poiStoreStaffVO = new PoiStoreStaffVO();
                 BeanUtils.copyProperties(poiStoreStaff, poiStoreStaffVO);
-                poiStoreStaffVO.setRole(Constants.SELLER_TYPE.get(poiStoreStaff.getType()));
+                poiStoreStaffVO.setRole(SELLER_TYPE.get(poiStoreStaff.getType()));
                 //拿到用户信息
                 PrivilageUserBean privilageUserBean = privilageUserDOMapper.selectUserInfoAndRoleInfo(poiStoreStaff.getUserId());
                 if(null != privilageUserBean){
@@ -251,7 +252,11 @@ public class PoiStoreStaffService {
         PoiStoreStaff poiStoreStaff = new PoiStoreStaff();
         BeanUtils.copyProperties(poiStoreStaffVO,poiStoreStaff);
         poiStoreStaff.setUserId(userId);
-        poiStoreStaff.setType(poiStoreStaffVO.getRoleId());
+        if (privilageRoleDO.getCode().equals("DZ-DIANZHANG")){
+            poiStoreStaff.setType(1L);
+        }else if(privilageRoleDO.getCode().equals("DG-DAOGOU")){
+            poiStoreStaff.setType(4L);
+        }
         poiStoreStaff.setEpWechatQrCode(qrCodeDO.getQrCode());
        return poiStoreStaffDOMapper.insert(poiStoreStaff);
     }