Commit 64cddf47 by Nepxion

修复ServerWebExchange为null的Bug

parent 4f318f1b
...@@ -9,6 +9,8 @@ package com.nepxion.discovery.plugin.strategy.gateway.adapter; ...@@ -9,6 +9,8 @@ package com.nepxion.discovery.plugin.strategy.gateway.adapter;
* @version 1.0 * @version 1.0
*/ */
import org.springframework.web.server.ServerWebExchange;
import com.nepxion.discovery.common.constant.DiscoveryConstant; import com.nepxion.discovery.common.constant.DiscoveryConstant;
import com.nepxion.discovery.plugin.strategy.adapter.AbstractDiscoveryEnabledAdapter; import com.nepxion.discovery.plugin.strategy.adapter.AbstractDiscoveryEnabledAdapter;
import com.nepxion.discovery.plugin.strategy.gateway.context.GatewayStrategyContext; import com.nepxion.discovery.plugin.strategy.gateway.context.GatewayStrategyContext;
...@@ -18,13 +20,23 @@ public class DefaultDiscoveryEnabledAdapter extends AbstractDiscoveryEnabledAdap ...@@ -18,13 +20,23 @@ public class DefaultDiscoveryEnabledAdapter extends AbstractDiscoveryEnabledAdap
protected String getVersionValue() { protected String getVersionValue() {
GatewayStrategyContext context = GatewayStrategyContext.getCurrentContext(); GatewayStrategyContext context = GatewayStrategyContext.getCurrentContext();
return context.getExchange().getRequest().getHeaders().getFirst(DiscoveryConstant.VERSION); ServerWebExchange exchange = context.getExchange();
if (exchange == null) {
return null;
}
return exchange.getRequest().getHeaders().getFirst(DiscoveryConstant.VERSION);
} }
@Override @Override
protected String getRegionValue() { protected String getRegionValue() {
GatewayStrategyContext context = GatewayStrategyContext.getCurrentContext(); GatewayStrategyContext context = GatewayStrategyContext.getCurrentContext();
return context.getExchange().getRequest().getHeaders().getFirst(DiscoveryConstant.REGION); ServerWebExchange exchange = context.getExchange();
if (exchange == null) {
return null;
}
return exchange.getRequest().getHeaders().getFirst(DiscoveryConstant.REGION);
} }
} }
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment