Send the persisting block to IPersistencePlugin (neo-project#2213)

cloud8little · Jan 24, 2021 · 93af6a5 · 93af6a5
1 parent 4aec7b2
commit 93af6a5
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 6 deletions.
diff --git a/src/neo/Ledger/Blockchain.cs b/src/neo/Ledger/Blockchain.cs
@@ -446,14 +446,14 @@ private void Persist(Block block)
                     all_application_executed.Add(application_executed);
                 }
                 foreach (IPersistencePlugin plugin in Plugin.PersistencePlugins)
-                    plugin.OnPersist(snapshot, all_application_executed);
+                    plugin.OnPersist(block, snapshot, all_application_executed);
                 snapshot.Commit();
                 List<Exception> commitExceptions = null;
                 foreach (IPersistencePlugin plugin in Plugin.PersistencePlugins)
                 {
                     try
                     {
-                        plugin.OnCommit(snapshot);
+                        plugin.OnCommit(block, snapshot);
                     }
                     catch (Exception ex)
                     {

diff --git a/src/neo/Plugins/IPersistencePlugin.cs b/src/neo/Plugins/IPersistencePlugin.cs
@@ -1,3 +1,4 @@
+using Neo.Network.P2P.Payloads;
 using Neo.Persistence;
 using System;
 using System.Collections.Generic;
@@ -7,8 +8,8 @@ namespace Neo.Plugins
 {
     public interface IPersistencePlugin
     {
-        void OnPersist(StoreView snapshot, IReadOnlyList<ApplicationExecuted> applicationExecutedList) { }
-        void OnCommit(StoreView snapshot) { }
+        void OnPersist(Block block, StoreView snapshot, IReadOnlyList<ApplicationExecuted> applicationExecutedList) { }
+        void OnCommit(Block block, StoreView snapshot) { }
         bool ShouldThrowExceptionFromCommit(Exception ex) => false;
     }
 }
diff --git a/tests/neo.UnitTests/Plugins/UT_Plugin.cs b/tests/neo.UnitTests/Plugins/UT_Plugin.cs
@@ -33,8 +33,8 @@ public void TestIPersistencePlugin()
 
             // With empty default implementation
 
-            pp.OnCommit(null);
-            pp.OnPersist(null, null);
+            pp.OnCommit(null, null);
+            pp.OnPersist(null, null, null);
         }
 
         [TestMethod]