Expose reflection metadata

splitgraph · Dec 30, 2023 · 3f365c1 · 3f365c1
1 parent a565470
commit 3f365c1
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/clade/build.rs b/clade/build.rs
@@ -1,5 +1,10 @@
+use std::env;
+use std::path::PathBuf;
+
 fn main() -> Result<(), Box<dyn std::error::Error>> {
+    let out_dir = PathBuf::from(env::var("OUT_DIR").unwrap());
     tonic_build::configure()
+        .file_descriptor_set_path(out_dir.join("clade_descriptor.bin"))
         .build_server(true)
         .build_client(true)
         .compile(&["proto/catalog.proto", "proto/schema.proto"], &["proto"])?;

diff --git a/clade/src/lib.rs b/clade/src/lib.rs
@@ -4,4 +4,5 @@ pub mod catalog {
 
 pub mod schema {
     tonic::include_proto!("clade.schema");
+    pub const FILE_DESCRIPTOR_SET: &[u8] = tonic::include_file_descriptor_set!("clade_descriptor");
 }
diff --git a/tests/clade/mod.rs b/tests/clade/mod.rs
@@ -1,6 +1,7 @@
 use clade::catalog::{CatalogReference, TableObject};
 use clade::schema::{
     schema_store_service_server::{SchemaStoreService, SchemaStoreServiceServer},
+    FILE_DESCRIPTOR_SET,
     ListSchemaResponse, SchemaObject,
 };
 use datafusion_common::assert_batches_eq;
@@ -84,8 +85,14 @@ async fn run_clade_server(addr: SocketAddr) {
 
     let svc = SchemaStoreServiceServer::new(metastore);
 
+    let reflection = tonic_reflection::server::Builder::configure()
+        .register_encoded_file_descriptor_set(FILE_DESCRIPTOR_SET)
+        .build()
+        .unwrap();
+
     Server::builder()
         .add_service(svc)
+        .add_service(reflection)
         .serve(addr)
         .await
         .unwrap();