src/org/sonews/ShutdownHook.java
author cli
Sun, 11 Sep 2011 15:05:04 +0200
changeset 48 b78e77619152
parent 37 74139325d305
child 49 8df94bfd3e2f
permissions -rwxr-xr-x
Merge Channel and Group classes.
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
chris@3
    19
package org.sonews;
chris@3
    20
chris@3
    21
import java.sql.SQLException;
chris@3
    22
import java.util.Map;
chris@3
    23
import org.sonews.daemon.AbstractDaemon;
chris@3
    24
chris@3
    25
/**
chris@3
    26
 * Will force all other threads to shutdown cleanly.
chris@3
    27
 * @author Christian Lins
chris@3
    28
 * @since sonews/0.5.0
chris@3
    29
 */
chris@3
    30
class ShutdownHook extends Thread
chris@3
    31
{
chris@3
    32
cli@37
    33
	/**
cli@37
    34
	 * Called when the JVM exits.
cli@37
    35
	 */
cli@37
    36
	@Override
cli@37
    37
	public void run()
cli@37
    38
	{
cli@37
    39
		System.out.println("sonews: Trying to shutdown all threads...");
chris@3
    40
cli@37
    41
		Map<Thread, StackTraceElement[]> threadsMap = Thread.getAllStackTraces();
cli@37
    42
		for (Thread thread : threadsMap.keySet()) {
cli@37
    43
			// Interrupt the thread if it's a AbstractDaemon
cli@37
    44
			AbstractDaemon daemon;
cli@37
    45
			if (thread instanceof AbstractDaemon && thread.isAlive()) {
cli@37
    46
				try {
cli@37
    47
					daemon = (AbstractDaemon) thread;
cli@37
    48
					daemon.shutdownNow();
cli@37
    49
				} catch (SQLException ex) {
cli@37
    50
					System.out.println("sonews: " + ex);
cli@37
    51
				}
cli@37
    52
			}
cli@37
    53
		}
cli@37
    54
cli@37
    55
		for (Thread thread : threadsMap.keySet()) {
cli@37
    56
			AbstractDaemon daemon;
cli@37
    57
			if (thread instanceof AbstractDaemon && thread.isAlive()) {
cli@37
    58
				daemon = (AbstractDaemon) thread;
cli@37
    59
				System.out.println("sonews: Waiting for " + daemon + " to exit...");
cli@37
    60
				try {
cli@37
    61
					daemon.join(500);
cli@37
    62
				} catch (InterruptedException ex) {
cli@37
    63
					System.out.println(ex.getLocalizedMessage());
cli@37
    64
				}
cli@37
    65
			}
cli@37
    66
		}
cli@37
    67
cli@37
    68
		// We have notified all not-sleeping AbstractDaemons of the shutdown;
cli@37
    69
		// all other threads can be simply purged on VM shutdown
cli@37
    70
cli@37
    71
		System.out.println("sonews: Clean shutdown.");
cli@37
    72
	}
chris@3
    73
}