[TRELLO-2880] Creating a website should be transactionnal

betagouv · Jan 28, 2025 · 6656e6c · 6656e6c
1 parent 8bb2140
commit 6656e6c
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/app/repositories/website/WebsiteRepository.scala b/app/repositories/website/WebsiteRepository.scala
@@ -38,9 +38,9 @@ class WebsiteRepository(
 
   import dbConfig._
 
-  override def validateAndCreate(newWebsite: Website): Future[Website] =
-    db.run(
-      table
+  override def validateAndCreate(newWebsite: Website): Future[Website] = db.run(
+    (for {
+      maybeWebsite <- table
         .filter(_.host === newWebsite.host)
         .filter { website =>
           val hasBeenAlreadyIdentifiedByConso =
@@ -55,10 +55,12 @@ class WebsiteRepository(
         }
         .result
         .headOption
-    ).flatMap {
-      case Some(website) => Future.successful(website)
-      case None          => create(newWebsite)
-    }
+      website <- maybeWebsite match {
+        case Some(website) => DBIO.successful(website)
+        case None          => table returning table += newWebsite
+      }
+    } yield website).transactionally
+  )
 
   override def searchValidAssociationByHost(host: String): Future[Seq[Website]] =
     db.run(