optimize code

pull/422/head
wulingxiao 3 years ago
parent c985dc6fb0
commit efc572dfee

@ -22,6 +22,7 @@ import java.util.Objects;
import java.util.concurrent.Executor; import java.util.concurrent.Executor;
import com.alibaba.ttl.threadpool.TtlExecutors; import com.alibaba.ttl.threadpool.TtlExecutors;
import com.alibaba.ttl.threadpool.agent.TtlAgent;
import com.tencent.cloud.common.metadata.MetadataContext; import com.tencent.cloud.common.metadata.MetadataContext;
import com.tencent.cloud.metadata.concurrent.MetadataWrap; import com.tencent.cloud.metadata.concurrent.MetadataWrap;
@ -38,9 +39,14 @@ class MetadataExecutor implements Executor, MetadataWrap<Executor> {
private final Executor delegate; private final Executor delegate;
MetadataExecutor(Executor delegate) { MetadataExecutor(@NonNull Executor delegate) {
if (TtlAgent.isTtlAgentLoaded() || TtlExecutors.isTtlWrapper(delegate)) {
this.delegate = delegate;
}
else {
this.delegate = TtlExecutors.getTtlExecutor(delegate); this.delegate = TtlExecutors.getTtlExecutor(delegate);
} }
}
@Override @Override
public void execute(@NonNull Runnable command) { public void execute(@NonNull Runnable command) {

@ -29,6 +29,7 @@ import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException; import java.util.concurrent.TimeoutException;
import com.alibaba.ttl.threadpool.TtlExecutors; import com.alibaba.ttl.threadpool.TtlExecutors;
import com.alibaba.ttl.threadpool.agent.TtlAgent;
import com.tencent.cloud.common.metadata.MetadataContext; import com.tencent.cloud.common.metadata.MetadataContext;
import org.springframework.lang.NonNull; import org.springframework.lang.NonNull;
@ -44,10 +45,15 @@ class MetadataExecutorService extends MetadataExecutor implements ExecutorServic
private final ExecutorService delegate; private final ExecutorService delegate;
MetadataExecutorService(ExecutorService delegate) { MetadataExecutorService(@NonNull ExecutorService delegate) {
super(delegate); super(delegate);
if (TtlAgent.isTtlAgentLoaded() || TtlExecutors.isTtlWrapper(delegate)) {
this.delegate = delegate;
}
else {
this.delegate = TtlExecutors.getTtlExecutorService(delegate); this.delegate = TtlExecutors.getTtlExecutorService(delegate);
} }
}
@Override @Override
public void shutdown() { public void shutdown() {

@ -25,6 +25,7 @@ import java.util.concurrent.ScheduledFuture;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
import com.alibaba.ttl.threadpool.TtlExecutors; import com.alibaba.ttl.threadpool.TtlExecutors;
import com.alibaba.ttl.threadpool.agent.TtlAgent;
import com.tencent.cloud.common.metadata.MetadataContext; import com.tencent.cloud.common.metadata.MetadataContext;
import org.springframework.lang.NonNull; import org.springframework.lang.NonNull;
@ -41,10 +42,15 @@ class MetadataScheduledExecutorService extends MetadataExecutorService
private final ScheduledExecutorService delegate; private final ScheduledExecutorService delegate;
MetadataScheduledExecutorService(ScheduledExecutorService delegate) { MetadataScheduledExecutorService(@NonNull ScheduledExecutorService delegate) {
super(delegate); super(delegate);
if (TtlAgent.isTtlAgentLoaded() || TtlExecutors.isTtlWrapper(delegate)) {
this.delegate = delegate;
}
else {
this.delegate = TtlExecutors.getTtlScheduledExecutorService(delegate); this.delegate = TtlExecutors.getTtlScheduledExecutorService(delegate);
} }
}
@Override @Override
@NonNull @NonNull

Loading…
Cancel
Save