oddjob之smooth关闭程序

java程序的smooth关闭策略可以采用hook跟观察者的模式实现

无限等状态，如果状态出现可以关闭的事件则进行关闭

虚拟机的关闭通过钩子调用关闭，如果关闭失败，在超时时间内强制杀掉jvm

状态类

public interface State {

    public boolean isStoppable();
    
}

package com.s.stat;



/**
 * A Stateful job implements this interface so that 
 * it's state can be discovered. State is used to 
 * control the flow of execution within Oddjob, as
 * well as being a way of informing client applications
 * of progress.
 * 
 * @author Rob Gordon
 */
public interface Stateful {

    /**
     * Add a job state listener.
     * 
     * @param listener The listener.
     * 
     */
    public void addStateListener(StateListener listener) ;

    /**
     * Remove a job state listener.
     * 
     * @param listener The listener.
     */    
    public void removeStateListener(StateListener listener);

    /**
     * Get the last state event.
     * 
     * @return The last State Event
     */
    public StateEvent lastStateEvent();
    
}

package com.s.stat;

import java.io.Serializable;
import java.util.Date;
import java.util.EventObject;

/**
 * An instance of this class is produced when a job state changes. It is
 * passed to all JobStateListeners.
 * 
 * @author Rob Gordon
 */

public class StateEvent extends EventObject 
implements Serializable {

    private static final long serialVersionUID = 20051026;

    static final String REPLACEMENT_EXCEPTION_TEXT = "Exception is not serializable, message is: ";
    
    private State state;
    private Date time;
    private Throwable exception;
    
    /**
     * Used to replace a non serializable exception.
     *
     */
    class ExceptionReplacement extends Exception {
        private static final long serialVersionUID = 20051217;
        public ExceptionReplacement(Throwable replacing) {
            super(REPLACEMENT_EXCEPTION_TEXT + replacing.getMessage());
            super.setStackTrace(exception.getStackTrace());
        }
    }
    
    /**
     * Constructor.
     * 
     * @param job The source of the event.
     * @param jobState The state.
     * @param time the Time of the event.
     * @param exception The exception if applicable, or null otherwise.
     */    
    public StateEvent(Stateful job, State jobState, Date time, Throwable exception) {
        super(job);
        if (jobState == null) {
            throw new NullPointerException("JobState can not be null!");
        }
        this.state = jobState;
        this.time = time;
        this.exception = exception;
    }

    /**
     * Constructor.
     * 
     * @param job The source of the event.
     * @param jobState The state.
     * @param exception The exception if applicable, or null otherwise.
     */
    public StateEvent(Stateful job, State jobState, Throwable exception) {
        this(job, jobState, new Date(), exception);
    }

    /**
     * Constructor.
     * 
     * @param job The source of the event.
     * @param jobState The state.
     */
    public StateEvent(Stateful job, State jobState) {
        this(job, jobState, null);
    }

    @Override
    public Stateful getSource() {
        return (Stateful) super.getSource();
    }
    
    /**
     * Get the job state.
     * 
     * @return The job state.
     */    
    public State getState() {
        return state;    
    }

    /**
     * Get the exception if applicable, null otherwise.
     * 
     * @return The exception of null.
     */    
    public Throwable getException() {
        return exception;
    }    

    /**
     * Get the time of the event..
     * 
     * @return The time.
     */
    public Date getTime() {
        return time;
    }

    /**
     * Override toString.
     */
    public String toString() {
        return "JobStateEvent, source=" + getSource() + ", " + state;
    }
    
    
}

package com.s.stat;



/**
 * Implementors of this interface are able to listen to state events.
 * 
 * @author Rob Gordon
 */

public interface StateListener {


    /**
     * Triggered when the job state changes.
     * 
     * @param event The job state event.
     */    
    public void jobStateChange(StateEvent event);
    
}

package com.s;

import com.s.stat.StateEvent;
import com.s.stat.StateListener;
import com.s.stat.Stateful;


public class Runner implements Runnable{
    public static final String KILLER_TIMEOUT_PROPERTY = "shutdown.killer.timeout";

    public static final long DEFAULT_KILLER_TIMEOUT = 15000L;

    /** Flag if program is being destroyed from the Shutdown Hook. */
    private volatile boolean destroying = false;
    
    /** The killer thread time out. */
    private final long killerTimeout;

    public Runner() {
        String timeoutProperty = System.getProperty(KILLER_TIMEOUT_PROPERTY);
        if (timeoutProperty == null) {
            killerTimeout = DEFAULT_KILLER_TIMEOUT;
        } else {
            killerTimeout = Long.parseLong(timeoutProperty);
        }
    }

    @Override
    public void run() {
        // TODO Auto-generated method stub
        Runtime.getRuntime().addShutdownHook(new ShutdownHook());        
        // Possibly wait for Oddjob to be in a stopped state.
        new StopWait(new Stateful() {
            
            @Override
            public void removeStateListener(StateListener listener) {
                // TODO Auto-generated method stub
                
            }
            
            @Override
            public StateEvent lastStateEvent() {
                // TODO Auto-generated method stub
                return null;
            }
            
            @Override
            public void addStateListener(StateListener listener) {
                // TODO Auto-generated method stub
                
            }
        }, Long.MAX_VALUE).run();
        
        // 调用程序的stop
        
        
    }

    /**
     * shutdown hook.
     * <p>
     * This Class has evolved quite a lot though trial and error due to
     * a lack of understanding of JVM shutdown. Should this thread be a
     * daemon? Current thinking is no because you don't want other daemon
     * threads to terminate until  has been shutdown properly.
     *
     */
    class ShutdownHook extends Thread {
        
        /** Killer thread will forcibly halt  if it hasn't terminated
         * cleanly. */
        private Thread killer;
        
        /*
         * (non-Javadoc)
         * @see java.lang.Thread#run()
         */
        public void run() {

        //    logger.info("Shutdown Hook Executing.");
            
            // killer will just kill process if we can't stop in 15 sec
            killer = new Thread(new Runnable() {
                public void run() {
                    try {
                        Thread.sleep(killerTimeout);
                    }
                    catch (InterruptedException e) {
                    //    logger.debug("Killer thread interrupted and terminating.");
                        return;
                    }
//                    logger.error("Failed to stop Oddjob nicely, using halt(-1)");
                    Runtime.getRuntime().halt(-1);
                }
            });
            
            // start the killer. Not sure it really need to be daemon but
            // it does no harm.
//            logger.debug("Starting killer thread.");
            killer.setDaemon(true);
            killer.start();
            
            // 调用程序的关闭
            
            
            // Nothing's hanging so we don't need our killer.
            //TODO 判断程序是否调用自身的关闭程序，如果成功则单端killer
//            killer.interrupt();
            
        }
    }
}

package com.s;

import java.util.concurrent.BlockingQueue;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.TimeUnit;

import com.s.stat.IsStoppable;
import com.s.stat.State;
import com.s.stat.StateEvent;
import com.s.stat.StateListener;
import com.s.stat.Stateful;

/**
 * A utility class to provide wait until stopped functionality.
 * <p>
 * The default timeout is 5 seconds before a {@link FailedToStopException}
 * is thrown.
 * 
 * @author rob
 *
 */
public class StopWait {

    private final Stateful stateful;
    
    private final long timeout;
    
    /**
     * Constructor with default timeout.
     * 
     * @param stateful The thing to wait until stopped.
     */
    public StopWait(Stateful stateful) {
        this(stateful, 5000);
    }
    
    /**
     * Constructor where timeout can be specified.
     * 
     * @param stateful The thing to wait until stopped.
     * @param timeout The timeout. Note that a timeout of 0 or less is
     * no timeout.
     */
    public StopWait(Stateful stateful, long timeout) {
        this.stateful = stateful;
        this.timeout = timeout;
    }
    
    /**
     * Run the stop wait. This will block until the job stops or the
     * timeout occurs.
     * 
     * @throws FailedToStopException If timeout occurs.
     */
    public void run()  {        

        if (new IsStoppable().test(
                stateful.lastStateEvent().getState())) {
            doWait();
        }
    }
    
    private void doWait() {        
        
        final BlockingQueue<State> handoff = new LinkedBlockingQueue<State>();
        
        class StopListener implements StateListener {
            
            @Override
            public void jobStateChange(StateEvent event) {
                handoff.add(event.getState());
            }
        };
        
        StopListener listener = new StopListener();
                
        stateful.addStateListener(listener);
        
        try {
            while (true) {

                State state = handoff.poll(timeout, TimeUnit.MILLISECONDS);
                if (state == null) {
                }
                if (!state.isStoppable()) {
                    return;
                }
            }
        }
        catch (InterruptedException e) {
            Thread.currentThread().interrupt();
        }
        finally {
            stateful.removeStateListener(listener);
        }
    }    
}