[type:fix] set timeout which from rule handle to dubbo rpc context (#…

…5778)
apache · Nov 22, 2024 · 4a4a1e5 · 4a4a1e5
1 parent cc76d4a
commit 4a4a1e5
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/...n-apache-dubbo/src/main/java/org/apache/shenyu/plugin/apache/dubbo/ApacheDubboPlugin.java b/...n-apache-dubbo/src/main/java/org/apache/shenyu/plugin/apache/dubbo/ApacheDubboPlugin.java
@@ -17,11 +17,14 @@
 
 package org.apache.shenyu.plugin.apache.dubbo;
 
+import org.apache.dubbo.common.constants.CommonConstants;
 import org.apache.dubbo.rpc.RpcContext;
 import org.apache.shenyu.common.constant.Constants;
 import org.apache.shenyu.common.dto.MetaData;
 import org.apache.shenyu.common.dto.RuleData;
 import org.apache.shenyu.common.dto.SelectorData;
+import org.apache.shenyu.common.dto.convert.rule.impl.DubboRuleHandle;
+import org.apache.shenyu.plugin.apache.dubbo.handler.ApacheDubboPluginDataHandler;
 import org.apache.shenyu.plugin.apache.dubbo.proxy.ApacheDubboProxyService;
 import org.apache.shenyu.plugin.api.ShenyuPluginChain;
 import org.apache.shenyu.plugin.dubbo.common.AbstractDubboPlugin;
@@ -63,6 +66,10 @@ protected Mono<Void> doDubboInvoker(final ServerWebExchange exchange,
                                         final RuleData rule,
                                         final MetaData metaData,
                                         final String param) {
+        DubboRuleHandle dubboRuleHandle = ApacheDubboPluginDataHandler.RULE_CACHED_HANDLE.get().obtainHandle(rule.getId());
+        if (dubboRuleHandle.getTimeout() > 0) {
+            RpcContext.getClientAttachment().setAttachment(CommonConstants.TIMEOUT_KEY, dubboRuleHandle.getTimeout());
+        }
         RpcContext.getClientAttachment().setAttachment(Constants.DUBBO_SELECTOR_ID, selector.getId());
         RpcContext.getClientAttachment().setAttachment(Constants.DUBBO_RULE_ID, rule.getId());
         RpcContext.getClientAttachment().setAttachment(Constants.DUBBO_REMOTE_ADDRESS, Objects.requireNonNull(exchange.getRequest().getRemoteAddress()).getAddress().getHostAddress());