mirror of
https://github.com/launchbadge/sqlx.git
synced 2025-12-29 21:00:54 +00:00
c57b46ceb6
* Make encode and encode_by_ref fallible This only changes the trait for now and makes it compile, calling .expect() on all users. Those will be removed in a later commit. * PgNumeric: Turn TryFrom Decimal to an infallible From * Turn panics in Encode implementations into errors * Add Encode error analogous to the Decode error * Propagate decode errors through Arguments::add This pushes the panics one level further to mostly bind calls. Those will also be removed later. * Only check argument encoding at the end * Use Result in Query internally * Implement query_with functions in terms of _with_result * Surface encode errors when executing a query. * Remove remaining panics in AnyConnectionBackend implementations * PostgreSQL BigDecimal: Return encode error immediately * Arguments: Add len method to report how many arguments were added * Query::bind: Report which argument failed to encode * IsNull: Add is_null method * MySqlArguments: Replace manual bitmap code with NullBitMap helper type * Roll back buffer in MySqlArguments if encoding fails * Roll back buffer in SqliteArguments if encoding fails * Roll back PgArgumentBuffer if encoding fails
62 lines
1.9 KiB
Rust
62 lines
1.9 KiB
Rust
//! Types and traits for passing arguments to SQL queries.
|
|
|
|
use crate::database::Database;
|
|
use crate::encode::Encode;
|
|
use crate::error::BoxDynError;
|
|
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) -> Result<(), BoxDynError>
|
|
where
|
|
T: 'q + Encode<'q, Self::Database> + Type<Self::Database>;
|
|
|
|
/// The number of arguments that were already added.
|
|
fn len(&self) -> usize;
|
|
|
|
fn format_placeholder<W: Write>(&self, writer: &mut W) -> fmt::Result {
|
|
writer.write_str("?")
|
|
}
|
|
}
|
|
|
|
pub trait IntoArguments<'q, DB: Database>: Sized + Send {
|
|
fn into_arguments(self) -> <DB as Database>::Arguments<'q>;
|
|
}
|
|
|
|
// NOTE: required due to lack of lazy normalization
|
|
#[macro_export]
|
|
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: Database>(pub <DB as Database>::Arguments<'q>);
|
|
|
|
impl<'q, DB: Database> IntoArguments<'q, DB> for ImmutableArguments<'q, DB> {
|
|
fn into_arguments(self) -> <DB as Database>::Arguments<'q> {
|
|
self.0
|
|
}
|
|
}
|
|
|
|
// TODO: Impl `IntoArguments` for &[&dyn Encode]
|
|
// TODO: Impl `IntoArguments` for (impl Encode, ...) x16
|