add /api/tables endpoint

wobcom · Jan 22, 2025 · 5314a24 · 5314a24
1 parent 33be446
commit 5314a24
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/src/api.rs b/src/api.rs
@@ -336,6 +336,9 @@ async fn query<T: Store>(
     Ok(Body::from_stream(stream))
 }
 
+async fn tables<T: Store>(State(AppState { store, .. }): State<AppState<T>>) -> impl IntoResponse {
+    serde_json::to_string(&store.get_tables()).unwrap()
+}
 async fn routers<T: Store>(State(AppState { store, .. }): State<AppState<T>>) -> impl IntoResponse {
     serde_json::to_string(&store.get_routers()).unwrap()
 }
@@ -371,6 +374,7 @@ async fn make_api<T: Store>(cfg: ApiServerConfig, store: T) -> anyhow::Result<Ro
     Ok(Router::new()
         .route("/query", get(query::<T>))
         .route("/routers", get(routers::<T>))
+        .route("/tables", get(tables::<T>))
         .route("/routing-instances", get(routing_instances::<T>))
         .with_state(AppState {
             cfg: Arc::new(cfg),

diff --git a/src/store.rs b/src/store.rs
@@ -195,6 +195,8 @@ pub trait Store: Clone + Send + Sync + 'static {
 
     fn get_routes(&self, query: Query) -> Pin<Box<dyn Stream<Item = QueryResult> + Send>>;
 
+    fn get_tables(&self) -> Vec<TableSelector>;
+
     fn get_routers(&self) -> HashMap<SocketAddr, Client>;
 
     fn get_routing_instances(&self) -> HashMap<SocketAddr, HashSet<RouteDistinguisher>>;

diff --git a/src/store_impl.rs b/src/store_impl.rs
@@ -214,6 +214,10 @@ impl Store for InMemoryStore {
         )
     }
 
+    fn get_tables(&self) -> Vec<TableSelector> {
+        self.tables.lock().unwrap().keys().cloned().collect()
+    }
+
     fn get_routers(&self) -> HashMap<SocketAddr, Client> {
         self.clients.lock().unwrap().clone()
     }