1.1 --- /dev/null	Thu Jan 01 00:00:00 1970 +0000
     1.2 +++ b/src/org/sonews/ShutdownHook.java	Sun Aug 29 17:28:58 2010 +0200
     1.3 @@ -0,0 +1,84 @@
     1.4 +/*
     1.5 + *   SONEWS News Server
     1.6 + *   see AUTHORS for the list of contributors
     1.7 + *
     1.8 + *   This program is free software: you can redistribute it and/or modify
     1.9 + *   it under the terms of the GNU General Public License as published by
    1.10 + *   the Free Software Foundation, either version 3 of the License, or
    1.11 + *   (at your option) any later version.
    1.12 + *
    1.13 + *   This program is distributed in the hope that it will be useful,
    1.14 + *   but WITHOUT ANY WARRANTY; without even the implied warranty of
    1.15 + *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    1.16 + *   GNU General Public License for more details.
    1.17 + *
    1.18 + *   You should have received a copy of the GNU General Public License
    1.19 + *   along with this program.  If not, see <http://www.gnu.org/licenses/>.
    1.20 + */
    1.21 +
    1.22 +package org.sonews;
    1.23 +
    1.24 +import java.sql.SQLException;
    1.25 +import java.util.Map;
    1.26 +import org.sonews.daemon.AbstractDaemon;
    1.27 +
    1.28 +/**
    1.29 + * Will force all other threads to shutdown cleanly.
    1.30 + * @author Christian Lins
    1.31 + * @since sonews/0.5.0
    1.32 + */
    1.33 +class ShutdownHook extends Thread
    1.34 +{
    1.35 +
    1.36 +  /**
    1.37 +   * Called when the JVM exits.
    1.38 +   */
    1.39 +  @Override
    1.40 +  public void run()
    1.41 +  {
    1.42 +    System.out.println("sonews: Trying to shutdown all threads...");
    1.43 +
    1.44 +    Map<Thread, StackTraceElement[]> threadsMap = Thread.getAllStackTraces();
    1.45 +    for(Thread thread : threadsMap.keySet())
    1.46 +    {
    1.47 +      // Interrupt the thread if it's a AbstractDaemon
    1.48 +      AbstractDaemon daemon;
    1.49 +      if(thread instanceof AbstractDaemon && thread.isAlive())
    1.50 +      {
    1.51 +        try
    1.52 +        {
    1.53 +          daemon = (AbstractDaemon)thread;
    1.54 +          daemon.shutdownNow();
    1.55 +        }
    1.56 +        catch(SQLException ex)
    1.57 +        {
    1.58 +          System.out.println("sonews: " + ex);
    1.59 +        }
    1.60 +      }
    1.61 +    }
    1.62 +    
    1.63 +    for(Thread thread : threadsMap.keySet())
    1.64 +    {
    1.65 +      AbstractDaemon daemon;
    1.66 +      if(thread instanceof AbstractDaemon && thread.isAlive())
    1.67 +      {
    1.68 +        daemon = (AbstractDaemon)thread;
    1.69 +        System.out.println("sonews: Waiting for " + daemon + " to exit...");
    1.70 +        try
    1.71 +        {
    1.72 +          daemon.join(500);
    1.73 +        }
    1.74 +        catch(InterruptedException ex)
    1.75 +        {
    1.76 +          System.out.println(ex.getLocalizedMessage());
    1.77 +        }
    1.78 +      }
    1.79 +    }
    1.80 +    
    1.81 +    // We have notified all not-sleeping AbstractDaemons of the shutdown;
    1.82 +    // all other threads can be simply purged on VM shutdown
    1.83 +    
    1.84 +    System.out.println("sonews: Clean shutdown.");
    1.85 +  }
    1.86 +  
    1.87 +}