Added more use of ProgressLogger (#8504)

src/Nethermind/Nethermind.Db/FullPruning/FullPruningDb.cs

@@ -24,6 +24,8 @@ public class FullPruningDb : IDb, IFullPruningDb, ITunableDb
         private readonly DbSettings _settings;
         private readonly IDbFactory _dbFactory;
         private readonly Action? _updateDuplicateWriteMetrics;
+        private readonly ILogManager _logManager;
+        private ProgressLogger? _progressLogger;
 
         // current main DB, will be written to and will be main source for reading
         private IDb _currentDb;
@@ -32,11 +34,12 @@ public class FullPruningDb : IDb, IFullPruningDb, ITunableDb
         // this will be null if no full pruning is in progress
         private PruningContext? _pruningContext;
 
-        public FullPruningDb(DbSettings settings, IDbFactory dbFactory, Action? updateDuplicateWriteMetrics = null)
+        public FullPruningDb(DbSettings settings, IDbFactory dbFactory, ILogManager logManager, Action? updateDuplicateWriteMetrics = null)
         {
             _settings = settings;
             _dbFactory = dbFactory;
             _updateDuplicateWriteMetrics = updateDuplicateWriteMetrics;
+            _logManager = logManager;
             _currentDb = CreateDb(_settings).WithEOACompressed();
         }
 
@@ -182,6 +185,8 @@ DbSettings ClonedDbSettings()
             context = pruningContext ?? newContext;
             if (pruningContext is null)
             {
+                _progressLogger = new ProgressLogger("Pruning", _logManager);
+                _progressLogger.Reset(0, 0); // Initialize with no upper bound
                 PruningStarted?.Invoke(this, new PruningEventArgs(context, true));
                 return true;
             }
@@ -203,6 +208,8 @@ private void FinishPruning()
             _pruningContext?.CloningDb?.Flush();
             IDb oldDb = Interlocked.Exchange(ref _currentDb, _pruningContext?.CloningDb);
             ClearOldDb(oldDb);
+            _progressLogger?.MarkEnd();
+            _progressLogger = null;
         }
 
         protected virtual void ClearOldDb(IDb oldDb)
@@ -218,39 +225,39 @@ private void FinishPruning(PruningContext pruningContext, bool success)
 
         private class PruningContext : IPruningContext
         {
-            private bool _committed = false;
-            private bool _disposed = false;
+            private readonly FullPruningDb _db;
+            private long _processedKeys;
             public IDb CloningDb { get; }
             public bool DuplicateReads { get; }
-            private readonly FullPruningDb _db;
 
             public PruningContext(FullPruningDb db, IDb cloningDb, bool duplicateReads)
             {
                 CloningDb = cloningDb;
                 DuplicateReads = duplicateReads;
                 _db = db;
+                _db._progressLogger?.Reset(0, 0); // Initialize with no upper bound
             }
 
             public void Set(ReadOnlySpan<byte> key, byte[]? value, WriteFlags flags = WriteFlags.None)
             {
-                _db.Duplicate(CloningDb, key, value, flags);
+                CloningDb.Set(key, value, flags);
+                _processedKeys++;
+                _db._progressLogger?.Update(_processedKeys);
+                if (_processedKeys % 100000 == 0)
+                {
+                    _db._progressLogger?.LogProgress();
+                }
             }
 
             public IWriteBatch StartWriteBatch()
             {
-                return CloningDb.StartWriteBatch();
-            }
-
-            public byte[]? Get(ReadOnlySpan<byte> key, ReadFlags flags = ReadFlags.None)
-            {
-                return CloningDb.Get(key, flags);
+                return new ProgressTrackingWriteBatch(CloningDb.StartWriteBatch(), this);
             }
 
             /// <inheritdoc />
             public void Commit()
             {
                 _db.FinishPruning();
-                _committed = true; // we mark the context as committed.
             }
 
             /// <inheritdoc />
@@ -264,19 +271,9 @@ public void MarkStart()
             /// <inheritdoc />
             public void Dispose()
             {
-                if (!_disposed)
-                {
-                    _db.FinishPruning(this, _committed);
-                    if (!_committed)
-                    {
-                        // if the context was not committed, then pruning failed and we delete the cloned DB
-                        CloningDb.Clear();
-                    }
-
-                    CancellationTokenSource.Dispose();
-                    Metrics.StateDbPruning = 0;
-                    _disposed = true;
-                }
+                _db.FinishPruning(this, true);
+                CancellationTokenSource.Dispose();
+                Metrics.StateDbPruning = 0;
             }
         }
 
@@ -312,6 +309,36 @@ public void Set(ReadOnlySpan<byte> key, byte[]? value, WriteFlags flags = WriteF
             }
         }
 
+        private class ProgressTrackingWriteBatch : IWriteBatch
+        {
+            private readonly IWriteBatch _writeBatch;
+            private readonly PruningContext _context;
+            private long _batchProcessedKeys;
+
+            public ProgressTrackingWriteBatch(IWriteBatch writeBatch, PruningContext context)
+            {
+                _writeBatch = writeBatch;
+                _context = context;
+            }
+
+            public void Set(ReadOnlySpan<byte> key, byte[]? value, WriteFlags flags = WriteFlags.None)
+            {
+                _writeBatch.Set(key, value, flags);
+                _batchProcessedKeys++;
+                if (_batchProcessedKeys % 1000 == 0)
+                {
+                    _context._db._progressLogger?.Update(_context._processedKeys + _batchProcessedKeys);
+                    _context._db._progressLogger?.LogProgress();
+                }
+            }
+
+            public void Dispose()
+            {
+                _context._processedKeys += _batchProcessedKeys;
+                _writeBatch.Dispose();
+            }
+        }
+
         public void Tune(ITunableDb.TuneType type)
         {
             if (_currentDb is ITunableDb tunableDb)