fix: Fixed internal error handling

Author: Andcool-Systems

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "mc-oauth"
-version = "0.1.5"
+version = "0.1.6"
 edition = "2021"
 
 [dependencies]

src/client.rs

@@ -19,7 +19,7 @@ use bytes::{BufMut, BytesMut};
 use cfb8::Encryptor;
 use std::{sync::Arc, time::Duration};
 use tokio::net::TcpStream;
-use tracing::{debug, info};
+use tracing::{debug, error, info};
 use uuid::Uuid;
 
 #[derive(Debug)]
@@ -77,7 +77,17 @@ impl MinecraftClient {
         }
     }
 
-    pub async fn run(&mut self) -> Result<()> {
+    pub async fn run(&mut self) {
+        match self._run().await {
+            Ok(_) => info!(
+                "Connection from {:?} closed successfully",
+                self.stream.peer_addr().unwrap()
+            ),
+            Err(e) => error!("Internal error occurred: {}", e),
+        }
+    }
+
+    pub async fn _run(&mut self) -> Result<()> {
         loop {
             let mut temp_buf = vec![0; 1024];
             self.stream.readable().await?;

src/main.rs

@@ -97,13 +97,10 @@ async fn main() -> anyhow::Result<()> {
 
                         tokio::spawn(async move {
                             // Setting client timeout
-                            match timeout(
+                            timeout(
                                 Duration::from_secs(config.server.timeout),
                                 client.run()
-                            ).await {
-                                Ok(_) => info!("Connection from {} closed", addr),
-                                Err(e) => error!("Client exceptionally closed connection: {}", e)
-                            }
+                            ).await.unwrap_or_else(|_| {error!("Connection from {} has been timed out", addr);})
                         });
                     },
                     Err(e) => {