Legg på transaction på utbetalingsstatistikk

Author: hestad

app/src/main/kotlin/no/nav/tiltakspenger/vedtak/context/UtbetalingContext.kt

@@ -40,6 +40,7 @@ open class UtbetalingContext(
             utbetalingsvedtakRepo = utbetalingsvedtakRepo,
             rammevedtakRepo = rammevedtakRepo,
             statistikkStønadRepo = statistikkStønadRepo,
+            sessionFactory = sessionFactory,
         )
     }
     val hentUtbetalingsvedtakService by lazy {

app/src/main/kotlin/no/nav/tiltakspenger/vedtak/repository/statistikk/stønad/StatistikkStønadPostgresRepo.kt

@@ -102,7 +102,7 @@ class StatistikkStønadPostgresRepo(
                     "arsak" to dto.årsak,
                     "posteringsDato" to dto.posteringDato,
                     "gyldigFraDato" to dto.gyldigFraDatoPostering,
-                    "gyldigTilDat" to dto.gyldigTilDatoPostering,
+                    "gyldigTilDato" to dto.gyldigTilDatoPostering,
                 ),
             ).asUpdate,
         )

app/src/main/kotlin/no/nav/tiltakspenger/vedtak/repository/utbetaling/UtbetalingsvedtakPostgresRepo.kt

@@ -8,6 +8,7 @@ import no.nav.tiltakspenger.libs.common.Fnr
 import no.nav.tiltakspenger.libs.common.MeldekortId
 import no.nav.tiltakspenger.libs.common.SakId
 import no.nav.tiltakspenger.libs.common.VedtakId
+import no.nav.tiltakspenger.libs.persistering.domene.TransactionContext
 import no.nav.tiltakspenger.libs.persistering.infrastruktur.PostgresSessionFactory
 import no.nav.tiltakspenger.meldekort.domene.Meldekort.UtfyltMeldekort
 import no.nav.tiltakspenger.meldekort.domene.Meldeperiode
@@ -21,8 +22,8 @@ import no.nav.tiltakspenger.vedtak.repository.meldekort.toMeldekortStatus
 internal class UtbetalingsvedtakPostgresRepo(
     private val sessionFactory: PostgresSessionFactory,
 ) : UtbetalingsvedtakRepo {
-    override fun lagre(vedtak: Utbetalingsvedtak) {
-        sessionFactory.withSession { session ->
+    override fun lagre(vedtak: Utbetalingsvedtak, context: TransactionContext?) {
+        sessionFactory.withSession(context) { session ->
             session.run(
                 queryOf(
                     """

domene/src/main/kotlin/no/nav/tiltakspenger/utbetaling/ports/UtbetalingsvedtakRepo.kt

@@ -3,12 +3,13 @@ package no.nav.tiltakspenger.utbetaling.ports
 import no.nav.tiltakspenger.libs.common.BehandlingId
 import no.nav.tiltakspenger.libs.common.SakId
 import no.nav.tiltakspenger.libs.common.VedtakId
+import no.nav.tiltakspenger.libs.persistering.domene.TransactionContext
 import no.nav.tiltakspenger.meldekort.domene.Meldekort
 import no.nav.tiltakspenger.saksbehandling.ports.SendtUtbetaling
 import no.nav.tiltakspenger.utbetaling.domene.Utbetalingsvedtak
 
 interface UtbetalingsvedtakRepo {
-    fun lagre(vedtak: Utbetalingsvedtak)
+    fun lagre(vedtak: Utbetalingsvedtak, context: TransactionContext? = null)
 
     fun markerSendtTilUtbetaling(
         vedtakId: VedtakId,

domene/src/main/kotlin/no/nav/tiltakspenger/utbetaling/service/OpprettUtbetalingsvedtakService.kt

@@ -2,6 +2,7 @@ package no.nav.tiltakspenger.utbetaling.service
 
 import arrow.core.Either
 import mu.KotlinLogging
+import no.nav.tiltakspenger.libs.persistering.domene.SessionFactory
 import no.nav.tiltakspenger.meldekort.domene.Meldekort.UtfyltMeldekort
 import no.nav.tiltakspenger.saksbehandling.domene.vedtak.Rammevedtak
 import no.nav.tiltakspenger.saksbehandling.ports.RammevedtakRepo
@@ -21,6 +22,7 @@ class OpprettUtbetalingsvedtakService(
     private val utbetalingsvedtakRepo: UtbetalingsvedtakRepo,
     private val rammevedtakRepo: RammevedtakRepo,
     private val statistikkStønadRepo: StatistikkStønadRepo,
+    private val sessionFactory: SessionFactory,
 ) {
     val logger = KotlinLogging.logger { }
 
@@ -34,9 +36,12 @@ class OpprettUtbetalingsvedtakService(
                     // TODO pre-mvp jah: Lag en type som passer på at disse er sortert riktig. Dette vil være en del av arbeidet å sentralisere Sak.kt
                     val eksisterendeUtbetalingsvedtak = utbetalingsvedtakRepo.hentForSakId(meldekort.sakId)
                     meldekort.tilUtbetalingsperiode(rammevedtak, eksisterendeUtbetalingsvedtak.lastOrNull()?.id).also {
-                        utbetalingsvedtakRepo.lagre(it)
-                        // TODO pre-mvp jah: lager en herlig transaksjon som lagrer statistikk og utbetalingsvedtak i samme transaksjon.
-                        statistikkStønadRepo.lagre(it.tilStatistikk())
+                        sessionFactory.withTransactionContext { tx ->
+
+                            utbetalingsvedtakRepo.lagre(it, tx)
+                            // TODO pre-mvp jah: lager en herlig transaksjon som lagrer statistikk og utbetalingsvedtak i samme transaksjon.
+                            statistikkStønadRepo.lagre(it.tilStatistikk(), tx)
+                        }
                     }
                 }.onLeft {
                     logger.error(it) { "Feil ved opprettelse av utbetalingsvedtak for meldekortId=${meldekort.id}" }

test-common/src/main/kotlin/no/nav/tiltakspenger/fakes/repos/UtbetalingsvedtakFakeRepo.kt

@@ -4,6 +4,7 @@ import arrow.atomic.Atomic
 import no.nav.tiltakspenger.libs.common.BehandlingId
 import no.nav.tiltakspenger.libs.common.SakId
 import no.nav.tiltakspenger.libs.common.VedtakId
+import no.nav.tiltakspenger.libs.persistering.domene.TransactionContext
 import no.nav.tiltakspenger.meldekort.domene.Meldekort.UtfyltMeldekort
 import no.nav.tiltakspenger.saksbehandling.ports.SendtUtbetaling
 import no.nav.tiltakspenger.utbetaling.domene.Utbetalingsvedtak
@@ -15,7 +16,7 @@ class UtbetalingsvedtakFakeRepo(
 ) : UtbetalingsvedtakRepo {
     private val data = Atomic(mutableMapOf<VedtakId, Utbetalingsvedtak>())
 
-    override fun lagre(vedtak: Utbetalingsvedtak) {
+    override fun lagre(vedtak: Utbetalingsvedtak, context: TransactionContext?) {
         data.get()[vedtak.id] = vedtak
     }
 
@@ -44,7 +45,9 @@ class UtbetalingsvedtakFakeRepo(
             data.get().values.none { utbetalingsvedtak -> utbetalingsvedtak.meldekortperiode.meldekortId == it.id }
         }
 
-    override fun hentUtbetalingsvedtakForUtsjekk(limit: Int): List<Utbetalingsvedtak> = data.get().values.filter { it.sendtTilUtbetaling }
+    override fun hentUtbetalingsvedtakForUtsjekk(limit: Int): List<Utbetalingsvedtak> =
+        data.get().values.filter { it.sendtTilUtbetaling }
 
-    override fun hentUtbetalingsvedtakForDokument(limit: Int): List<Utbetalingsvedtak> = data.get().values.filter { it.sendtTilDokument }
+    override fun hentUtbetalingsvedtakForDokument(limit: Int): List<Utbetalingsvedtak> =
+        data.get().values.filter { it.sendtTilDokument }
 }