ExecutorService 종료 마스터링: ThreadPool 종료 추적

어떤 작업을 실행하고 싶다고 가정해 보겠습니다. 단일 스레드를 통해 실행하면 결과를 얻는 데 상당한 시간이 걸릴 수 있으므로 신뢰할 수 있는 ExecutorService를 사용하여 여러 스레드를 통해 이를 처리하기로 결정합니다.

샘플은 다음과 같습니다.

public static void main(String[] args) {
    ExecutorService executorService = Executors.newFixedThreadPool(5);
    for (int i = 0; i < 5; i++) {
        int temp = i;
        executorService.submit(() -> {
            task(temp);
        });
    }
    executorService.shutdown();
    System.out.println("ExecutorService is shutdown");
}

private static void task(int temp) {
    try {
        TimeUnit.SECONDS.sleep(1L);
        System.out.println("Task " + temp + " completed");
    } catch (InterruptedException e) {
        throw new RuntimeException(e);
    }
}

물론 평소와 같이 "sleep"을 작업 실행의 원형으로 사용하지 않고는 스레드의 어떤 예도 완성되지 않습니다.

출력됩니다.

ExecutorService is shutdown
Task 1 completed
Task 2 completed
Task 0 completed
Task 4 completed
Task 3 completed

이제 자신도 모르는 작업의 끝없는 대기열이 있다고 상상해 보세요. 아마도 동적으로 추가되는 데이터베이스의 항목 수에 따라 결정될 수도 있습니다.

예를 들어 하루 종일 수많은 거래를 처리해야 하는 은행이 있습니다. 거래 종료 시간은 오후 5시이며, 그 이후에는 추가 작업을 받지 않습니다.

그러나 작업 수는 한정되어 있으며 특정 시점에 종료됩니다.

모든 작업이 완료되는 시점을 어떻게 알 수 있나요?

위의 코드 조각을 보면 ExecutorService.shutdown()을 사용하여 기본 스레드를 즉시 종료할 수 있지만 백그라운드 스레드는 여전히 허용된 작업을 완료까지 처리합니다. 완료 알림을 받을 수 있는 방법이 있나요?

두 가지 솔루션이 떠오릅니다.

1. CountDownLatch를 사용하여 작업 수를 계산하지만 작업 수를 모르기 때문에 사용하는 것이 비현실적입니다.
2. ExecutorService.awaitTermination()을 사용하세요. 그러나 여기의 시간은 아직 결정적이지 않습니다. 매우 자유로운 ExecutorService.awaitTermination(Long.MAX_VALUE, TimeUnit.DAYS) 또는 이와 유사한 것을 사용할 수 있습니다. 그런데 그게 또 차단전화네요.

이 문제를 해결할 수 있는 더 좋은 방법이 있나요?

Java는 이 문제를 해결할 수 있는 더 좋고 상대적으로 알려지지 않은 방법을 제공합니다. 여기서 "비결"은 Executors.newFixedThreadPool이 본질적으로 사전 정의된 값을 가진 ThreadPoolExecutor라는 것을 아는 것입니다. Executors.newFixedThreadPool의 구현을 확인해 보겠습니다.

public static ExecutorService newFixedThreadPool(int nThreads) {
    return new ThreadPoolExecutor(nThreads, nThreads,
                                  0L, TimeUnit.MILLISECONDS,
                                  new LinkedBlockingQueue<Runnable>());
}

여기에서 ThreadPoolExecutor에 대한 문서를 읽어 보시기 바랍니다. ExecutorService는 ThreadPoolExecutor에 대한 편리한 래퍼입니다.

... 프로그래머는 보다 편리한 Executors 팩토리 메소드 Executors.newCachedThreadPool()(자동 스레드 회수 기능이 있는 무제한 스레드 풀), Executors.newFixedThreadPool(int)(고정 크기 스레드 풀) 및 Executors.newSingleThreadExecutor( )(단일 백그라운드 스레드), 가장 일반적인 사용 시나리오에 대한 설정을 사전 구성합니다.

문제 해결에 도움이 되는 섹션은 다음과 같습니다.

후크 방식

이 클래스는 각 작업 실행 전후에 호출되는 재정의 가능한 beforeExecute(Thread, Runnable) 및 afterExecute(Runnable, Throwable) 메서드를 제공합니다. 이는 실행 환경을 조작하는 데 사용될 수 있습니다. 예를 들어 ThreadLocals 다시 초기화, 통계 수집 또는 로그 항목 추가 등이 있습니다. 또한, Executor가 완전히 종료된 후 수행되어야 하는 특수 처리를 수행하기 위해 종료() 메소드를 재정의할 수 있습니다.

해지 방법을 사용하여 동일한 내용을 통보할 수 있습니다! 그런데 어떻게 사용하나요?

public static void main(String[] args) {
    ExecutorService executorService = Executors.newFixedThreadPool(5);
    for (int i = 0; i < 5; i++) {
        int temp = i;
        executorService.submit(() -> {
            task(temp);
        });
    }
    executorService.shutdown();
    System.out.println("ExecutorService is shutdown");
}

private static void task(int temp) {
    try {
        TimeUnit.SECONDS.sleep(1L);
        System.out.println("Task " + temp + " completed");
    } catch (InterruptedException e) {
        throw new RuntimeException(e);
    }
}

나처럼 익명 클래스를 선호하지 않는 경우 언제든지 ThreadPoolExecutor를 직접 확장하여 사용자 정의 클래스를 만들 수 있습니다.

ExecutorService is shutdown
Task 1 completed
Task 2 completed
Task 0 completed
Task 4 completed
Task 3 completed

예상대로 작동하는지 확인하기 위한 출력은 다음과 같습니다.

public static ExecutorService newFixedThreadPool(int nThreads) {
    return new ThreadPoolExecutor(nThreads, nThreads,
                                  0L, TimeUnit.MILLISECONDS,
                                  new LinkedBlockingQueue<Runnable>());
}

당신이 사용하는 상대적으로 알려지지 않은 다른 스니펫은 무엇입니까? 댓글로 알려주세요!

위 내용은 ExecutorService 종료 마스터링: ThreadPool 종료 추적의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!