src/org/sonews/ShutdownHook.java
author cli
Sun, 11 Sep 2011 17:01:19 +0200
changeset 49 8df94bfd3e2f
parent 37 74139325d305
permissions -rwxr-xr-x
Fix for #14
chris@3
     1
/*
chris@3
     2
 *   SONEWS News Server
chris@3
     3
 *   see AUTHORS for the list of contributors
chris@3
     4
 *
chris@3
     5
 *   This program is free software: you can redistribute it and/or modify
chris@3
     6
 *   it under the terms of the GNU General Public License as published by
chris@3
     7
 *   the Free Software Foundation, either version 3 of the License, or
chris@3
     8
 *   (at your option) any later version.
chris@3
     9
 *
chris@3
    10
 *   This program is distributed in the hope that it will be useful,
chris@3
    11
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
chris@3
    12
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
chris@3
    13
 *   GNU General Public License for more details.
chris@3
    14
 *
chris@3
    15
 *   You should have received a copy of the GNU General Public License
chris@3
    16
 *   along with this program.  If not, see <http://www.gnu.org/licenses/>.
chris@3
    17
 */
chris@3
    18
package org.sonews;
chris@3
    19
chris@3
    20
import java.sql.SQLException;
chris@3
    21
import java.util.Map;
chris@3
    22
import org.sonews.daemon.AbstractDaemon;
chris@3
    23
chris@3
    24
/**
chris@3
    25
 * Will force all other threads to shutdown cleanly.
chris@3
    26
 * @author Christian Lins
chris@3
    27
 * @since sonews/0.5.0
chris@3
    28
 */
cli@49
    29
class ShutdownHook extends Thread {
chris@3
    30
cli@37
    31
	/**
cli@37
    32
	 * Called when the JVM exits.
cli@37
    33
	 */
cli@37
    34
	@Override
cli@49
    35
	public void run() {
cli@37
    36
		System.out.println("sonews: Trying to shutdown all threads...");
chris@3
    37
cli@37
    38
		Map<Thread, StackTraceElement[]> threadsMap = Thread.getAllStackTraces();
cli@37
    39
		for (Thread thread : threadsMap.keySet()) {
cli@37
    40
			// Interrupt the thread if it's a AbstractDaemon
cli@37
    41
			AbstractDaemon daemon;
cli@37
    42
			if (thread instanceof AbstractDaemon && thread.isAlive()) {
cli@37
    43
				try {
cli@37
    44
					daemon = (AbstractDaemon) thread;
cli@37
    45
					daemon.shutdownNow();
cli@37
    46
				} catch (SQLException ex) {
cli@37
    47
					System.out.println("sonews: " + ex);
cli@37
    48
				}
cli@37
    49
			}
cli@37
    50
		}
cli@37
    51
cli@37
    52
		for (Thread thread : threadsMap.keySet()) {
cli@37
    53
			AbstractDaemon daemon;
cli@37
    54
			if (thread instanceof AbstractDaemon && thread.isAlive()) {
cli@37
    55
				daemon = (AbstractDaemon) thread;
cli@37
    56
				System.out.println("sonews: Waiting for " + daemon + " to exit...");
cli@37
    57
				try {
cli@37
    58
					daemon.join(500);
cli@37
    59
				} catch (InterruptedException ex) {
cli@37
    60
					System.out.println(ex.getLocalizedMessage());
cli@37
    61
				}
cli@37
    62
			}
cli@37
    63
		}
cli@37
    64
cli@37
    65
		// We have notified all not-sleeping AbstractDaemons of the shutdown;
cli@37
    66
		// all other threads can be simply purged on VM shutdown
cli@37
    67
cli@37
    68
		System.out.println("sonews: Clean shutdown.");
cli@37
    69
	}
chris@3
    70
}