Implement read/write timeout (#57)

DennisOSRM · Dec 6, 2023 · 603fead · 603fead
1 parent 50216b1
commit 603fead
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/hms2mqtt/src/inverter.rs b/hms2mqtt/src/inverter.rs
@@ -1,6 +1,6 @@
 use crate::protos::hoymiles::RealData::{HMSStateResponse, RealDataResDTO};
 use crc16::{State, MODBUS};
-use log::{debug, info};
+use log::{debug, info, warn};
 use protobuf::Message;
 use std::io::{Read, Write};
 use std::net::TcpStream;
@@ -71,6 +71,12 @@ impl<'a> Inverter<'a> {
         }
 
         let mut stream = stream.unwrap();
+        if let Err(e) = stream.set_write_timeout(Some(Duration::new(5, 0))) {
+            warn!("could not set write timeout: {e}");
+        }
+        if let Err(e) = stream.set_read_timeout(Some(Duration::new(5, 0))) {
+            warn!("could not set read timeout: {e}");
+        }
         if let Err(e) = stream.write(&message) {
             debug!(r#"{e}"#);
             self.set_state(NetworkState::Offline);