expose generate_message fn as public

jdeepee · Nov 3, 2020 · c336fbc · c336fbc
1 parent b46af05
commit c336fbc
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/gremlin-client/src/client.rs b/gremlin-client/src/client.rs
@@ -130,7 +130,7 @@ impl GremlinClient {
         Ok(GResultSet::new(self.clone(), results, response, conn))
     }
 
-    pub(crate) fn submit_traversal(&self, bytecode: &Bytecode) -> GremlinResult<GResultSet> {
+    pub fn generate_message(&self, bytecode: &Bytecode) -> GremlinResult<Message<serde_json::Value>> {
         let mut args = HashMap::new();
 
         args.insert(String::from("gremlin"), GValue::Bytecode(bytecode.clone()));
@@ -150,12 +150,17 @@ impl GremlinClient {
 
         let args = self.options.serializer.write(&GValue::from(args))?;
 
-        let message = message_with_args(String::from("bytecode"), String::from("traversal"), args);
+        Ok(message_with_args(String::from("bytecode"), String::from("traversal"), args))
+    }
+
+    pub(crate) fn submit_traversal(&self, bytecode: &Bytecode) -> GremlinResult<GResultSet> {
+        let message = self.generate_message(bytecode)?;
 
         let conn = self.pool.get()?;
 
         self.send_message(conn, message)
     }
+
     pub(crate) fn read_response(
         &self,
         conn: &mut r2d2::PooledConnection<GremlinConnectionManager>,

diff --git a/gremlin-client/src/lib.rs b/gremlin-client/src/lib.rs
@@ -126,6 +126,7 @@ pub use connection::{ConnectionOptions, TlsOptions};
 pub use conversion::{BorrowFromGValue, FromGValue, ToGValue};
 pub use error::GremlinError;
 pub use io::GraphSON;
+pub use message::Message;
 
 pub type GremlinResult<T> = Result<T, error::GremlinError>;