Sendable

Author: wickwirew

Sources/Compiler/Gen/SwiftLanguage.swift

@@ -265,9 +265,11 @@ public struct SwiftLanguage: Language {
     }
 
     private func queries(name: String, queries: [GeneratedQuery]) {
-        writer.write(line: "struct ", name, " {")
+        writer.write(line: "struct ", name, ": ConnectionWrapper, Sendable {")
         writer.indent()
 
+        writer.write(line: "let connection: any Connection")
+        
         for query in queries {
             writer.write(line: "var ", query.variableName, ": any ", query.typealiasName)
         }
@@ -289,6 +291,8 @@ public struct SwiftLanguage: Language {
         writer.write(line: "return ", name, "(")
         writer.indent()
 
+        writer.write(line: "connection: connection,")
+        
         for (position, query) in queries.positional() {
             writer.write(line: query.variableName, ": ")
 
@@ -343,6 +347,8 @@ public struct SwiftLanguage: Language {
         writer.write(line: name, "(")
         writer.indent()
 
+        writer.write(line: "connection: NoopConnection(),")
+        
         for (position, query) in queries.positional() {
             writer.write(line: query.variableName, ": ", query.variableName)
 

Tests/CompilerTests/Gen/Swift.output

@@ -214,7 +214,8 @@ struct SelectWithManyInputsInput: Hashable, Sendable {
     let textNotNull: String
 }
 
-struct QueriesQueries {
+struct QueriesQueries: ConnectionWrapper, Sendable {
+    let connection: any Connection
     var insertFooReturningFoo: any InsertFooReturningFooQuery
     var insertBarReturningIntPk: any InsertBarReturningIntPkQuery
     var insertBarReturningExtraColumn: any InsertBarReturningExtraColumnQuery
@@ -233,6 +234,7 @@ struct QueriesQueries {
         selectWithManyInputs: any SelectWithManyInputsQuery = Queries.Just()
     ) -> QueriesQueries {
         QueriesQueries(
+            connection: NoopConnection(),
             insertFooReturningFoo: insertFooReturningFoo,
             insertBarReturningIntPk: insertBarReturningIntPk,
             insertBarReturningExtraColumn: insertBarReturningExtraColumn,
@@ -245,6 +247,7 @@ struct QueriesQueries {
 
     static func live(connection: Connection, adapters: DB.Adapters) -> QueriesQueries {
         return QueriesQueries(
+            connection: connection,
             insertFooReturningFoo: DatabaseQuery<InsertFooReturningFooInput, InsertFooReturningFooOutput>(
                 .write,
                 in: connection,