get by position in general endpoint is now working properly well

This commit is contained in:
filipriec
2025-03-05 18:59:19 +01:00
parent 1a3c063fdd
commit c27fa586d3
6 changed files with 187 additions and 4 deletions

View File

@@ -10,7 +10,7 @@ service TablesData {
rpc DeleteTableData (DeleteTableDataRequest) returns (DeleteTableDataResponse); rpc DeleteTableData (DeleteTableDataRequest) returns (DeleteTableDataResponse);
rpc GetTableData(GetTableDataRequest) returns (GetTableDataResponse); rpc GetTableData(GetTableDataRequest) returns (GetTableDataResponse);
rpc GetTableDataCount(GetTableDataCountRequest) returns (multieko2.common.CountResponse); rpc GetTableDataCount(GetTableDataCountRequest) returns (multieko2.common.CountResponse);
rpc GetTableDataByPosition(GetTableDataByPositionRequest) returns (GetTableDataResponse);
} }
message PostTableDataRequest { message PostTableDataRequest {
@@ -62,3 +62,9 @@ message GetTableDataCountRequest {
string profile_name = 1; string profile_name = 1;
string table_name = 2; string table_name = 2;
} }
message GetTableDataByPositionRequest {
string profile_name = 1;
string table_name = 2;
int32 position = 3;
}

Binary file not shown.

View File

@@ -81,6 +81,15 @@ pub struct GetTableDataCountRequest {
#[prost(string, tag = "2")] #[prost(string, tag = "2")]
pub table_name: ::prost::alloc::string::String, pub table_name: ::prost::alloc::string::String,
} }
#[derive(Clone, PartialEq, ::prost::Message)]
pub struct GetTableDataByPositionRequest {
#[prost(string, tag = "1")]
pub profile_name: ::prost::alloc::string::String,
#[prost(string, tag = "2")]
pub table_name: ::prost::alloc::string::String,
#[prost(int32, tag = "3")]
pub position: i32,
}
/// Generated client implementations. /// Generated client implementations.
pub mod tables_data_client { pub mod tables_data_client {
#![allow( #![allow(
@@ -308,6 +317,35 @@ pub mod tables_data_client {
); );
self.inner.unary(req, path, codec).await self.inner.unary(req, path, codec).await
} }
pub async fn get_table_data_by_position(
&mut self,
request: impl tonic::IntoRequest<super::GetTableDataByPositionRequest>,
) -> std::result::Result<
tonic::Response<super::GetTableDataResponse>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::unknown(
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/multieko2.tables_data.TablesData/GetTableDataByPosition",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(
GrpcMethod::new(
"multieko2.tables_data.TablesData",
"GetTableDataByPosition",
),
);
self.inner.unary(req, path, codec).await
}
} }
} }
/// Generated server implementations. /// Generated server implementations.
@@ -358,6 +396,13 @@ pub mod tables_data_server {
tonic::Response<super::super::common::CountResponse>, tonic::Response<super::super::common::CountResponse>,
tonic::Status, tonic::Status,
>; >;
async fn get_table_data_by_position(
&self,
request: tonic::Request<super::GetTableDataByPositionRequest>,
) -> std::result::Result<
tonic::Response<super::GetTableDataResponse>,
tonic::Status,
>;
} }
#[derive(Debug)] #[derive(Debug)]
pub struct TablesDataServer<T> { pub struct TablesDataServer<T> {
@@ -661,6 +706,55 @@ pub mod tables_data_server {
}; };
Box::pin(fut) Box::pin(fut)
} }
"/multieko2.tables_data.TablesData/GetTableDataByPosition" => {
#[allow(non_camel_case_types)]
struct GetTableDataByPositionSvc<T: TablesData>(pub Arc<T>);
impl<
T: TablesData,
> tonic::server::UnaryService<super::GetTableDataByPositionRequest>
for GetTableDataByPositionSvc<T> {
type Response = super::GetTableDataResponse;
type Future = BoxFuture<
tonic::Response<Self::Response>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::GetTableDataByPositionRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as TablesData>::get_table_data_by_position(
&inner,
request,
)
.await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = GetTableDataByPositionSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.unary(method, req).await;
Ok(res)
};
Box::pin(fut)
}
_ => { _ => {
Box::pin(async move { Box::pin(async move {
let mut response = http::Response::new(empty_body()); let mut response = http::Response::new(empty_body());

View File

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

View File

@@ -4,9 +4,11 @@ pub mod put_table_data;
pub mod delete_table_data; pub mod delete_table_data;
pub mod get_table_data; pub mod get_table_data;
pub mod get_table_data_count; pub mod get_table_data_count;
pub mod get_table_data_by_position;
pub use post_table_data::post_table_data; pub use post_table_data::post_table_data;
pub use put_table_data::put_table_data; pub use put_table_data::put_table_data;
pub use delete_table_data::delete_table_data; pub use delete_table_data::delete_table_data;
pub use get_table_data::get_table_data; pub use get_table_data::get_table_data;
pub use get_table_data_count::get_table_data_count; pub use get_table_data_count::get_table_data_count;
pub use get_table_data_by_position::get_table_data_by_position;

View File

@@ -0,0 +1,72 @@
// src/tables_data/handlers/get_table_data_by_position.rs
use tonic::Status;
use sqlx::PgPool;
use common::proto::multieko2::tables_data::{
GetTableDataByPositionRequest, GetTableDataRequest, GetTableDataResponse
};
use super::get_table_data;
pub async fn get_table_data_by_position(
db_pool: &PgPool,
request: GetTableDataByPositionRequest,
) -> Result<GetTableDataResponse, Status> {
let profile_name = request.profile_name;
let table_name = request.table_name;
if request.position < 1 {
return Err(Status::invalid_argument("Position must be at least 1"));
}
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;
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"));
}
let id: i64 = sqlx::query_scalar(
&format!(
r#"SELECT id FROM "{}"
WHERE deleted = FALSE
ORDER BY id ASC
OFFSET $1
LIMIT 1"#,
table_name
)
)
.bind(request.position - 1)
.fetch_optional(db_pool)
.await
.map_err(|e| Status::internal(format!("Position query failed: {}", e)))?
.ok_or_else(|| Status::not_found("Position out of bounds"))?;
get_table_data(
db_pool,
GetTableDataRequest {
profile_name,
table_name,
id,
}
).await
}