4545import com .evolvedbinary .j8fu .function .SupplierE ;
4646import org .exist .util .sanity .SanityCheck ;
4747
48+ import javax .annotation .Nullable ;
49+
4850/**
4951 * Database recovery. This class is used once during startup to check
5052 * if the database is in a consistent state. If not, the class attempts to recover
@@ -59,11 +61,13 @@ public class RecoveryManager {
5961 private final DBBroker broker ;
6062 private final JournalRecoveryAccessor journalRecovery ;
6163 private final boolean restartOnError ;
64+ private final boolean hideProgressBar ;
6265
6366 public RecoveryManager (final DBBroker broker , final JournalManager journalManager , final boolean restartOnError ) {
6467 this .broker = broker ;
6568 this .journalRecovery = journalManager .getRecoveryAccessor (this );
6669 this .restartOnError = restartOnError ;
70+ this .hideProgressBar = Boolean .getBoolean ("exist.recovery.progressbar.hide" );
6771 }
6872
6973 /**
@@ -120,10 +124,10 @@ public boolean recover() throws LogException {
120124 Lsn lastLsn = Lsn .LSN_INVALID ;
121125 Loggable next ;
122126 try {
123- final ProgressBar progress = new ProgressBar ("Scanning journal " , FileUtils .sizeQuietly (last ));
127+ final long lastSize = FileUtils .sizeQuietly (last );
128+ @ Nullable final ProgressBar scanProgressBar = hideProgressBar ? null : new ProgressBar ("Scanning journal " , lastSize );
124129 while ((next = reader .nextEntry ()) != null ) {
125130// LOG.debug(next.dump());
126- progress .set (next .getLsn ().getOffset ());
127131 if (next .getLogType () == LogEntryTypes .TXN_START ) {
128132 // new transaction starts: add it to the transactions table
129133 txnsStarted .put (next .getTransactionId (), next );
@@ -135,6 +139,14 @@ public boolean recover() throws LogException {
135139 lastCheckpoint = (Checkpoint ) next ;
136140 }
137141 lastLsn = next .getLsn ();
142+
143+ if (scanProgressBar != null ) {
144+ scanProgressBar .set (next .getLsn ().getOffset ());
145+ }
146+ }
147+
148+ if (scanProgressBar != null ) {
149+ scanProgressBar .set (lastSize ); // 100%
138150 }
139151 } catch (final LogException e ) {
140152 if (LOG .isDebugEnabled ()) {
@@ -146,7 +158,7 @@ public boolean recover() throws LogException {
146158 // if the last checkpoint record is not the last record in the file
147159 // we need a recovery.
148160 if ((lastCheckpoint == null || !lastCheckpoint .getLsn ().equals (lastLsn )) &&
149- txnsStarted .size () > 0 ) {
161+ ! txnsStarted .isEmpty () ) {
150162 LOG .info ("Dirty transactions: {}" , txnsStarted .size ());
151163 // starting recovery: reposition the log reader to the last checkpoint
152164 if (lastCheckpoint == null ) {
@@ -250,10 +262,11 @@ private void doRecovery(final int txnCount, final Path last, final JournalReader
250262 if (LOG .isInfoEnabled ())
251263 {
252264 LOG .info ("First pass: redoing {} transactions..." , txnCount );}
253- final ProgressBar progress = new ProgressBar ("Redo " , FileUtils .sizeQuietly (last ));
254265 Loggable next = null ;
255266 int redoCnt = 0 ;
256267 try {
268+ final long lastSize = FileUtils .sizeQuietly (last );
269+ @ Nullable final ProgressBar redoProgressBar = hideProgressBar ? null : new ProgressBar ("Redo " , lastSize );
257270 while ((next = reader .nextEntry ()) != null ) {
258271 SanityCheck .ASSERT (next .getLogType () != LogEntryTypes .CHECKPOINT ,
259272 "Found a checkpoint during recovery run! This should not ever happen." );
@@ -271,9 +284,19 @@ private void doRecovery(final int txnCount, final Path last, final JournalReader
271284 // LOG.debug("Redo: " + next.dump());
272285 // redo the log entry
273286 next .redo ();
274- progress .set (next .getLsn ().getOffset ());
275- if (next .getLsn ().equals (lastLsn ))
276- {break ;} // last readable entry reached. Stop here.
287+
288+ if (redoProgressBar != null ) {
289+ redoProgressBar .set (next .getLsn ().getOffset ());
290+ }
291+
292+ if (next .getLsn ().equals (lastLsn )) {
293+ // last readable entry reached. Stop here.
294+ break ;
295+ }
296+ }
297+
298+ if (redoProgressBar != null ) {
299+ redoProgressBar .set (lastSize ); // 100% done
277300 }
278301 } catch (final Exception e ) {
279302 LOG .error ("Exception caught while redoing transactions. Aborting recovery to avoid possible damage. " +
@@ -291,16 +314,19 @@ private void doRecovery(final int txnCount, final Path last, final JournalReader
291314 {
292315 LOG .info ("Second pass: undoing dirty transactions. Uncommitted transactions: {}" , runningTxns .size ());}
293316 // see if there are uncommitted transactions pending
294- if (runningTxns .size () > 0 ) {
317+ if (! runningTxns .isEmpty () ) {
295318 // do a reverse scan of the log, undoing all uncommitted transactions
296319 try {
297- while ((next = reader .previousEntry ()) != null ) {
320+ final long lastSize = FileUtils .sizeQuietly (last );
321+ final ProgressBar undoProgressBar = hideProgressBar ? null : new ProgressBar ("Undo " , lastSize );
322+ while ((next = reader .previousEntry ()) != null ) {
298323 if (next .getLogType () == LogEntryTypes .TXN_START ) {
299324 if (runningTxns .get (next .getTransactionId ()) != null ) {
300325 runningTxns .remove (next .getTransactionId ());
301- if (runningTxns .size () == 0 )
326+ if (runningTxns .isEmpty ()) {
302327 // all dirty transactions undone
303- {break ;}
328+ break ;
329+ }
304330 }
305331 } else if (next .getLogType () == LogEntryTypes .TXN_COMMIT ) {
306332 // ignore already committed transaction
@@ -314,6 +340,14 @@ private void doRecovery(final int txnCount, final Path last, final JournalReader
314340 // LOG.debug("Undo: " + next.dump());
315341 next .undo ();
316342 }
343+
344+ if (undoProgressBar != null ) {
345+ undoProgressBar .set (lastSize - next .getLsn ().getOffset ());
346+ }
347+ }
348+
349+ if (undoProgressBar != null ) {
350+ undoProgressBar .set (lastSize ); // 100% done
317351 }
318352 } catch (final Exception e ) {
319353 LOG .warn ("Exception caught while undoing dirty transactions. Remaining transactions to be undone: {}. Aborting recovery to avoid possible damage. Before starting again, make sure to run a check via the emergency export tool." , runningTxns .size (), e );
0 commit comments