Implementer get/setPeriodeArray

Author: peterbb

dbconnect/src/main/kotlin/no/nav/aap/komponenter/dbconnect/Params.kt

@@ -105,4 +105,9 @@ public class Params internal constructor(
         val array = connection.createArrayOf("BIGINT", longs.toTypedArray())
         preparedStatement.setArray(index, array)
     }
+
+    public fun setPeriodeArray(index: Int, perioder: List<Periode>) {
+        val array = connection.createArrayOf("daterange", perioder.map(DaterangeParser::toSQL).toTypedArray())
+        preparedStatement.setArray(index, array)
+    }
 }

dbconnect/src/main/kotlin/no/nav/aap/komponenter/dbconnect/Row.kt

@@ -162,6 +162,11 @@ public class Row internal constructor(private val resultSet: ResultSet) {
         return DaterangeParser.fromSQL(dateRange)
     }
 
+    public fun getPeriodeArray(columnLabel: String): List<Periode> {
+        return (resultSet.getArray(columnLabel).array as Array<*>)
+            .map { DaterangeParser.fromSQL(it.toString()) }
+    }
+
     /**
      * Feltet [elementType] må være av en type JDBC returnerer.
      * Example:

dbconnect/src/test/kotlin/no/nav/aap/komponenter/dbconnect/ParamsOgRowTest.kt

@@ -402,6 +402,35 @@ internal class ParamsOgRowTest {
         }
     }
 
+    @Test
+    fun `Skriver og leser med periode-array`() {
+        InitTestDatabase.dataSource.transaction { connection ->
+            val perioderInn = listOf(
+                Periode(LocalDate.of(1990, 6, 1), LocalDate.of(2200, 1, 1)),
+                Periode(LocalDate.of(2020, 1, 5), LocalDate.of(2020, 5, 3)),
+            )
+            connection.execute(
+                """
+                    INSERT INTO TEST_PERIODE_ARRAY(TEST)
+                    VALUES (?::daterange[])
+                """,
+            ) {
+                setParams {
+                    setPeriodeArray(1, perioderInn)
+                }
+            }
+
+
+            val perioderUt = connection.queryFirst("SELECT TEST FROM TEST_PERIODE_ARRAY") {
+                setRowMapper { row ->
+                    row.getPeriodeArray("TEST")
+                }
+            }
+            assertThat(perioderUt)
+                .isEqualTo(perioderInn)
+        }
+    }
+
     @Test
     fun `CURRENT_TIMESTAMP i postgres (UTC) matcher LocalDateTime now() i pod (Europe Oslo)`() {
         InitTestDatabase.dataSource.transaction { connection ->

dbconnect/src/test/resources/flyway/V0.4__test_periode_array.sql

@@ -0,0 +1,5 @@
+CREATE TABLE TEST_PERIODE_ARRAY
+(
+    TEST daterange[] NOT NULL
+);
+