Overload UseAzureBlobLockProvider with connection string/container name

elsa-workflows · Nov 23, 2023 · d524fc9 · d524fc9
1 parent f015e85
commit d524fc9
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/src/locking/Elsa.DistributedLocking.AzureBlob/DistributedLockingOptionsExtensions.cs b/src/locking/Elsa.DistributedLocking.AzureBlob/DistributedLockingOptionsExtensions.cs
@@ -21,6 +21,12 @@ public static DistributedLockingOptionsBuilder UseAzureBlobLockProvider(this Dis
             return options;
         }
 
+        public static DistributedLockingOptionsBuilder UseAzureBlobLockProvider(this DistributedLockingOptionsBuilder options, string connectionString, string blobContainerName, BlobClientOptions? blobClientOptions = null)
+        {
+            options.UseProviderFactory(sp => CreateAzureDistributedLockFactory(sp, connectionString, blobContainerName, blobClientOptions));
+            return options;
+        }
+
         private static Func<string, IDistributedLock> CreateAzureDistributedLockFactory(IServiceProvider services, Uri blobContainerUrl)
         {
             var container = new BlobContainerClient(blobContainerUrl);
@@ -32,5 +38,11 @@ public static DistributedLockingOptionsBuilder UseAzureBlobLockProvider(this Dis
             var container = new BlobContainerClient(blobContainerUrl, tokenCredential, blobClientOptions);
             return name => new AzureBlobLeaseDistributedLock(container, name);
         }
+
+        private static Func<string, IDistributedLock> CreateAzureDistributedLockFactory(IServiceProvider services, string connectionString, string blobContainerName, BlobClientOptions? blobClientOptions)
+        {
+            var container = new BlobContainerClient(connectionString, blobContainerName, blobClientOptions);
+            return name => new AzureBlobLeaseDistributedLock(container, name);
+        }
     }
-}
+}