Catch errors in migration

navikt/dagpenger#397

Author: geiralund

src/main/kotlin/no/nav/dagpenger/inntekt/db/ArenaMappingMigrator.kt

@@ -24,6 +24,7 @@ internal class ArenaMappingMigrator(private val datasource: DataSource) {
                 """.trimIndent()
 
     fun migrate(): Int {
+        val empty = 0
         try {
             val rowsMigrated = using(sessionOf(datasource)) { session ->
                 session.transaction { transaction ->
@@ -32,13 +33,16 @@ internal class ArenaMappingMigrator(private val datasource: DataSource) {
                         transaction.run(queryOf(statement).asUpdate)
                     } else {
                         logger.info { "Could not obtain lock for migrator" }
-                        0
+                        empty
                     }
                 }
             }
 
             logger.info { "Migrated $rowsMigrated rows from inntekt_v1_arena_mapping to inntekt_v1_person_mapping" }
             return rowsMigrated
+        } catch (err: Throwable) {
+            logger.error(err) { "Failed to migrate from inntekt_v1_arena_mapping to inntekt_v1_person_mapping" }
+            return empty
         } finally {
             val unlocked = unlock()
             logger.info { "Unlocked = $unlocked for migrator" }

src/main/kotlin/no/nav/dagpenger/inntekt/db/PostgresInntektStore.kt

@@ -89,7 +89,7 @@ internal class PostgresInntektStore(private val dataSource: DataSource) : Inntek
         }
     }
 
-    private fun fetchInntektIdFromPersonMappingTable(inntektparametre: Inntektparametre): InntektId? {
+    internal fun fetchInntektIdFromPersonMappingTable(inntektparametre: Inntektparametre): InntektId? {
         @Language("sql")
         val statement: String = """
                 SELECT inntektId

src/test/kotlin/no/nav/dagpenger/inntekt/db/ArenaMappingMigratorTest.kt

@@ -42,11 +42,19 @@ internal class ArenaMappingMigratorTest() {
             val inntektStore = PostgresInntektStore(TestDataSource.instance)
 
             val ids = commands.mapNotNull { (_, command) ->
+                inntektStore.fetchInntektIdFromPersonMappingTable(
+                    inntektparametre = command.inntektparametre
+                )
+            }
+
+            val fromOld = commands.mapNotNull { (_, command) ->
                 inntektStore.getInntektId(
                     inntektparametre = command.inntektparametre
                 )
             }
 
+            fromOld shouldContainAll ids
+
             rowsMigrated shouldBeExactly numberOfInserts
             ids.size shouldBeExactly numberOfInserts
             ids shouldContainAll commands.map { it.first }
@@ -68,7 +76,7 @@ internal class ArenaMappingMigratorTest() {
             val inntektStore = PostgresInntektStore(TestDataSource.instance)
 
             val ids = commandsWithSameVedtakId.mapNotNull { (_, command) ->
-                inntektStore.getInntektId(
+                inntektStore.fetchInntektIdFromPersonMappingTable(
                     inntektparametre = command.inntektparametre
                 )
             }
@@ -102,7 +110,7 @@ internal class ArenaMappingMigratorTest() {
             val inntektStore = PostgresInntektStore(TestDataSource.instance)
 
             val ids = commands.mapNotNull { (_, command) ->
-                inntektStore.getInntektId(
+                inntektStore.fetchInntektIdFromPersonMappingTable(
                     inntektparametre = command.inntektparametre
                 )
             }