org.apache.tika.batch.BatchProcess Maven / Gradle / Ivy
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.tika.batch;
import static java.nio.charset.StandardCharsets.UTF_8;
import java.io.IOException;
import java.io.PrintStream;
import java.util.List;
import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.Callable;
import java.util.concurrent.CompletionService;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorCompletionService;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
* This is the main processor class for a single process.
* This class can only be run once.
*
* It requires a {@link FileResourceCrawler} and {@link FileResourceConsumer}s, and it can also
* support a {@link StatusReporter} and an {@link Interrupter}.
*
* This is designed to shutdown if a parser has timed out or if there is
* an OutOfMemoryError. Consider using {@link BatchProcessDriverCLI}
* as a daemon/watchdog that monitors and can restart this batch process;
*
* Note that this classs redirects stderr to stdout so that it can
* communicate without interference with the parent process on stderr.
*/
public class BatchProcess implements Callable {
private static final Logger LOG = LoggerFactory.getLogger(BatchProcess.class);
private final long consumersManagerMaxMillis;
private final FileResourceCrawler fileResourceCrawler;
private final ConsumersManager consumersManager;
private final StatusReporter reporter;
private final Interrupter interrupter;
private final ArrayBlockingQueue timedOuts;
private PrintStream outputStreamWriter;
// If a file hasn't been processed in this amount of time,
// report it to the console. When the directory crawler has stopped, the thread will
// be terminated and the file name will be logged
private long timeoutThresholdMillis = 5 * 60 * 1000; // 5 minutes
private long timeoutCheckPulseMillis = 2 * 60 * 1000; //2 minutes
//if there was an early termination via the Interrupter
//or because of an uncaught runtime throwable, pause
//this long before shutting down to allow parsers to finish
private long pauseOnEarlyTerminationMillis = 30 * 1000; //30 seconds
//maximum time that this process should stay alive
//to avoid potential memory leaks, not a bad idea to shutdown
//every hour or so.
private int maxAliveTimeSeconds = -1;
private boolean alreadyExecuted = false;
public BatchProcess(FileResourceCrawler fileResourceCrawler, ConsumersManager consumersManager, StatusReporter reporter, Interrupter interrupter) {
this.fileResourceCrawler = fileResourceCrawler;
this.consumersManager = consumersManager;
this.reporter = reporter;
this.interrupter = interrupter;
timedOuts = new ArrayBlockingQueue<>(consumersManager
.getConsumers()
.size());
this.consumersManagerMaxMillis = consumersManager.getConsumersManagerMaxMillis();
}
/**
* Runs main execution loop.
*
* Redirects stdout to stderr to keep clean communications
* over stdout with parent process
*
* @return result of the processing
* @throws InterruptedException
*/
public ParallelFileProcessingResult call() throws InterruptedException {
if (alreadyExecuted) {
throw new IllegalStateException("Can only execute BatchRunner once.");
}
//redirect streams; all organic warnings should go to System.err;
//System.err should be redirected to System.out
PrintStream sysErr = System.err;
try {
outputStreamWriter = new PrintStream(sysErr, true, UTF_8.toString());
} catch (IOException e) {
throw new RuntimeException("Can't redirect streams");
}
System.setErr(System.out);
ParallelFileProcessingResult result = null;
try {
int numConsumers = consumersManager
.getConsumers()
.size();
// fileResourceCrawler, statusReporter, the Interrupter, timeoutChecker
int numNonConsumers = 2;
if (interrupter != null) {
numNonConsumers++;
}
if (reporter != null) {
numNonConsumers++;
}
ExecutorService ex = Executors.newFixedThreadPool(numConsumers + numNonConsumers);
CompletionService completionService = new ExecutorCompletionService<>(ex);
TimeoutChecker timeoutChecker = new TimeoutChecker();
try {
startConsumersManager();
} catch (BatchNoRestartError e) {
return new ParallelFileProcessingResult(0, 0, 0, 0, 0, BatchProcessDriverCLI.PROCESS_NO_RESTART_EXIT_CODE,
CAUSE_FOR_TERMINATION.CONSUMERS_MANAGER_DIDNT_INIT_IN_TIME_NO_RESTART.toString());
}
State state = mainLoop(completionService, timeoutChecker);
result = shutdown(ex, completionService, timeoutChecker, state);
} finally {
shutdownConsumersManager();
}
LOG.trace("finishing up");
return result;
}
private State mainLoop(CompletionService completionService, TimeoutChecker timeoutChecker) {
alreadyExecuted = true;
State state = new State();
LOG.info("BatchProcess starting up");
state.start = System.currentTimeMillis();
if (interrupter != null) {
completionService.submit(interrupter);
}
if (reporter != null) {
completionService.submit(reporter);
}
completionService.submit(fileResourceCrawler);
completionService.submit(timeoutChecker);
for (FileResourceConsumer consumer : consumersManager.getConsumers()) {
completionService.submit(consumer);
}
state.numConsumers = consumersManager
.getConsumers()
.size();
CAUSE_FOR_TERMINATION causeForTermination = null;
//main processing loop
while (true) {
try {
Future futureResult = completionService.poll(1, TimeUnit.SECONDS);
if (futureResult != null) {
state.removed++;
IFileProcessorFutureResult result = futureResult.get();
LOG.trace("result: " + result);
if (result instanceof FileConsumerFutureResult) {
state.consumersRemoved++;
} else if (result instanceof FileResourceCrawlerFutureResult) {
state.crawlersRemoved++;
if (fileResourceCrawler.wasTimedOut()) {
causeForTermination = CAUSE_FOR_TERMINATION.CRAWLER_TIMED_OUT;
break;
}
} else if (result instanceof InterrupterFutureResult) {
causeForTermination = CAUSE_FOR_TERMINATION.PARENT_SHUTDOWN;
break;
} else if (result instanceof TimeoutFutureResult) {
causeForTermination = CAUSE_FOR_TERMINATION.TIMED_OUT_CONSUMER;
break;
} //only thing left should be StatusReporterResult
}
if (state.consumersRemoved >= state.numConsumers) {
causeForTermination = CAUSE_FOR_TERMINATION.COMPLETED_NORMALLY;
break;
}
if (aliveTooLong(state.start)) {
causeForTermination = CAUSE_FOR_TERMINATION.BATCH_PROCESS_ALIVE_TOO_LONG;
break;
}
} catch (Throwable e) {
if (isNonRestart(e)) {
causeForTermination = CAUSE_FOR_TERMINATION.MAIN_LOOP_EXCEPTION_NO_RESTART;
} else {
causeForTermination = CAUSE_FOR_TERMINATION.MAIN_LOOP_EXCEPTION;
}
LOG.error("Main loop execution exception: {}", e.getMessage(), e);
break;
}
}
state.causeForTermination = causeForTermination;
return state;
}
private ParallelFileProcessingResult shutdown(ExecutorService ex, CompletionService completionService, TimeoutChecker timeoutChecker, State state) {
if (reporter != null) {
reporter.setIsShuttingDown(true);
}
int added = fileResourceCrawler.getAdded();
int considered = fileResourceCrawler.getConsidered();
//TODO: figure out safe way to shutdown resource crawler
//if it isn't. Does it need to add poison at this point?
//fileResourceCrawler.pleaseShutdown();
LOG.trace("about to shutdown");
//Step 1: prevent uncalled threads from being started
ex.shutdown();
//Step 2: ask consumers to shutdown politely.
//Under normal circumstances, they should all have completed by now.
for (FileResourceConsumer consumer : consumersManager.getConsumers()) {
consumer.pleaseShutdown();
}
//The resourceCrawler should shutdown now. No need for poison.
fileResourceCrawler.shutDownNoPoison();
//if there are any active/asked-to-shutdown consumers, wait
//a bit for those parsers to finish.
//This can happen if the parent process dies
//of if the crawler stops early, or ...
politelyAwaitTermination(state.causeForTermination);
//Step 3: Gloves come off. We've tried to ask kindly before.
//Now it is time shut down. This will corrupt
//nio channels via thread interrupts! Hopefully, everything
//has shut down by now.
LOG.trace("About to shutdownNow()");
List neverCalled = ex.shutdownNow();
LOG.trace("TERMINATED {} : {} : {}", ex.isTerminated(), state.consumersRemoved, state.crawlersRemoved);
int end = state.numConsumers + state.numNonConsumers - state.removed - neverCalled.size();
for (int t = 0; t < end; t++) {
Future future = null;
try {
future = completionService.poll(10, TimeUnit.MILLISECONDS);
} catch (InterruptedException e) {
LOG.warn("thread interrupt while polling in final shutdown loop");
break;
}
LOG.trace("In while future==null loop in final shutdown loop");
if (future == null) {
break;
}
try {
IFileProcessorFutureResult result = future.get();
if (result instanceof FileConsumerFutureResult) {
FileConsumerFutureResult consumerResult = (FileConsumerFutureResult) result;
FileStarted fileStarted = consumerResult.getFileStarted();
LOG.trace("file started " + fileStarted);
if (fileStarted != null && fileStarted.getElapsedMillis() > timeoutThresholdMillis) {
LOG.warn("{} caused a file processor to hang or crash. You may need to remove " + "this file from your input set and rerun.", fileStarted.getResourceId());
}
} else if (result instanceof FileResourceCrawlerFutureResult) {
FileResourceCrawlerFutureResult crawlerResult = (FileResourceCrawlerFutureResult) result;
considered += crawlerResult.getConsidered();
added += crawlerResult.getAdded();
} //else ...we don't care about anything else stopping at this point
} catch (ExecutionException e) {
LOG.error("Execution exception trying to shutdown after shutdownNow", e);
} catch (InterruptedException e) {
LOG.error("Interrupted exception trying to shutdown after shutdownNow", e);
}
}
//do we need to restart?
String restartMsg = null;
if (state.causeForTermination == CAUSE_FOR_TERMINATION.PARENT_SHUTDOWN || state.causeForTermination == CAUSE_FOR_TERMINATION.MAIN_LOOP_EXCEPTION_NO_RESTART) {
//do not restart!!!
} else if (state.causeForTermination == CAUSE_FOR_TERMINATION.MAIN_LOOP_EXCEPTION) {
restartMsg = "Uncaught consumer throwable";
} else if (state.causeForTermination == CAUSE_FOR_TERMINATION.TIMED_OUT_CONSUMER) {
if (areResourcesPotentiallyRemaining()) {
restartMsg = "Consumer timed out with resources remaining";
}
} else if (state.causeForTermination == CAUSE_FOR_TERMINATION.BATCH_PROCESS_ALIVE_TOO_LONG) {
restartMsg = BATCH_CONSTANTS.BATCH_PROCESS_EXCEEDED_MAX_ALIVE_TIME.toString();
} else if (state.causeForTermination == CAUSE_FOR_TERMINATION.CRAWLER_TIMED_OUT) {
restartMsg = "Crawler timed out.";
} else if (fileResourceCrawler.wasTimedOut()) {
restartMsg = "Crawler was timed out.";
} else if (fileResourceCrawler.isActive()) {
restartMsg = "Crawler is still active.";
} else if (!fileResourceCrawler.isQueueEmpty()) {
restartMsg = "Resources still exist for processing";
}
LOG.trace("restart msg: " + restartMsg);
int exitStatus = getExitStatus(state.causeForTermination, restartMsg);
//need to re-check, report, mark timed out consumers
timeoutChecker.checkForTimedOutConsumers();
for (FileStarted fs : timedOuts) {
LOG.warn("A parser was still working on >{}< for {} milliseconds after it started. " + "This exceeds the maxTimeoutMillis parameter", fs.getResourceId(),
fs.getElapsedMillis());
}
double elapsed = ((double) System.currentTimeMillis() - (double) state.start) / 1000.0;
int processed = 0;
int numExceptions = 0;
for (FileResourceConsumer c : consumersManager.getConsumers()) {
processed += c.getNumResourcesConsumed();
numExceptions += c.getNumHandledExceptions();
}
LOG.trace("returning " + state.causeForTermination);
return new ParallelFileProcessingResult(considered, added, processed, numExceptions, elapsed, exitStatus, state.causeForTermination.toString());
}
private void startConsumersManager() {
if (consumersManagerMaxMillis < 0) {
consumersManager.init();
return;
}
Thread timed = new Thread(() -> {
LOG.trace("about to start consumers manager");
consumersManager.init();
LOG.trace("finished starting consumers manager");
});
//don't allow this thread to keep process alive
timed.setDaemon(true);
timed.start();
try {
timed.join(consumersManagerMaxMillis);
} catch (InterruptedException e) {
LOG.warn("interruption exception during consumers manager shutdown");
}
if (timed.isAlive()) {
LOG.error("ConsumersManager did not start within {}ms", consumersManagerMaxMillis);
throw new BatchNoRestartError("ConsumersManager did not start within " + consumersManagerMaxMillis + "ms");
}
}
private void shutdownConsumersManager() {
if (consumersManagerMaxMillis < 0) {
consumersManager.shutdown();
return;
}
Thread timed = new Thread(() -> {
LOG.trace("starting to shutdown consumers manager");
consumersManager.shutdown();
LOG.trace("finished shutting down consumers manager");
});
timed.setDaemon(true);
timed.start();
try {
timed.join(consumersManagerMaxMillis);
} catch (InterruptedException e) {
LOG.warn("interruption exception during consumers manager shutdown");
}
if (timed.isAlive()) {
LOG.error("ConsumersManager was still alive during shutdown!");
throw new BatchNoRestartError("ConsumersManager did not shutdown within: " + consumersManagerMaxMillis + "ms");
}
}
/**
* This is used instead of awaitTermination(), because that interrupts
* the thread and then waits for its termination. This politely waits.
*
* @param causeForTermination reason for termination.
*/
private void politelyAwaitTermination(CAUSE_FOR_TERMINATION causeForTermination) {
if (causeForTermination == CAUSE_FOR_TERMINATION.COMPLETED_NORMALLY) {
return;
}
long start = System.currentTimeMillis();
while (countActiveConsumers() > 0) {
try {
Thread.sleep(500);
} catch (InterruptedException e) {
LOG.warn("Thread interrupted while trying to politelyAwaitTermination");
return;
}
long elapsed = System.currentTimeMillis() - start;
if (pauseOnEarlyTerminationMillis > -1 && elapsed > pauseOnEarlyTerminationMillis) {
LOG.warn("Waited after an early termination for {}ms, but there was at least one active consumer", elapsed);
return;
}
}
}
private boolean isNonRestart(Throwable e) {
if (e instanceof BatchNoRestartError) {
return true;
}
Throwable cause = e.getCause();
return cause != null && isNonRestart(cause);
}
private int getExitStatus(CAUSE_FOR_TERMINATION causeForTermination, String restartMsg) {
if (causeForTermination == CAUSE_FOR_TERMINATION.MAIN_LOOP_EXCEPTION_NO_RESTART) {
LOG.info(CAUSE_FOR_TERMINATION.MAIN_LOOP_EXCEPTION_NO_RESTART.name());
return BatchProcessDriverCLI.PROCESS_NO_RESTART_EXIT_CODE;
}
if (restartMsg != null) {
if (restartMsg.equals(BATCH_CONSTANTS.BATCH_PROCESS_EXCEEDED_MAX_ALIVE_TIME.toString())) {
LOG.warn(restartMsg);
} else {
LOG.error(restartMsg);
}
//send over stdout wrapped in outputStreamWriter
outputStreamWriter.println(BATCH_CONSTANTS.BATCH_PROCESS_FATAL_MUST_RESTART.toString() + " >> " + restartMsg);
outputStreamWriter.flush();
return BatchProcessDriverCLI.PROCESS_RESTART_EXIT_CODE;
}
return 0;
}
//could new FileResources be consumed from the Queue?
//Because of race conditions, this can return a true
//when the real answer is false.
//This should never return false, though, if the answer is true!
private boolean areResourcesPotentiallyRemaining() {
if (fileResourceCrawler.isActive()) {
return true;
}
return !fileResourceCrawler.isQueueEmpty();
}
private boolean aliveTooLong(long started) {
if (maxAliveTimeSeconds < 0) {
return false;
}
double elapsedSeconds = (double) (System.currentTimeMillis() - started) / (double) 1000;
return elapsedSeconds > (double) maxAliveTimeSeconds;
}
//snapshot of non-retired consumers; actual number may be smaller by the time
//this returns a value!
private int countActiveConsumers() {
int active = 0;
for (FileResourceConsumer consumer : consumersManager.getConsumers()) {
if (consumer.isStillActive()) {
active++;
}
}
return active;
}
/**
* If there is an early termination via an interrupt or too many timed out consumers
* or because a consumer or other Runnable threw a Throwable, pause this long
* before interrupting the consumers and other threads.
*
* Typically makes sense for this to be the same or slightly larger than
* timeoutThresholdMillis
*
* @param pauseOnEarlyTerminationMillis how long to pause if there is an early termination
*/
public void setPauseOnEarlyTerminationMillis(long pauseOnEarlyTerminationMillis) {
this.pauseOnEarlyTerminationMillis = pauseOnEarlyTerminationMillis;
}
/**
* The amount of time allowed before a consumer should be timed out.
*
* @param timeoutThresholdMillis threshold in milliseconds before declaring a consumer timed out
*/
public void setTimeoutThresholdMillis(long timeoutThresholdMillis) {
this.timeoutThresholdMillis = timeoutThresholdMillis;
}
public void setTimeoutCheckPulseMillis(long timeoutCheckPulseMillis) {
this.timeoutCheckPulseMillis = timeoutCheckPulseMillis;
}
/**
* The maximum amount of time that this process can be alive. To avoid
* memory leaks, it is sometimes beneficial to shutdown (and restart) the
* process periodically.
*
* If the value is < 0, the process will run until completion, interruption or exception.
*
* @param maxAliveTimeSeconds maximum amount of time in seconds to remain alive
*/
public void setMaxAliveTimeSeconds(int maxAliveTimeSeconds) {
this.maxAliveTimeSeconds = maxAliveTimeSeconds;
}
public enum BATCH_CONSTANTS {
BATCH_PROCESS_EXCEEDED_MAX_ALIVE_TIME, BATCH_PROCESS_FATAL_MUST_RESTART
}
private enum CAUSE_FOR_TERMINATION {
COMPLETED_NORMALLY, MAIN_LOOP_EXCEPTION_NO_RESTART, CONSUMERS_MANAGER_DIDNT_INIT_IN_TIME_NO_RESTART, MAIN_LOOP_EXCEPTION, CRAWLER_TIMED_OUT, TIMED_OUT_CONSUMER,
PARENT_SHUTDOWN, BATCH_PROCESS_ALIVE_TOO_LONG,
}
private static class State {
long start = -1;
int numConsumers = 0;
int numNonConsumers = 0;
int removed = 0;
int consumersRemoved = 0;
int crawlersRemoved = 0;
CAUSE_FOR_TERMINATION causeForTermination = null;
}
private static class TimeoutFutureResult implements IFileProcessorFutureResult {
//used to be used when more than one timeout was allowed
//TODO: get rid of this?
private final int timedOutCount;
private TimeoutFutureResult(final int timedOutCount) {
this.timedOutCount = timedOutCount;
}
protected int getTimedOutCount() {
return timedOutCount;
}
}
private class TimeoutChecker implements Callable {
@Override
public TimeoutFutureResult call() throws Exception {
while (timedOuts.size() == 0) {
try {
Thread.sleep(timeoutCheckPulseMillis);
} catch (InterruptedException e) {
LOG.debug("Thread interrupted exception in TimeoutChecker");
break;
//just stop.
}
checkForTimedOutConsumers();
if (countActiveConsumers() == 0) {
LOG.info("No activeConsumers in TimeoutChecker");
break;
}
}
LOG.debug("TimeoutChecker quitting: {}", timedOuts.size());
return new TimeoutFutureResult(timedOuts.size());
}
private void checkForTimedOutConsumers() {
for (FileResourceConsumer consumer : consumersManager.getConsumers()) {
FileStarted fs = consumer.checkForTimedOutMillis(timeoutThresholdMillis);
if (fs != null) {
timedOuts.add(fs);
}
}
}
}
}