Merge pull request #2591 from get10101/chore/update-hodl-invoice-stat…

…e-on-lnd-event

chore: Set hodl invoice to `Settled` on lnd event

coordinator/src/db/hodl_invoice.rs

@@ -69,6 +69,19 @@ pub fn get_r_hash_by_order_id(conn: &mut PgConnection, order_id: Uuid) -> QueryR
         .get_result(conn)
 }
 
+/// Returns the pre image of the hodl invoice associated with the order id
+/// If the hodl invoice can not be found a [`Not Found`] error is returned
+/// If the hodl invoice is found the pre_image is optional, as it might have not yet been set.
+pub fn get_pre_image_by_order_id(
+    conn: &mut PgConnection,
+    order_id: Uuid,
+) -> QueryResult<Option<String>> {
+    hodl_invoices::table
+        .filter(hodl_invoices::order_id.eq(order_id))
+        .select(hodl_invoices::pre_image)
+        .get_result(conn)
+}
+
 pub fn update_hodl_invoice_to_accepted(
     conn: &mut PgConnection,
     hash: &str,
@@ -91,10 +104,10 @@ pub fn update_hodl_invoice_to_accepted(
 
 pub fn update_hodl_invoice_to_settled(
     conn: &mut PgConnection,
-    order_id: Uuid,
+    r_hash: String,
 ) -> QueryResult<Option<String>> {
     diesel::update(hodl_invoices::table)
-        .filter(hodl_invoices::order_id.eq(order_id))
+        .filter(hodl_invoices::r_hash.eq(r_hash))
         .set((
             hodl_invoices::updated_at.eq(OffsetDateTime::now_utc()),
             hodl_invoices::invoice_state.eq(InvoiceState::Settled),

coordinator/src/node/invoice.rs

@@ -33,6 +33,24 @@ pub fn spawn_invoice_watch(
                     }
                     InvoiceState::Settled => {
                         tracing::info!(%trader_pubkey, r_hash, "Accepted hodl invoice has been settled.");
+                        if let Err(e) = spawn_blocking({
+                            let r_hash = r_hash.clone();
+                            move || {
+                                let mut conn = pool.get()?;
+                                db::hodl_invoice::update_hodl_invoice_to_settled(
+                                    &mut conn, r_hash,
+                                )?;
+                                anyhow::Ok(())
+                            }
+                        })
+                        .await
+                        .expect("task to finish")
+                        {
+                            tracing::error!(
+                                r_hash,
+                                "Failed to set hodl invoice to failed. Error: {e:#}"
+                            );
+                        }
                         break;
                     }
                     InvoiceState::Canceled => {

coordinator/src/trade/mod.rs

@@ -225,8 +225,7 @@ impl TradeExecutor {
             let pool = self.node.pool.clone();
             move || {
                 let mut conn = pool.get()?;
-                let pre_image =
-                    db::hodl_invoice::update_hodl_invoice_to_settled(&mut conn, order_id)?;
+                let pre_image = db::hodl_invoice::get_pre_image_by_order_id(&mut conn, order_id)?;
 
                 anyhow::Ok(pre_image)
             }