提交 1e32aaa7 authored 作者: YunaiV's avatar YunaiV

优化请求头的 tenant-id 的解析逻辑,避免前端传错类型

上级 9832947d
package cn.iocoder.yudao.framework.web.core.util; package cn.iocoder.yudao.framework.web.core.util;
import cn.hutool.core.util.NumberUtil;
import cn.hutool.core.util.StrUtil; import cn.hutool.core.util.StrUtil;
import cn.iocoder.yudao.framework.common.enums.RpcConstants; import cn.iocoder.yudao.framework.common.enums.RpcConstants;
import cn.iocoder.yudao.framework.common.enums.UserTypeEnum; import cn.iocoder.yudao.framework.common.enums.UserTypeEnum;
...@@ -41,7 +42,7 @@ public class WebFrameworkUtils { ...@@ -41,7 +42,7 @@ public class WebFrameworkUtils {
*/ */
public static Long getTenantId(HttpServletRequest request) { public static Long getTenantId(HttpServletRequest request) {
String tenantId = request.getHeader(HEADER_TENANT_ID); String tenantId = request.getHeader(HEADER_TENANT_ID);
return StrUtil.isNotEmpty(tenantId) ? Long.valueOf(tenantId) : null; return NumberUtil.isNumber(tenantId) ? Long.valueOf(tenantId) : null;
} }
public static void setLoginUserId(ServletRequest request, Long userId) { public static void setLoginUserId(ServletRequest request, Long userId) {
......
...@@ -2,6 +2,7 @@ package cn.iocoder.yudao.gateway.util; ...@@ -2,6 +2,7 @@ package cn.iocoder.yudao.gateway.util;
import cn.hutool.core.net.NetUtil; import cn.hutool.core.net.NetUtil;
import cn.hutool.core.util.ArrayUtil; import cn.hutool.core.util.ArrayUtil;
import cn.hutool.core.util.NumberUtil;
import cn.hutool.extra.servlet.ServletUtil; import cn.hutool.extra.servlet.ServletUtil;
import cn.iocoder.yudao.framework.common.util.json.JsonUtils; import cn.iocoder.yudao.framework.common.util.json.JsonUtils;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
...@@ -44,7 +45,7 @@ public class WebFrameworkUtils { ...@@ -44,7 +45,7 @@ public class WebFrameworkUtils {
public static Long getTenantId(ServerWebExchange exchange) { public static Long getTenantId(ServerWebExchange exchange) {
String tenantId = exchange.getRequest().getHeaders().getFirst(HEADER_TENANT_ID); String tenantId = exchange.getRequest().getHeaders().getFirst(HEADER_TENANT_ID);
return tenantId != null ? Long.parseLong(tenantId) : null; return NumberUtil.isNumber(tenantId) ? Long.valueOf(tenantId) : null;
} }
/** /**
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论