use crate::backend::sql_dialect;
use crate::expression::SqlLiteral;
use crate::query_builder::*;
use crate::query_source::Column;
mod sealed {
pub trait Sealed {}
}
impl sealed::Sealed for NoConflictTarget {}
impl<T> sealed::Sealed for ConflictTarget<T> {}
pub trait OnConflictTarget<Table>: sealed::Sealed {}
#[doc(hidden)]
#[derive(Debug, Clone, Copy, QueryId)]
pub struct NoConflictTarget;
impl<DB> QueryFragment<DB> for NoConflictTarget
where
DB: Backend,
DB::OnConflictClause: sql_dialect::on_conflict_clause::SupportsOnConflictClause,
{
fn walk_ast<'b>(&'b self, _: AstPass<'_, 'b, DB>) -> QueryResult<()> {
Ok(())
}
}
impl<Table> OnConflictTarget<Table> for NoConflictTarget {}
#[doc(hidden)]
#[derive(Debug, Clone, Copy, QueryId)]
pub struct ConflictTarget<T>(T);
impl<T> ConflictTarget<T> {
#[diesel_derives::__diesel_public_if(
feature = "i-implement-a-third-party-backend-and-opt-into-breaking-changes"
)]
pub(crate) fn new(target: T) -> Self {
ConflictTarget(target)
}
#[allow(dead_code)]
pub(crate) fn target(&self) -> &T {
&self.0
}
}
impl<DB, T> QueryFragment<DB> for ConflictTarget<T>
where
DB: Backend,
Self: QueryFragment<DB, DB::OnConflictClause>,
{
fn walk_ast<'b>(&'b self, pass: AstPass<'_, 'b, DB>) -> QueryResult<()> {
<Self as QueryFragment<DB, DB::OnConflictClause>>::walk_ast(self, pass)
}
}
impl<DB, T, SP> QueryFragment<DB, SP> for ConflictTarget<T>
where
DB: Backend<OnConflictClause = SP>,
SP: sql_dialect::on_conflict_clause::PgLikeOnConflictClause,
T: Column,
{
fn walk_ast<'b>(&'b self, mut out: AstPass<'_, 'b, DB>) -> QueryResult<()> {
out.push_sql(" (");
out.push_identifier(T::NAME)?;
out.push_sql(")");
Ok(())
}
}
impl<T> OnConflictTarget<T::Table> for ConflictTarget<T> where T: Column {}
impl<DB, ST, SP> QueryFragment<DB, SP> for ConflictTarget<SqlLiteral<ST>>
where
DB: Backend<OnConflictClause = SP>,
SP: sql_dialect::on_conflict_clause::PgLikeOnConflictClause,
SqlLiteral<ST>: QueryFragment<DB>,
{
fn walk_ast<'b>(&'b self, mut out: AstPass<'_, 'b, DB>) -> QueryResult<()> {
out.push_sql(" ");
self.0.walk_ast(out.reborrow())?;
Ok(())
}
}
impl<Tab, ST> OnConflictTarget<Tab> for ConflictTarget<SqlLiteral<ST>> {}
impl<DB, T, SP> QueryFragment<DB, SP> for ConflictTarget<(T,)>
where
DB: Backend<OnConflictClause = SP>,
SP: sql_dialect::on_conflict_clause::PgLikeOnConflictClause,
T: Column,
{
fn walk_ast<'b>(&'b self, mut out: AstPass<'_, 'b, DB>) -> QueryResult<()> {
out.push_sql(" (");
out.push_identifier(T::NAME)?;
out.push_sql(")");
Ok(())
}
}
impl<T> OnConflictTarget<T::Table> for ConflictTarget<(T,)> where T: Column {}
macro_rules! on_conflict_tuples {
($(
$Tuple:tt {
$(($idx:tt) -> $T:ident, $ST:ident, $TT:ident,)*
}
)+) => {
$(
impl<_DB, _T, _SP, $($T),*> QueryFragment<_DB, _SP> for ConflictTarget<(_T, $($T),*)> where
_DB: Backend<OnConflictClause = _SP>,
_SP: sql_dialect::on_conflict_clause::PgLikeOnConflictClause,
_T: Column,
$($T: Column<Table=_T::Table>,)*
{
fn walk_ast<'b>(&'b self, mut out: AstPass<'_, 'b, _DB>) -> QueryResult<()>
{
out.push_sql(" (");
out.push_identifier(_T::NAME)?;
$(
out.push_sql(", ");
out.push_identifier($T::NAME)?;
)*
out.push_sql(")");
Ok(())
}
}
impl<_T, $($T),*> OnConflictTarget<_T::Table> for ConflictTarget<(_T, $($T),*)> where
_T: Column,
$($T: Column<Table=_T::Table>,)*
{
}
)*
}
}
crate::for_each_tuple!(on_conflict_tuples);