More extensions

Author: wickwirew

Sources/Compiler/Gen/SwiftLanguage.swift

@@ -576,6 +576,7 @@ public struct SwiftLanguage: Language {
         extensionOn("Query") {
             self.writer.write("Input == ", query.inputName)
         } builder: {
+            // Execute no tx
             writer.write("func execute(")
             writeInput()
             writer.write(") async throws -> Output ")
@@ -586,6 +587,19 @@ public struct SwiftLanguage: Language {
                 writer.write("))")
             }
 
+            // Execute with tx
+            writer.blankLine()
+            writer.write(line: "func execute(")
+            writeInput()
+            writer.write(", tx: borrowing Transaction) async throws -> Output ")
+            
+            writer.braces {
+                writer.write(line: "try await execute(with: ", query.inputName, "(")
+                initInput()
+                writer.write("), tx: tx)")
+            }
+            
+            // Observe
             writer.blankLine()
             writer.write(line: "func observe(")
             writeInput()

Tests/CompilerTests/Gen/Swift.output

@@ -416,6 +416,10 @@ extension Query where Input == InsertUserInput {
         try await execute(with: InsertUserInput(id: id, firstName: firstName, lastName: lastName, preference: preference, favoriteNumber: favoriteNumber, randomValue: randomValue, bornOn: bornOn))
     }
 
+    func execute(id: Int, firstName: String, lastName: String, preference: Bool?, favoriteNumber: Int?, randomValue: SQLAny?, bornOn: Date?, tx: borrowing Transaction) async throws -> Output {
+        try await execute(with: InsertUserInput(id: id, firstName: firstName, lastName: lastName, preference: preference, favoriteNumber: favoriteNumber, randomValue: randomValue, bornOn: bornOn), tx: tx)
+    }
+    
     func observe(id: Int, firstName: String, lastName: String, preference: Bool?, favoriteNumber: Int?, randomValue: SQLAny?, bornOn: Date?) -> any QueryObservation<Output> {
         observe(with: InsertUserInput(id: id, firstName: firstName, lastName: lastName, preference: preference, favoriteNumber: favoriteNumber, randomValue: randomValue, bornOn: bornOn))
     }
@@ -431,6 +435,10 @@ extension Query where Input == SelectUserWithManyInputsInput {
         try await execute(with: SelectUserWithManyInputsInput(id: id, firstName: firstName))
     }
 
+    func execute(id: Int, firstName: String, tx: borrowing Transaction) async throws -> Output {
+        try await execute(with: SelectUserWithManyInputsInput(id: id, firstName: firstName), tx: tx)
+    }
+    
     func observe(id: Int, firstName: String) -> any QueryObservation<Output> {
         observe(with: SelectUserWithManyInputsInput(id: id, firstName: firstName))
     }

Tests/CompilerTests/GenTests.swift

@@ -24,8 +24,6 @@ struct GenTests {
             schema: compiler.schema
         )
 
-        print(rawOutput)
-        
         for diagnostics in migrations.1 {
             Issue.record(diagnostics)
         }