Полезный пример ловушки выключения в Java?

126

Я пытаюсь убедиться, что мое приложение Java принимает разумные меры для обеспечения надежности, и частично это связано с корректным завершением работы. Я читаю о хуках завершения работы и не понимаю, как их использовать на практике.

Есть ли там практический пример?

Допустим, у меня было действительно простое приложение, подобное приведенному ниже, которое записывает числа в файл, от 10 до строки, партиями по 100 штук, и я хочу убедиться, что данная партия завершится, если программа будет прервана. Я понимаю, как зарегистрировать ловушку выключения, но понятия не имею, как интегрировать ее в свое приложение. Какие-либо предложения?

package com.example.test.concurrency;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.PrintWriter;

public class GracefulShutdownTest1 {
    final private int N;
    final private File f;
    public GracefulShutdownTest1(File f, int N) { this.f=f; this.N = N; }

    public void run()
    {
        PrintWriter pw = null;
        try {
            FileOutputStream fos = new FileOutputStream(this.f);
            pw = new PrintWriter(fos);
            for (int i = 0; i < N; ++i)
                writeBatch(pw, i);
        }
        catch (FileNotFoundException e) {
            e.printStackTrace();
        }
        finally
        {
            pw.close();
        }       
    }

    private void writeBatch(PrintWriter pw, int i) {
        for (int j = 0; j < 100; ++j)
        {
            int k = i*100+j;
            pw.write(Integer.toString(k));
            if ((j+1)%10 == 0)
                pw.write('\n');
            else
                pw.write(' ');
        }
    }

    static public void main(String[] args)
    {
        if (args.length < 2)
        {
            System.out.println("args = [file] [N] "
                    +"where file = output filename, N=batch count");
        }
        else
        {
            new GracefulShutdownTest1(
                    new File(args[0]), 
                    Integer.parseInt(args[1])
            ).run();
        }
    }
}
Джейсон С
источник
См. Также stackoverflow.com/questions/8722826/…
flow2k

Ответы:

160

Вы могли сделать следующее:

  • Пусть обработчик выключения устанавливает для некоторого AtomicBoolean (или изменчивого логического) "keepRunning" значение false.
  • (Необязательно, .interruptрабочие потоки, если они ждут данных в каком-то блокирующем вызове)
  • Дождитесь завершения рабочих потоков (выполняемых writeBatchв вашем случае), вызвав Thread.join()метод в рабочих потоках.
  • Завершить программу

Какой-то схематичный код:

  • Добавить static volatile boolean keepRunning = true;
  • В run () вы меняете на

    for (int i = 0; i < N && keepRunning; ++i)
        writeBatch(pw, i);
  • В main () вы добавляете:

    final Thread mainThread = Thread.currentThread();
    Runtime.getRuntime().addShutdownHook(new Thread() {
        public void run() {
            keepRunning = false;
            mainThread.join();
        }
    });

Примерно так я изящно «отклоняю всех клиентов при нажатии Control-C» в терминале.


Из документов :

Когда виртуальная машина начинает свою последовательность выключения, она запускает все зарегистрированные перехватчики выключения в некотором неопределенном порядке и позволяет им работать одновременно. Когда все перехватчики завершены, он запустит все неактивированные финализаторы, если финализация-при-выходе была включена. Наконец, виртуальная машина остановится.

То есть ловушка выключения поддерживает работу JVM до тех пор, пока ловушка не будет завершена (возвращена из метода run () .

aioobe
источник
14
Ах, так что, если я правильно понимаю, суть того, что вы говорите, заключается в том, что ловушка выключения имеет возможность взаимодействовать с другими запущенными в данный момент потоками и удерживать выключение виртуальной машины (например, с помощью Thread.join () ), пока не убедится, что быстрая очистка завершена. Это то, чего я упустил. Спасибо!
Джейсон С.
2
Да. Ты понял. Обновил ответ несколькими предложениями из документов.
aioobe
1
Заранее извините, но разве GracefulShutdownTest1 не должен реализовывать Runnable?
Виктор
Будет ли это работать, если графический интерфейс был запущен в основном потоке? Я знаю, что это не лучший способ (вместо этого используйте EDT), но я работаю над старым кодом.
Agostino
1
@ MartynasJusevičius, JVM завершает работу, когда завершаются все потоки, не являющиеся демонами. Если mainThreadэто поток демона, то joinмы гарантируем, что мы дождемся его завершения, прежде чем позволить JVM завершиться.
aioobe
-9

Обработчики выключения - это незапущенные потоки, которые зарегистрированы с помощью Runtime.addShutdownHook (). JVM не дает никаких гарантий относительно порядка, в котором запускаются обработчики выключения. Для получения дополнительной информации см. Http://techno-terminal.blogspot.in/2015/08 /shutdown-hooks.html

Сатиш
источник