sqlx_postgres/
statement.rs

1use super::{PgColumn, PgTypeInfo};
2use crate::column::ColumnIndex;
3use crate::error::Error;
4use crate::ext::ustr::UStr;
5use crate::{PgArguments, Postgres};
6use std::sync::Arc;
7
8use sqlx_core::sql_str::SqlStr;
9pub(crate) use sqlx_core::statement::Statement;
10use sqlx_core::{Either, HashMap};
11
12#[derive(Debug, Clone)]
13pub struct PgStatement {
14    pub(crate) sql: SqlStr,
15    pub(crate) metadata: Arc<PgStatementMetadata>,
16}
17
18#[derive(Debug, Default)]
19pub(crate) struct PgStatementMetadata {
20    pub(crate) columns: Vec<PgColumn>,
21    // This `Arc` is not redundant; it's used to avoid deep-copying this map for the `Any` backend.
22    // See `sqlx-postgres/src/any.rs`
23    pub(crate) column_names: Arc<HashMap<UStr, usize>>,
24    pub(crate) parameters: Vec<PgTypeInfo>,
25}
26
27impl Statement for PgStatement {
28    type Database = Postgres;
29
30    fn into_sql(self) -> SqlStr {
31        self.sql
32    }
33
34    fn sql(&self) -> &SqlStr {
35        &self.sql
36    }
37
38    fn parameters(&self) -> Option<Either<&[PgTypeInfo], usize>> {
39        Some(Either::Left(&self.metadata.parameters))
40    }
41
42    fn columns(&self) -> &[PgColumn] {
43        &self.metadata.columns
44    }
45
46    impl_statement_query!(PgArguments);
47}
48
49impl ColumnIndex<PgStatement> for &'_ str {
50    fn index(&self, statement: &PgStatement) -> Result<usize, Error> {
51        statement
52            .metadata
53            .column_names
54            .get(*self)
55            .ok_or_else(|| Error::ColumnNotFound((*self).into()))
56            .copied()
57    }
58}
59
60// #[cfg(feature = "any")]
61// impl<'q> From<PgStatement<'q>> for crate::any::AnyStatement<'q> {
62//     #[inline]
63//     fn from(statement: PgStatement<'q>) -> Self {
64//         crate::any::AnyStatement::<'q> {
65//             columns: statement
66//                 .metadata
67//                 .columns
68//                 .iter()
69//                 .map(|col| col.clone().into())
70//                 .collect(),
71//             column_names: statement.metadata.column_names.clone(),
72//             parameters: Some(Either::Left(
73//                 statement
74//                     .metadata
75//                     .parameters
76//                     .iter()
77//                     .map(|ty| ty.clone().into())
78//                     .collect(),
79//             )),
80//             sql: statement.sql,
81//         }
82//     }
83// }
sqlx_postgres/statement.rs

sqlx_postgres/
statement.rs
