itsscb/sqlx
1
0
Fork 0
mirror of https://github.com/launchbadge/sqlx.git synced 2026-03-25 03:20:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

support arbitrary numbers of bind parameters in query!() et al

Browse Source
This commit is contained in:
Austin Bonander
2020-01-15 00:03:05 -08:00
parent da5c538d22
commit 0fdb875c20
10 changed files with 66 additions and 124 deletions
Download Patch File Download Diff File Expand all files Collapse all files

119
sqlx-core/src/arguments.rs
View File

@@ -38,123 +38,22 @@ where
fn into_arguments(self) -> DB::Arguments;
}
impl<DB> IntoArguments<DB> for DB::Arguments
impl<A> IntoArguments<A::Database> for A
where
DB: Database,
A: Arguments,
A::Database: Database<Arguments = Self> + Sized,
{
#[inline]
fn into_arguments(self) -> DB::Arguments {
fn into_arguments(self) -> Self {
self
}
}
#[allow(unused)]
macro_rules! impl_into_arguments {
($B:ident: $( ($idx:tt) -> $T:ident );+;) => {
impl<$($T,)+> crate::arguments::IntoArguments<$B> for ($($T,)+)
where
$($B: crate::types::HasSqlType<$T>,)+
$($T: crate::encode::Encode<$B>,)+
{
fn into_arguments(self) -> <$B as crate::database::Database>::Arguments {
use crate::arguments::Arguments;
#[doc(hidden)]
pub struct ImmutableArguments<DB: Database>(pub DB::Arguments);
let mut arguments = <$B as crate::database::Database>::Arguments::default();
let binds = 0 $(+ { $idx; 1 } )+;
let bytes = 0 $(+ crate::encode::Encode::size_hint(&self.$idx))+;
arguments.reserve(binds, bytes);
$(crate::arguments::Arguments::add(&mut arguments, self.$idx);)+
arguments
}
}
};
}
#[allow(unused)]
macro_rules! impl_into_arguments_for_database {
($B:ident) => {
impl crate::arguments::IntoArguments<$B> for ()
{
#[inline]
fn into_arguments(self) -> <$B as crate::database::Database>::Arguments {
Default::default()
}
}
impl_into_arguments!($B:
(0) -> T1;
);
impl_into_arguments!($B:
(0) -> T1;
(1) -> T2;
);
impl_into_arguments!($B:
(0) -> T1;
(1) -> T2;
(2) -> T3;
);
impl_into_arguments!($B:
(0) -> T1;
(1) -> T2;
(2) -> T3;
(3) -> T4;
);
impl_into_arguments!($B:
(0) -> T1;
(1) -> T2;
(2) -> T3;
(3) -> T4;
(4) -> T5;
);
impl_into_arguments!($B:
(0) -> T1;
(1) -> T2;
(2) -> T3;
(3) -> T4;
(4) -> T5;
(5) -> T6;
);
impl_into_arguments!($B:
(0) -> T1;
(1) -> T2;
(2) -> T3;
(3) -> T4;
(4) -> T5;
(5) -> T6;
(6) -> T7;
);
impl_into_arguments!($B:
(0) -> T1;
(1) -> T2;
(2) -> T3;
(3) -> T4;
(4) -> T5;
(5) -> T6;
(6) -> T7;
(7) -> T8;
);
impl_into_arguments!($B:
(0) -> T1;
(1) -> T2;
(2) -> T3;
(3) -> T4;
(4) -> T5;
(5) -> T6;
(6) -> T7;
(7) -> T8;
(8) -> T9;
);
impl<DB: Database> IntoArguments<DB> for ImmutableArguments<DB> {
fn into_arguments(self) -> <DB as Database>::Arguments {
self.0
}
}

2
sqlx-core/src/mysql/database.rs
View File

@@ -14,5 +14,3 @@ impl Database for MySql {
type TableId = Box<str>;
}
impl_into_arguments_for_database!(MySql);

2
sqlx-core/src/postgres/arguments.rs
View File

@@ -1,6 +1,6 @@
use byteorder::{ByteOrder, NetworkEndian};
use crate::arguments::Arguments;
use crate::arguments::{Arguments, IntoArguments};
use crate::encode::{Encode, IsNull};
use crate::io::BufMut;
use crate::types::HasSqlType;

2
sqlx-core/src/postgres/database.rs
View File

@@ -14,5 +14,3 @@ impl Database for Postgres {
type TableId = u32;
}
impl_into_arguments_for_database!(Postgres);

9
sqlx-core/src/query_as.rs
View File

@@ -1,7 +1,7 @@
use futures_core::Stream;
use futures_util::{future, TryStreamExt};
use crate::arguments::Arguments;
use crate::arguments::{Arguments, ImmutableArguments};
use crate::{
arguments::IntoArguments, database::Database, encode::Encode, executor::Executor, row::FromRow,
types::HasSqlType,
@@ -128,13 +128,10 @@ where
// used by query!() and friends
#[doc(hidden)]
pub fn bind_all<I>(self, values: I) -> QueryAs<'q, DB, R, I>
where
I: IntoArguments<DB>,
{
pub fn bind_all(self, values: DB::Arguments) -> QueryAs<'q, DB, R, ImmutableArguments<DB>> {
QueryAs {
query: self.query,
args: values,
args: ImmutableArguments(values),
map_row: self.map_row,
}
}