Add service lifecycle support via package trait (#6)

- implemented service lifecycle support
- add new package trait
- updated tests

Author: tib

Package.resolved

@@ -34,10 +34,19 @@ let package = Package(
     products: [
         .library(name: "FeatherSQLiteDatabase", targets: ["FeatherSQLiteDatabase"]),
     ],
+    traits: [
+        "ServiceLifecycleSupport",
+        .default(
+            enabledTraits: [
+                "ServiceLifecycleSupport",
+            ]
+        ),
+    ],
     dependencies: [
         .package(url: "https://github.com/apple/swift-log", from: "1.6.0"),
         .package(url: "https://github.com/vapor/sqlite-nio", from: "1.12.0"),
         .package(url: "https://github.com/feather-framework/feather-database", exact: "1.0.0-beta.3"),
+        .package(url: "https://github.com/swift-server/swift-service-lifecycle", from: "2.8.0"),
         // [docc-plugin-placeholder]
     ],
     targets: [
@@ -52,8 +61,13 @@ let package = Package(
         .target(
             name: "FeatherSQLiteDatabase",
             dependencies: [
-                .product(name: "FeatherDatabase", package: "feather-database"),
                 .target(name: "SQLiteNIOExtras"),
+                .product(name: "FeatherDatabase", package: "feather-database"),
+                .product(
+                    name: "ServiceLifecycle",
+                    package: "swift-service-lifecycle",
+                    condition: .when(traits: ["ServiceLifecycleSupport"])
+                ),
             ],
             swiftSettings: defaultSwiftSettings
         ),

Package.swift

@@ -45,6 +45,30 @@ Then add `FeatherSQLiteDatabase` to your target dependencies:
 .product(name: "FeatherSQLiteDatabase", package: "feather-sqlite-database"),
 ```
 
+<<<<<<< HEAD
+### Package traits
+
+This package offers additional integrations you can enable using [package traits](https://docs.swift.org/swiftpm/documentation/packagemanagerdocs/addingdependencies#Packages-with-Traits).
+To enable an additional trait on the package, update the package dependency:
+
+```diff
+.package(
+    url: "https://github.com/feather-framework/feather-sqlite-database",
+    exact: "1.0.0-beta.3",
++   traits: [
++       .defaults, 
++       "ServiceLifecycleSupport",
++   ]
+)
+```
+
+Available traits:
+
+- `ServiceLifecycleSupport` (default): Adds support for `SQLiteClientService`, a `ServiceLifecycle.Service` implementation for managing SQLite clients.
+
+
+=======
+>>>>>>> main
 ## Usage
 
 API documentation is available at the link below:

README.md

@@ -0,0 +1,41 @@
+//
+//  SQLiteDatabaseService.swift
+//  feather-sqlite-database
+//
+//  Created by Tibor Bödecs on 2026. 01. 29..
+//
+
+#if ServiceLifecycleSupport
+
+import SQLiteNIOExtras
+import ServiceLifecycle
+
+/// A `Service` wrapper around an `SQLiteClient`.
+public struct SQLiteDatabaseService: Service {
+
+    /// The underlying SQLite client instance.
+    public var client: SQLiteClient
+
+    /// Creates a new SQLite client service.
+    ///
+    /// - Parameter client: The SQLite client to manage for the service lifecycle.
+    public init(
+        _ client: SQLiteClient
+    ) {
+        self.client = client
+    }
+
+    /// Runs the SQLite client service.
+    ///
+    /// This method starts the SQLite client, waits for a graceful shutdown
+    /// signal, and then shuts down the client in an orderly manner.
+    ///
+    /// - Throws: Rethrows any error produced while starting the SQLite client.
+    public func run() async throws {
+        try await client.run()
+        try? await gracefulShutdown()
+        await client.shutdown()
+    }
+
+}
+#endif

Sources/FeatherSQLiteDatabase/SQLiteDatabaseService.swift

@@ -208,7 +208,9 @@ actor SQLiteConnectionPool {
             catch {
                 configuration.logger.warning(
                     "Failed to close SQLite connection after setup error",
-                    metadata: ["error": "\(error)"]
+                    metadata: [
+                        "error": "\(error)"
+                    ]
                 )
             }
             throw error
@@ -225,7 +227,9 @@ actor SQLiteConnectionPool {
         catch {
             configuration.logger.warning(
                 "Failed to close SQLite connection",
-                metadata: ["error": "\(error)"]
+                metadata: [
+                    "error": "\(error)"
+                ]
             )
         }
     }

Sources/SQLiteNIOExtras/SQLiteConnectionPool.swift

@@ -1,5 +1,5 @@
 //
-//  SQLiteDatabaseTestSuite.swift
+//  FeatherSQLiteDatabaseTestSuite.swift
 //  feather-sqlite-database
 //
 //  Created by Tibor Bödecs on 2026. 01. 10..
@@ -14,7 +14,7 @@ import Testing
 @testable import FeatherSQLiteDatabase
 
 @Suite
-struct SQLiteDatabaseTestSuite {
+struct FeatherSQLiteDatabaseTestSuite {
 
     private func runUsingTestDatabaseClient(
         _ closure: ((SQLiteDatabaseClient) async throws -> Void)
@@ -775,3 +775,62 @@ struct SQLiteDatabaseTestSuite {
         }
     }
 }
+
+#if ServiceLifecycleSupport
+import ServiceLifecycle
+
+extension FeatherSQLiteDatabaseTestSuite {
+
+    @Test
+    func serviceLifecycleSupport() async throws {
+        var logger = Logger(label: "test")
+        logger.logLevel = .info
+
+        let configuration = SQLiteClient.Configuration(
+            storage: .memory,
+            logger: logger,
+        )
+        let client = SQLiteClient(configuration: configuration)
+        let database = SQLiteDatabaseClient(client: client, logger: logger)
+        let service = SQLiteDatabaseService(client)
+
+        let serviceGroup = ServiceGroup(
+            services: [service],
+            logger: logger
+        )
+
+        try await withThrowingTaskGroup(of: Void.self) { group in
+            group.addTask {
+                try await serviceGroup.run()
+            }
+            group.addTask {
+                let result = try await database.withConnection { connection in
+                    try await connection.run(
+                        query: #"""
+                            SELECT 
+                                sqlite_version() AS "version" 
+                            WHERE 
+                                1=\#(1);
+                            """#
+                    )
+                }
+
+                let resultArray = try await result.collect()
+                #expect(resultArray.count == 1)
+
+                let item = resultArray[0]
+                let version = try item.decode(
+                    column: "version",
+                    as: String.self
+                )
+                #expect(version.split(separator: ".").count == 3)
+            }
+            try await group.next()
+
+            try await Task.sleep(for: .milliseconds(100))
+
+            await serviceGroup.triggerGracefulShutdown()
+        }
+    }
+}
+#endif