From 3b8aafe1018ed626dbb4c19759195a05be93a1bf Mon Sep 17 00:00:00 2001
From: MaxVerevkin <maxxverrr@gmail.com>
Date: Sun, 2 Apr 2023 22:32:52 +0300
Subject: small changes

---
 src/main.rs | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index befd750..72d635c 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -62,7 +62,7 @@ fn main() -> anyhow::Result<()> {
                     conn.flush(IoMode::Blocking)?;
                 }
                 Err(e) if e.kind() == ErrorKind::WouldBlock => (),
-                Err(e) => return Err(e.into()),
+                Err(e) => bail!(e),
             }
         }
 
@@ -72,7 +72,13 @@ fn main() -> anyhow::Result<()> {
         }
 
         if fds.len() > 2 && fds[2].any().unwrap_or(false) {
-            match state.shared_state.status_cmd.as_mut().unwrap().read() {
+            match state
+                .shared_state
+                .status_cmd
+                .as_mut()
+                .unwrap()
+                .receive_blocks()
+            {
                 Ok(None) => (),
                 Ok(Some(blocks)) => {
                     state.set_blocks(&mut conn, blocks);
-- 
cgit v1.3.1