net.dongliu.prettypb.rpc.utils.CleanShutdownHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of prettypb-rpc Show documentation

proto rpc libs, compatible with proto-rpc-pro

There is a newer version: 0.3.5

/**
 *   Copyright 2010-2014 Peter Klauser
 *
 *   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.dongliu.prettypb.rpc.utils;

import io.netty.util.concurrent.EventExecutorGroup;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.LinkedList;
import java.util.List;
import java.util.concurrent.ExecutorService;

/**
 * Registers a JVM shutdown hook to cleanly shutdown any
 * Client or Server Bootstraps, and TimeoutCheckers or TimeoutExecutors.
 * 
 * @author Peter Klauser
 *
 */
public class CleanShutdownHandler {

	private static Logger log = LoggerFactory.getLogger(CleanShutdownHandler.class);
	
	private List bootstraps = new LinkedList();
	private List executors = new LinkedList();
	
	public CleanShutdownHandler() {
		Runtime.getRuntime().addShutdownHook(new Thread(new Runnable() {
			
			@Override
			public void run() {
				log.debug("Releasing " + bootstraps.size() + " Client Bootstrap.");
				for( EventExecutorGroup bootstrap : getBootstraps() ) {
					bootstrap.shutdownGracefully();
				}
				
				log.debug("Releasing " + executors.size() + " Executors.");
				for( ExecutorService executor : getExecutors() ) {
					executor.shutdown();
				}
			}
		} ));
	}
	
	public void addResource( EventExecutorGroup bootstrap ) {
		bootstraps.add(bootstrap);
	}
	
	public void addResource( ExecutorService executor ) {
		executors.add(executor);
	}
	
	public void removeResource( ExecutorService executor ) {
		executors.remove(executor);
	}

	/**
	 * @return the executors
	 */
	public List getExecutors() {
		return executors;
	}

	/**
	 * @param executors the executors to set
	 */
	public void setExecutors(List executors) {
		this.executors = executors;
	}

	/**
	 * @return the bootstraps
	 */
	public List getBootstraps() {
		return bootstraps;
	}

	/**
	 * @param bootstraps the bootstraps to set
	 */
	public void setBootstraps(List bootstraps) {
		this.bootstraps = bootstraps;
	}
}