.

Azure · Aug 13, 2024 · 7ad41ca · 7ad41ca
1 parent f520b51
commit 7ad41ca
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/code/KustoCopyConsole/Entity/InMemory/RowItemInMemoryCache.cs b/code/KustoCopyConsole/Entity/InMemory/RowItemInMemoryCache.cs
@@ -13,8 +13,6 @@ internal class RowItemInMemoryCache
         private volatile IImmutableDictionary<(Uri, string), SourceDatabaseCache> _sourceDatabaseMap
             = ImmutableDictionary<(Uri, string), SourceDatabaseCache>.Empty;
 
-        public event EventHandler<RowItem>? NewRowItem;
-
         public RowItemInMemoryCache(IEnumerable<RowItem> items)
         {
             lock (_lock)
@@ -45,10 +43,6 @@ public void AddItem(RowItem item)
                         AddItemToCache(item));
                 }
             }
-            if (NewRowItem != null)
-            {
-                NewRowItem(this, item);
-            }
         }
 
         private IImmutableDictionary<(Uri, string), SourceDatabaseCache> AddItemToCache(RowItem item)

diff --git a/code/KustoCopyConsole/Storage/RowItemGateway.cs b/code/KustoCopyConsole/Storage/RowItemGateway.cs
@@ -34,6 +34,8 @@ private record QueueItem(byte[] Buffer, TaskCompletionSource TaskSource);
         private readonly AsyncLock _asyncLock = new();
         private TaskCompletionSource _persistanceTaskSource = new();
 
+        public event EventHandler<RowItemAppend>? RowItemAppended;
+
         #region Construction
         private RowItemGateway(IAppendStorage appendStorage, RowItemInMemoryCache cache)
         {
@@ -148,6 +150,10 @@ public async Task<RowItemAppend> AppendAsync(RowItem item, CancellationToken ct)
                     {
                         QueueCurrentStream();
                     }
+                    if (RowItemAppended != null)
+                    {
+                        RowItemAppended(this, package);
+                    }
 
                     return package;
                 }