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