io.micrometer.core.instrument.internal.TimedExecutorService Maven / Gradle / Ivy
Show all versions of micrometer-core Show documentation
/**
* Copyright 2017 VMware, 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
*
* https://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. This class is for internal use.
*
* @author Jon Schneider
* @see io.micrometer.core.instrument.binder.jvm.ExecutorServiceMetrics
*/
public class TimedExecutorService implements ExecutorService {
private final MeterRegistry registry;
private final ExecutorService delegate;
private final Timer executionTimer;
private final Timer idleTimer;
public TimedExecutorService(MeterRegistry registry, ExecutorService delegate, String executorServiceName,
String metricPrefix, Iterable tags) {
this.registry = registry;
this.delegate = delegate;
Tags finalTags = Tags.concat(tags, "name", executorServiceName);
this.executionTimer = registry.timer(metricPrefix + "executor", finalTags);
this.idleTimer = registry.timer(metricPrefix + "executor.idle", finalTags);
}
@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(wrap(task));
}
@Override
public Future submit(Runnable task, T result) {
return delegate.submit(wrap(task), result);
}
@Override
public Future submit(Runnable task) {
return delegate.submit(wrap(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(wrap(command));
}
private Runnable wrap(Runnable task) {
return new TimedRunnable(registry, executionTimer, idleTimer, task);
}
private Callable wrap(Callable task) {
return new TimedCallable<>(registry, executionTimer, idleTimer, task);
}
private Collection> wrapAll(Collection> tasks) {
return tasks.stream().map(this::wrap).collect(toList());
}
}