From 3b3c35f131a02c027dcff25f8d8e7f2e68c434a8 Mon Sep 17 00:00:00 2001
From: Evie Viau <evie@uwueviee.live>
Date: Thu, 17 Feb 2022 19:41:03 -0500
Subject: [PATCH] Allow a custom heartbeat interval

---
 src/main.rs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/main.rs b/src/main.rs
index 038b6be..2d94afd 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -70,7 +70,10 @@ async fn handle_conn(peer: SocketAddr, stream: TcpStream) -> tokio_tungstenite::
             &SocketMessage {
                 op: HELLO,
                 d: MessageData::HELLO {
-                    heartbeat_interval: 1000,
+                    heartbeat_interval: env::var("HEARTBEAT_INTERVAL").
+                        unwrap_or("1".to_string())
+                        .parse::<i32>()
+                        .unwrap_or(1),
                     nonce
                 }
             }