evm_discard_context removes the context queue

DeFiCh · Jun 9, 2023 · 96e3f5c · 96e3f5c
1 parent 94b2380
commit 96e3f5c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 11 deletions.
diff --git a/lib/ain-evm/src/evm.rs b/lib/ain-evm/src/evm.rs
@@ -239,6 +239,10 @@ impl EVMHandler {
         self.tx_queues.clear(context)?;
         Ok(())
     }
+
+    pub fn remove(&self, context: u64) {
+        self.tx_queues.remove(context);
+    }
 }
 
 impl EVMHandler {

diff --git a/lib/ain-rs-exports/src/lib.rs b/lib/ain-rs-exports/src/lib.rs
@@ -68,7 +68,7 @@ pub mod ffi {
         fn evm_try_prevalidate_raw_tx(result: &mut RustRes, tx: &str) -> Result<ValidateTxResult>;
 
         fn evm_get_context() -> u64;
-        fn evm_discard_context(context: u64) -> Result<()>;
+        fn evm_discard_context(context: u64);
         fn evm_try_queue_tx(
             result: &mut RustRes,
             context: u64,
@@ -340,16 +340,8 @@ pub fn evm_get_context() -> u64 {
 ///
 /// * `context` - The context queue number.
 ///
-/// # Errors
-///
-/// Returns an Error if the context does not match any existing queue.
-/// # Returns
-///
-/// Returns `Ok(())` on success.
-fn evm_discard_context(context: u64) -> Result<(), Box<dyn Error>> {
-    // TODO discard
-    RUNTIME.handlers.evm.clear(context)?;
-    Ok(())
+fn evm_discard_context(context: u64) {
+    RUNTIME.handlers.evm.remove(context)
 }
 
 /// Add an EVM transaction to a specific queue.