feat: terminator agent (#29)

configs/stable_pool.toml

@@ -8,4 +8,7 @@ PoolAdmin = {}
 TokenAdmin = { token_data.TKN0 = { name = "TKN0", symbol = "Token 0", decimals = 18 } }
 
 [[changer]]
-PriceChanger = { mu = 1, theta = 1, sigma = 1, cursor = 0, value = 1, seed = 69 }
+PriceChanger = { mu = 1, theta = 1, sigma = 1, cursor = 0, value = 1, seed = 69 }
+
+[[terminator]]
+Terminator = { steps = 10, current_step = 1 }

src/behaviors/mod.rs

@@ -5,17 +5,20 @@ use serde::{Deserialize, Serialize};
 pub mod deployer;
 pub mod pool_admin;
 pub mod price_changer;
+pub mod terminator;
 pub mod token_admin;
 
 use deployer::Deployer;
 use pool_admin::PoolAdmin;
 use price_changer::PriceChanger;
+use terminator::Terminator;
 use token_admin::TokenAdmin;
 
 #[derive(Behaviors, Debug, Serialize, Deserialize)]
 pub enum Behaviors {
     Deployer(Deployer),
     TokenAdmin(TokenAdmin),
+    Terminator(Terminator),
     PoolAdmin(PoolAdmin),
     PriceChanger(PriceChanger),
 }

src/behaviors/terminator.rs

@@ -0,0 +1,41 @@
+use std::sync::Arc;
+
+use anyhow::Result;
+use arbiter_core::middleware::ArbiterMiddleware;
+use arbiter_engine::messager::{Message, Messager};
+
+use super::*;
+use crate::behaviors::price_changer::PriceUpdate;
+
+#[derive(Debug, Deserialize, Serialize)]
+pub struct Terminator {
+    pub steps: usize,
+    pub current_step: usize,
+}
+
+#[async_trait::async_trait]
+impl Behavior<Message> for Terminator {
+    async fn startup(
+        &mut self,
+        _client: Arc<ArbiterMiddleware>,
+        messager: Messager,
+    ) -> Result<Option<EventStream<Message>>> {
+        Ok(Some(messager.clone().stream().unwrap()))
+    }
+
+    async fn process(&mut self, event: Message) -> Result<ControlFlow> {
+        if self.current_step >= self.steps {
+            return Ok(ControlFlow::Halt);
+        }
+
+        let _query: PriceUpdate = match serde_json::from_str::<PriceUpdate>(&event.data) {
+            Ok(_) => {
+                self.current_step += 1;
+                return Ok(ControlFlow::Continue);
+            }
+            Err(_) => {
+                return Ok(ControlFlow::Continue);
+            }
+        };
+    }
+}