org/sonews/ShutdownHook.java
author cli
Fri, 21 Aug 2009 17:33:15 +0200
changeset 18 7e527fdf0fa8
parent 3 2fdc9cc89502
permissions -rw-r--r--
Fix for #549.
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
chris@3
    33
  /**
chris@3
    34
   * Called when the JVM exits.
chris@3
    35
   */
chris@3
    36
  @Override
chris@3
    37
  public void run()
chris@3
    38
  {
chris@3
    39
    System.out.println("sonews: Trying to shutdown all threads...");
chris@3
    40
chris@3
    41
    Map<Thread, StackTraceElement[]> threadsMap = Thread.getAllStackTraces();
chris@3
    42
    for(Thread thread : threadsMap.keySet())
chris@3
    43
    {
chris@3
    44
      // Interrupt the thread if it's a AbstractDaemon
chris@3
    45
      AbstractDaemon daemon;
chris@3
    46
      if(thread instanceof AbstractDaemon && thread.isAlive())
chris@3
    47
      {
chris@3
    48
        try
chris@3
    49
        {
chris@3
    50
          daemon = (AbstractDaemon)thread;
chris@3
    51
          daemon.shutdownNow();
chris@3
    52
        }
chris@3
    53
        catch(SQLException ex)
chris@3
    54
        {
chris@3
    55
          System.out.println("sonews: " + ex);
chris@3
    56
        }
chris@3
    57
      }
chris@3
    58
    }
chris@3
    59
    
chris@3
    60
    for(Thread thread : threadsMap.keySet())
chris@3
    61
    {
chris@3
    62
      AbstractDaemon daemon;
chris@3
    63
      if(thread instanceof AbstractDaemon && thread.isAlive())
chris@3
    64
      {
chris@3
    65
        daemon = (AbstractDaemon)thread;
chris@3
    66
        System.out.println("sonews: Waiting for " + daemon + " to exit...");
chris@3
    67
        try
chris@3
    68
        {
chris@3
    69
          daemon.join(500);
chris@3
    70
        }
chris@3
    71
        catch(InterruptedException ex)
chris@3
    72
        {
chris@3
    73
          System.out.println(ex.getLocalizedMessage());
chris@3
    74
        }
chris@3
    75
      }
chris@3
    76
    }
chris@3
    77
    
chris@3
    78
    // We have notified all not-sleeping AbstractDaemons of the shutdown;
chris@3
    79
    // all other threads can be simply purged on VM shutdown
chris@3
    80
    
chris@3
    81
    System.out.println("sonews: Clean shutdown.");
chris@3
    82
  }
chris@3
    83
  
chris@3
    84
}