设置ShutdownHook并退出应用程序

Question

我有以下代码：

public static void main(String[] args) {

    // login event
    String event = "login";
    System.out.printf("Handling event: %s %s\n",event,getCurrentLogin());
    sendMessage(event, getCurrentLogin());

    // logout or shutdown event
    Runtime.getRuntime().addShutdownHook(new Thread(new Runnable() {
        @Override
        public void run() {
                String event = "logout";
                System.out.printf("Handling event: %s %s\n",event,getCurrentLogin());
                sendMessage(event, getCurrentLogin());
        }
    }));
 }

这是一个非常简单的程序，用于记录用户的登录和注销。问题是程序在到达函数 main() 末尾时退出。

我是否正确使用了关闭事件挂钩？ 我不想创建一个复杂的 Windows 服务，它必须是一个非常简单的应用程序，因为它将用于远程连接的 Windows 会话。

您对后台等待登录终止有什么建议吗？

Answer 1

ShutdownHook 是一个线程，当 JVM 退出并且程序在执行“sendMessage”后没有执行任何操作时执行：

sendMessage(event, getCurrentLogin());

您应该等待退出信号，例如控制台上的 Ctrl+C。
只需等待锁或信号量来避免程序终止，并且当您需要完成它时应该释放该锁。

Answer 2

我已在您的示例中添加了一个闩锁，以尝试使该示例正常工作。不幸的是，我没有 IDE 来测试，但这应该足以提供一个想法。

编辑：下面的示例将打印登录和注销。

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

    public class Test {
        public static void main(String[] args) throws Exception {

            // login event
            String event = "login";
            System.out.printf("Handling event: %s %s\n", event, getCurrentLogin());
            sendMessage(event, getCurrentLogin());
            final CountDownLatch latch = new CountDownLatch(1);

            // logout or shutdown event
            Runtime.getRuntime().addShutdownHook(new Thread(new Runnable() {
                @Override
                public void run() {
                    String event = "logout";
                    System.out.printf("Handling event: %s %s\n", event, getCurrentLogin());
                    sendMessage(event, getCurrentLogin());
                    latch.countDown();
                }
            }));
            latch.await(2, TimeUnit.SECONDS);
        }

        private static void sendMessage(String event, Object currentLogin) {
        }

        private static Object getCurrentLogin() {
            return "X";
        }
    }

Answer 3

如果我理解正确的话，您想无限等待直到操作系统停止您的程序，对吗？如果是这样，您需要至少运行一个守护线程：

public static void main(String[] args) throws InterruptedException {
    //...
    final Object o = new Object();
    synchronized(o) {
        while(true) {
            o.wait();
        }
    }
}

我不确定操作系统是否会足够温和地终止 JVM，但关闭挂钩是否会执行。测试一下。

Answer 4

当应用程序完成时，关闭钩子被调用。因此，如果你想让你的应用程序永远运行，你需要实现一些东西：

while (true) {
    //Your logic here
}

如果你想编写一个守护进程或调度程序，也许你需要使用一些框架，如 Quartz http://quartz-scheduler.org

如果你想实现一个守护进程，你可以使用像 http://yajsw.sourceforge.net 或 http://wrapper.tanukisoftware。 com/

Answer 5

这里有一些很好的答案，但我仍然看到一些缺失的信息。

您的代码在 main 中注册一个关闭钩子，然后 main 到达末尾并返回。当最后一个非守护线程退出时，关闭钩子被调用。我怀疑 main 是唯一的非守护线程，因此当 main 退出时，会立即调用钩子。

如果您不希望程序立即退出，则 main 将必须循环并等待某种信号。从您问题的上下文来看，我想它应该监听来自 sendMessage 调用的某种响应？如果你想等待 control-c 那么我喜欢 @JB Nizet 的回答：

public static void main(String[] args) throws InterruptedException {
    ...
    Runtime.getRuntime().addShutdownHook(...);
    // wait until control-c is called
    final Object o = new Object();
    synchronized(o) {
        while(true) {
            o.wait();
        }
    }
}

你想在注册关闭钩子之后开始等待，以便执行关闭代码。

Answer 6

这是一种使用 CountDownLatch< /code>让主线程保持等待：

public static void main(String[] args) {

    // login event
    String event = "login";
    System.out.printf("Handling event: %s %s\n",event,getCurrentLogin());
    sendMessage(event, getCurrentLogin());

    final CountDownLatch latch = new CountDownLatch(1);

    // logout or shutdown event
    Runtime.getRuntime().addShutdownHook(new Thread(new Runnable() {
        @Override
        public void run() {
                String event = "logout";
                System.out.printf("Handling event: %s %s\n",event,getCurrentLogin());
                sendMessage(event, getCurrentLogin());
                latch.countDown();
        }
    }));

    latch.await();
 }