perf: #![forbid(clippy::unwrap_used)]

Cargo.toml

@@ -28,7 +28,7 @@ futures-util = "0.3.28"
 # juniper_subscriptions = "0.16.0"
 # juniper_warp = { version = "0.7.0", features = ["subscriptions"] }
 lazy_static = "1.4.0"
-log = "0.4.20"
+# log = "0.4.20"
 poem = "2.0.0"
 prost = "0.11.9"
 prost-types = "0.11.9"
@@ -40,6 +40,8 @@ tokio = { version = "1.31.0", features = ["full"] }
 tokio-tungstenite = { version = "0.20.0", features = [
     "rustls-tls-native-roots"
 ] }
+tracing = { version = "0.1.40", features = ["log"] }
+tracing-subscriber = { version = "0.3.18", features = ["chrono", "env-filter"] }
 uuid = { version = "1.7", features = ["serde", "v4"] }
 # warp = { version = "0.3.5", features = ["tokio-rustls"] }
 

src/connection.rs

@@ -2,7 +2,8 @@ use futures_util::{stream::{SplitStream, SplitSink}, StreamExt, SinkExt, Stream}
 use prost::Message as _;
 use tokio::net::TcpStream;
 use tokio_tungstenite::{WebSocketStream, MaybeTlsStream, tungstenite::Message};
-use crate::{proto::{models, packet}};
+use tracing::error;
+use crate::proto::{models, packet};
 
 #[derive(Debug)]
 pub struct TAConnection {
@@ -18,7 +19,7 @@ impl TAConnection {
         let (ws_stream, _) = match tokio_tungstenite::connect_async(uri).await {
             Ok(c) => c,
             Err(_) => {
-                log::error!("Failed to connect to server. Are you sure you're connecting to the overlay websocket?");
+                error!("Failed to connect to server. Are you sure you're connecting to the overlay websocket?");
                 return Err(anyhow::anyhow!("Failed to connect to server. Are you sure you're connecting to the overlay websocket?"));
             },
         };
@@ -42,7 +43,13 @@ impl TAConnection {
             }))
         };
 
-        ws_tx.send(Message::Binary(connect.encode_to_vec())).await.unwrap();
+        match ws_tx.send(Message::Binary(connect.encode_to_vec())).await {
+            Ok(_) => {},
+            Err(e) => {
+                error!("Failed to send connect packet. {:#?}", e);
+                return Err(anyhow::anyhow!("Failed to send connect packet. {:#?}", e));
+            },
+        };
 
         Ok(TAConnection {
             ws_rx,

src/gql.rs

@@ -10,11 +10,11 @@ pub struct Query;
 
 #[Object]
 impl Query {
-    async fn state<'ctx>(&self, _ctx: &Context<'ctx>) ->  GQLTAState {
+    async fn state<'ctx>(&self, _ctx: &Context<'ctx>) ->  anyhow::Result<GQLTAState> {
         TA_STATE.read().await.into_gql().await
     }
 
-    async fn match_by_id<'ctx>(&self, _ctx: &Context<'ctx>, id: Uuid) -> Option<Match> {
+    async fn match_by_id<'ctx>(&self, _ctx: &Context<'ctx>, id: Uuid) -> anyhow::Result<Option<Match>> {
         TA_STATE.read().await.get_single_match_gql(id).await
     }