Support bytecode-based requests for interactive engine

python/graphscope/deploy/tests/test_demo_script.py

@@ -209,7 +209,7 @@ def test_multiple_session(data_dir):
     sess.close()
 
 
-def test_load_modern_graph(modern_graph_data_dir):
+def test_query_modern_graph(modern_graph_data_dir):
     image = get_gs_image_on_ci_env()
     sess = graphscope.session(
         show_log=True,
@@ -236,3 +236,52 @@ def test_load_modern_graph(modern_graph_data_dir):
     for q in queries:
         result = interactive.execute(q).all().result()[0]
         assert result == 1
+
+
+def test_traversal_modern_graph(modern_graph_data_dir):
+    from gremlin_python.process.traversal import Order
+    from gremlin_python.process.traversal import P
+
+    image = get_gs_image_on_ci_env()
+    sess = graphscope.session(
+        show_log=True,
+        num_workers=1,
+        k8s_gs_image=image,
+        k8s_coordinator_cpu=0.5,
+        k8s_coordinator_mem="2500Mi",
+        k8s_vineyard_cpu=0.1,
+        k8s_vineyard_mem="512Mi",
+        k8s_engine_cpu=0.1,
+        k8s_engine_mem="1500Mi",
+        k8s_vineyard_shared_mem="2Gi",
+    )
+    graph = load_modern_graph(sess, modern_graph_data_dir)
+    interactive = sess.gremlin(graph)
+    g = interactive.traversal_source()
+    assert g.V().has("name", "marko").count().toList()[0] == 1
+    assert g.V().has("person", "name", "marko").count().toList()[0] == 1
+    assert g.V().has("person", "name", "marko").outE("created").count().toList()[0] == 1
+    assert (
+        g.V().has("person", "name", "marko").outE("created").inV().count().toList()[0]
+        == 1
+    )
+    assert g.V().has("person", "name", "marko").out("created").count().toList()[0] == 1
+    assert (
+        g.V()
+        .has("person", "name", "marko")
+        .out("created")
+        .values("name")
+        .count()
+        .toList()[0]
+        == 1
+    )
+    assert (
+        g.V()
+        .hasLabel("person")
+        .has("age", P.gt(30))
+        .order()
+        .by("age", Order.desc)
+        .count()
+        .toList()[0]
+        == 2
+    )

python/graphscope/interactive/query.py

@@ -22,6 +22,8 @@
 from concurrent.futures import ThreadPoolExecutor
 
 from gremlin_python.driver.client import Client
+from gremlin_python.driver.driver_remote_connection import DriverRemoteConnection
+from gremlin_python.process.anonymous_traversal import traversal
 
 from graphscope.framework.loader import Loader
 
@@ -32,11 +34,15 @@ class InteractiveQuery(object):
     """`InteractiveQuery` class, is a simple wrapper around
     `Gremlin-Python <https://pypi.org/project/gremlinpython/>`_,
     which implements Gremlin within the Python language.
-    It also can expose gremlin endpoint which can be used by any other standard gremlin console.
+    It also can expose gremlin endpoint which can be used by
+    any other standard gremlin console, with the method `graph_url()`.
 
     It also has a method called `subgraph` which can extract some fragments
     from origin graph, produce a new, smaller but concise graph stored in vineyard,
     which lifetime is independent from the origin graph.
+
+    User can either use `execute()` to submit a script, or use `traversal_source()`
+    to get a `GraphTraversalSource` for further traversal.
     """
 
     def __init__(self, graphscope_session, object_id, front_ip, front_port):
@@ -141,6 +147,27 @@ def execute(self, query):
             raise RuntimeError("Interactive query is closed.")
         return self._client.submit(query)
 
+    def traversal_source(self):
+        """Create a GraphTraversalSource and return.
+        Once `g` has been created using a connection, we can start to write
+        Gremlin traversals to query the remote graph.
+
+        Examples:
+
+            .. code:: python
+
+                sess = graphscope.session()
+                graph = load_modern_graph(sess, modern_graph_data_dir)
+                interactive = sess.gremlin(graph)
+                g = interactive.traversal_source()
+                print(g.V().both()[1:3].toList())
+                print(g.V().both().name.toList())
+
+        Returns:
+            `GraphTraversalSource`
+        """
+        return traversal().withRemote(DriverRemoteConnection(self._graph_url, "g"))
+
     def close(self):
         """Close interactive instance and release resources"""
         if not self.closed():