add sideEffect step

wolf4ood · Mar 30, 2023 · a0dff5f · a0dff5f
1 parent b08ea08
commit a0dff5f
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/gremlin-client/src/process/traversal/builder.rs b/gremlin-client/src/process/traversal/builder.rs
@@ -13,6 +13,7 @@ use crate::process::traversal::step::not::NotStep;
 use crate::process::traversal::step::or::OrStep;
 use crate::process::traversal::step::repeat::RepeatStep;
 use crate::process::traversal::step::select::SelectStep;
+use crate::process::traversal::step::side_effect::IntoSideEffectStep;
 use crate::process::traversal::step::to::ToStep;
 use crate::process::traversal::step::until::UntilStep;
 use crate::process::traversal::step::where_step::WhereStep;
@@ -136,6 +137,15 @@ impl TraversalBuilder {
         self
     }
 
+    pub fn side_effect<A>(mut self, step: A) -> Self
+    where
+        A: IntoSideEffectStep,
+    {
+        self.bytecode
+            .add_step(String::from("sideEffect"), step.into_step());
+        self
+    }
+
     pub fn with_side_effect<A>(mut self, step: (&'static str, A)) -> Self
     where
         A: Into<GValue> + FromGValue,

diff --git a/gremlin-client/src/process/traversal/step/mod.rs b/gremlin-client/src/process/traversal/step/mod.rs
@@ -12,6 +12,7 @@ pub mod not;
 pub mod or;
 pub mod repeat;
 pub mod select;
+pub mod side_effect;
 pub mod to;
 pub mod until;
 pub mod where_step;
diff --git a/gremlin-client/src/process/traversal/step/side_effect.rs b/gremlin-client/src/process/traversal/step/side_effect.rs
@@ -0,0 +1,13 @@
+use crate::{process::traversal::TraversalBuilder, GValue};
+
+
+pub trait IntoSideEffectStep {
+    fn into_step(self) -> Vec<GValue>;
+}
+
+impl IntoSideEffectStep for TraversalBuilder {
+    fn into_step(self) -> Vec<GValue> {
+        vec![self.bytecode.into()]
+    }
+}
+