cdbc_pg/
statement.rs

1use super::{PgColumn, PgTypeInfo};
2use cdbc::column::ColumnIndex;
3use cdbc::error::Error;
4use cdbc::utils::ustr::UStr;
5use crate::{PgArguments, Postgres};
6use cdbc::statement::Statement;
7use cdbc::HashMap;
8use either::Either;
9use std::borrow::Cow;
10use std::sync::Arc;
11
12#[derive(Debug, Clone)]
13pub struct PgStatement {
14    pub(crate) sql: String,
15    pub(crate) metadata: Arc<PgStatementMetadata>,
16}
17
18#[derive(Debug, Default)]
19pub(crate) struct PgStatementMetadata {
20    pub(crate) columns: Vec<PgColumn>,
21    pub(crate) column_names: HashMap<UStr, usize>,
22    pub(crate) parameters: Vec<PgTypeInfo>,
23}
24
25impl Statement for PgStatement {
26    type Database = Postgres;
27
28    fn to_owned(&self) -> PgStatement {
29        PgStatement {
30            sql: self.sql.clone(),
31            metadata: self.metadata.clone(),
32        }
33    }
34
35    fn sql(&self) -> &str {
36        &self.sql
37    }
38
39    fn sql_mut(&mut self) -> &mut String {
40        &mut self.sql
41    }
42
43    fn parameters(&self) -> Option<Either<&[PgTypeInfo], usize>> {
44        Some(Either::Left(&self.metadata.parameters))
45    }
46
47    fn columns(&self) -> &[PgColumn] {
48        &self.metadata.columns
49    }
50
51    impl_statement_query!(PgArguments);
52}
53
54impl ColumnIndex<PgStatement> for &'_ str {
55    fn index(&self, statement: &PgStatement) -> Result<usize, Error> {
56        statement
57            .metadata
58            .column_names
59            .get(*self)
60            .ok_or_else(|| Error::ColumnNotFound((*self).into()))
61            .map(|v| *v)
62    }
63}
cdbc_pg/statement.rs

cdbc_pg/
statement.rs
