|
@@ -9,13 +9,16 @@ import com.usky.common.core.constants.CommonConst;
|
|
|
import com.usky.common.core.exception.BusinessErrorCode;
|
|
|
import com.usky.common.core.util.StringUtils;
|
|
|
import com.usky.system.domain.SysMenu;
|
|
|
+import com.usky.system.domain.SysUser;
|
|
|
import com.usky.system.domain.constants.UserConstants;
|
|
|
import com.usky.system.model.LoginUser;
|
|
|
import com.usky.system.service.ISysMenuService;
|
|
|
+import com.usky.system.service.ISysUserService;
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
|
import org.springframework.validation.annotation.Validated;
|
|
|
import org.springframework.web.bind.annotation.*;
|
|
|
|
|
|
+import java.util.ArrayList;
|
|
|
import java.util.HashMap;
|
|
|
import java.util.List;
|
|
|
import java.util.Map;
|
|
@@ -33,6 +36,9 @@ public class SysMenuController extends BaseController
|
|
|
@Autowired
|
|
|
private ISysMenuService menuService;
|
|
|
|
|
|
+ @Autowired
|
|
|
+ private ISysUserService userService;
|
|
|
+
|
|
|
@Autowired
|
|
|
private TokenService tokenService;
|
|
|
|
|
@@ -151,7 +157,14 @@ public class SysMenuController extends BaseController
|
|
|
public ApiResult getRouters()
|
|
|
{
|
|
|
Long userId = SecurityUtils.getUserId();
|
|
|
- List<SysMenu> menus = menuService.selectMenuTreeByUserId(userId);
|
|
|
+
|
|
|
+ SysUser sysUser = userService.selectUserById(userId);
|
|
|
+ List<SysMenu> menus = new ArrayList<>();
|
|
|
+ if ("01".equals(sysUser.getUserType())){
|
|
|
+ menus = menuService.selectMenuTreeByUserIdOne(SecurityUtils.getTenantId());
|
|
|
+ }else {
|
|
|
+ menus = menuService.selectMenuTreeByUserId(userId);
|
|
|
+ }
|
|
|
return ApiResult.success(menuService.buildMenus(menus));
|
|
|
}
|
|
|
}
|