#2: Implementing receipts in squeryl storage

softwaremill · Dec 6, 2012 · 27f11c3 · 27f11c3
1 parent fa25c39
commit 27f11c3
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/...-database/src/main/scala/org/elasticmq/storage/squeryl/SquerylMessagesStorageModule.scala b/...-database/src/main/scala/org/elasticmq/storage/squeryl/SquerylMessagesStorageModule.scala
@@ -77,11 +77,17 @@ trait SquerylMessagesStorageModule {
 
     private def updateNextDelivery(message: MessageData, nextDelivery: MillisNextDelivery) = {
       inTransaction {
+        val deliveryReceipt = DeliveryReceipt.generate
+
         val updatedCount = update(messages)(m =>
           where(m.id === message.id.id and m.nextDelivery === message.nextDelivery.millis)
-                  set(m.nextDelivery := nextDelivery.millis))
+                  set(m.nextDelivery := nextDelivery.millis, m.receipt := Some(deliveryReceipt.receipt)))
 
-        if (updatedCount == 0) None else Some(message.copy(nextDelivery = nextDelivery))
+        if (updatedCount == 0) None else {
+          Some(message
+            .copy(nextDelivery = nextDelivery)
+            .copy(deliveryReceipt = Some(deliveryReceipt)))
+        }
       }
     }
   }