org.cometd.benchmark.MonitoringThreadPoolExecutor Maven / Gradle / Ivy
/*
* Copyright (c) 2008-2016 the original author or authors.
*
* 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 org.cometd.benchmark;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.RejectedExecutionHandler;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicLong;
public class MonitoringThreadPoolExecutor extends ThreadPoolExecutor {
private final AtomicLong tasks = new AtomicLong();
private final AtomicLong maxTaskLatency = new AtomicLong();
private final AtomicLong totalTaskLatency = new AtomicLong();
private final MonitoringLinkedBlockingQueue queue;
private final AtomicLong maxQueueLatency = new AtomicLong();
private final AtomicLong totalQueueLatency = new AtomicLong();
private final AtomicInteger threads = new AtomicInteger();
private final AtomicInteger maxThreads = new AtomicInteger();
public MonitoringThreadPoolExecutor(int maximumPoolSize, long keepAliveTime, TimeUnit unit) {
this(maximumPoolSize, keepAliveTime, unit, new AbortPolicy());
}
public MonitoringThreadPoolExecutor(int maximumPoolSize, long keepAliveTime, TimeUnit unit, RejectedExecutionHandler handler) {
super(maximumPoolSize, maximumPoolSize, keepAliveTime, unit, new MonitoringLinkedBlockingQueue(), handler);
queue = (MonitoringLinkedBlockingQueue)getQueue();
}
public void reset() {
tasks.set(0);
maxTaskLatency.set(0);
totalTaskLatency.set(0);
queue.reset();
maxQueueLatency.set(0);
totalQueueLatency.set(0);
threads.set(0);
maxThreads.set(0);
}
public long getTasks() {
return tasks.get();
}
public long getMaxTaskLatency() {
return maxTaskLatency.get();
}
public long getAverageTaskLatency() {
long count = tasks.get();
return count == 0 ? -1 : totalTaskLatency.get() / count;
}
public long getMaxQueueLatency() {
return maxQueueLatency.get();
}
public long getAverageQueueLatency() {
long count = tasks.get();
return count == 0 ? -1 : totalQueueLatency.get() / count;
}
public int getMaxQueueSize() {
return queue.maxSize.get();
}
public int getMaxActiveThreads() {
return maxThreads.get();
}
@Override
public void execute(final Runnable task) {
final long begin = System.nanoTime();
super.execute(new Runnable() {
public void run() {
long latency = System.nanoTime() - begin;
tasks.incrementAndGet();
Atomics.updateMax(maxQueueLatency, latency);
totalQueueLatency.addAndGet(latency);
Atomics.updateMax(maxThreads, threads.incrementAndGet());
long start = System.nanoTime();
try {
task.run();
} finally {
long taskLatency = System.nanoTime() - start;
threads.decrementAndGet();
Atomics.updateMax(maxTaskLatency, taskLatency);
totalTaskLatency.addAndGet(taskLatency);
}
}
});
}
private static class MonitoringLinkedBlockingQueue extends LinkedBlockingQueue {
private final AtomicInteger size = new AtomicInteger();
private final AtomicInteger maxSize = new AtomicInteger();
public void reset() {
size.set(0);
maxSize.set(0);
}
@Override
public void clear() {
reset();
super.clear();
}
@Override
public boolean offer(Runnable task) {
boolean added = super.offer(task);
if (added) {
increment();
}
return added;
}
private void increment() {
Atomics.updateMax(maxSize, size.incrementAndGet());
}
@Override
public Runnable poll() {
Runnable task = super.poll();
if (task != null) {
decrement();
}
return task;
}
@Override
public Runnable poll(long timeout, TimeUnit unit) throws InterruptedException {
Runnable task = super.poll(timeout, unit);
if (task != null) {
decrement();
}
return task;
}
@Override
public Runnable take() throws InterruptedException {
Runnable task = super.take();
decrement();
return task;
}
private void decrement() {
size.decrementAndGet();
}
}
}