beta 4

Author: tanner0101

Sources/FluentSQLiteDriver/FluentSQLiteDatabase.swift

@@ -32,6 +32,23 @@ extension _FluentSQLiteDatabase: Database {
                 }
         }
     }
+
+    func transaction<T>(_ closure: @escaping (Database) -> EventLoopFuture<T>) -> EventLoopFuture<T> {
+        self.database.withConnection { conn in
+            conn.query("BEGIN TRANSACTION").flatMap { _ in
+                let db = _FluentSQLiteDatabase(database: conn, context: self.context)
+                return closure(db).flatMap { result in
+                    conn.query("COMMIT TRANSACTION").map { _ in
+                        result
+                    }
+                }.flatMapError { error in
+                    conn.query("ROLLBACK TRANSACTION").flatMapThrowing { _ in
+                        throw error
+                    }
+                }
+            }
+        }
+    }
 
     func execute(schema: DatabaseSchema) -> EventLoopFuture<Void> {
         let sql = SQLSchemaConverter(delegate: SQLiteConverterDelegate()).convert(schema)

Tests/FluentSQLiteDriverTests/FluentSQLiteDriverTests.swift

@@ -32,10 +32,6 @@ final class FluentSQLiteDriverTests: XCTestCase {
         try self.benchmarker.testEagerLoadParent()
     }
 
-    func testEagerLoadParentJoin() throws {
-        try self.benchmarker.testEagerLoadParentJoin()
-    }
-
     func testEagerLoadParentJSON() throws {
         try self.benchmarker.testEagerLoadParentJSON()
     }
@@ -100,6 +96,10 @@ final class FluentSQLiteDriverTests: XCTestCase {
         try self.benchmarker.testTimestampable()
     }
 
+    func testTransaction() throws {
+        try self.benchmarker.testTransaction()
+    }
+
     func testModelMiddleware() throws {
         try self.benchmarker.testModelMiddleware()
     }
@@ -164,10 +164,18 @@ final class FluentSQLiteDriverTests: XCTestCase {
         try self.benchmarker.testRange()
     }
 
+    func testRelationMethods() throws {
+        try self.benchmarker.testRelationMethods()
+    }
+
     func testNonstandardIDKey() throws {
         try self.benchmarker.testNonstandardIDKey()
     }
 
+    func testGroup() throws {
+        try self.benchmarker.testGroup()
+    }
+
     var benchmarker: FluentBenchmarker {
         return .init(database: self.database)
     }