More flexible signal handling

pulse-sdk/src/lib.rs

@@ -122,12 +122,7 @@ pub trait Risk {
             RiskEngineMessage::Initialize => self.initialize().await,
             RiskEngineMessage::Command { command, args } => self.command(command, args).await,
             RiskEngineMessage::WatchList(watchlist) => self.watchlist(watchlist).await,
-            RiskEngineMessage::Signal(signal) => {
+            RiskEngineMessage::Signal(signal) => self.signal_request(signal).await,
-                self.send(&prelude::RiskMessage::Signal(
-                    self.signal_request(signal).await?,
-                ))
-                .await
-            }
         }
     }
 
@@ -143,10 +138,7 @@ pub trait Risk {
         unimplemented!("Risk::watchlist")
     }
 
-    async fn signal_request(
+    async fn signal_request(&self, _signal: prelude::StrategySignal) -> tokio::io::Result<()> {
-        &self,
-        _signal: prelude::StrategySignal,
-    ) -> tokio::io::Result<prelude::RiskSignal> {
         unimplemented!("Risk::signal_request")
     }
 }

pulse-sdk/src/plugin.rs

@@ -97,6 +97,7 @@ pub struct StrategySignal {
 pub enum RiskSignal {
     Approve(Signal),
     Reject {
+        signal: StrategySignal,
         rejection_confidence: f32,
         reason: String,
     },