diesel_dynamic_schema/
table.rs

Help
1use diesel::backend::Backend;
2use diesel::expression::expression_types;
3use diesel::internal::table_macro::{FromClause, SelectStatement};
4use diesel::prelude::*;
5use diesel::query_builder::*;
6use std::borrow::Borrow;
7
8use crate::column::Column;
9use crate::dummy_expression::*;
10
11#[derive(Debug, Clone, Copy, Hash, PartialEq, Eq)]
12/// A database table.
13/// This type is created by the [`table`](crate::table()) function.
14pub struct Table<T, U = T> {
15    name: T,
16    schema: Option<U>,
17}
18
19impl<T, U> Table<T, U> {
20    pub(crate) fn new(name: T) -> Self {
21        Self { name, schema: None }
22    }
23
24    pub(crate) fn with_schema(schema: U, name: T) -> Self {
25        Self {
26            name,
27            schema: Some(schema),
28        }
29    }
30
31    /// Create a column with this table.
32    pub fn column<ST, V>(&self, name: V) -> Column<Self, V, ST>
33    where
34        Self: Clone,
35    {
36        Column::new(self.clone(), name)
37    }
38
39    /// Gets the name of the table, as specified on creation.
40    pub fn name(&self) -> &T {
41        &self.name
42    }
43}
44
45impl<T, U> QuerySource for Table<T, U>
46where
47    Self: Clone,
48{
49    type FromClause = Self;
50    type DefaultSelection = DummyExpression;
51
52    fn from_clause(&self) -> Self {
53        self.clone()
54    }
55
56    fn default_selection(&self) -> Self::DefaultSelection {
57        DummyExpression::new()
58    }
59}
60
61impl<T, U> AsQuery for Table<T, U>
62where
63    T: Clone,
64    U: Clone,
65    SelectStatement<FromClause<Self>>: Query<SqlType = expression_types::NotSelectable>,
66{
67    type SqlType = expression_types::NotSelectable;
68    type Query = SelectStatement<FromClause<Self>>;
69
70    fn as_query(self) -> Self::Query {
71        SelectStatement::simple(self)
72    }
73}
74
75impl<T, U> diesel::Table for Table<T, U>
76where
77    Self: QuerySource + AsQuery,
78{
79    type PrimaryKey = DummyExpression;
80    type AllColumns = DummyExpression;
81
82    fn primary_key(&self) -> Self::PrimaryKey {
83        DummyExpression::new()
84    }
85
86    fn all_columns() -> Self::AllColumns {
87        DummyExpression::new()
88    }
89}
90
91impl<T, U, DB> QueryFragment<DB> for Table<T, U>
92where
93    DB: Backend,
94    T: Borrow<str>,
95    U: Borrow<str>,
96{
97    fn walk_ast<'b>(&'b self, mut out: AstPass<'_, 'b, DB>) -> QueryResult<()> {
98        out.unsafe_to_cache_prepared();
99
100        if let Some(ref schema) = self.schema {
101            out.push_identifier(schema.borrow())?;
102            out.push_sql(".");
103        }
104
105        out.push_identifier(self.name.borrow())?;
106        Ok(())
107    }
108}
109
110impl<T, U> QueryId for Table<T, U> {
111    type QueryId = ();
112    const HAS_STATIC_QUERY_ID: bool = false;
113}
diesel_dynamic_schema/table.rs

diesel_dynamic_schema/
table.rs
