enable foreign keys (#70)

Sources/SQLiteKit/SQLiteConnectionSource.swift

@@ -34,7 +34,14 @@ public struct SQLiteConnectionSource: ConnectionPoolSource {
             threadPool: self.threadPool,
             logger: logger,
             on: eventLoop
-        )
+        ).flatMap { conn in
+            if self.configuration.enableForeignKeys {
+                return conn.query("PRAGMA foreign_keys = ON")
+                    .map { _ in conn }
+            } else {
+                return eventLoop.makeSucceededFuture(conn)
+            }
+        }
     }
 }
 
@@ -46,13 +53,11 @@ public struct SQLiteConfiguration {
     }
 
     public var storage: Storage
-
-    public init(file: String) {
-        self.init(storage: .file(path: file))
-    }
+    public var enableForeignKeys: Bool
 
-    public init(storage: Storage) {
+    public init(storage: Storage, enableForeignKeys: Bool = true) {
         self.storage = storage
+        self.enableForeignKeys = enableForeignKeys
     }
 }
 

Tests/SQLiteKitTests/SQLKitTests.swift

@@ -85,6 +85,11 @@ class SQLiteTests: XCTestCase {
             .run().wait()
     }
 
+    func testForeignKeysEnabled() throws {
+        let res = try self.connection.query("PRAGMA foreign_keys").wait()
+        XCTAssertEqual(res[0].column("foreign_keys"), .integer(1))
+    }
+
     var db: SQLDatabase {
         self.connection.sql()
     }
@@ -101,11 +106,10 @@ class SQLiteTests: XCTestCase {
         self.eventLoopGroup = MultiThreadedEventLoopGroup(numberOfThreads: 2)
         self.threadPool = NIOThreadPool(numberOfThreads: 2)
         self.threadPool.start()
-        self.connection = try! SQLiteConnection.open(
-            storage: .memory,
-            threadPool: self.threadPool,
-            on: self.eventLoopGroup.next()
-        ).wait()
+        self.connection = try! SQLiteConnectionSource(
+            configuration: .init(storage: .memory, enableForeignKeys: true),
+            threadPool: self.threadPool
+        ).makeConnection(logger: .init(label: "test"), on: self.eventLoopGroup.next()).wait()
     }
 
     override func tearDown() {