wangys/urbanLifeline
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

解决dubbo上下文问题

Browse Source
This commit is contained in:
wangys
2025-12-20 17:06:45 +08:00
parent 81508a1fdc
commit dfd9bb8b95
3 changed files with 80 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
urbanLifelineServ/common/common-auth/src/main/java/org/xyzh/common/auth/filter/DubboConsumerContextFilter.java Normal file
View File

@@ -0,0 +1,36 @@
package org.xyzh.common.auth.filter;
import org.apache.dubbo.common.constants.CommonConstants;
import org.apache.dubbo.common.extension.Activate;
import org.apache.dubbo.rpc.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.util.StringUtils;
import org.xyzh.common.auth.utils.LoginUtil;
/**
* Dubbo消费者端Filter：自动传递登录Token到RpcContext
*
* @author yslg
* @since 2025-12-20
*/
@Activate(group = CommonConstants.CONSUMER, order = -9000)
public class DubboConsumerContextFilter implements Filter {
private static final Logger logger = LoggerFactory.getLogger(DubboConsumerContextFilter.class);
private static final String AUTH_TOKEN_KEY = "auth-token";
@Override
public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {
// 从当前HTTP请求获取Token
String token = LoginUtil.getToken();
if (StringUtils.hasText(token)) {
// 将Token放入RpcContext，传递给提供者
RpcContext.getClientAttachment().setAttachment(AUTH_TOKEN_KEY, token);
logger.debug("Dubbo消费者传递Token: {}", token.substring(0, Math.min(20, token.length())) + "...");
}
return invoker.invoke(invocation);
}
}

42
urbanLifelineServ/common/common-auth/src/main/java/org/xyzh/common/auth/filter/DubboProviderContextFilter.java Normal file
View File

@@ -0,0 +1,42 @@
package org.xyzh.common.auth.filter;
import org.apache.dubbo.common.constants.CommonConstants;
import org.apache.dubbo.common.extension.Activate;
import org.apache.dubbo.rpc.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.util.StringUtils;
/**
* Dubbo提供者端Filter：从RpcContext接收Token并存储到ThreadLocal
*
* @author yslg
* @since 2025-12-20
*/
@Activate(group = CommonConstants.PROVIDER, order = -9000)
public class DubboProviderContextFilter implements Filter {
private static final Logger logger = LoggerFactory.getLogger(DubboProviderContextFilter.class);
private static final String AUTH_TOKEN_KEY = "auth-token";
public static final ThreadLocal<String> TOKEN_HOLDER = new ThreadLocal<>();
@Override
public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {
try {
// 从RpcContext获取消费者传递的Token
String token = RpcContext.getServerAttachment().getAttachment(AUTH_TOKEN_KEY);
if (StringUtils.hasText(token)) {
// 存储到ThreadLocal供LoginUtil使用
TOKEN_HOLDER.set(token);
logger.debug("Dubbo提供者接收Token: {}", token.substring(0, Math.min(20, token.length())) + "...");
}
return invoker.invoke(invocation);
} finally {
// 清理ThreadLocal避免内存泄漏
TOKEN_HOLDER.remove();
}
}
}

2
urbanLifelineServ/common/common-auth/src/main/resources/META-INF/dubbo/org.apache.dubbo.rpc.Filter Normal file
View File

@@ -0,0 +1,2 @@
dubboConsumerContextFilter=org.xyzh.common.auth.filter.DubboConsumerContextFilter
dubboProviderContextFilter=org.xyzh.common.auth.filter.DubboProviderContextFilter