group by partition key

tforkmann · Jun 21, 2024 · e963b6a · e963b6a
1 parent 4ea76f8
commit e963b6a
Showing 1 changed file with 13 additions and 11 deletions.
diff --git a/src/AzureTackle/AzureTackle.fs b/src/AzureTackle/AzureTackle.fs
@@ -451,17 +451,19 @@ module AzureTable =
             try
                 let azureTable = getTable props
 
-                let chunks =
-                    entities
-                    |> Array.chunkBySize 100
-                for chunks in chunks do
-                    let actions =
-                        chunks
-                        |> Array.map (fun entity ->
-                            TableTransactionAction(TableTransactionActionType.UpsertReplace, entity))
-
-                    let! _ = azureTable.SubmitTransactionAsync(actions)
-                    return ()
+                for _key, entities in entities |> Array.groupBy (fun x -> x.PartitionKey) do
+                    let chunks =
+                        entities
+                        |> Array.chunkBySize 100
+
+                    for chunks in chunks do
+                        let actions =
+                            chunks
+                            |> Array.map (fun entity ->
+                                TableTransactionAction(TableTransactionActionType.UpsertReplace, entity))
+
+                        let! _ = azureTable.SubmitTransactionAsync(actions)
+                        ()
             with exn ->
                 return failwithf "UpsertBatch failed with exn: %s" exn.Message
         }