net.hasor.rsf.utils.ExecutesManager Maven / Gradle / Ivy
/*
* Copyright 2008-2009 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 net.hasor.rsf.utils;
import net.hasor.utils.NameThreadFactory;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.*;
import java.util.function.Supplier;
/**
* 业务线程
* @version : 2014年11月11日
* @author 赵永春 ([email protected])
*/
public class ExecutesManager {
protected Logger logger = LoggerFactory.getLogger(getClass());
private Supplier defaultExecutorProvider;
private ThreadPoolExecutor defaultExecutor;
private final ConcurrentMap servicePoolCache;
public ExecutesManager(final int minCorePoolSize, final int maxCorePoolSize, final int queueSize, final long keepAliveTime, final ClassLoader loader) {
logger.info("executesManager init ->> minCorePoolSize ={}, maxCorePoolSize ={}, queueSize ={}, keepAliveTime ={}",//
minCorePoolSize, maxCorePoolSize, queueSize, keepAliveTime);
//
final BlockingQueue inWorkQueue = new LinkedBlockingQueue<>(queueSize);
this.defaultExecutorProvider = () -> new ThreadPoolExecutor(//
minCorePoolSize, //
maxCorePoolSize, //
keepAliveTime, //
TimeUnit.SECONDS, //
inWorkQueue, //
new NameThreadFactory("RSF-Biz-%s", loader),//
new ThreadPoolExecutor.AbortPolicy()//
);
this.defaultExecutor = this.defaultExecutorProvider.get();
this.servicePoolCache = new ConcurrentHashMap<>();
}
public Executor getExecute(String serviceUniqueName) {
if (!this.servicePoolCache.isEmpty() && serviceUniqueName != null) {
ThreadPoolExecutor executor = this.servicePoolCache.get(serviceUniqueName);
if (executor != null) {
return executor;
}
}
return this.defaultExecutor;
}
/**停止应用服务。*/
public void shutdown() {
List executorList = new ArrayList<>(this.servicePoolCache.values());
executorList.add(this.defaultExecutor);
this.servicePoolCache.clear();
this.defaultExecutor = this.defaultExecutorProvider.get();
//
for (ThreadPoolExecutor exec : executorList) {
if (exec == null)
continue;
exec.shutdown();
}
while (true) {
boolean jump = true;
for (ThreadPoolExecutor exec : executorList) {
if (exec == null) {
continue;
}
if (!exec.isShutdown()) {
jump = false;
break;
}
}
if (jump) {
break;
}
try {
Thread.sleep(100);
} catch (Exception e) { /**/ }
}
}
}