diesel_derives/
diesel_numeric_ops.rs

Help
1use proc_macro2::TokenStream;
2use quote::quote;
3use syn::parse_quote;
4use syn::DeriveInput;
5
6use crate::util::wrap_in_dummy_mod;
7
8pub fn derive(mut item: DeriveInput) -> TokenStream {
9    let struct_name = &item.ident;
10
11    {
12        let where_clause = item
13            .generics
14            .where_clause
15            .get_or_insert(parse_quote!(where));
16        where_clause.predicates.push(parse_quote!(Self: Expression));
17        where_clause.predicates.push_punct(Default::default());
18    }
19
20    let (_, ty_generics, where_clause) = item.generics.split_for_impl();
21    let mut impl_generics = item.generics.clone();
22    impl_generics.params.push(parse_quote!(__Rhs));
23    let (impl_generics, _, _) = impl_generics.split_for_impl();
24
25    wrap_in_dummy_mod(quote! {
26        use diesel::internal::derives::numeric_ops as ops;
27        use diesel::expression::{Expression, AsExpression};
28        use diesel::sql_types::ops::{Add, Sub, Mul, Div};
29        use diesel::sql_types::{SqlType, SingleValue};
30
31        impl #impl_generics ::std::ops::Add<__Rhs> for #struct_name #ty_generics
32        #where_clause
33            Self: Expression,
34            <Self as Expression>::SqlType: Add,
35            <<Self as Expression>::SqlType as Add>::Rhs: SqlType + SingleValue,
36            __Rhs: AsExpression<<<Self as Expression>::SqlType as Add>::Rhs>,
37        {
38            type Output = ops::Add<Self, __Rhs::Expression>;
39
40            fn add(self, rhs: __Rhs) -> Self::Output {
41                ops::Add::new(self, rhs.as_expression())
42            }
43        }
44
45        impl #impl_generics ::std::ops::Sub<__Rhs> for #struct_name #ty_generics
46        #where_clause
47            Self: Expression,
48            <Self as Expression>::SqlType: Sub,
49            <<Self as Expression>::SqlType as Sub>::Rhs: SqlType + SingleValue,
50            __Rhs: AsExpression<<<Self as Expression>::SqlType as Sub>::Rhs>,
51        {
52            type Output = ops::Sub<Self, __Rhs::Expression>;
53
54            fn sub(self, rhs: __Rhs) -> Self::Output {
55                ops::Sub::new(self, rhs.as_expression())
56            }
57        }
58
59        impl #impl_generics ::std::ops::Mul<__Rhs> for #struct_name #ty_generics
60        #where_clause
61            Self: Expression,
62            <Self as Expression>::SqlType: Mul,
63            <<Self as Expression>::SqlType as Mul>::Rhs: SqlType + SingleValue,
64            __Rhs: AsExpression<<<Self as Expression>::SqlType as Mul>::Rhs>,
65        {
66            type Output = ops::Mul<Self, __Rhs::Expression>;
67
68            fn mul(self, rhs: __Rhs) -> Self::Output {
69                ops::Mul::new(self, rhs.as_expression())
70            }
71        }
72
73        impl #impl_generics ::std::ops::Div<__Rhs> for #struct_name #ty_generics
74        #where_clause
75            Self: Expression,
76            <Self as Expression>::SqlType: Div,
77            <<Self as Expression>::SqlType as Div>::Rhs: SqlType + SingleValue,
78            __Rhs: AsExpression<<<Self as Expression>::SqlType as Div>::Rhs>,
79        {
80            type Output = ops::Div<Self, __Rhs::Expression>;
81
82            fn div(self, rhs: __Rhs) -> Self::Output {
83                ops::Div::new(self, rhs.as_expression())
84            }
85        }
86    })
87}
diesel_derives/diesel_numeric_ops.rs

diesel_derives/
diesel_numeric_ops.rs
