.

Azure · Sep 14, 2024 · 47b1857 · 47b1857
1 parent 0f88b91
commit 47b1857
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/code/KustoCopyConsole/Orchestration/SourceTableExportingOrchestration.cs b/code/KustoCopyConsole/Orchestration/SourceTableExportingOrchestration.cs
@@ -129,6 +129,18 @@ public IImmutableList<string> GetOperationIds()
                 }
             }
 
+            public RowItem RemoveBlockItem(string operationId)
+            {
+                lock (_operationIdToBlockMap)
+                {
+                    var blockItem = _operationIdToBlockMap[operationId];
+
+                    _operationIdToBlockMap.Remove(operationId);
+
+                    return blockItem;
+                }
+            }
+
             public void AddOperationId(RowItem blockItem)
             {
                 lock (_operationIdToBlockMap)
@@ -460,7 +472,13 @@ private async Task OnExportedAsync(CancellationToken ct)
                             switch (status.State)
                             {
                                 case "Completed":
-                                    throw new NotSupportedException();
+                                    var blockItem = clusterQueue.RemoveBlockItem(status.OperationId)
+                                        .Clone();
+
+                                    blockItem.State = SourceBlockState.Exported.ToString();
+                                    await RowItemGateway.AppendAsync(blockItem, ct);
+
+                                    break;
                                 case "Failed":
                                     throw new CopyException(
                                         $"Failed export '{status.OperationId}':  '{status.Status}'",