Use a sequence in copy in (#136)

* Fix test results

Fix test results

Use DateTimePeriod for Interval

* Update copy

Co-authored-by: hfhbd <[email protected]>

Author: hfhbd

postgres-native-sqldelight-driver/src/commonMain/kotlin/app/softwork/sqldelight/postgresdriver/PostgresNativeDriver.kt

@@ -276,10 +276,15 @@ public class PostgresNativeDriver(
 
     // Custom functions
 
-    public fun copy(stdin: String): Long {
-        val status = PQputCopyData(conn, stdin, stdin.encodeToByteArray().size)
-        check(status == 1) {
-            conn.error()
+    /**
+     * Each element of stdin can be up to 2 GB.
+     */
+    public fun copy(stdin: Sequence<String>): Long {
+        for (stdin in stdin) {
+            val status = PQputCopyData(conn, stdin, stdin.encodeToByteArray().size)
+            check(status == 1) {
+                conn.error()
+            }
         }
         val end = PQputCopyEnd(conn, null)
         check(end == 1) {

testing-sqldelight/src/commonTest/kotlin/app/softwork/sqldelight/postgresdriver/PostgresNativeDriverTest.kt renamed to testing-sqldelight/src/commonTest/kotlin/app/softwork/sqldelight/postgresdriver/PostgresNativeSqldelightDriverTest.kt

@@ -10,7 +10,7 @@ import kotlin.test.*
 import kotlin.time.Duration.Companion.seconds
 
 @ExperimentalCoroutinesApi
-class PostgresNativeDriverTest {
+class PostgresNativeSqldelightDriverTest {
     private val driver = PostgresNativeDriver(
         host = "localhost",
         port = 5432,
@@ -53,8 +53,11 @@ class PostgresNativeDriverTest {
         val queries = NativePostgres(driver).fooQueries
         NativePostgres.Schema.migrate(driver, 0, NativePostgres.Schema.version)
         queries.startCopy()
-        val result =
-            driver.copy("42,answer,2020-12-12,12:42:00.0000,2014-08-01T12:01:02.0000,1970-01-01T00:00:00.010Z,P45Y6M42DT42H42M42.424242S,00000000-0000-0000-0000-000000000000")
+        val result = driver.copy(
+            sequenceOf(
+                "42,answer,2020-12-12,12:42:00.0000,2014-08-01T12:01:02.0000,1970-01-01T00:00:00.010Z,P45Y6M42DT42H42M42.424242S,00000000-0000-0000-0000-000000000000"
+            )
+        )
         assertEquals(1, result)
         val foo = Foo(
             a = 42,

testing/src/commonTest/kotlin/app/softwork/sqldelight/postgresdriver/PostgresNativeSqlDelightDriverTest.kt renamed to testing/src/commonTest/kotlin/app/softwork/sqldelight/postgresdriver/PostgresNativeDriverTest.kt

@@ -8,7 +8,7 @@ import kotlin.test.*
 import kotlin.time.Duration.Companion.seconds
 
 @ExperimentalCoroutinesApi
-class PostgresNativeSqlDelightDriverTest {
+class PostgresNativeDriverTest {
     @Test
     fun simpleTest() = runTest {
         val driver = PostgresNativeDriver(
@@ -227,8 +227,18 @@ class PostgresNativeSqlDelightDriverTest {
         assertEquals(0, driver.execute(null, "DROP TABLE IF EXISTS copying;", parameters = 0).value)
         assertEquals(0, driver.execute(null, "CREATE TABLE copying(a int primary key);", parameters = 0).value)
         driver.execute(-42, "COPY copying FROM STDIN (FORMAT CSV);", 0)
-        val results = driver.copy("1\n2\n")
-        assertEquals(2, results)
+        val results = driver.copy(sequenceOf("1\n2\n", "3\n4\n"))
+        assertEquals(4, results)
+        assertEquals(
+            listOf(1, 2, 3, 4),
+            driver.executeQuery(null, "SELECT * FROM copying", parameters = 0, binders = null, mapper = {
+                buildList {
+                    while (it.next()) {
+                        add(it.getLong(0)!!.toInt())
+                    }
+                }
+            }).value
+        )
     }
 
     @Test