1use super::MySqlColumn;
2use crate::column::ColumnIndex;
3use crate::error::Error;
4use crate::ext::ustr::UStr;
5use crate::HashMap;
6use crate::{MySql, MySqlArguments, MySqlTypeInfo};
7use either::Either;
8use std::borrow::Cow;
9use std::sync::Arc;
10
11pub(crate) use sqlx_core::statement::*;
12
13#[derive(Debug, Clone)]
14pub struct MySqlStatement<'q> {
15 pub(crate) sql: Cow<'q, str>,
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<'q> Statement<'q> for MySqlStatement<'q> {
27 type Database = MySql;
28
29 fn to_owned(&self) -> MySqlStatement<'static> {
30 MySqlStatement::<'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 fn parameters(&self) -> Option<Either<&[MySqlTypeInfo], usize>> {
41 Some(Either::Right(self.metadata.parameters))
42 }
43
44 fn columns(&self) -> &[MySqlColumn] {
45 &self.metadata.columns
46 }
47
48 impl_statement_query!(MySqlArguments);
49}
50
51impl ColumnIndex<MySqlStatement<'_>> for &'_ str {
52 fn index(&self, statement: &MySqlStatement<'_>) -> Result<usize, Error> {
53 statement
54 .metadata
55 .column_names
56 .get(*self)
57 .ok_or_else(|| Error::ColumnNotFound((*self).into()))
58 .copied()
59 }
60}