登录

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;
 }

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);
             }

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);
 }

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,