provide take state interface

nekomoto911 · nekomoto911 · commit 0928c1c577ec · 2024-11-14T14:56:18.000+08:00
diff --git a/src/scheduler.rs b/src/scheduler.rs
@@ -681,4 +681,10 @@ where
     pub fn force_sequential_execute(&mut self) -> Result<ExecuteOutput, GrevmError<DB::Error>> {
         self.evm_execute(Some(true), false, None)
     }
+
+    /// Take the state of the scheduler.
+    /// It is typically called after the execution.
+    pub fn take_state(self) -> Box<State> {
+        Arc::try_unwrap(self.database).ok().unwrap().state
+    }
 }

Original file line number	Diff line number	Diff line change
`@@ -681,4 +681,10 @@ where`
`681`	`681`	`pub fn force_sequential_execute(&mut self) -> Result<ExecuteOutput, GrevmError<DB::Error>> {`
`682`	`682`	`self.evm_execute(Some(true), false, None)`
`683`	`683`	`}`
	`684`	`+`
	`685`	`+ /// Take the state of the scheduler.`
	`686`	`+ /// It is typically called after the execution.`
	`687`	`+ pub fn take_state(self) -> Box<State> {`
	`688`	`+ Arc::try_unwrap(self.database).ok().unwrap().state`
	`689`	`+ }`
`684`	`690`	`}`