Revert Postgres connection handling

microsoft · Oct 3, 2024 · adb9c01 · adb9c01
1 parent ffd90fa
commit adb9c01
Show file tree

Hide file tree

Showing 3 changed files with 311 additions and 299 deletions.
diff --git a/extensions/Postgres/Postgres.FunctionalTests/ConcurrencyTests.cs b/extensions/Postgres/Postgres.FunctionalTests/ConcurrencyTests.cs
@@ -69,7 +69,7 @@ last_update   TIMESTAMP WITH TIME ZONE DEFAULT CURRENT_TIMESTAMP NOT NULL
         var indexName = "create_index_test";
         var vectorSize = 1536;
 
-        var target = new PostgresMemory(config, new FakeEmbeddingGenerator());
+        using var target = new PostgresMemory(config, new FakeEmbeddingGenerator());
 
         var tasks = new List<Task>();
         for (int i = 0; i < concurrency; i++)
@@ -96,7 +96,7 @@ public async Task UpsertConcurrencyTest()
         var vectorSize = 4;
         var indexName = "upsert_test" + Guid.NewGuid().ToString("D");
 
-        var target = new PostgresMemory(this.PostgresConfig, new FakeEmbeddingGenerator());
+        using var target = new PostgresMemory(this.PostgresConfig, new FakeEmbeddingGenerator());
 
         await target.CreateIndexAsync(indexName, vectorSize);