Commit 18300239 authored by zhouwei's avatar zhouwei

请求转发携带model和model key

parent 633730f7
......@@ -25,5 +25,7 @@ public class FailoverProperties {
private String primaryUrl;
private String fallbackHost;
private String fallbackUrl;
private String model;
private String modelKey;
}
}
\ No newline at end of file
......@@ -55,6 +55,8 @@ public class FailoverGatewayFilterFactory extends AbstractGatewayFilterFactory<F
uriConfig.getPrimaryUrl(),
sourceUri,
uriConfig.getPrimaryHost(),
uriConfig.getModel(),
uriConfig.getModelKey(),
exchange,
true
).onErrorResume(primaryError -> {
......@@ -67,6 +69,8 @@ public class FailoverGatewayFilterFactory extends AbstractGatewayFilterFactory<F
uriConfig.getFallbackUrl(),
targetUri,
uriConfig.getFallbackHost(),
uriConfig.getModel(),
uriConfig.getModelKey(),
exchange,
false
);
......@@ -109,7 +113,7 @@ public class FailoverGatewayFilterFactory extends AbstractGatewayFilterFactory<F
.defaultIfEmpty("");
}
private Mono<Void> tryRequest(String baseUrl, String uri, String host,
private Mono<Void> tryRequest(String baseUrl, String uri, String host, String model, String modelKey,
ServerWebExchange exchange, boolean isSourceRequest) {
String cachedBody = REQUEST_BODY_CACHE.get();
ServerHttpRequest request = exchange.getRequest();
......@@ -127,6 +131,8 @@ public class FailoverGatewayFilterFactory extends AbstractGatewayFilterFactory<F
requestBodySpec.headers(headers -> {
headers.addAll(request.getHeaders());
headers.set(HttpHeaders.HOST, host);
headers.set("model", model);
headers.set("modelKey", modelKey);
});
// 设置请求体
......
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