address comments

dogstatsd-client/src/lib.rs

@@ -71,7 +71,7 @@ pub enum DogStatsDAction<'a, T: AsRef<str>, V: IntoIterator<Item = &'a Tag>> {
 /// A dogstatsd-client that flushes stats to a given endpoint.
 #[derive(Debug, Default)]
 pub struct Client {
-    client: Arc<Mutex<Option<StatsdClient>>>,
+    client: Mutex<Arc<Option<StatsdClient>>>,
     endpoint: Option<Endpoint>,
 }
 
@@ -162,19 +162,21 @@ impl Client {
         }
     }
 
-    fn get_or_init_client(&self) -> anyhow::Result<std::sync::MutexGuard<Option<StatsdClient>>> {
-        let mut client_guard = self
-            .client
-            .lock()
-            .map_err(|e| anyhow!("Failed to acquire dogstatsd client lock: {}", e))?;
-
-        if client_guard.is_none() {
-            if let Some(endpoint) = &self.endpoint {
-                *client_guard = Some(create_client(endpoint)?);
-            }
+    fn get_or_init_client(&self) -> anyhow::Result<Arc<Option<StatsdClient>>> {
+        if let Some(endpoint) = &self.endpoint {
+            let mut client_guard = self.client.lock().map_err(|e| {
+                anyhow!("Failed to acquire dogstatsd client lock: {}", e.to_string())
+            })?;
+            return if client_guard.is_some() {
+                Ok(client_guard.clone())
+            } else {
+                let client = Arc::new(Some(create_client(endpoint)?));
+                *client_guard = client.clone();
+                Ok(client)
+            };
         }
 
-        Ok(client_guard)
+        Ok(None.into())
     }
 }