Formatting

criminosis · Oct 25, 2024 · 815767a · 815767a
1 parent 32b47ac
commit 815767a
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 6 deletions.
diff --git a/gremlin-client/src/io/graph_binary_v1.rs b/gremlin-client/src/io/graph_binary_v1.rs
@@ -401,7 +401,9 @@ impl GraphBinaryV1Deser for bool {
         match bytes.next() {
             Some(0x00) => Ok(false),
             Some(0x01) => Ok(true),
-            other => Err(GremlinError::Cast(format!("No boolean value byte {other:?}"))),
+            other => Err(GremlinError::Cast(format!(
+                "No boolean value byte {other:?}"
+            ))),
         }
     }
 }
@@ -492,7 +494,9 @@ impl GraphBinaryV1Deser for GValue {
                 //Need to confirm the null-ness with the next byte being a 1
                 match bytes.next().cloned() {
                     Some(VALUE_NULL_FLAG) => Ok(GValue::Null),
-                    other => Err(GremlinError::Cast(format!("Not expected null value byte {other:?}"))),
+                    other => Err(GremlinError::Cast(format!(
+                        "Not expected null value byte {other:?}"
+                    ))),
                 }
             }
             other => unimplemented!("TODO {other}"),
@@ -508,7 +512,9 @@ impl GraphBinaryV1Deser for T {
             Some(GValue::String(literal)) if literal.eq_ignore_ascii_case("key") => Ok(T::Id),
             Some(GValue::String(literal)) if literal.eq_ignore_ascii_case("label") => Ok(T::Id),
             Some(GValue::String(literal)) if literal.eq_ignore_ascii_case("value") => Ok(T::Id),
-            other => Err(GremlinError::Cast(format!("Unexpected T literal {other:?}"))),
+            other => Err(GremlinError::Cast(format!(
+                "Unexpected T literal {other:?}"
+            ))),
         }
     }
 }

diff --git a/gremlin-client/src/structure/gid.rs b/gremlin-client/src/structure/gid.rs
@@ -41,7 +41,9 @@ impl TryFrom<GValue> for GID {
             GValue::Int32(v) => Ok(GID::Int32(v)),
             GValue::Int64(v) => Ok(GID::Int64(v)),
             GValue::String(v) => Ok(GID::String(v)),
-            other => Err(GremlinError::Cast(format!("Invalid GValue for GID {other:?}")))
+            other => Err(GremlinError::Cast(format!(
+                "Invalid GValue for GID {other:?}"
+            ))),
         }
     }
 }

diff --git a/gremlin-client/src/structure/value.rs b/gremlin-client/src/structure/value.rs
@@ -266,6 +266,8 @@ impl From<GKey> for GValue {
             GKey::Token(s) => GValue::String(s.value().clone()),
             GKey::Vertex(v) => GValue::Vertex(v),
             GKey::Edge(v) => GValue::Edge(v),
+            GKey::Int64(v) => GValue::Int64(v),
+            GKey::Int32(v) => GValue::Int32(v),
         }
     }
 }

diff --git a/gremlin-client/tests/graph_binary_read_write_cycle.rs b/gremlin-client/tests/graph_binary_read_write_cycle.rs
@@ -1,5 +1,5 @@
 use std::array::IntoIter;
-use std::collections::{HashSet, HashMap};
+use std::collections::{HashMap, HashSet};
 
 use chrono::{DateTime, TimeZone, Utc};
 use common::io::graph_serializer;
@@ -10,8 +10,8 @@ use gremlin_client::{
     GValue, IoProtocol,
 };
 use rstest::rstest;
-use uuid::Uuid;
 use std::iter::FromIterator;
+use uuid::Uuid;
 
 mod common;