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