1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60
use crate::any::{Any, AnyArguments, AnyColumn, AnyColumnIndex, AnyTypeInfo}; use crate::column::ColumnIndex; use crate::error::Error; use crate::ext::ustr::UStr; use crate::statement::Statement; use crate::HashMap; use either::Either; use std::borrow::Cow; use std::sync::Arc; pub struct AnyStatement<'q> { pub(crate) sql: Cow<'q, str>, pub(crate) parameters: Option<Either<Vec<AnyTypeInfo>, usize>>, pub(crate) column_names: Arc<HashMap<UStr, usize>>, pub(crate) columns: Vec<AnyColumn>, } impl<'q> Statement<'q> for AnyStatement<'q> { type Database = Any; fn to_owned(&self) -> AnyStatement<'static> { AnyStatement::<'static> { sql: Cow::Owned(self.sql.clone().into_owned()), column_names: self.column_names.clone(), parameters: self.parameters.clone(), columns: self.columns.clone(), } } fn sql(&self) -> &str { &self.sql } fn parameters(&self) -> Option<Either<&[AnyTypeInfo], usize>> { match &self.parameters { Some(Either::Left(types)) => Some(Either::Left(&types)), Some(Either::Right(count)) => Some(Either::Right(*count)), None => None, } } fn columns(&self) -> &[AnyColumn] { &self.columns } impl_statement_query!(AnyArguments<'_>); } impl<'i> ColumnIndex<AnyStatement<'_>> for &'i str where &'i str: AnyColumnIndex, { fn index(&self, statement: &AnyStatement<'_>) -> Result<usize, Error> { statement .column_names .get(*self) .ok_or_else(|| Error::ColumnNotFound((*self).into())) .map(|v| *v) } }