浏览代码

token验证

523096025 2 年之前
父节点
当前提交
7e53d172aa
共有 1 个文件被更改,包括 7 次插入2 次删除
  1. 7 2
      huimv-cattle/src/main/java/com/huimv/cattle/service/impl/SysMenuServiceImpl.java

+ 7 - 2
huimv-cattle/src/main/java/com/huimv/cattle/service/impl/SysMenuServiceImpl.java

@@ -1,5 +1,6 @@
 package com.huimv.cattle.service.impl;
 
+import cn.hutool.core.util.ObjectUtil;
 import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
@@ -14,6 +15,8 @@ import com.huimv.cattle.service.SysMenuService;
 import com.huimv.cattle.token.TokenSign;
 import com.huimv.cattle.utils.ConstUtil;
 import com.huimv.cattle.utils.ResultUtil;
+import com.huimv.common.exception.ExceptionEnum;
+import com.huimv.common.exception.MiException;
 import com.huimv.common.utils.Result;
 import com.huimv.common.utils.ResultCode;
 //import com.huimv.saas.admin.mapper.SysMenuMapper;
@@ -168,8 +171,10 @@ public class SysMenuServiceImpl extends ServiceImpl<SysMenuMapper, SysMenu> impl
     @Override
     public List<String>  getMyButton(HttpServletRequest request,Map<String, String> paramMap) {
         Integer userId = TokenSign.getMemberIdByJwtToken(request);
-
-        if (userId == 1){
+        if (ObjectUtil.isEmpty(userId)){
+            throw new MiException(ExceptionEnum.TOKEN_NULL);
+        }
+        if ( 1 == userId){
             List<SysMenu> menus = sysMenuMapper.selectList(null);
             List<String> list = new ArrayList<>();
             for (SysMenu menu : menus) {