|
|
@ -20,6 +20,7 @@ package com.tencent.cloud.rpc.enhancement.zuul;
|
|
|
|
import java.util.ArrayList;
|
|
|
|
import java.util.ArrayList;
|
|
|
|
import java.util.Collection;
|
|
|
|
import java.util.Collection;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
import com.netflix.client.ClientException;
|
|
|
|
import com.netflix.zuul.ZuulFilter;
|
|
|
|
import com.netflix.zuul.ZuulFilter;
|
|
|
|
import com.netflix.zuul.context.RequestContext;
|
|
|
|
import com.netflix.zuul.context.RequestContext;
|
|
|
|
import com.netflix.zuul.exception.ZuulException;
|
|
|
|
import com.netflix.zuul.exception.ZuulException;
|
|
|
@ -30,6 +31,7 @@ import com.tencent.cloud.rpc.enhancement.plugin.EnhancedPluginRunner;
|
|
|
|
import com.tencent.cloud.rpc.enhancement.plugin.EnhancedPluginType;
|
|
|
|
import com.tencent.cloud.rpc.enhancement.plugin.EnhancedPluginType;
|
|
|
|
import com.tencent.cloud.rpc.enhancement.plugin.EnhancedResponseContext;
|
|
|
|
import com.tencent.cloud.rpc.enhancement.plugin.EnhancedResponseContext;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
import org.springframework.cloud.netflix.zuul.util.ZuulRuntimeException;
|
|
|
|
import org.springframework.core.env.Environment;
|
|
|
|
import org.springframework.core.env.Environment;
|
|
|
|
import org.springframework.http.HttpHeaders;
|
|
|
|
import org.springframework.http.HttpHeaders;
|
|
|
|
import org.springframework.util.StringUtils;
|
|
|
|
import org.springframework.util.StringUtils;
|
|
|
@ -86,6 +88,10 @@ public class EnhancedErrorZuulFilter extends ZuulFilter {
|
|
|
|
Object startTimeMilliObject = context.get(POLARIS_PRE_ROUTE_TIME);
|
|
|
|
Object startTimeMilliObject = context.get(POLARIS_PRE_ROUTE_TIME);
|
|
|
|
Throwable throwable = context.getThrowable();
|
|
|
|
Throwable throwable = context.getThrowable();
|
|
|
|
if (throwable != null && startTimeMilliObject != null && startTimeMilliObject instanceof Long) {
|
|
|
|
if (throwable != null && startTimeMilliObject != null && startTimeMilliObject instanceof Long) {
|
|
|
|
|
|
|
|
if (checkNoInstanceException(throwable)) {
|
|
|
|
|
|
|
|
enhancedPluginContext.setThrowable(throwable);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
else {
|
|
|
|
HttpHeaders responseHeaders = new HttpHeaders();
|
|
|
|
HttpHeaders responseHeaders = new HttpHeaders();
|
|
|
|
Collection<String> names = context.getResponse().getHeaderNames();
|
|
|
|
Collection<String> names = context.getResponse().getHeaderNames();
|
|
|
|
for (String name : names) {
|
|
|
|
for (String name : names) {
|
|
|
@ -106,6 +112,21 @@ public class EnhancedErrorZuulFilter extends ZuulFilter {
|
|
|
|
// Run finally enhanced plugins.
|
|
|
|
// Run finally enhanced plugins.
|
|
|
|
pluginRunner.run(EnhancedPluginType.Client.FINALLY, enhancedPluginContext);
|
|
|
|
pluginRunner.run(EnhancedPluginType.Client.FINALLY, enhancedPluginContext);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
return null;
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
protected boolean checkNoInstanceException(Throwable throwable) {
|
|
|
|
|
|
|
|
if (throwable.getCause() instanceof ZuulRuntimeException) {
|
|
|
|
|
|
|
|
Throwable cause = null;
|
|
|
|
|
|
|
|
if (throwable.getCause().getCause() != null) {
|
|
|
|
|
|
|
|
cause = throwable.getCause().getCause().getCause();
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
if (cause instanceof ClientException && StringUtils.hasText(cause.getMessage()) &&
|
|
|
|
|
|
|
|
cause.getMessage().contains("Load balancer does not have available server for client")) {
|
|
|
|
|
|
|
|
return true;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
return false;
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|