mirror of
https://github.com/launchbadge/sqlx.git
synced 2025-10-05 16:54:50 +00:00

* Add Query builder * Make query_builder.rs in sqlx-core * Add QueryBuilder::new() * Add QueryBuilder::push() * Define questions for documentation * Get new, push, push_bind working with types * Handle postgres' numbered bind varaibles * Add a test for QueryBuilder#build * Move arguments into Option * Refactor query builder * Finish testing QueryBuilder#build * Remove design doc * Add a test for pushing strings with push_bind * Integration test green * Adjust some tests * Make query builder generic about placeholder segmenent ('$N' or '?') * Run fmt * Redesign Arguments#format_placeholder in line with code review * Use write! to push sql to QueryBuilder * Add QueryBuilder::reset to allow for QueryBuilder reuse * Run cargo fmt
58 lines
1.8 KiB
Rust
58 lines
1.8 KiB
Rust
//! Types and traits for passing arguments to SQL queries.
|
|
|
|
use crate::database::{Database, HasArguments};
|
|
use crate::encode::Encode;
|
|
use crate::types::Type;
|
|
use std::fmt::{self, Write};
|
|
|
|
/// A tuple of arguments to be sent to the database.
|
|
pub trait Arguments<'q>: Send + Sized + Default {
|
|
type Database: Database;
|
|
|
|
/// Reserves the capacity for at least `additional` more values (of `size` total bytes) to
|
|
/// be added to the arguments without a reallocation.
|
|
fn reserve(&mut self, additional: usize, size: usize);
|
|
|
|
/// Add the value to the end of the arguments.
|
|
fn add<T>(&mut self, value: T)
|
|
where
|
|
T: 'q + Send + Encode<'q, Self::Database> + Type<Self::Database>;
|
|
|
|
fn format_placeholder<W: Write>(&self, writer: &mut W) -> fmt::Result {
|
|
writer.write_str("?")
|
|
}
|
|
}
|
|
|
|
pub trait IntoArguments<'q, DB: HasArguments<'q>>: Sized + Send {
|
|
fn into_arguments(self) -> <DB as HasArguments<'q>>::Arguments;
|
|
}
|
|
|
|
// NOTE: required due to lack of lazy normalization
|
|
#[allow(unused_macros)]
|
|
macro_rules! impl_into_arguments_for_arguments {
|
|
($Arguments:path) => {
|
|
impl<'q>
|
|
crate::arguments::IntoArguments<
|
|
'q,
|
|
<$Arguments as crate::arguments::Arguments<'q>>::Database,
|
|
> for $Arguments
|
|
{
|
|
fn into_arguments(self) -> $Arguments {
|
|
self
|
|
}
|
|
}
|
|
};
|
|
}
|
|
|
|
/// used by the query macros to prevent supernumerary `.bind()` calls
|
|
pub struct ImmutableArguments<'q, DB: HasArguments<'q>>(pub <DB as HasArguments<'q>>::Arguments);
|
|
|
|
impl<'q, DB: HasArguments<'q>> IntoArguments<'q, DB> for ImmutableArguments<'q, DB> {
|
|
fn into_arguments(self) -> <DB as HasArguments<'q>>::Arguments {
|
|
self.0
|
|
}
|
|
}
|
|
|
|
// TODO: Impl `IntoArguments` for &[&dyn Encode]
|
|
// TODO: Impl `IntoArguments` for (impl Encode, ...) x16
|