Fix overwriting state with undefined on exceptions

voltrevo · Feb 29, 2024 · b6baaf7 · b6baaf7
1 parent add823c
commit b6baaf7
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/vstc/src/db_host.rs b/vstc/src/db_host.rs
@@ -191,10 +191,12 @@ impl Handler<DbRequest> for DbActor {
       }
     };
 
-    self
-      .storage
-      .set_head(storage_head_ptr(b"state"), &instance)
-      .unwrap();
+    if res.is_ok() {
+      self
+        .storage
+        .set_head(storage_head_ptr(b"state"), &instance)
+        .unwrap();
+    }
 
     // TODO: Consider more cache retention
     self.storage.clear_read_cache();