diff --git a/server/src/server/services/table_definition_service.rs b/server/src/server/services/table_definition_service.rs index d3b4227..a96cd6a 100644 --- a/server/src/server/services/table_definition_service.rs +++ b/server/src/server/services/table_definition_service.rs @@ -1,13 +1,14 @@ -// src/server/services/table_definition_service.rs +// server/src/server/services/table_definition_service.rs use tonic::{Request, Response, Status}; -use common::proto::multieko2::table_definition::{ - table_definition_server::TableDefinition, - PostTableDefinitionRequest, TableDefinitionResponse +use common::proto::multieko2::{ + common::Empty, + table_definition::{ + table_definition_server::TableDefinition, + PostTableDefinitionRequest, TableDefinitionResponse, ProfileTreeResponse + } }; -use common::proto::multieko2::table_definition::ProfileTreeResponse; use sqlx::PgPool; -use crate::table_definition::handlers::post_table_definition; -use crate::table_definition::handlers::get_profile_tree; +use crate::table_definition::handlers::{post_table_definition, get_profile_tree}; #[derive(Debug)] pub struct TableDefinitionService { @@ -23,9 +24,10 @@ impl TableDefinition for TableDefinitionService { let response = post_table_definition(&self.db_pool, request.into_inner()).await?; Ok(Response::new(response)) } + async fn get_profile_tree( &self, - request: Request<()>, + request: Request, // Changed from Request<()> ) -> Result, Status> { get_profile_tree::get_profile_tree(&self.db_pool, request).await } diff --git a/server/src/table_definition/handlers.rs b/server/src/table_definition/handlers.rs index 2803921..db5b807 100644 --- a/server/src/table_definition/handlers.rs +++ b/server/src/table_definition/handlers.rs @@ -1,4 +1,4 @@ -// src/table_definition/handlers.rs +// server/src/table_definition/handlers.rs pub mod post_table_definition; pub mod get_profile_tree; diff --git a/server/src/table_definition/handlers/get_profile_tree.rs b/server/src/table_definition/handlers/get_profile_tree.rs index 1c875f4..73bfedf 100644 --- a/server/src/table_definition/handlers/get_profile_tree.rs +++ b/server/src/table_definition/handlers/get_profile_tree.rs @@ -1,11 +1,17 @@ // server/src/table_definition/handlers/get_profile_tree.rs use tonic::{Request, Response, Status}; use sqlx::PgPool; -use common::proto::multieko2::table_definition::ProfileTreeResponse; +use common::proto::multieko2::{ + common::Empty, + table_definition::{ + ProfileTreeResponse, + profile_tree_response::{Table, Profile} + } +}; pub async fn get_profile_tree( db_pool: &PgPool, - _request: Request<()>, + _request: Request, ) -> Result, Status> { let mut profiles = Vec::new(); @@ -45,13 +51,13 @@ pub async fn get_profile_tree( // Convert to protobuf format let proto_tables = table_map.into_iter() - .map(|(name, depends_on)| ProfileTreeResponse::Table { + .map(|(name, depends_on)| Table { name, depends_on }) .collect(); - profiles.push(ProfileTreeResponse::Profile { + profiles.push(Profile { name: profile.name, tables: proto_tables });