third_codes/PowerJob
1
0
Fork 1
mirror of https://github.com/PowerJob/PowerJob.git synced 2025-07-17 00:00:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: offical HttpProcessor support custom timeout

Browse Source
This commit is contained in:
tjq 2021-01-31 19:49:16 +08:00
parent 02eed5e15e
commit 2c3e35a742
1 changed files with 22 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
powerjob-offical-processors/src/main/java/tech/powerjob/offical/processors/impl/HttpProcessor.java
View File

@ -11,6 +11,7 @@ import org.apache.commons.lang3.StringUtils;
import tech.powerjob.offical.processors.CommonBasicProcessor; import tech.powerjob.offical.processors.CommonBasicProcessor;
import java.util.Map; import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
/** /**
@ -21,14 +22,9 @@ import java.util.concurrent.TimeUnit;
*/ */
public class HttpProcessor extends CommonBasicProcessor { public class HttpProcessor extends CommonBasicProcessor {
private static final OkHttpClient client; // 60 seconds
private static final int DEFAULT_TIMEOUT = 60;
static { private static final Map<Integer, OkHttpClient> CLIENT_STORE = new ConcurrentHashMap<>();
client = new OkHttpClient.Builder()
.connectTimeout(10, TimeUnit.SECONDS)
.readTimeout(60, TimeUnit.SECONDS)
.build();
}
@Override @Override
public ProcessResult process0(TaskContext taskContext) throws Exception { public ProcessResult process0(TaskContext taskContext) throws Exception {
@ -61,6 +57,13 @@ public class HttpProcessor extends CommonBasicProcessor {
} }
} }
// set default timeout
if (httpParams.timeout == null) {
httpParams.timeout = DEFAULT_TIMEOUT;
}
omsLogger.info("[HttpProcessor] request timeout: {} seconds", httpParams.timeout);
OkHttpClient client = getClient(httpParams.timeout);
Request.Builder builder = new Request.Builder().url(httpParams.url); Request.Builder builder = new Request.Builder().url(httpParams.url);
if (httpParams.headers != null) { if (httpParams.headers != null) {
httpParams.headers.forEach((k, v) -> { httpParams.headers.forEach((k, v) -> {
@ -118,5 +121,16 @@ public class HttpProcessor extends CommonBasicProcessor {
private String body; private String body;
private Map<String, String> headers; private Map<String, String> headers;
/**
* timeout for complete calls
*/
private Integer timeout;
}
private static OkHttpClient getClient(Integer timeout) {
return CLIENT_STORE.computeIfAbsent(timeout, ignore -> new OkHttpClient.Builder()
.callTimeout(timeout, TimeUnit.SECONDS)
.build());
} }
} }