add shutdown event for plugins

neo-project · Oct 28, 2019 · 3f632ea · 3f632ea
1 parent cd85039
commit 3f632ea
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/neo/NeoSystem.cs b/neo/NeoSystem.cs
@@ -44,6 +44,7 @@ public void Dispose()
         {
             RpcServer?.Dispose();
             EnsureStoped(LocalNode);
+            Plugin.NotifyPluginsToShutdown();
             // Dispose will call ActorSystem.Terminate()
             ActorSystem.Dispose();
             ActorSystem.WhenTerminated.Wait();

diff --git a/neo/Plugins/Plugin.cs b/neo/Plugins/Plugin.cs
@@ -62,6 +62,10 @@ protected virtual void OnPluginsLoaded()
         {
         }
 
+        protected virtual void OnShutdown()
+        {
+        }
+
         private static void ConfigWatcher_Changed(object sender, FileSystemEventArgs e)
         {
             foreach (var plugin in Plugins)
@@ -112,6 +116,12 @@ internal static void NotifyPluginsLoadedAfterSystemConstructed()
                 plugin.OnPluginsLoaded();
         }
 
+        internal static void NotifyPluginsToShutdown()
+        {
+            foreach (var plugin in Plugins)
+                plugin.OnShutdown();
+        }
+
         protected void Log(string message, LogLevel level = LogLevel.Info)
         {
             Log($"{nameof(Plugin)}:{Name}", level, message);