(#121) Distributed Lock

Author: phongnguyend

src/Monolith/ClassifiedAds.Application/EmailMessages/Services/EmailMessageService.cs

@@ -1,4 +1,5 @@
 using ClassifiedAds.CrossCuttingConcerns.CircuitBreakers;
+using ClassifiedAds.CrossCuttingConcerns.Locks;
 using ClassifiedAds.CrossCuttingConcerns.OS;
 using ClassifiedAds.Domain.Notification;
 using ClassifiedAds.Domain.Repositories;
@@ -17,18 +18,21 @@ public class EmailMessageService
         private readonly IEmailNotification _emailNotification;
         private readonly IDateTimeProvider _dateTimeProvider;
         private readonly ICircuitBreakerManager _circuitBreakerManager;
+        private readonly IDistributedLock _distributedLock;
 
         public EmailMessageService(ILogger<EmailMessageService> logger,
             IEmailMessageRepository repository,
             IEmailNotification emailNotification,
             IDateTimeProvider dateTimeProvider,
-            ICircuitBreakerManager circuitBreakerManager)
+            ICircuitBreakerManager circuitBreakerManager,
+            IDistributedLock distributedLock)
         {
             _logger = logger;
             _repository = repository;
             _emailNotification = emailNotification;
             _dateTimeProvider = dateTimeProvider;
             _circuitBreakerManager = circuitBreakerManager;
+            _distributedLock = distributedLock;
         }
 
         public async Task<int> SendEmailMessagesAsync()

src/Monolith/ClassifiedAds.Application/SmsMessages/Services/SmsMessageService.cs

@@ -1,4 +1,5 @@
 using ClassifiedAds.CrossCuttingConcerns.CircuitBreakers;
+using ClassifiedAds.CrossCuttingConcerns.Locks;
 using ClassifiedAds.CrossCuttingConcerns.OS;
 using ClassifiedAds.Domain.Notification;
 using ClassifiedAds.Domain.Repositories;
@@ -17,18 +18,21 @@ public class SmsMessageService
         private readonly ISmsNotification _smsNotification;
         private readonly IDateTimeProvider _dateTimeProvider;
         private readonly ICircuitBreakerManager _circuitBreakerManager;
+        private readonly IDistributedLock _distributedLock;
 
         public SmsMessageService(ILogger<SmsMessageService> logger,
             ISmsMessageRepository repository,
             ISmsNotification smsNotification,
             IDateTimeProvider dateTimeProvider,
-            ICircuitBreakerManager circuitBreakerManager)
+            ICircuitBreakerManager circuitBreakerManager,
+            IDistributedLock distributedLock)
         {
             _logger = logger;
             _repository = repository;
             _smsNotification = smsNotification;
             _dateTimeProvider = dateTimeProvider;
             _circuitBreakerManager = circuitBreakerManager;
+            _distributedLock = distributedLock;
         }
 
         public async Task<int> SendSmsMessagesAsync()

src/Monolith/ClassifiedAds.CrossCuttingConcerns/Locks/IDistributedLock.cs

@@ -1,6 +1,8 @@
-namespace ClassifiedAds.CrossCuttingConcerns.Locks
+using System;
+
+namespace ClassifiedAds.CrossCuttingConcerns.Locks
 {
-    public interface IDistributedLock
+    public interface IDistributedLock : IDisposable
     {
         IDistributedLockScope Acquire(string lockName);
 

src/Monolith/ClassifiedAds.Persistence/Locks/SqlDistributedLock.cs

@@ -1,4 +1,5 @@
 using ClassifiedAds.CrossCuttingConcerns.Locks;
+using EntityFrameworkCore.SqlServer.SimpleBulks.Extensions;
 using Microsoft.Data.SqlClient;
 using System;
 using System.Data;
@@ -11,6 +12,7 @@ public class SqlDistributedLock : IDistributedLock
 
         private readonly SqlConnection _connection;
         private readonly SqlTransaction _transaction;
+        private readonly string _connectionString;
 
         public bool HasTransaction
         {
@@ -31,6 +33,12 @@ public SqlDistributedLock(SqlTransaction transaction)
             _connection = _transaction.Connection;
         }
 
+        public SqlDistributedLock(string connectionString)
+        {
+            _connectionString = connectionString;
+            _connection = new SqlConnection(connectionString);
+        }
+
         public IDistributedLockScope Acquire(string lockName)
         {
             SqlParameter returnValue;
@@ -67,6 +75,8 @@ public IDistributedLockScope TryAcquire(string lockName)
 
         private SqlCommand CreateAcquireCommand(int commandTimeout, string lockName, int lockTimeout, out SqlParameter returnValue)
         {
+            _connection.EnsureOpen();
+
             SqlCommand command = _connection.CreateCommand();
             command.Transaction = _transaction;
 
@@ -120,5 +130,13 @@ public static bool ParseReturnCode(int returnCode)
 
             return false;
         }
+
+        public void Dispose()
+        {
+            if (!string.IsNullOrEmpty(_connectionString))
+            {
+                _connection.Dispose();
+            }
+        }
     }
 }

src/Monolith/ClassifiedAds.Persistence/PersistenceExtensions.cs

@@ -25,6 +25,9 @@ public static IServiceCollection AddPersistence(this IServiceCollection services
                     }))
                     .AddDbContextFactory<AdsDbContext>((Action<DbContextOptionsBuilder>)null, ServiceLifetime.Scoped)
                     .AddRepositories();
+
+            services.AddScoped(typeof(IDistributedLock), _ => new SqlDistributedLock(connectionString));
+
             return services;
         }
 
@@ -39,6 +42,12 @@ public static IServiceCollection AddMultiTenantPersistence(this IServiceCollecti
                         return services.GetRequiredService<AdsDbContextMultiTenant>();
                     })
                     .AddRepositories();
+
+            services.AddScoped(typeof(IDistributedLock), services =>
+            {
+                return new SqlDistributedLock(services.GetRequiredService<IConnectionStringResolver<AdsDbContextMultiTenant>>().ConnectionString);
+            });
+
             return services;
         }