From 6e734cc461718745c7328cd085dc12401f194a60 Mon Sep 17 00:00:00 2001
From: jiangyunjie <woainiduoduo999@163.com>
Date: Fri, 24 Apr 2020 17:21:40 +0800
Subject: [PATCH] =?UTF-8?q?=E7=99=BB=E5=BD=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/main/java/com/kiisoo/ic/login/bean/LoginBean.java        | 5 ++++-
 .../com/kiisoo/ic/login/service/impl/LoginServiceImpl.java   | 3 ++-
 .../com/kiisoo/ic/system/mapper/PrivilageUserDOMapper.java   | 2 ++
 src/main/resources/mapper/PrivilageUserDOMapper.xml          | 3 +++
 4 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/kiisoo/ic/login/bean/LoginBean.java b/src/main/java/com/kiisoo/ic/login/bean/LoginBean.java
index cffb1ba..7fa2fab 100644
--- a/src/main/java/com/kiisoo/ic/login/bean/LoginBean.java
+++ b/src/main/java/com/kiisoo/ic/login/bean/LoginBean.java
@@ -63,5 +63,8 @@ public class LoginBean {
      */
     private Map<String, Object> loginInfo;
 
-
+    /**
+     * 店铺id
+     */
+    private Long storeId;
 }
diff --git a/src/main/java/com/kiisoo/ic/login/service/impl/LoginServiceImpl.java b/src/main/java/com/kiisoo/ic/login/service/impl/LoginServiceImpl.java
index be7e743..bbd3898 100644
--- a/src/main/java/com/kiisoo/ic/login/service/impl/LoginServiceImpl.java
+++ b/src/main/java/com/kiisoo/ic/login/service/impl/LoginServiceImpl.java
@@ -140,12 +140,13 @@ public class LoginServiceImpl implements LoginService {
             Map<String, Object> loginInfoMap = new HashMap<>();
             PrivilageUserDO privilageUserDO = privilageUserDOMapper.selectById(privilageAccountDOS.get(0).getUserId());
             PrivilageRoleDO roleDO = privilageRoleDOMapper.selectRoleName(privilageAccountDOS.get(0).getUserId());
-
+            Long storeId = privilageUserDOMapper.selectStoreIdByUserId(privilageAccountDOS.get(0).getUserId());
             //成功设置信息
             loginBean.setUserName(privilageUserDO.getName());
             loginBean.setUserId(privilageUserDO.getId());
             loginBean.setRoleId(privilageUserDO.getId());
             loginBean.setRoleCode(roleDO.getCode());
+            loginBean.setStoreId(storeId);
             if (StringUtils.isNotBlank(roleDO.getCode()) && !MINI_ROLE_CODE.contains(roleDO.getCode())){
                 return new LoginBean(Constants.ERROR_ROLE_NO_PERMISSION);
             }
diff --git a/src/main/java/com/kiisoo/ic/system/mapper/PrivilageUserDOMapper.java b/src/main/java/com/kiisoo/ic/system/mapper/PrivilageUserDOMapper.java
index b013940..afb277d 100644
--- a/src/main/java/com/kiisoo/ic/system/mapper/PrivilageUserDOMapper.java
+++ b/src/main/java/com/kiisoo/ic/system/mapper/PrivilageUserDOMapper.java
@@ -16,5 +16,7 @@ public interface PrivilageUserDOMapper extends BaseMapper<PrivilageUserDO> {
 
     PrivilageUserBean selectUserInfoAndRoleInfo(long userId);
 
+    Long selectStoreIdByUserId(Long userId);
+
     PrivilageRoleDO selecctRoleByUserId(long userId);
 }
diff --git a/src/main/resources/mapper/PrivilageUserDOMapper.xml b/src/main/resources/mapper/PrivilageUserDOMapper.xml
index 97e6ea1..cd39953 100644
--- a/src/main/resources/mapper/PrivilageUserDOMapper.xml
+++ b/src/main/resources/mapper/PrivilageUserDOMapper.xml
@@ -10,6 +10,9 @@
         and t1.id = #{userId}
         limit 1
     </select>
+    <select id="selectStoreIdByUserId" resultType="java.lang.Long">
+        SELECT store_id from poi_store_staff where user_id = #{userId}
+    </select>
     <select id="selecctRoleByUserId" resultType="com.kiisoo.ic.system.entity.PrivilageRoleDO">
         select t2.*
         from privilage_user_role t1,