sqlx_xugu/
statement.rs

1use super::XuguColumn;
2use crate::column::ColumnIndex;
3use crate::error::Error;
4use crate::{Xugu, XuguArguments, XuguTypeInfo};
5use sqlx_core::ext::ustr::UStr;
6use sqlx_core::{impl_statement_query, Either, HashMap};
7use std::borrow::Cow;
8use std::sync::Arc;
9
10use crate::protocol::statement::ParameterDef;
11pub(crate) use sqlx_core::statement::*;
12
13#[derive(Debug, Clone)]
14pub struct XuguStatement<'q> {
15    pub(crate) sql: Cow<'q, str>,
16    pub(crate) metadata: XuguStatementMetadata,
17}
18
19#[derive(Debug, Default, Clone)]
20pub(crate) struct XuguStatementMetadata {
21    pub(crate) columns: Arc<Vec<XuguColumn>>,
22    pub(crate) column_names: Arc<HashMap<UStr, usize>>,
23    pub(crate) parameters: Arc<Vec<ParameterDef>>,
24}
25
26impl<'q> Statement<'q> for XuguStatement<'q> {
27    type Database = Xugu;
28
29    fn to_owned(&self) -> XuguStatement<'static> {
30        XuguStatement::<'static> {
31            sql: Cow::Owned(self.sql.clone().into_owned()),
32            metadata: self.metadata.clone(),
33        }
34    }
35
36    fn sql(&self) -> &str {
37        &self.sql
38    }
39
40    /// 获取此语句的预期参数。
41    ///
42    /// 返回的信息取决于驱动程序提供的信息。SQLite 只能告诉我们参数的数量。PostgreSQL 可以为我们提供完整的类型信息。
43    fn parameters(&self) -> Option<Either<&[XuguTypeInfo], usize>> {
44        Some(Either::Right(self.metadata.parameters.len()))
45    }
46
47    fn columns(&self) -> &[XuguColumn] {
48        &self.metadata.columns
49    }
50
51    impl_statement_query!(XuguArguments);
52}
53
54impl ColumnIndex<XuguStatement<'_>> for &'_ str {
55    fn index(&self, statement: &XuguStatement<'_>) -> Result<usize, Error> {
56        statement
57            .metadata
58            .column_names
59            .get(*self)
60            .ok_or_else(|| Error::ColumnNotFound((*self).into()))
61            .copied()
62    }
63}
sqlx_xugu/statement.rs

sqlx_xugu/
statement.rs
