From c336fbc575ffd213613d4d6139a9b9f11e39a8ff Mon Sep 17 00:00:00 2001
From: jdeepee <joshuadparkin@gmail.com>
Date: Tue, 3 Nov 2020 16:07:03 +0000
Subject: [PATCH] expose generate_message fn as public

---
 gremlin-client/src/client.rs | 9 +++++++--
 gremlin-client/src/lib.rs    | 1 +
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/gremlin-client/src/client.rs b/gremlin-client/src/client.rs
index cbdf6932..afe94ed5 100644
--- 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
index fa6fb66a..42078a7c 100644
--- 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>;