cdbc_mysql/
statement.rs

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