Load plugins before initializing the system and call OnPluginsLoaded …

…afterward. (#633)
neo-project · Mar 12, 2019 · 432a3f6 · 432a3f6
1 parent d3094c2
commit 432a3f6
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/neo/NeoSystem.cs b/neo/NeoSystem.cs
@@ -34,10 +34,11 @@ public class NeoSystem : IDisposable
         public NeoSystem(Store store)
         {
             this.store = store;
+            Plugin.LoadPlugins(this);
             this.Blockchain = ActorSystem.ActorOf(Ledger.Blockchain.Props(this, store));
             this.LocalNode = ActorSystem.ActorOf(Network.P2P.LocalNode.Props(this));
             this.TaskManager = ActorSystem.ActorOf(Network.P2P.TaskManager.Props(this));
-            Plugin.LoadPlugins(this);
+            Plugin.NotifyPluginsLoadedAfterSystemConstructed();
         }
 
         public void Dispose()

diff --git a/neo/Plugins/Plugin.cs b/neo/Plugins/Plugin.cs
@@ -67,6 +67,10 @@ public static bool CheckPolicy(Transaction tx)
 
         public abstract void Configure();
 
+        protected virtual void OnPluginsLoaded()
+        {
+        }
+
         private static void ConfigWatcher_Changed(object sender, FileSystemEventArgs e)
         {
             foreach (var plugin in Plugins)
@@ -110,6 +114,12 @@ internal static void LoadPlugins(NeoSystem system)
             }
         }
 
+        internal static void NotifyPluginsLoadedAfterSystemConstructed()
+        {
+            foreach (var plugin in Plugins)
+                plugin.OnPluginsLoaded();
+        }
+
         protected void Log(string message, LogLevel level = LogLevel.Info)
         {
             Log($"{nameof(Plugin)}:{Name}", level, message);