refactor(接口测试): 优化资源池调用重试机制

--bug=1039749 --user=陈建星 【接口测试】场景列表-批量执行-串行独立报告-执行失败 https://www.tapd.cn/55049933/s/1503398
This commit is contained in:
AgAngle 2024-04-21 14:45:18 +08:00 committed by Craftsman
parent da8d506311
commit f3330726ec
1 changed files with 41 additions and 15 deletions

View File

@ -3,9 +3,12 @@ package io.metersphere.system.utils;
import com.bastiaanjansen.otp.TOTPGenerator;
import io.metersphere.sdk.constants.MsHttpHeaders;
import io.metersphere.sdk.dto.api.task.TaskRequestDTO;
import io.metersphere.sdk.util.LogUtils;
import io.metersphere.system.controller.handler.ResultHolder;
import io.metersphere.system.controller.handler.result.MsHttpResultCode;
import jakarta.annotation.Resource;
import org.apache.hc.client5.http.ConnectTimeoutException;
import org.apache.hc.core5.http.NoHttpResponseException;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
@ -14,8 +17,9 @@ import org.springframework.http.client.HttpComponentsClientHttpRequestFactory;
import org.springframework.stereotype.Component;
import org.springframework.web.client.RestTemplate;
import java.io.IOException;
import java.net.SocketException;
import java.util.List;
import java.util.concurrent.TimeUnit;
@Component
public class TaskRunnerClient {
@ -27,7 +31,8 @@ public class TaskRunnerClient {
private static final String API_STOP = "/api/stop";
private static final RestTemplate restTemplateWithTimeOut = new RestTemplate();
private static final int retryCount = 3;
private final static int RETRY_COUNT = 3;
private final static long RETRY_INTERVAL_TIME = 1000L;
static {
HttpComponentsClientHttpRequestFactory httpRequestFactory = new HttpComponentsClientHttpRequestFactory();
@ -80,24 +85,45 @@ public class TaskRunnerClient {
return retry(url, requestBody, action);
}
private static ResultHolder retry(String url, Object requestBody, Action action) throws Exception {
private static ResultHolder retry(String url, Object requestBody, Action action) throws IOException {
ResultHolder body;
try {
// 首次调用
ResultHolder body = action.execute(url, requestBody);
body = action.execute(url, requestBody);
if (body != null && body.getCode() == MsHttpResultCode.SUCCESS.getCode()) {
return body;
}
} catch (NoHttpResponseException | ConnectTimeoutException | SocketException e) {
return doRetry(url, requestBody, action, e);
}
return body;
}
private static ResultHolder doRetry(String url, Object requestBody, Action action, IOException e) throws IOException {
ResultHolder body;
// 增加token失败重试
for (int i = 0; i < retryCount; i++) {
TimeUnit.MILLISECONDS.sleep(300);
for (int i = 1; i <= RETRY_COUNT; i++) {
LogUtils.error(e);
LogUtils.error("retry count {}", i);
try {
//重试延迟
Thread.sleep(i * RETRY_INTERVAL_TIME);
} catch (InterruptedException interruptedException) {
LogUtils.error(interruptedException);
}
try {
body = action.execute(url, requestBody);
// 重试后检查是否成功
if (body != null && body.getCode() == MsHttpResultCode.SUCCESS.getCode()) {
return body;
}
} catch (NoHttpResponseException | ConnectTimeoutException | SocketException retryException) {
LogUtils.error("retry count {} fail", i);
}
return body;
}
throw e;
}
@Resource
@ -107,7 +133,7 @@ public class TaskRunnerClient {
@FunctionalInterface
private interface Action {
ResultHolder execute(String url, Object body);
ResultHolder execute(String url, Object body) throws IOException;
}
}