ThreadPool: Add a blocking thread pool, closes 74.
This commit is contained in:
parent
2551fb9fce
commit
fc47bd6f2f
|
@ -0,0 +1,71 @@
|
||||||
|
/*
|
||||||
|
* Licensed to Elastic Search and Shay Banon under one
|
||||||
|
* or more contributor license agreements. See the NOTICE file
|
||||||
|
* distributed with this work for additional information
|
||||||
|
* regarding copyright ownership. Elastic Search licenses this
|
||||||
|
* file to you 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.elasticsearch.threadpool.blocking;
|
||||||
|
|
||||||
|
import com.google.inject.Inject;
|
||||||
|
import org.elasticsearch.threadpool.support.AbstractThreadPool;
|
||||||
|
import org.elasticsearch.util.SizeUnit;
|
||||||
|
import org.elasticsearch.util.SizeValue;
|
||||||
|
import org.elasticsearch.util.TimeValue;
|
||||||
|
import org.elasticsearch.util.concurrent.DynamicExecutors;
|
||||||
|
import org.elasticsearch.util.settings.Settings;
|
||||||
|
|
||||||
|
import java.util.concurrent.Executors;
|
||||||
|
|
||||||
|
import static org.elasticsearch.util.TimeValue.*;
|
||||||
|
import static org.elasticsearch.util.settings.ImmutableSettings.Builder.*;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* A thread pool that will will block the execute if all threads are busy.
|
||||||
|
*
|
||||||
|
* @author kimchy (shay.banon)
|
||||||
|
*/
|
||||||
|
public class BlockingThreadPool extends AbstractThreadPool {
|
||||||
|
|
||||||
|
private final int min;
|
||||||
|
private final int max;
|
||||||
|
private final int capacity;
|
||||||
|
private final TimeValue waitTime;
|
||||||
|
private final TimeValue keepAlive;
|
||||||
|
|
||||||
|
private final int scheduledSize;
|
||||||
|
|
||||||
|
public BlockingThreadPool() {
|
||||||
|
this(EMPTY_SETTINGS);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Inject public BlockingThreadPool(Settings settings) {
|
||||||
|
super(settings);
|
||||||
|
this.min = componentSettings.getAsInt("min", 1);
|
||||||
|
this.max = componentSettings.getAsInt("max", 100);
|
||||||
|
this.capacity = (int) componentSettings.getAsSize("capacity", new SizeValue(1, SizeUnit.KB)).bytes();
|
||||||
|
this.waitTime = componentSettings.getAsTime("waitTime", timeValueSeconds(60));
|
||||||
|
this.keepAlive = componentSettings.getAsTime("keepAlive", timeValueSeconds(60));
|
||||||
|
this.scheduledSize = componentSettings.getAsInt("scheduledSize", 20);
|
||||||
|
logger.debug("Initializing {} thread pool with min[{}], max[{}], keepAlive[{}], capacity[{}], waitTime[{}], scheduledSize[{}]", new Object[]{getType(), min, max, keepAlive, capacity, waitTime, scheduledSize});
|
||||||
|
executorService = DynamicExecutors.newBlockingThreadPool(min, max, keepAlive.millis(), capacity, waitTime.millis(), DynamicExecutors.daemonThreadFactory(settings, "[tp]"));
|
||||||
|
scheduledExecutorService = Executors.newScheduledThreadPool(scheduledSize, DynamicExecutors.daemonThreadFactory(settings, "[sc]"));
|
||||||
|
started = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override public String getType() {
|
||||||
|
return "blocking";
|
||||||
|
}
|
||||||
|
}
|
|
@ -34,7 +34,9 @@ import static org.elasticsearch.util.TimeValue.*;
|
||||||
import static org.elasticsearch.util.settings.ImmutableSettings.Builder.*;
|
import static org.elasticsearch.util.settings.ImmutableSettings.Builder.*;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @author kimchy (Shay Banon)
|
* A thread pool that will create an unbounded number of threads.
|
||||||
|
*
|
||||||
|
* @author kimchy (shay.banon)
|
||||||
*/
|
*/
|
||||||
public class CachedThreadPool extends AbstractThreadPool {
|
public class CachedThreadPool extends AbstractThreadPool {
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue