enable grpc server reflection

Signed-off-by: John Pedrie <[email protected]>
crawler-commons · Oct 30, 2024 · 52998a8 · 52998a8
1 parent 0eb6110
commit 52998a8
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/API/pom.xml b/API/pom.xml
@@ -41,6 +41,11 @@
 			<artifactId>grpc-stub</artifactId>
 			<version>${grpc.version}</version>
 		</dependency>
+		<dependency>
+			<groupId>io.grpc</groupId>
+			<artifactId>grpc-services</artifactId>
+			<version>${grpc.version}</version>
+		</dependency>
 
 	</dependencies>
 	<dependencyManagement>

diff --git a/service/src/main/java/crawlercommons/urlfrontier/service/URLFrontierServer.java b/service/src/main/java/crawlercommons/urlfrontier/service/URLFrontierServer.java
@@ -4,6 +4,7 @@
 package crawlercommons.urlfrontier.service;
 
 import crawlercommons.urlfrontier.service.rocksdb.RocksDBService;
+import io.grpc.protobuf.services.ProtoReflectionService;
 import io.grpc.Server;
 import io.grpc.ServerBuilder;
 import io.prometheus.client.exporter.HTTPServer;
@@ -170,7 +171,10 @@ public void start() throws Exception {
             }
         }
 
-        this.server = ServerBuilder.forPort(port).addService(service).build();
+        this.server = ServerBuilder.forPort(port)
+            .addService(service)
+            .addService(ProtoReflectionService.newInstance())
+            .build();
         server.start();
         LOG.info(
                 "Started URLFrontierServer [{}] on port {} as {}",