com.netflix.zeno.util.SimultaneousExecutor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of netflix-zeno Show documentation

netflix-zeno developed by Netflix

There is a newer version: 2.22.3

/*
 *
 *  Copyright 2013 Netflix, 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 com.netflix.zeno.util;

import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

/**
 *
 * A convenience wrapper around ThreadPoolExecutor.  Provides sane defaults to constructor arguments and
 * allows for awaitUninterruptibly().
 *
 * @author dkoszewnik
 *
 */
public class SimultaneousExecutor extends ThreadPoolExecutor {

    private static final String DEFAULT_THREAD_NAME = "zeno-simultaneous-executor";

    public SimultaneousExecutor() {
        this(2.5d);
    }

    public SimultaneousExecutor(double threadsPerCpu) {
        this(threadsPerCpu, DEFAULT_THREAD_NAME);
    }

    public SimultaneousExecutor(double threadsPerCpu, String threadName) {
        this((int)((double)Runtime.getRuntime().availableProcessors() * threadsPerCpu), threadName);
    }

    public SimultaneousExecutor(int numThreads) {
        this(numThreads, DEFAULT_THREAD_NAME);
    }

    public SimultaneousExecutor(int numThreads, final String threadName) {
        super(numThreads, numThreads, numThreads, TimeUnit.SECONDS, new LinkedBlockingQueue(), new ThreadFactory() {
            public Thread newThread(Runnable r) {
                Thread t = new Thread(r, threadName);
                t.setDaemon(true);
                return t;
            }
        });
    }


    public void awaitUninterruptibly() {
        shutdown();
        while(!isTerminated()) {
            try {
                awaitTermination(1, TimeUnit.DAYS);
            } catch (final InterruptedException e) { }
        }
    }

}