Refactor import/export database code

2020-06-11 21:11:50 +02:00
parent 3d69a00716
commit 39f92309b7
6 changed files with 381 additions and 229 deletions
--- a/PluralKit.Core/Utils/DatabaseUtils.cs
+++ b/PluralKit.Core/Utils/DatabaseUtils.cs
@@ -155,7 +155,7 @@ namespace PluralKit.Core
        }
    }

-    public class PerformanceTrackingConnection: IDbConnection
+    public class PerformanceTrackingConnection: IDbConnection, IAsyncDisposable
    {
        // Simple delegation of everything.
        internal NpgsqlConnection _impl;
@@ -226,6 +226,7 @@ namespace PluralKit.Core
        public string Database => _impl.Database;

        public ConnectionState State => _impl.State;
+        public ValueTask DisposeAsync() => _impl.DisposeAsync();
    }

    public class DbConnectionCountHolder