Remove PostgresClientService

hummingbird-project · May 14, 2024 · dd1504e · dd1504e
1 parent 5769176
commit dd1504e
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 14 deletions.
diff --git a/.gitignore b/.gitignore
@@ -7,4 +7,5 @@ DerivedData/
 .swiftpm/xcode/package.xcworkspace/contents.xcworkspacedata
 .netrc
 .vscode
+.devcontainer
 Package.resolved
diff --git a/Tests/HummingbirdPostgresTests/JobsTests.swift b/Tests/HummingbirdPostgresTests/JobsTests.swift
@@ -77,7 +77,7 @@ final class JobsTests: XCTestCase {
             return try await withThrowingTaskGroup(of: Void.self) { group in
                 let serviceGroup = ServiceGroup(
                     configuration: .init(
-                        services: [PostgresClientService(client: jobQueue.queue.client), jobQueue],
+                        services: [jobQueue.queue.client, jobQueue],
                         gracefulShutdownSignals: [.sigterm, .sigint],
                         logger: jobQueue.queue.logger
                     )
@@ -352,7 +352,7 @@ final class JobsTests: XCTestCase {
         try await withThrowingTaskGroup(of: Void.self) { group in
             let serviceGroup = ServiceGroup(
                 configuration: .init(
-                    services: [PostgresClientService(client: postgresClient), jobQueue, jobQueue2],
+                    services: [postgresClient, jobQueue, jobQueue2],
                     gracefulShutdownSignals: [.sigterm, .sigint],
                     logger: logger
                 )

diff --git a/Tests/HummingbirdPostgresTests/PersistTests.swift b/Tests/HummingbirdPostgresTests/PersistTests.swift
@@ -71,7 +71,7 @@ final class PersistTests: XCTestCase {
         }
         updateRouter(router, persist)
         var app = Application(responder: router.buildResponder())
-        app.addServices(PostgresClientService(client: postgresClient), persist)
+        app.addServices(postgresClient, persist)
         app.runBeforeServerStart {
             try await postgresMigrations.apply(client: postgresClient, groups: [.persist], logger: logger, dryRun: false)
         }

diff --git a/Tests/HummingbirdPostgresTests/TestUtils.swift b/Tests/HummingbirdPostgresTests/TestUtils.swift
@@ -2,17 +2,6 @@ import Hummingbird
 import PostgresNIO
 import ServiceLifecycle
 
-/// Manage the lifecycle of a PostgresClient
-struct PostgresClientService: Service {
-    let client: PostgresClient
-
-    func run() async {
-        await cancelOnGracefulShutdown {
-            await self.client.run()
-        }
-    }
-}
-
 func getPostgresConfiguration() async throws -> PostgresClient.Configuration {
     let env = try await Environment.shared.merging(with: .dotEnv())
     return .init(