Page MenuHomeSoftware Heritage
Files F7123625

D8464.id30505.diff
No OneTemporary
Actions

D8464.id30505.diff
View Options

diff --git a/swh/graph/cli.py b/swh/graph/cli.py
--- a/swh/graph/cli.py
+++ b/swh/graph/cli.py
@@ -134,6 +134,53 @@
aiohttp.web.run_app(app, host=host, port=port)
+@graph_cli_group.command(name="grpc-serve")
+@click.option(
+ "--port",
+ "-p",
+ default=50091,
+ type=click.INT,
+ metavar="PORT",
+ show_default=True,
+ help=(
+ "port to bind the server on (note: host is not configurable "
+ "for now and will be 0.0.0.0)"
+ ),
+)
+@click.option(
+ "--java-home",
+ "-j",
+ default=None,
+ metavar="JAVA_HOME",
+ help="absolute path to the Java Runtime Environment (JRE)",
+)
+@click.option(
+ "--graph", "-g", required=True, metavar="GRAPH", help="compressed graph basename"
+)
+@click.pass_context
+def grpc_serve(ctx, host, port, java_home, graph):
+ """start the graph GRPC service
+
+ This command uses execve to execute the java GRPC service.
+ """
+ import os
+ from pathlib import Path
+
+ from swh.graph.grpc_server import build_grpc_server_cmdline
+
+ config = ctx.obj["config"]
+ config.setdefault("graph", {})
+ config["graph"]["path"] = graph
+ cmd, port = build_grpc_server_cmdline(**config["graph"])
+
+ java_bin = cmd[0]
+ if java_home is not None:
+ java_bin = str(Path(java_home) / "bin" / java_bin)
+
+ print(f"Starting the GRPC server on 0.0.0.0:{port}")
+ os.execvp(java_bin, cmd)
+
+
@graph_cli_group.command()
@click.option(
"--input-dataset",
diff --git a/swh/graph/grpc_server.py b/swh/graph/grpc_server.py
--- a/swh/graph/grpc_server.py
+++ b/swh/graph/grpc_server.py
@@ -17,7 +17,7 @@
from swh.graph.config import check_config
-def spawn_java_grpc_server(**config):
+def build_grpc_server_cmdline(**config):
port = config.pop("port", None)
if port is None:
port = aiohttp.test_utils.unused_port()
@@ -32,6 +32,11 @@
str(port),
str(config["path"]),
]
+ return cmd, port
+
+
+def spawn_java_grpc_server(**config):
+ cmd, port = build_grpc_server_cmdline(**config)
print(cmd)
# XXX: shlex.join() is in 3.8
# logging.info("Starting RPC server: %s", shlex.join(cmd))

File Metadata

Mime Type
text/plain
Expires
Dec 19 2024, 3:15 PM (11 w, 4 d ago)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
3229312

Event Timeline

About · Developer information · Legal