update onpersist and oncommit (neo-project#461)

* update method * More fixes Co-authored-by: Erik Zhang <[email protected]>
joeqian10 · Apr 7, 2021 · 527b5c6 · 527b5c6
1 parent 2cc5f87
commit 527b5c6
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/src/RpcNep17Tracker/RpcNep17Tracker.cs b/src/RpcNep17Tracker/RpcNep17Tracker.cs
@@ -144,7 +144,7 @@ private void HandleNotification(StoreView snapshot, IVerifiable scriptContainer,
             }
         }
 
-        public void OnPersist(StoreView snapshot, IReadOnlyList<Blockchain.ApplicationExecuted> applicationExecutedList)
+        void IPersistencePlugin.OnPersist(Block block, StoreView snapshot, IReadOnlyList<Blockchain.ApplicationExecuted> applicationExecutedList)
         {
             // Start freshly with a new DBCache for each block.
             ResetBatch();
@@ -192,7 +192,7 @@ public void OnPersist(StoreView snapshot, IReadOnlyList<Blockchain.ApplicationEx
             }
         }
 
-        public void OnCommit(StoreView snapshot)
+        void IPersistencePlugin.OnCommit(Block block, StoreView snapshot)
         {
             _balances.Commit();
             if (_shouldTrackHistory)
@@ -204,7 +204,7 @@ public void OnCommit(StoreView snapshot)
             _db.Write(WriteOptions.Default, _writeBatch);
         }
 
-        public bool ShouldThrowExceptionFromCommit(Exception ex)
+        bool IPersistencePlugin.ShouldThrowExceptionFromCommit(Exception ex)
         {
             return true;
         }

diff --git a/src/StatesDumper/StatesDumper.cs b/src/StatesDumper/StatesDumper.cs
@@ -3,6 +3,7 @@
 using Neo.IO.Caching;
 using Neo.IO.Json;
 using Neo.Ledger;
+using Neo.Network.P2P.Payloads;
 using Neo.Persistence;
 using System;
 using System.Collections.Generic;
@@ -49,7 +50,7 @@ private void OnDumpStorage(UInt160 key = null)
                 : Blockchain.Singleton.View.Storages.Find());
         }
 
-        public void OnPersist(StoreView snapshot, IReadOnlyList<Blockchain.ApplicationExecuted> applicationExecutedList)
+        void IPersistencePlugin.OnPersist(Block block, StoreView snapshot, IReadOnlyList<Blockchain.ApplicationExecuted> applicationExecutedList)
         {
             if (Settings.Default.PersistAction.HasFlag(PersistActions.StorageChanges))
                 OnPersistStorage(snapshot);
@@ -96,7 +97,7 @@ private void OnPersistStorage(StoreView snapshot)
             }
         }
 
-        public void OnCommit(StoreView snapshot)
+        void IPersistencePlugin.OnCommit(Block block, StoreView snapshot)
         {
             if (Settings.Default.PersistAction.HasFlag(PersistActions.StorageChanges))
                 OnCommitStorage(snapshot);
@@ -119,7 +120,7 @@ public void OnCommitStorage(StoreView snapshot)
             }
         }
 
-        public bool ShouldThrowExceptionFromCommit(Exception ex)
+        bool IPersistencePlugin.ShouldThrowExceptionFromCommit(Exception ex)
         {
             Console.WriteLine($"Error writing States with StatesDumper.{Environment.NewLine}{ex}");
             return true;