chris@3: /*
chris@3:  *   SONEWS News Server
chris@3:  *   see AUTHORS for the list of contributors
chris@3:  *
chris@3:  *   This program is free software: you can redistribute it and/or modify
chris@3:  *   it under the terms of the GNU General Public License as published by
chris@3:  *   the Free Software Foundation, either version 3 of the License, or
chris@3:  *   (at your option) any later version.
chris@3:  *
chris@3:  *   This program is distributed in the hope that it will be useful,
chris@3:  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
chris@3:  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
chris@3:  *   GNU General Public License for more details.
chris@3:  *
chris@3:  *   You should have received a copy of the GNU General Public License
chris@3:  *   along with this program.  If not, see <http://www.gnu.org/licenses/>.
chris@3:  */
chris@3: package org.sonews;
chris@3: 
chris@3: import java.sql.SQLException;
chris@3: import java.util.Map;
chris@3: import org.sonews.daemon.AbstractDaemon;
chris@3: 
chris@3: /**
chris@3:  * Will force all other threads to shutdown cleanly.
chris@3:  * @author Christian Lins
chris@3:  * @since sonews/0.5.0
chris@3:  */
cli@49: class ShutdownHook extends Thread {
chris@3: 
cli@37: 	/**
cli@37: 	 * Called when the JVM exits.
cli@37: 	 */
cli@37: 	@Override
cli@49: 	public void run() {
cli@37: 		System.out.println("sonews: Trying to shutdown all threads...");
chris@3: 
cli@37: 		Map<Thread, StackTraceElement[]> threadsMap = Thread.getAllStackTraces();
cli@37: 		for (Thread thread : threadsMap.keySet()) {
cli@37: 			// Interrupt the thread if it's a AbstractDaemon
cli@37: 			AbstractDaemon daemon;
cli@37: 			if (thread instanceof AbstractDaemon && thread.isAlive()) {
cli@37: 				try {
cli@37: 					daemon = (AbstractDaemon) thread;
cli@37: 					daemon.shutdownNow();
cli@37: 				} catch (SQLException ex) {
cli@37: 					System.out.println("sonews: " + ex);
cli@37: 				}
cli@37: 			}
cli@37: 		}
cli@37: 
cli@37: 		for (Thread thread : threadsMap.keySet()) {
cli@37: 			AbstractDaemon daemon;
cli@37: 			if (thread instanceof AbstractDaemon && thread.isAlive()) {
cli@37: 				daemon = (AbstractDaemon) thread;
cli@37: 				System.out.println("sonews: Waiting for " + daemon + " to exit...");
cli@37: 				try {
cli@37: 					daemon.join(500);
cli@37: 				} catch (InterruptedException ex) {
cli@37: 					System.out.println(ex.getLocalizedMessage());
cli@37: 				}
cli@37: 			}
cli@37: 		}
cli@37: 
cli@37: 		// We have notified all not-sleeping AbstractDaemons of the shutdown;
cli@37: 		// all other threads can be simply purged on VM shutdown
cli@37: 
cli@37: 		System.out.println("sonews: Clean shutdown.");
cli@37: 	}
chris@3: }