feat: add count route

server/src/api/count.rs

@@ -0,0 +1,80 @@
+use crate::app_state::AppState;
+use crate::error::CustomRouteResult;
+use crate::identifier::is_valid_table_identifier;
+use crate::response::build_response;
+use actix_web::http::StatusCode;
+use actix_web::{
+    post,
+    web::{self, Path},
+    HttpResponse,
+};
+use serde_json::json;
+
+#[post("/v1/table/{name}/count")]
+pub async fn handler(
+    path: Path<String>,
+    app_state: web::Data<AppState>,
+) -> CustomRouteResult<HttpResponse> {
+    let before = std::time::Instant::now();
+
+    let table_name = path.into_inner();
+
+    if table_name.starts_with('_') {
+        return Ok(build_response(
+            before.elapsed(),
+            StatusCode::BAD_REQUEST,
+            "Invalid table name",
+            &json!(null),
+        ));
+    }
+
+    if !is_valid_table_identifier(&table_name) {
+        return Ok(build_response(
+            before.elapsed(),
+            StatusCode::BAD_REQUEST,
+            "Invalid table name",
+            &json!(null),
+        ));
+    }
+
+    let tables = app_state.tables.read().await;
+
+    if let Some(table) = tables.get(&table_name) {
+        let (row_count, cell_count) = {
+            let table = table.clone();
+
+            tokio::task::spawn_blocking(move || table.count())
+                .await
+                .expect("should join")
+        }?;
+
+        let dur = before.elapsed();
+
+        let micros_total = dur.as_micros();
+
+        let micros_per_row = if row_count == 0 {
+            None
+        } else {
+            Some(micros_total / row_count as u128)
+        };
+
+        Ok(build_response(
+            dur,
+            StatusCode::OK,
+            "Count successful",
+            &json!({
+                "row_count": row_count,
+                "cell_count": cell_count,
+                "micros": micros_total,
+                "micros_per_row": micros_per_row,
+            }),
+        ))
+    } else {
+        Ok(build_response(
+            before.elapsed(),
+            StatusCode::NOT_FOUND,
+            "Table not found",
+            &json!(null),
+        ))
+    }
+}

server/src/api/delete_row.rs

@@ -29,8 +29,6 @@ pub async fn handler(
 ) -> CustomRouteResult<HttpResponse> {
     let before = std::time::Instant::now();
 
-    let tables = app_state.tables.read().await;
-
     let table_name = path.into_inner();
 
     if table_name.starts_with('_') {
@@ -53,6 +51,8 @@ pub async fn handler(
 
     let req_body = req_body.into_inner();
 
+    let tables = app_state.tables.read().await;
+
     if let Some(table) = tables.get(&table_name) {
         let count = {
             let table = table.clone();

server/src/api/delete_table.rs

@@ -17,8 +17,6 @@ pub async fn handler(
 ) -> CustomRouteResult<HttpResponse> {
     let before = std::time::Instant::now();
 
-    let mut tables = app_state.tables.write().await;
-
     let table_name = path.into_inner();
 
     if table_name.starts_with('_') {
@@ -39,6 +37,8 @@ pub async fn handler(
         ));
     }
 
+    let mut tables = app_state.tables.write().await;
+
     if let Some(table) = tables.get(&table_name).cloned() {
         app_state.manifest_table.delete_user_table(&table_name)?;
         tables.remove(&table_name);

server/src/api/get_rows.rs

@@ -26,8 +26,6 @@ pub async fn handler(
 ) -> CustomRouteResult<HttpResponse> {
     let before = std::time::Instant::now();
 
-    let tables = app_state.tables.read().await;
-
     let table_name = path.into_inner();
 
     if table_name.starts_with('_') {
@@ -48,6 +46,8 @@ pub async fn handler(
         ));
     }
 
+    let tables = app_state.tables.read().await;
+
     if let Some(table) = tables.get(&table_name) {
         let result = {
             let table = table.clone();

server/src/api/metrics.rs

@@ -16,12 +16,12 @@ pub async fn handler(
 ) -> CustomRouteResult<HttpResponse> {
     let before = std::time::Instant::now();
 
-    let tables = app_state.tables.write().await;
-
     let table_name = path.into_inner();
 
     let actual_name = format!("usr_{table_name}");
 
+    let tables = app_state.tables.write().await;
+
     if tables.get(&actual_name).is_some() {
         /* let rows = app_state
         .metrics_table

server/src/api/mod.rs

@@ -1,3 +1,4 @@
+pub mod count;
 pub mod create_column_family;
 pub mod create_table;
 pub mod delete_row;

server/src/api/scan.rs

@@ -20,8 +20,6 @@ pub async fn handler(
 ) -> CustomRouteResult<HttpResponse> {
     let before = std::time::Instant::now();
 
-    let tables = app_state.tables.read().await;
-
     let table_name = path.into_inner();
 
     if table_name.starts_with('_') {
@@ -42,6 +40,8 @@ pub async fn handler(
         ));
     }
 
+    let tables = app_state.tables.read().await;
+
     if let Some(table) = tables.get(&table_name) {
         let result = {
             let table = table.clone();

server/src/api/write.rs

@@ -32,8 +32,6 @@ pub async fn handler(
         return bad_request(before, "Items array should not be empty");
     }
 
-    let tables = app_state.tables.read().await;
-
     let table_name = path.into_inner();
 
     if table_name.starts_with('_') {
@@ -54,6 +52,8 @@ pub async fn handler(
         ));
     }
 
+    let tables = app_state.tables.read().await;
+
     if let Some(table) = tables.get(&table_name).cloned() {
         // TODO: use spawn_blocking
 

server/src/main.rs

@@ -135,6 +135,7 @@ async fn main() -> smoltable::Result<()> {
             .service(api::list_tables::handler)
             .service(api::create_table::handler)
             .service(api::write::handler)
+            .service(api::count::handler)
             .service(api::get_rows::handler)
             .service(api::delete_row::handler)
             .service(api::scan::handler)