From d4ba9f948f5acc5f6c22db0b5f2a1666262acc59 Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom.smeding@gmail.com>
Date: Sat, 28 Mar 2020 21:23:36 +0100
Subject: controller: Set keepalive on sockets

---
 controller/src/lib.rs | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'controller/src')

diff --git a/controller/src/lib.rs b/controller/src/lib.rs
index b53502f..60b3639 100644
--- a/controller/src/lib.rs
+++ b/controller/src/lib.rs
@@ -120,6 +120,15 @@ enum ThreadCollect {
 async fn thread_handshake_handler(mut listener: TcpListener, sink: mpsc::Sender<ThreadCollect>) {
     loop {
         let (mut sock, _) = listener.accept().await.expect("Accept failed on TCP server socket");
+
+        // Error setting keepalive is not fatal
+        match sock.set_keepalive(Some(std::time::Duration::from_secs(60))) {
+            Ok(()) => {}
+            Err(e) => {
+                eprintln!("WARNING: Error setting keepalive on worker socket: {}", e);
+            }
+        }
+
         let mut sink = sink.clone();
         task::spawn(async move {
             let payload = encode_message(MessageBody::Version(1));
-- 
cgit v1.2.3-70-g09d2