implemented action req handling

2024-04-10 15:37:24 +02:00
parent 443995fe5e
commit 5d87cab9cc
4 changed files with 113 additions and 14 deletions
@@ -0,0 +1,84 @@
+use derive_new::new;
+use num_enum::TryFromPrimitive;
+use satrs::{
+    action::ActionRequest,
+    pus::action::{ActionReplyVariant, PusActionReply},
+    request::{GenericMessage, MessageMetadata},
+};
+use std::sync::{atomic::AtomicBool, mpsc, Arc};
+
+use ops_sat_rs::config::{action_err::INVALID_ACTION_ID, STOP_FILE_NAME};
+
+use crate::requests::CompositeRequest;
+
+#[derive(Debug, Clone, Copy, TryFromPrimitive)]
+#[repr(u32)]
+pub enum ActionId {
+    StopExperiment = 1,
+}
+
+#[derive(new)]
+pub struct ExperimentController {
+    pub composite_request_rx: mpsc::Receiver<GenericMessage<CompositeRequest>>,
+    pub action_reply_tx: mpsc::Sender<GenericMessage<PusActionReply>>,
+    pub stop_signal: Arc<AtomicBool>,
+}
+
+impl ExperimentController {
+    pub fn perform_operation(&mut self) {
+        match self.composite_request_rx.try_recv() {
+            Ok(msg) => match msg.message {
+                CompositeRequest::Hk(_) => todo!(),
+                CompositeRequest::Action(action_req) => {
+                    self.handle_action_request(msg.requestor_info, action_req);
+                }
+            },
+            Err(_) => todo!(),
+        }
+        self.check_stop_file();
+    }
+
+    pub fn handle_action_request(&mut self, requestor: MessageMetadata, action_req: ActionRequest) {
+        let action_id = ActionId::try_from(action_req.action_id);
+        if action_id.is_err() {
+            let result = self.action_reply_tx.send(GenericMessage::new_action_reply(
+                requestor,
+                action_req.action_id,
+                ActionReplyVariant::CompletionFailed {
+                    error_code: INVALID_ACTION_ID,
+                    params: None,
+                },
+            ));
+            if result.is_err() {
+                log::error!("sending action reply failed");
+            }
+            return;
+        }
+        let action_id = action_id.unwrap();
+        match action_id {
+            ActionId::StopExperiment => {
+                self.stop_signal
+                    .store(true, std::sync::atomic::Ordering::Relaxed);
+                let result = self.action_reply_tx.send(GenericMessage::new_action_reply(
+                    requestor,
+                    action_req.action_id,
+                    ActionReplyVariant::Completed,
+                ));
+                if result.is_err() {
+                    log::error!("sending action reply failed");
+                }
+            }
+        }
+    }
+
+    pub fn check_stop_file(&self) {
+        if std::path::Path::new(STOP_FILE_NAME).exists() {
+            log::warn!(
+                "Detected stop file name at {}. Initiating experiment shutdown",
+                STOP_FILE_NAME
+            );
+            self.stop_signal
+                .store(true, std::sync::atomic::Ordering::Relaxed);
+        }
+    }
+}