sqlx_core_oldapi/mysql/
statement.rs1use super::MySqlColumn;
2use crate::column::ColumnIndex;
3use crate::error::Error;
4use crate::ext::ustr::UStr;
5use crate::mysql::{MySql, MySqlArguments, MySqlTypeInfo};
6use crate::statement::Statement;
7use crate::HashMap;
8use either::Either;
9use std::borrow::Cow;
10use std::sync::Arc;
11
12#[derive(Debug, Clone)]
13pub struct MySqlStatement<'q> {
14 pub(crate) sql: Cow<'q, str>,
15 pub(crate) metadata: MySqlStatementMetadata,
16}
17
18#[derive(Debug, Default, Clone)]
19pub(crate) struct MySqlStatementMetadata {
20 pub(crate) columns: Arc<Vec<MySqlColumn>>,
21 pub(crate) column_names: Arc<HashMap<UStr, usize>>,
22 pub(crate) parameters: usize,
23}
24
25impl<'q> Statement<'q> for MySqlStatement<'q> {
26 type Database = MySql;
27
28 fn to_owned(&self) -> MySqlStatement<'static> {
29 MySqlStatement::<'static> {
30 sql: Cow::Owned(self.sql.clone().into_owned()),
31 metadata: self.metadata.clone(),
32 }
33 }
34
35 fn sql(&self) -> &str {
36 &self.sql
37 }
38
39 fn parameters(&self) -> Option<Either<&[MySqlTypeInfo], usize>> {
40 Some(Either::Right(self.metadata.parameters))
41 }
42
43 fn columns(&self) -> &[MySqlColumn] {
44 &self.metadata.columns
45 }
46
47 impl_statement_query!(MySqlArguments);
48}
49
50impl ColumnIndex<MySqlStatement<'_>> for &'_ str {
51 fn index(&self, statement: &MySqlStatement<'_>) -> Result<usize, Error> {
52 statement
53 .metadata
54 .column_names
55 .get(*self)
56 .ok_or_else(|| Error::ColumnNotFound((*self).into()))
57 .map(|v| *v)
58 }
59}
60
61#[cfg(feature = "any")]
62impl<'q> From<MySqlStatement<'q>> for crate::any::AnyStatement<'q> {
63 #[inline]
64 fn from(statement: MySqlStatement<'q>) -> Self {
65 crate::any::AnyStatement::<'q> {
66 columns: statement
67 .metadata
68 .columns
69 .iter()
70 .map(|col| col.clone().into())
71 .collect(),
72 column_names: statement.metadata.column_names,
73 parameters: Some(Either::Right(statement.metadata.parameters)),
74 sql: statement.sql,
75 }
76 }
77}