Formatting

wolf4ood · May 20, 2023 · 2eaab4f · 2eaab4f
1 parent 7336d46
commit 2eaab4f
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 6 deletions.
diff --git a/gremlin-client/src/aio/connection.rs b/gremlin-client/src/aio/connection.rs
@@ -148,9 +148,23 @@ impl Conn {
         let websocket_config = opts.websocket_options.as_ref().map(WebSocketConfig::from);
 
         #[cfg(feature = "async-std-runtime")]
-        let (client, _) = { connect_async_with_tls_connector_and_config(url, tls::connector(&opts), websocket_config).await? };
+        let (client, _) = {
+            connect_async_with_tls_connector_and_config(
+                url,
+                tls::connector(&opts),
+                websocket_config,
+            )
+            .await?
+        };
         #[cfg(feature = "tokio-runtime")]
-        let (client, _) = { connect_async_with_tls_connector_and_config(url, tls::connector(&opts), websocket_config).await? };
+        let (client, _) = {
+            connect_async_with_tls_connector_and_config(
+                url,
+                tls::connector(&opts),
+                websocket_config,
+            )
+            .await?
+        };
 
         let (sink, stream) = client.split();
         let (sender, receiver) = channel(20);

diff --git a/gremlin-client/src/connection.rs b/gremlin-client/src/connection.rs
@@ -5,8 +5,9 @@ use native_tls::TlsConnector;
 use tungstenite::{
     client::{uri_mode, IntoClientRequest},
     client_tls_with_config,
+    protocol::WebSocketConfig,
     stream::{MaybeTlsStream, Mode, NoDelay},
-    Connector, Message, WebSocket, protocol::WebSocketConfig,
+    Connector, Message, WebSocket,
 };
 
 struct ConnectionStream(WebSocket<MaybeTlsStream<TcpStream>>);
@@ -47,7 +48,10 @@ impl ConnectionStream {
         NoDelay::set_nodelay(&mut stream, true)
             .map_err(|e| GremlinError::Generic(e.to_string()))?;
 
-        let websocket_config = options.websocket_options.as_ref().map(WebSocketConfig::from);
+        let websocket_config = options
+            .websocket_options
+            .as_ref()
+            .map(WebSocketConfig::from);
 
         let (client, _response) =
             client_tls_with_config(options.websocket_url(), stream, websocket_config, connector)
@@ -202,7 +206,6 @@ impl Default for WebSocketOptions {
     }
 }
 
-
 impl From<WebSocketOptions> for tungstenite::protocol::WebSocketConfig {
     fn from(value: WebSocketOptions) -> Self {
         (&value).into()

diff --git a/gremlin-client/src/lib.rs b/gremlin-client/src/lib.rs
@@ -122,7 +122,10 @@ mod message;
 mod pool;
 
 pub use client::GremlinClient;
-pub use connection::{ConnectionOptions, ConnectionOptionsBuilder, TlsOptions, WebSocketOptions, WebSocketOptionsBuilder};
+pub use connection::{
+    ConnectionOptions, ConnectionOptionsBuilder, TlsOptions, WebSocketOptions,
+    WebSocketOptionsBuilder,
+};
 pub use conversion::{BorrowFromGValue, FromGValue, ToGValue};
 pub use error::GremlinError;
 pub use io::GraphSON;