mirror of
https://gitee.com/dromara/RuoYi-Cloud-Plus.git
synced 2025-09-04 03:26:31 +00:00
update 优化 移除无用的重写 已无必要
This commit is contained in:
@@ -1,51 +0,0 @@
|
||||
package cn.dev33.satoken.context.dubbo3.filter;
|
||||
|
||||
import cn.dev33.satoken.SaManager;
|
||||
import cn.dev33.satoken.context.SaTokenContextDefaultImpl;
|
||||
import cn.dev33.satoken.same.SaSameUtil;
|
||||
import cn.dev33.satoken.spring.SaBeanInject;
|
||||
import cn.dev33.satoken.stp.StpUtil;
|
||||
import cn.dev33.satoken.util.SaTokenConsts;
|
||||
import org.dromara.common.core.utils.SpringUtils;
|
||||
import org.apache.dubbo.common.constants.CommonConstants;
|
||||
import org.apache.dubbo.common.extension.Activate;
|
||||
import org.apache.dubbo.rpc.*;
|
||||
|
||||
/**
|
||||
*
|
||||
* Sa-Token 整合 Dubbo Consumer端过滤器
|
||||
*
|
||||
* 此过滤器为覆盖 Sa-Token 包内过滤器
|
||||
*
|
||||
* @author kong
|
||||
*
|
||||
*/
|
||||
@Activate(group = {CommonConstants.CONSUMER}, order = Integer.MIN_VALUE)
|
||||
public class SaTokenDubbo3ConsumerFilter implements Filter {
|
||||
|
||||
@Override
|
||||
public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {
|
||||
// 强制初始化 Sa-Token 相关配置 解决内网鉴权元数据加载报错问题
|
||||
SpringUtils.getBean(SaBeanInject.class);
|
||||
|
||||
// 追加 Id-Token 参数
|
||||
if(SaManager.getConfig().getCheckSameToken()) {
|
||||
RpcContext.getServiceContext().setAttachment(SaSameUtil.SAME_TOKEN, SaSameUtil.getToken());
|
||||
}
|
||||
|
||||
// 1. 调用前,向下传递会话Token
|
||||
if(SaManager.getSaTokenContextOrSecond() != SaTokenContextDefaultImpl.defaultContext) {
|
||||
RpcContext.getServiceContext().setAttachment(SaTokenConsts.JUST_CREATED, StpUtil.getTokenValueNotCut());
|
||||
}
|
||||
|
||||
// 2. 开始调用
|
||||
Result invoke = invoker.invoke(invocation);
|
||||
|
||||
// 3. 调用后,解析回传的Token值
|
||||
StpUtil.setTokenValue(invoke.getAttachment(SaTokenConsts.JUST_CREATED_NOT_PREFIX));
|
||||
|
||||
// note
|
||||
return invoke;
|
||||
}
|
||||
|
||||
}
|
@@ -1,42 +0,0 @@
|
||||
package cn.dev33.satoken.context.dubbo3.filter;
|
||||
|
||||
import cn.dev33.satoken.SaManager;
|
||||
import cn.dev33.satoken.same.SaSameUtil;
|
||||
import cn.dev33.satoken.spring.SaBeanInject;
|
||||
import org.dromara.common.core.utils.SpringUtils;
|
||||
import org.apache.dubbo.common.constants.CommonConstants;
|
||||
import org.apache.dubbo.common.extension.Activate;
|
||||
import org.apache.dubbo.rpc.*;
|
||||
|
||||
/**
|
||||
*
|
||||
* Sa-Token 整合 Dubbo Provider端过滤器
|
||||
*
|
||||
* 此过滤器为覆盖 Sa-Token 包内过滤器
|
||||
*
|
||||
* @author kong
|
||||
*
|
||||
*/
|
||||
@Activate(group = {CommonConstants.PROVIDER}, order = Integer.MIN_VALUE)
|
||||
public class SaTokenDubbo3ProviderFilter implements Filter {
|
||||
|
||||
@Override
|
||||
public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {
|
||||
// 强制初始化 Sa-Token 相关配置 解决内网鉴权元数据加载报错问题
|
||||
SpringUtils.getBean(SaBeanInject.class);
|
||||
|
||||
// RPC 调用鉴权
|
||||
if(SaManager.getConfig().getCheckSameToken()) {
|
||||
String idToken = invocation.getAttachment(SaSameUtil.SAME_TOKEN);
|
||||
// dubbo部分协议会将参数变为小写,详细参考:https://gitee.com/dromara/sa-token/issues/I4WXQG
|
||||
if(idToken == null) {
|
||||
idToken = invocation.getAttachment(SaSameUtil.SAME_TOKEN.toLowerCase());
|
||||
}
|
||||
SaSameUtil.checkToken(idToken);
|
||||
}
|
||||
|
||||
// 开始调用
|
||||
return invoker.invoke(invocation);
|
||||
}
|
||||
|
||||
}
|
Reference in New Issue
Block a user