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.*;
|
||||
|
||||
/**
|
||||
* @author kimchy (Shay Banon)
|
||||
* A thread pool that will create an unbounded number of threads.
|
||||
*
|
||||
* @author kimchy (shay.banon)
|
||||
*/
|
||||
public class CachedThreadPool extends AbstractThreadPool {
|
||||
|
||||
|
|
Loading…
Reference in New Issue