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}