zgty/zt-dsc
8
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

1. 调整 databus api 策略，添加每次重试新建请求的逻辑

Browse Source
This commit is contained in:
chenbowen
2025-11-06 23:08:55 +08:00
parent ec473a00d4
commit f25ee5091e
1 changed files with 25 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
zt-module-databus/zt-module-databus-server/src/main/java/com/zt/plat/module/databus/framework/integration/gateway/step/impl/HttpStepHandler.java
View File

@@ -50,14 +50,14 @@ public class HttpStepHandler implements ApiStepHandler {
private static final int RETRY_ATTEMPTS = 3;
private static final Set<String> DEFAULT_FORWARDED_HEADERS = Set.of(
"authorization",
"zt-auth-token",
"tenant-id",
"visit-tenant-id",
"visit-company-id",
"visit-company-name",
"visit-dept-id",
"visit-dept-name"
"authorization",
"zt-auth-token",
"tenant-id",
"visit-tenant-id",
"visit-company-id",
"visit-company-name",
"visit-dept-id",
"visit-dept-name"
);
@Override
@@ -112,8 +112,17 @@ public class HttpStepHandler implements ApiStepHandler {
Map<String, String> headerMap = resolveHeaders(stepDefinition, payload);
Duration timeout = resolveTimeout(stepDefinition);
WebClient client = webClientBuilder.build();
WebClient.RequestHeadersSpec<?> requestSpec = buildRequest(client, callSpec, requestPayload, headerMap, supportsBody);
Mono<Object> responseMono = requestSpec.retrieve().bodyToMono(Object.class);
final HttpRequestPayload resolvedPayload = requestPayload;
boolean finalSupportsBody = supportsBody;
Mono<Object> responseMono = Mono.defer(() -> {
// 每次订阅（含重试）都会重新构建请求，避免缓存第一次结果
HttpRequestPayload payloadForAttempt = resolvedPayload == null
? HttpRequestPayload.of(null, Collections.emptyMap())
: resolvedPayload;
return buildRequest(client, callSpec, payloadForAttempt, headerMap, finalSupportsBody)
.retrieve()
.bodyToMono(Object.class);
});
responseMono = applyResilientRetry(responseMono, stepDefinition);
Object response = timeout == null ? responseMono.block() : responseMono.block(timeout);
payload.addStepResult(ApiStepResult.builder()
@@ -353,12 +362,12 @@ public class HttpStepHandler implements ApiStepHandler {
private record HttpRequestPayload(Object body, Map<String, Object> queryParams) {
private HttpRequestPayload {
Map<String, Object> safeQuery = queryParams == null
? Collections.emptyMap()
: Collections.unmodifiableMap(new LinkedHashMap<>(queryParams));
queryParams = safeQuery;
}
private HttpRequestPayload {
Map<String, Object> safeQuery = queryParams == null
? Collections.emptyMap()
: Collections.unmodifiableMap(new LinkedHashMap<>(queryParams));
queryParams = safeQuery;
}
static HttpRequestPayload of(Object body, Map<String, Object> queryParams) {
return new HttpRequestPayload(body, queryParams);