org.apache.rocketmq.shaded.io.opentelemetry.context.ContextExecutorService Maven / Gradle / Ivy
/*
* Copyright The OpenTelemetry Authors
* SPDX-License-Identifier: Apache-2.0
*/
package org.apache.rocketmq.shaded.io.opentelemetry.context;
import java.util.Collection;
import java.util.List;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;
class ContextExecutorService extends ForwardingExecutorService {
private final Context context;
ContextExecutorService(Context context, ExecutorService delegate) {
super(delegate);
this.context = context;
}
final Context context() {
return context;
}
@Override
public Future submit(Callable task) {
return delegate().submit(context.wrap(task));
}
@Override
public Future submit(Runnable task, T result) {
return delegate().submit(context.wrap(task), result);
}
@Override
public Future submit(Runnable task) {
return delegate().submit(context.wrap(task));
}
@Override
public List> invokeAll(Collection> tasks)
throws InterruptedException {
return delegate().invokeAll(wrap(context, tasks));
}
@Override
public List> invokeAll(
Collection> tasks, long timeout, TimeUnit unit)
throws InterruptedException {
return delegate().invokeAll(wrap(context, tasks), timeout, unit);
}
@Override
public T invokeAny(Collection> tasks)
throws InterruptedException, ExecutionException {
return delegate().invokeAny(wrap(context, tasks));
}
@Override
public T invokeAny(Collection> tasks, long timeout, TimeUnit unit)
throws InterruptedException, ExecutionException, TimeoutException {
return delegate().invokeAny(wrap(context, tasks), timeout, unit);
}
@Override
public void execute(Runnable command) {
delegate().execute(context.wrap(command));
}
}