beredskapstatistikk (#14)

* beredskapstatistikk

Author: rannveigskjerve

src/main/kotlin/no/nav/tms/statistikk/varsel/VarselAktivertSink.kt

@@ -4,11 +4,12 @@ import com.fasterxml.jackson.module.kotlin.readValue
 import io.github.oshai.kotlinlogging.KotlinLogging
 import no.nav.helse.rapids_rivers.*
 import no.nav.tms.statistikk.defaultDeserializer
+import no.nav.tms.statistikk.varsel.BeredskapMetadata.Companion.beredskapMetadata
 
 class VarselAktivertSink(
     rapidsConnection: RapidsConnection,
     private val repository: VarselRepository
-): River.PacketListener {
+) : River.PacketListener {
 
     private val log = KotlinLogging.logger {}
 
@@ -27,18 +28,38 @@ class VarselAktivertSink(
                     "innhold",
                     "sensitivitet"
                 )
-                it.interestedIn("aktivFremTil","eksternVarslingBestilling")
+                it.interestedIn("aktivFremTil", "eksternVarslingBestilling", "metadata")
             }
         }.register(this)
     }
 
     override fun onPacket(packet: JsonMessage, context: MessageContext) {
         val aktivertVarsel: AktivertVarsel = objectMapper.readValue(packet.toJson())
-
         repository.insertVarsel(aktivertVarsel)
+        packet.beredskapMetadata()?.let {
+            repository.insertBeredskapReference(it, packet["varselId"].asText())
+        }
     }
 
     override fun onError(problems: MessageProblems, context: MessageContext) {
         log.info { problems.toString() }
     }
 }
+
+class BeredskapMetadata private constructor(
+    val tittel: String?,
+    val ref: String?
+) {
+    val isEmpty = tittel == null && ref == null
+
+    companion object {
+        fun JsonMessage.beredskapMetadata() = BeredskapMetadata(
+            this["metadata"].findValue("beredskap_tittel")?.asText(),
+            this["metadata"].findValue("beredskap_ref")?.asText()
+        ).let {
+            if (it.isEmpty)
+                null
+            else it
+        }
+    }
+}

src/main/kotlin/no/nav/tms/statistikk/varsel/VarselRepository.kt

@@ -5,8 +5,10 @@ import no.nav.tms.statistikk.database.Database
 import no.nav.tms.statistikk.toUtcLocalDateTime
 
 class VarselRepository(private val database: Database) {
-    fun insertVarsel(varsel: AktivertVarsel) = database.update {
-        queryOf("""
+    fun insertVarsel(varsel: AktivertVarsel) {
+        database.update {
+            queryOf(
+                """
             insert into varsel(
                 eventId,
                 ident,
@@ -35,25 +37,27 @@ class VarselRepository(private val database: Database) {
                 :eksternVarslingBestilt
             ) on conflict do nothing
         """,
-            mapOf(
-                "eventId" to varsel.varselId,
-                "ident" to varsel.ident,
-                "type" to varsel.type,
-                "namespace" to varsel.produsent.namespace,
-                "appnavn" to varsel.produsent.appnavn,
-                "tekstlengde" to varsel.tekstLengde,
-                "lenke" to varsel.harLenke,
-                "sikkerhetsnivaa" to varsel.sensitivitet.loginLevel,
-                "aktiv" to true,
-                "frist" to varsel.frist,
-                "forstBehandlet" to varsel.opprettet.toUtcLocalDateTime(),
-                "eksternVarslingBestilt" to varsel.eksternVarslingBestilt
+                mapOf(
+                    "eventId" to varsel.varselId,
+                    "ident" to varsel.ident,
+                    "type" to varsel.type,
+                    "namespace" to varsel.produsent.namespace,
+                    "appnavn" to varsel.produsent.appnavn,
+                    "tekstlengde" to varsel.tekstLengde,
+                    "lenke" to varsel.harLenke,
+                    "sikkerhetsnivaa" to varsel.sensitivitet.loginLevel,
+                    "aktiv" to true,
+                    "frist" to varsel.frist,
+                    "forstBehandlet" to varsel.opprettet.toUtcLocalDateTime(),
+                    "eksternVarslingBestilt" to varsel.eksternVarslingBestilt
+                )
             )
-        )
+        }
     }
 
     fun updateVarsel(varselInaktivert: VarselInaktivert) = database.update {
-        queryOf("""
+        queryOf(
+            """
             update varsel set aktiv = false, inaktivertTidspunkt = :tidspunkt, inaktivertKilde = :kilde
             where eventId = :eventId
         """,
@@ -66,7 +70,8 @@ class VarselRepository(private val database: Database) {
     }
 
     fun registerVarselPerDag(varselPerDag: VarselPerDag) = database.update {
-            queryOf("""
+        queryOf(
+            """
                 insert into varsler_per_dag(dato, ident, type, ekstern_varsling, antall)
                     values(:dato, :ident, :type, :eksternVarsling, 1)
                 on conflict (dato, ident, type, ekstern_varsling) do update set antall = varsler_per_dag.antall + 1
@@ -79,4 +84,32 @@ class VarselRepository(private val database: Database) {
             )
         )
     }
+
+    fun insertBeredskapReference(beredskapMetadata: BeredskapMetadata, varselId: String) {
+        database.update {
+            queryOf(
+                """
+               insert into beredskapsvarsel (beredskap_ref,beredskap_tittel)
+               values(:ref, :tittel)
+               on conflict do nothing
+           """.trimIndent(),
+                mapOf(
+                    "ref" to beredskapMetadata.ref,
+                    "tittel" to beredskapMetadata.tittel
+                )
+            )
+        }
+
+        database.update {
+            queryOf(
+                """update varsel set beredskap_ref=:ref 
+                |where eventId = :varselId""".trimMargin(),
+                mapOf(
+                    "ref" to beredskapMetadata.ref,
+                    "varselId" to varselId
+                )
+            )
+        }
+
+    }
 }

src/main/kotlin/no/nav/tms/statistikk/varsel/varsel.kt

@@ -13,7 +13,7 @@ data class AktivertVarsel(
     val produsent: Produsent,
     val eksternVarslingBestilling: EksternVarslingBestilling? = null,
     val opprettet: ZonedDateTime,
-    val aktivFremTil: ZonedDateTime? = null
+    val aktivFremTil: ZonedDateTime? = null,
 ) {
     val tekstLengde get() = innhold.defaultTekst().tekst.length
     val harLenke get() = innhold.link.isNullOrBlank().not()

src/main/resources/db/migration/V1.0.10__beredskapsvarsel.sql

@@ -0,0 +1,7 @@
+create table beredskapsvarsel
+(
+    id             serial primary key,
+    beredskap_tittel varchar not null,
+    varselId       varchar not null references varsel (eventid)
+)
+

src/main/resources/db/migration/V1.0.11__beredskapsvarselref.sql

@@ -0,0 +1,14 @@
+drop table if exists beredskapsvarsel;
+
+create table beredskapsvarsel
+(
+    id             serial primary key,
+    beredskap_tittel varchar not null,
+    beredskap_ref varchar not null unique
+);
+
+alter table varsel
+add column beredskap_ref varchar default null;
+
+
+

src/test/kotlin/LocalPostgresDatabase.kt

@@ -17,7 +17,9 @@ class LocalPostgresDatabase private constructor() : Database {
         }
 
         fun cleanDb(): LocalPostgresDatabase {
+
             instance.cleanTables(
+                "beredskapsvarsel",
                 "varsel",
                 "microfrontends",
                 "utkast",

src/test/kotlin/no/nav/tms/statistikk/varsel/VarselAktivertSinkTest.kt

@@ -1,11 +1,13 @@
 package no.nav.tms.statistikk.varsel
 
+import assert
 import io.kotest.matchers.shouldBe
 import io.kotest.matchers.shouldNotBe
 import kotliquery.queryOf
 import no.nav.helse.rapids_rivers.testsupport.TestRapid
 import no.nav.tms.statistikk.database.DateTimeHelper
 import no.nav.tms.statistikk.varsel.VarselTestData.VarselType.beskjed
+import no.nav.tms.statistikk.varsel.VarselTestData.addBeredskapMetadata
 import org.junit.jupiter.api.AfterEach
 import org.junit.jupiter.api.BeforeAll
 import org.junit.jupiter.api.Test
@@ -23,6 +25,7 @@ internal class VarselAktivertSinkTest {
 
     @AfterEach
     fun cleanDb() {
+        database.update { queryOf("delete from beredskapsvarsel") }
         database.update { queryOf("delete from varsel") }
     }
 
@@ -93,13 +96,38 @@ internal class VarselAktivertSinkTest {
         database.getVarsel(eventId2)?.lenke shouldBe true
     }
 
+    @Test
+    fun `lagrer beredskapmetadata`() {
+        val eventId1 = UUID.randomUUID().toString()
+        val eventId2 = UUID.randomUUID().toString()
+
+        val medBeredskapsTittel =
+            VarselTestData.varselAktivertMessage(varselId = eventId1).addBeredskapMetadata("Something happened")
+        val utenBeredskapsTittel = VarselTestData.varselAktivertMessage(varselId = eventId2)
+
+        testRapid.sendTestMessage(utenBeredskapsTittel)
+        testRapid.sendTestMessage(medBeredskapsTittel)
+
+        database.getVarsel(eventId2).assert {
+            require(this != null)
+            beredskapstittel shouldBe null
+        }
+
+        database.getVarsel(eventId1).assert {
+            require(this != null)
+            beredskapstittel shouldBe "Something happened"
+            beredskapsRef shouldBe "123"
+        }
+    }
+
     @Test
     fun `tolker frist riktig`() {
         val eventId1 = UUID.randomUUID().toString()
         val eventId2 = UUID.randomUUID().toString()
 
         val utenFrist = VarselTestData.varselAktivertMessage(varselId = eventId1, aktivFremTil = null)
-        val medFrist = VarselTestData.varselAktivertMessage(varselId = eventId2, aktivFremTil = DateTimeHelper.nowAtUtcZ())
+        val medFrist =
+            VarselTestData.varselAktivertMessage(varselId = eventId2, aktivFremTil = DateTimeHelper.nowAtUtcZ())
 
         testRapid.sendTestMessage(utenFrist)
         testRapid.sendTestMessage(medFrist)

src/test/kotlin/no/nav/tms/statistikk/varsel/VarselInaktivertSinkTest.kt

@@ -4,6 +4,7 @@ import io.kotest.matchers.shouldBe
 import kotliquery.queryOf
 import no.nav.helse.rapids_rivers.testsupport.TestRapid
 import no.nav.tms.statistikk.database.DateTimeHelper
+import no.nav.tms.statistikk.varsel.VarselTestData.addBeredskapMetadata
 import org.junit.jupiter.api.AfterEach
 import org.junit.jupiter.api.BeforeAll
 import org.junit.jupiter.api.Test
@@ -24,6 +25,7 @@ internal class VarselInaktivertSinkTest {
 
     @AfterEach
     fun cleanDb() {
+        database.update { queryOf("delete from beredskapsvarsel") }
         database.update { queryOf("delete from varsel") }
     }
 
@@ -33,13 +35,21 @@ internal class VarselInaktivertSinkTest {
         val kilde = "bruker"
         val inaktivertTidspunkt = DateTimeHelper.nowAtUtcZ().minusHours(1)
 
-        testRapid.sendTestMessage(VarselTestData.varselAktivertMessage(varselId = eventId))
-        testRapid.sendTestMessage(VarselTestData.varselInaktivertMessage(varselId = eventId, kilde = kilde, tidspunkt = inaktivertTidspunkt))
+        testRapid.sendTestMessage(VarselTestData.varselAktivertMessage(varselId = eventId).addBeredskapMetadata("oups"))
+        testRapid.sendTestMessage(
+            VarselTestData.varselInaktivertMessage(
+                varselId = eventId,
+                kilde = kilde,
+                tidspunkt = inaktivertTidspunkt
+            )
+        )
 
         val varsel = database.getVarsel(eventId)!!
 
         varsel.aktiv shouldBe false
         varsel.inaktivertTidspunkt shouldBe inaktivertTidspunkt.toLocalDateTime()
         varsel.inaktivertKilde shouldBe kilde
+        varsel.beredskapstittel shouldBe "oups"
+        varsel.beredskapsRef shouldBe "123"
     }
 }

src/test/kotlin/no/nav/tms/statistikk/varsel/VarselTestData.kt

@@ -22,18 +22,18 @@ object VarselTestData {
         sensitivitet: Sensitivitet = Sensitivitet.Substantial,
         sistOppdatert: ZonedDateTime = nowAtUtcZ(),
         aktivFremTil: ZonedDateTime? = nowAtUtcZ().plusWeeks(2),
-    )
-    = """
+    ) = """
         {
             "@event_name": "opprettet",
             "ident": "$ident",
             "type": "${type.name}",
-            ${if (eksternVarsling) {
-                    """"eksternVarslingBestilling": { "prefererteKanaler": [] },"""
-                } else {
-                    ""
-                }
-            }
+            ${
+        if (eksternVarsling) {
+            """"eksternVarslingBestilling": { "prefererteKanaler": [] },"""
+        } else {
+            ""
+        }
+    }
             "opprettet": "$opprettet",
             "varselId": "$varselId",
             "produsent": {
@@ -62,6 +62,18 @@ object VarselTestData {
         }
     """
 
+    internal fun String.addBeredskapMetadata(beredskapTittel: String, beredskapRef:String = "123") =
+        this.substring(0, lastIndexOf("}")).let {
+            """$it,                    
+            "metadata": {
+              "beredskap_tittel": "$beredskapTittel",
+              "beredskap_ref": "$beredskapRef"
+              }    
+            }
+                """.trimMargin()
+        }
+
+
     fun varselInaktivertMessage(
         type: VarselType = beskjed,
         varselId: String = UUID.randomUUID().toString(),
@@ -70,8 +82,7 @@ object VarselTestData {
         appnavn: String = "appnavn",
         kilde: String = "produsent",
         tidspunkt: ZonedDateTime = nowAtUtcZ()
-    )
-    = """
+    ) = """
         {
             "@event_name": "inaktivert",
             "varselId": "$varselId",
@@ -94,8 +105,7 @@ object VarselTestData {
         namespace: String = "namespace",
         appnavn: String = "appnavn",
         tidspunkt: ZonedDateTime = nowAtUtcZ()
-    )
-    = """
+    ) = """
        {
             "@event_name": "eksternStatusOppdatert",
             "status": "sendt",