Add address to NoToken and NoGas status properties

golemfactory · Nov 7, 2023 · f355998 · f355998
1 parent 484ccd2
commit f355998
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/crates/erc20_payment_lib/src/runtime.rs b/crates/erc20_payment_lib/src/runtime.rs
@@ -191,10 +191,12 @@ pub enum StatusProperty {
     },
     NoGas {
         chain_id: i64,
+        address: String,
         missing_gas: Decimal,
     },
     NoToken {
         chain_id: i64,
+        address: String,
         missing_token: Decimal,
     },
 }
@@ -235,27 +237,31 @@ impl StatusTracker {
                 (
                     NoGas {
                         chain_id: id1,
+                        address: addr1,
                         missing_gas: old_missing,
                     },
                     NoGas {
                         chain_id: id2,
+                        address: addr2,
                         missing_gas: new_missing,
                     },
-                ) if id1 == id2 => {
+                ) if id1 == id2 && addr1 == addr2 => {
                     *old_missing = *new_missing;
                     return true;
                 }
                 // NoToken statuses add up
                 (
                     NoToken {
                         chain_id: id1,
+                        address: addr1,
                         missing_token: old_missing,
                     },
                     NoToken {
                         chain_id: id2,
+                        address: addr2,
                         missing_token: new_missing,
                     },
-                ) if id1 == id2 => {
+                ) if id1 == id2 && addr1 == addr2 => {
                     *old_missing = *new_missing;
                     return true;
                 }
@@ -315,6 +321,7 @@ impl StatusTracker {
                             status2.lock().await.deref_mut(),
                             StatusProperty::NoGas {
                                 chain_id: details.tx.chain_id,
+                                address: details.tx.from_addr.clone(),
                                 missing_gas,
                             },
                         )
@@ -327,6 +334,7 @@ impl StatusTracker {
                             status2.lock().await.deref_mut(),
                             StatusProperty::NoToken {
                                 chain_id: details.tx.chain_id,
+                                address: details.tx.from_addr.clone(),
                                 missing_token,
                             },
                         )