add wsi_retry_limit

assyst-common/src/config/config.rs

@@ -100,4 +100,5 @@ pub struct DevAttributes {
     pub dev_guild: u64,
     pub dev_channel: u64,
     pub dev_message: bool,
+    pub wsi_retry_limit: u64,
 }

assyst-core/src/wsi_handler/mod.rs

@@ -46,15 +46,20 @@ impl WsiHandler {
         let socket = socket.to_owned();
 
         spawn(async move {
+            let mut retries = 0;
             loop {
                 let stream = match TcpStream::connect(&socket).await {
                     Ok(stream) => stream,
                     Err(e) => {
+                        if retries > CONFIG.dev.wsi_retry_limit {
+                            break;
+                        }
                         err!(
                             "Failed to connect to WSI server ({}), attempting reconnection in 10 sec...",
                             e.to_string()
                         );
                         sleep(Duration::from_secs(10)).await;
+                        retries += 1;
                         continue;
                     },
                 };

config.template.toml

@@ -88,3 +88,6 @@ dev_guild = 0
 
 # The channel to send the dev message in.
 dev_channel = 0
+
+# The amount of times we try connecting to WSI before giving up after failure.
+wsi_retry_limit = 50