Implement function to notify the awaiter of an active session

Author: davidsteiner

crates/hotfix/src/session.rs

@@ -689,6 +689,8 @@ impl<M: FixMessage, S: MessageStore> Session<M, S> {
         if !is_active {
             // we are currently outside scheduled session time
             self.initiate_graceful_logout("End of session time").await;
+        } else {
+            self.state.notify_session_awaiter();
         }
 
         let deadline = Instant::now() + Duration::from_secs(SCHEDULE_CHECK_INTERVAL);

crates/hotfix/src/session/state.rs

@@ -103,7 +103,7 @@ impl SessionState {
     }
 
     pub fn register_session_awaiter(
-        &self,
+        &mut self,
         responder: oneshot::Sender<AwaitingActiveSessionResponse>,
     ) {
         match self {
@@ -118,7 +118,7 @@ impl SessionState {
                         error!("failed to send session awaiter response: {err:?}");
                     }
                 } else {
-                    // set the session_awaiter on the current state
+                    *session_awaiter = Some(responder);
                 }
             }
             _ => {
@@ -129,6 +129,19 @@ impl SessionState {
             }
         }
     }
+
+    pub fn notify_session_awaiter(&mut self) {
+        if let SessionState::Disconnected {
+            session_awaiter, ..
+        } = self
+        {
+            if let Some(awaiter) = session_awaiter.take() {
+                if let Err(err) = awaiter.send(AwaitingActiveSessionResponse::Active) {
+                    error!("failed to send session awaiter response: {err:?}");
+                }
+            }
+        }
+    }
 }
 
 /// Session state we're in while processing messages we requested to be resent.