Adds multirow insert method (#153)

* feat: Adds multirow insert method
vapor · Sep 23, 2024 · e0b35ff · e0b35ff
1 parent 3c8c6aa
commit e0b35ff
Show file tree

Hide file tree

Showing 2 changed files with 96 additions and 0 deletions.
diff --git a/Sources/SQLKit/Builders/Implementations/SQLInsertBuilder.swift b/Sources/SQLKit/Builders/Implementations/SQLInsertBuilder.swift
@@ -283,6 +283,16 @@ public final class SQLInsertBuilder: SQLQueryBuilder, SQLReturningBuilder/*, SQL
         self.insert.values.append(values)
         return self
     }
+
+    /// Add multiple sequences of values each inserted as a separate row.
+    @inlinable
+    @discardableResult
+    public func rows<S1, S2>(_ valueSets: S1) -> Self
+        where S1: Sequence, S2: Sequence,
+              S1.Element == S2, S2.Element == any SQLExpression
+    {
+        valueSets.reduce(self) { $0.values(Array($1)) }
+    }
 
     /// Specify a `SELECT` query to generate rows to insert.
     ///

diff --git a/Tests/SQLKitTests/SQLInsertUpsertTests.swift b/Tests/SQLKitTests/SQLInsertUpsertTests.swift
@@ -41,6 +41,92 @@ final class SQLInsertUpsertTests: XCTestCase {
             is: "INSERT INTO ``planets`` (``id``, ``name``) SELECT ``id``, ``name`` FROM ``other_planets``"
         )
     }
+
+    func testInsertValuesEncodable() throws {
+        // Test variadic values method
+        XCTAssertSerialization(
+            of: self.db.insert(into: "planets")
+                .columns(["name", "color"])
+                .values("Jupiter", "orange"),
+            is: "INSERT INTO ``planets`` (``name``, ``color``) VALUES (&1, &2)"
+        )
+
+        // Test array values method
+        XCTAssertSerialization(
+            of: db.insert(into: "planets")
+                .columns(["name", "color"])
+                .values(["Jupiter", "orange"]),
+            is: "INSERT INTO ``planets`` (``name``, ``color``) VALUES (&1, &2)"
+        )
+
+        // Test nested array values method
+        XCTAssertSerialization(
+            of: db.insert(into: "planets")
+                .columns(["name", "color"])
+                .values([["Jupiter", "orange"],["Mars", "red"]]),
+            is: "INSERT INTO ``planets`` (``name``, ``color``) VALUES (&1, &2)"
+        )
+
+        // Test multiple values calls make multiple rows
+        XCTAssertSerialization(
+            of: self.db.insert(into: "planets")
+                .columns(["name", "color"])
+                .values(["Jupiter", "orange"])
+                .values(["Mars", "red"]),
+            is: "INSERT INTO ``planets`` (``name``, ``color``) VALUES (&1, &2), (&3, &4)"
+        )
+
+        // Test single-value input method
+        XCTAssertSerialization(
+            of: self.db.insert(into: "planets")
+                .columns(["name"])
+                .values(["Jupiter"]),
+            is: "INSERT INTO ``planets`` (``name``) VALUES (&1)"
+        )
+    }
+
+    func testInsertValuesExpression() throws {
+        // Test variadic values method
+        XCTAssertSerialization(
+            of: self.db.insert(into: "planets")
+                .columns(["name", "color"])
+                .values(SQLBind("Jupiter"), SQLBind("orange")),
+            is: "INSERT INTO ``planets`` (``name``, ``color``) VALUES (&1, &2)"
+        )
+
+        // Test array values method
+        XCTAssertSerialization(
+            of: self.db.insert(into: "planets")
+                .columns(["name", "color"])
+                .values([SQLBind("Jupiter"), SQLBind("orange")]),
+            is: "INSERT INTO ``planets`` (``name``, ``color``) VALUES (&1, &2)"
+        )
+
+        // Test nested array values method
+        XCTAssertSerialization(
+            of: self.db.insert(into: "planets")
+                .columns(["name", "color"])
+                .rows([[SQLBind("Jupiter"), SQLBind("orange")],[SQLBind("Mars"), SQLBind("red")]]),
+            is: "INSERT INTO ``planets`` (``name``, ``color``) VALUES (&1, &2), (&3, &4)"
+        )
+
+        // Test multiple values calls make multiple rows
+        XCTAssertSerialization(
+            of: self.db.insert(into: "planets")
+                .columns(["name", "color"])
+                .values(["Jupiter", "orange"])
+                .values([SQLBind("Mars"), SQLBind("red")]),
+            is: "INSERT INTO ``planets`` (``name``, ``color``) VALUES (&1, &2), (&3, &4)"
+        )
+
+        // Test single-value input method
+        XCTAssertSerialization(
+            of: self.db.insert(into: "planets")
+                .columns(["name"])
+                .values([SQLBind("Jupiter")]),
+            is: "INSERT INTO ``planets`` (``name``) VALUES (&1)"
+        )
+    }
 
     // MARK: - Upsert