use crate::{
db::session::sql::SqlStatementResult,
value::{OutputValue, Value},
};
type SqlProjectionPayloadParts = (Vec<String>, Vec<Option<u32>>, Vec<Vec<Value>>, u32);
#[derive(Debug)]
pub(in crate::db::session::sql) struct SqlProjectionPayload {
columns: Vec<String>,
fixed_scales: Vec<Option<u32>>,
rows: Vec<Vec<Value>>,
row_count: u32,
}
impl SqlProjectionPayload {
#[must_use]
pub(in crate::db::session::sql) const fn new(
columns: Vec<String>,
fixed_scales: Vec<Option<u32>>,
rows: Vec<Vec<Value>>,
row_count: u32,
) -> Self {
Self {
columns,
fixed_scales,
rows,
row_count,
}
}
#[must_use]
pub(in crate::db::session::sql) fn into_parts(self) -> SqlProjectionPayloadParts {
(self.columns, self.fixed_scales, self.rows, self.row_count)
}
#[must_use]
pub(in crate::db::session::sql) fn into_statement_result(self) -> SqlStatementResult {
sql_projection_statement_result_from_value_rows(
self.columns,
self.fixed_scales,
self.rows,
self.row_count,
)
}
}
#[must_use]
pub(in crate::db::session::sql) fn sql_projection_statement_result_from_value_rows(
columns: Vec<String>,
fixed_scales: Vec<Option<u32>>,
rows: impl IntoIterator<Item = Vec<Value>>,
row_count: u32,
) -> SqlStatementResult {
SqlStatementResult::Projection {
columns,
fixed_scales,
rows: rows
.into_iter()
.map(sql_output_row_from_value_row)
.collect(),
row_count,
}
}
pub(in crate::db::session::sql) fn sql_projection_statement_result_from_fallible_value_rows<E>(
columns: Vec<String>,
fixed_scales: Vec<Option<u32>>,
rows: impl IntoIterator<Item = Result<Vec<Value>, E>>,
row_count: u32,
) -> Result<SqlStatementResult, E> {
let rows = rows
.into_iter()
.map(|row| row.map(sql_output_row_from_value_row))
.collect::<Result<Vec<_>, _>>()?;
Ok(SqlStatementResult::Projection {
columns,
fixed_scales,
rows,
row_count,
})
}
fn sql_output_row_from_value_row(row: Vec<Value>) -> Vec<OutputValue> {
row.into_iter().map(OutputValue::from).collect()
}