Commit 18300239 authored by zhouwei's avatar zhouwei

请求转发携带model和model key

parent 633730f7
...@@ -25,5 +25,7 @@ public class FailoverProperties { ...@@ -25,5 +25,7 @@ public class FailoverProperties {
private String primaryUrl; private String primaryUrl;
private String fallbackHost; private String fallbackHost;
private String fallbackUrl; 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 ...@@ -55,6 +55,8 @@ public class FailoverGatewayFilterFactory extends AbstractGatewayFilterFactory<F
uriConfig.getPrimaryUrl(), uriConfig.getPrimaryUrl(),
sourceUri, sourceUri,
uriConfig.getPrimaryHost(), uriConfig.getPrimaryHost(),
uriConfig.getModel(),
uriConfig.getModelKey(),
exchange, exchange,
true true
).onErrorResume(primaryError -> { ).onErrorResume(primaryError -> {
...@@ -67,6 +69,8 @@ public class FailoverGatewayFilterFactory extends AbstractGatewayFilterFactory<F ...@@ -67,6 +69,8 @@ public class FailoverGatewayFilterFactory extends AbstractGatewayFilterFactory<F
uriConfig.getFallbackUrl(), uriConfig.getFallbackUrl(),
targetUri, targetUri,
uriConfig.getFallbackHost(), uriConfig.getFallbackHost(),
uriConfig.getModel(),
uriConfig.getModelKey(),
exchange, exchange,
false false
); );
...@@ -109,7 +113,7 @@ public class FailoverGatewayFilterFactory extends AbstractGatewayFilterFactory<F ...@@ -109,7 +113,7 @@ public class FailoverGatewayFilterFactory extends AbstractGatewayFilterFactory<F
.defaultIfEmpty(""); .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) { ServerWebExchange exchange, boolean isSourceRequest) {
String cachedBody = REQUEST_BODY_CACHE.get(); String cachedBody = REQUEST_BODY_CACHE.get();
ServerHttpRequest request = exchange.getRequest(); ServerHttpRequest request = exchange.getRequest();
...@@ -127,6 +131,8 @@ public class FailoverGatewayFilterFactory extends AbstractGatewayFilterFactory<F ...@@ -127,6 +131,8 @@ public class FailoverGatewayFilterFactory extends AbstractGatewayFilterFactory<F
requestBodySpec.headers(headers -> { requestBodySpec.headers(headers -> {
headers.addAll(request.getHeaders()); headers.addAll(request.getHeaders());
headers.set(HttpHeaders.HOST, host); 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