io.micrometer.core.instrument.internal.TimedExecutorService Maven / Gradle / Ivy
Show all versions of micrometer-core Show documentation
/**
* Copyright 2017 Pivotal Software, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package io.micrometer.core.instrument.internal;
import io.micrometer.core.instrument.MeterRegistry;
import io.micrometer.core.instrument.Tag;
import io.micrometer.core.instrument.Tags;
import io.micrometer.core.instrument.Timer;
import java.util.Collection;
import java.util.List;
import java.util.concurrent.*;
import static java.util.stream.Collectors.toList;
/**
* An {@link java.util.concurrent.ExecutorService} that is timed
*
* @author Jon Schneider
*/
public class TimedExecutorService implements ExecutorService {
private final ExecutorService delegate;
private final Timer timer;
public TimedExecutorService(MeterRegistry registry, ExecutorService delegate, String executorServiceName, Iterable tags) {
this.delegate = delegate;
this.timer = registry.timer("executor", Tags.concat(tags ,"name", executorServiceName));
}
@Override
public void shutdown() {
delegate.shutdown();
}
@Override
public List shutdownNow() {
return delegate.shutdownNow();
}
@Override
public boolean isShutdown() {
return delegate.isShutdown();
}
@Override
public boolean isTerminated() {
return delegate.isTerminated();
}
@Override
public boolean awaitTermination(long timeout, TimeUnit unit) throws InterruptedException {
return delegate.awaitTermination(timeout, unit);
}
@Override
public Future submit(Callable task) {
return delegate.submit(timer.wrap(task));
}
@Override
public Future submit(Runnable task, T result) {
return delegate.submit(() -> timer.record(task), result);
}
@Override
public Future submit(Runnable task) {
return delegate.submit(() -> timer.record(task));
}
@Override
public List> invokeAll(Collection> tasks) throws InterruptedException {
return delegate.invokeAll(wrapAll(tasks));
}
@Override
public List> invokeAll(Collection> tasks, long timeout, TimeUnit unit) throws InterruptedException {
return delegate.invokeAll(wrapAll(tasks), timeout, unit);
}
@Override
public T invokeAny(Collection> tasks) throws InterruptedException, ExecutionException {
return delegate.invokeAny(wrapAll(tasks));
}
@Override
public T invokeAny(Collection> tasks, long timeout, TimeUnit unit) throws InterruptedException, ExecutionException, TimeoutException {
return delegate.invokeAny(wrapAll(tasks), timeout, unit);
}
@Override
public void execute(Runnable command) {
delegate.execute(timer.wrap(command));
}
private Collection> wrapAll(Collection> tasks) {
return tasks.stream().map(timer::wrap).collect(toList());
}
}