working get count endpoint

server/src/server/services/tables_data_service.rs

@@ -1,13 +1,15 @@
 // src/server/services/tables_data_service.rs
 use tonic::{Request, Response, Status};
 use common::proto::multieko2::tables_data::tables_data_server::TablesData;
+use common::proto::multieko2::common::CountResponse;
 use common::proto::multieko2::tables_data::{
     PostTableDataRequest, PostTableDataResponse,
     PutTableDataRequest, PutTableDataResponse,
     DeleteTableDataRequest, DeleteTableDataResponse,
     GetTableDataRequest, GetTableDataResponse,
+    GetTableDataCountRequest,
 };
-use crate::tables_data::handlers::{post_table_data, put_table_data, delete_table_data, get_table_data,};
+use crate::tables_data::handlers::{post_table_data, put_table_data, delete_table_data, get_table_data, get_table_data_count};
 use sqlx::PgPool;
 
 #[derive(Debug)]
@@ -53,4 +55,13 @@ impl TablesData for TablesDataService {
         let response = get_table_data(&self.db_pool, request).await?;
         Ok(Response::new(response))
     }
+
+    async fn get_table_data_count(
+        &self,
+        request: Request<GetTableDataCountRequest>,
+    ) -> Result<Response<CountResponse>, Status> {
+        let request = request.into_inner();
+        let response = get_table_data_count(&self.db_pool, request).await?;
+        Ok(Response::new(response))
+    }
 }

server/src/tables_data/docs/get_data_count.txt

@@ -0,0 +1,5 @@
+grpcurl -plaintext -d '{"profile_name": "default", "table_name": "2025_adresar"}' localhost:50051 multieko2.tables_data.TablesData/GetTableDataCount
+
+{
+  "count": "1"
+}

server/src/tables_data/handlers.rs

@@ -3,8 +3,10 @@ pub mod post_table_data;
 pub mod put_table_data;
 pub mod delete_table_data;
 pub mod get_table_data;
+pub mod get_table_data_count;
 
 pub use post_table_data::post_table_data;
 pub use put_table_data::put_table_data;
 pub use delete_table_data::delete_table_data;
 pub use get_table_data::get_table_data;
+pub use get_table_data_count::get_table_data_count;

server/src/tables_data/handlers/get_table_data_count.rs

@@ -0,0 +1,61 @@
+// src/tables_data/handlers/get_table_data_count.rs
+use tonic::Status;
+use sqlx::PgPool;
+use common::proto::multieko2::common::{CountResponse, Empty};
+use common::proto::multieko2::tables_data::GetTableDataCountRequest;
+
+pub async fn get_table_data_count(
+    db_pool: &PgPool,
+    request: GetTableDataCountRequest,
+) -> Result<CountResponse, Status> {
+    let profile_name = request.profile_name;
+    let table_name = request.table_name;
+
+    // Lookup profile
+    let profile = sqlx::query!(
+        "SELECT id FROM profiles WHERE name = $1",
+        profile_name
+    )
+    .fetch_optional(db_pool)
+    .await
+    .map_err(|e| Status::internal(format!("Profile lookup error: {}", e)))?;
+
+    let profile_id = profile.ok_or_else(|| Status::not_found("Profile not found"))?.id;
+
+    // Verify table exists and belongs to profile
+    let table_exists = sqlx::query!(
+        r#"SELECT EXISTS(
+            SELECT 1 FROM table_definitions
+            WHERE profile_id = $1 AND table_name = $2
+        )"#,
+        profile_id,
+        table_name
+    )
+    .fetch_one(db_pool)
+    .await
+    .map_err(|e| Status::internal(format!("Table verification error: {}", e)))?
+    .exists
+    .unwrap_or(false);
+
+    if !table_exists {
+        return Err(Status::not_found("Table not found"));
+    }
+
+    // Get count of non-deleted records
+    let query = format!(
+        r#"
+        SELECT COUNT(*) AS count
+        FROM "{}"
+        WHERE deleted = FALSE
+        "#,
+        table_name
+    );
+
+    let count: i64 = sqlx::query_scalar::<_, Option<i64>>(&query)
+        .fetch_one(db_pool)
+        .await
+        .map_err(|e| Status::internal(format!("Count query failed: {}", e)))?
+        .unwrap_or(0);
+
+    Ok(CountResponse { count })
+}