fix from tungstenite error (#198)

wolf4ood · Oct 19, 2023 · b1c8902 · b1c8902
1 parent 1add881
commit b1c8902
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/gremlin-client/src/error.rs b/gremlin-client/src/error.rs
@@ -54,3 +54,17 @@ impl From<mobc::Error<GremlinError>> for GremlinError {
         }
     }
 }
+
+#[cfg(not(feature = "async_gremlin"))]
+impl From<tungstenite::error::Error> for GremlinError {
+    fn from(e: tungstenite::error::Error) -> GremlinError {
+        let error = match e {
+            tungstenite::error::Error::AlreadyClosed => tungstenite::error::Error::AlreadyClosed,
+            tungstenite::error::Error::ConnectionClosed => {
+                tungstenite::error::Error::ConnectionClosed
+            }
+            _ => return GremlinError::Generic(format!("Error from ws {}", e)),
+        };
+        GremlinError::WebSocket(error)
+    }
+}