From 1df7952986f7cf0d65a4aa3f27da7658ad1b55d2 Mon Sep 17 00:00:00 2001
From: Bastian Gruber <foreach@me.com>
Date: Sun, 21 May 2023 14:48:42 +0200
Subject: [PATCH] awaiting the channel send

---
 problem_06/src/heartbeat.rs | 11 ++++++++---
 problem_06/src/server.rs    |  2 +-
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/problem_06/src/heartbeat.rs b/problem_06/src/heartbeat.rs
index 9a4245c..f0c9b47 100644
--- a/problem_06/src/heartbeat.rs
+++ b/problem_06/src/heartbeat.rs
@@ -1,7 +1,9 @@
-use crate::frame::ServerFrames;
 use std::time::Duration;
+
 use tokio::sync::mpsc;
-use tracing::debug;
+use tracing::{debug, error};
+
+use crate::frame::ServerFrames;
 
 pub(crate) struct Heartbeat {
 	is_running: bool,
@@ -35,7 +37,10 @@ impl Heartbeat {
 		loop {
 			debug!("Heartbeat");
 			interval.tick().await;
-			let _ = self.message.send(ServerFrames::Heartbeat);
+			if let Err(e) = self.message.send(ServerFrames::Heartbeat).await {
+				error!("Error sending heartbeat: {}", e);
+				return;
+			}
 		}
 	}
 }
diff --git a/problem_06/src/server.rs b/problem_06/src/server.rs
index bff9827..0f50bef 100644
--- a/problem_06/src/server.rs
+++ b/problem_06/src/server.rs
@@ -145,7 +145,7 @@ impl Handler {
 
 				}
 				message = receive_message.recv() => {
-					debug!(?message);
+					info!("Received a message through the channel");
 					match message {
 						Some(message) => {
 							let _ = self.connection.write_frame(message).await;