Reopen the connection immediately after it's closed

OoLunar · May 5, 2023 · 1b7ae6d · 1b7ae6d
1 parent e49479a
commit 1b7ae6d
Showing 1 changed file with 24 additions and 5 deletions.
diff --git a/src/CookieTracker.cs b/src/CookieTracker.cs
@@ -34,7 +34,7 @@ public CookieTracker(IConfiguration configuration, ILogger<CookieTracker> logger
             _timer = new PeriodicTimer(TimeSpan.FromSeconds(configuration.GetValue("CookieTracker:Period", 30)));
 
             _databaseConnection = NpgsqlDataSource.Create(CookieDatabaseContext.GetConnectionString(configuration)).CreateConnection();
-            _databaseConnection.Open();
+            _databaseConnection.StateChange += InitConnection;
             _databaseCommands = new Dictionary<DatabaseOperation, NpgsqlCommand>
             {
                 [DatabaseOperation.Create] = GetInsertCommand(_databaseConnection),
@@ -87,12 +87,9 @@ public ulong Click(Ulid cookieId)
 
         private async Task StartBakingAsync()
         {
+            InitConnection(null, new StateChangeEventArgs(ConnectionState.Closed, ConnectionState.Closed));
             DbCommand createCommand = _databaseCommands[DatabaseOperation.Create];
             DbCommand updateCommand = _databaseCommands[DatabaseOperation.Update];
-            foreach (DbCommand command in _databaseCommands.Values)
-            {
-                await command.PrepareAsync();
-            }
 
             while (await _timer.WaitForNextTickAsync())
             {
@@ -172,6 +169,28 @@ public async ValueTask DisposeAsync()
             _semaphore.Dispose();
         }
 
+        private void InitConnection(object? sender, StateChangeEventArgs eventArgs)
+        {
+            if (eventArgs.CurrentState is not ConnectionState.Closed and not ConnectionState.Broken)
+            {
+                return;
+            }
+
+            _semaphore.Wait();
+            try
+            {
+                _databaseConnection.Open();
+                foreach (DbCommand command in _databaseCommands.Values)
+                {
+                    command.Prepare();
+                }
+            }
+            finally
+            {
+                _semaphore.Release();
+            }
+        }
+
         private static NpgsqlCommand GetSelectCommand(NpgsqlConnection connection)
         {
             NpgsqlCommand command = connection.CreateCommand();