dynoxide/actions/
execute_statement.rs

1use crate::errors::{DynoxideError, Result};
2use crate::partiql;
3use crate::storage_backend::StorageBackend;
4use crate::types::{AttributeValue, Item};
5use serde::{Deserialize, Serialize};
6
7#[derive(Debug, Default, Deserialize)]
8pub struct ExecuteStatementRequest {
9    #[serde(rename = "Statement")]
10    pub statement: String,
11    #[serde(rename = "Parameters", default)]
12    pub parameters: Option<Vec<AttributeValue>>,
13    #[serde(rename = "Limit", default)]
14    pub limit: Option<usize>,
15    #[serde(rename = "NextToken", default)]
16    pub next_token: Option<String>,
17    /// Accepted for API compatibility. Has no behavioural effect — SQLite
18    /// reads are always consistent.
19    #[serde(rename = "ConsistentRead", default)]
20    pub consistent_read: Option<bool>,
21    #[serde(rename = "ReturnConsumedCapacity", default)]
22    pub return_consumed_capacity: Option<String>,
23}
24
25#[derive(Debug, Default, Serialize)]
26pub struct ExecuteStatementResponse {
27    #[serde(rename = "Items", skip_serializing_if = "Option::is_none")]
28    pub items: Option<Vec<Item>>,
29    #[serde(rename = "NextToken", skip_serializing_if = "Option::is_none")]
30    pub next_token: Option<String>,
31    #[serde(rename = "ConsumedCapacity", skip_serializing_if = "Option::is_none")]
32    pub consumed_capacity: Option<crate::types::ConsumedCapacity>,
33}
34
35pub async fn execute<S: StorageBackend>(
36    storage: &S,
37    request: ExecuteStatementRequest,
38) -> Result<ExecuteStatementResponse> {
39    let stmt = partiql::parser::parse(&request.statement).map_err(|e| {
40        DynoxideError::ValidationException(format!(
41            "Statement wasn't well formed, can't be processed: {e}"
42        ))
43    })?;
44
45    let params = request.parameters.unwrap_or_default();
46    let (items, size) =
47        partiql::executor::execute_measured(storage, &stmt, &params, request.limit).await?;
48
49    // ConsumedCapacity is returned whenever ReturnConsumedCapacity is requested,
50    // unlike some emulators that omit it. A SELECT is charged read units (an
51    // eventually consistent read unless ConsistentRead is set); INSERT, UPDATE
52    // and DELETE are charged write units. The unit (single object, not an array)
53    // comes from the shared `types.rs` helpers.
54    let consumed_capacity = partiql::parser::table_name(&stmt).and_then(|table| {
55        let units = if matches!(stmt, partiql::parser::Statement::Select { .. }) {
56            crate::types::read_capacity_units_with_consistency(
57                size,
58                request.consistent_read.unwrap_or(false),
59            )
60        } else {
61            crate::types::write_capacity_units(size)
62        };
63        crate::types::consumed_capacity(table, units, &request.return_consumed_capacity)
64    });
65
66    Ok(ExecuteStatementResponse {
67        items,
68        next_token: None,
69        consumed_capacity,
70    })
71}
dynoxide/actions/execute_statement.rs

dynoxide/actions/
execute_statement.rs
