feat: added lifecycle start/stop to plugins

DamianoPellegrini · Nov 22, 2023 · 8fac1aa · 8fac1aa
1 parent 1500f6f
commit 8fac1aa
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/src/app.rs b/src/app.rs
@@ -42,6 +42,11 @@ impl Argo {
         let listener = std::net::TcpListener::bind(self.socket_addr)?;
 
         log::info!("Listening on {}", self.socket_addr);
+
+        log::info!("Starting plugins...");
+        for plugin in self.plugins.read().unwrap().iter() {
+            plugin.lock().expect("Cannot acquire lock").startup();
+        }
 
         let mut handles = vec![];
 
@@ -87,6 +92,11 @@ impl Argo {
             }
         }
 
+        log::info!("Shutting down plugins...");
+        for plugin in self.plugins.read().unwrap().iter() {
+            plugin.lock().expect("Cannot acquire lock").shutdown();
+        }
+
         Ok(())
     }
 }

diff --git a/src/plugin.rs b/src/plugin.rs
@@ -55,6 +55,9 @@ impl From<&AVLDatagram> for TeltonikaEvent {
 ///
 /// _See the [`PluginHandle`](crate::PluginHandle) and [`PluginHandles`](crate::PluginHandles) for more info._
 pub trait Plugin {
+    fn startup(&mut self) {}
+    fn shutdown(&mut self) {}
+
     #[allow(unused_variables)]
     /// Check if a device is allowed to connect
     fn can_teltonika_connect(&mut self, imei: &str) -> bool {