Retry PutAsync if table not found

devlooped · May 4, 2022 · 1c260c4 · 1c260c4
1 parent 93325fb
commit 1c260c4
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/src/TableStorage/TableRepository`1.cs b/src/TableStorage/TableRepository`1.cs
@@ -165,9 +165,19 @@ pair.Value is DateOnly date ?
             values[nameof(ITableEntity.PartitionKey)] = partitionKey;
             values[nameof(ITableEntity.RowKey)] = rowKey;
 
-            var result = await table.UpsertEntityAsync(new TableEntity(values), UpdateMode, cancellation)
-                .ConfigureAwait(false);
-
+            try
+            {
+                await table.UpsertEntityAsync(new TableEntity(values), UpdateMode, cancellation)
+                    .ConfigureAwait(false);
+            }
+            catch (RequestFailedException e) when (e.ErrorCode == "TableNotFound")
+            {
+                // Retry after ensuring table exists.
+                await table.CreateIfNotExistsAsync();
+                await table.UpsertEntityAsync(new TableEntity(values), UpdateMode, cancellation)
+                    .ConfigureAwait(false);
+            }
+
             return (await GetAsync(partitionKey, rowKey, cancellation).ConfigureAwait(false))!;
         }