derive_sql/generics/
paginate.rs

1use super::*;
2
3pub struct Paginate<T, I>
4where T: traits::AsStatement<I>,
5{
6  inner: T,
7  limit: Option<usize>,
8  offset: Option<usize>,
9  phantom_i: std::marker::PhantomData<I>,
10}
11
12impl<T, I> Paginate<T, I>
13where T: traits::AsStatement<I>,
14{
15  pub fn with_limit(mut self, limit: usize) -> Self { self.limit = Some(limit); self }
16  pub fn with_offset(mut self, offset: usize) -> Self { self.offset = Some(offset); self }
17}
18
19impl<T, I> std::convert::From<T> for Paginate<T, I>
20where T: traits::AsStatement<I>,
21{
22  fn from(v: T) -> Self {
23    Paginate {
24      inner: v,
25      limit: None,
26      offset: None,
27      phantom_i: std::marker::PhantomData::default(),
28    }
29  }
30}
31
32impl<I, T> traits::AsStatement<I> for Paginate<T, I>
33where T: traits::AsStatement<I>,
34{
35  fn as_statement(&self) -> DeriveSqlResult<String> {
36    Ok(format!("{statement} {limit} {offset}",
37      statement = self.inner.as_statement()?,
38      limit = self.limit.as_ref().map(|v| format!("LIMIT {v}")).unwrap_or_default(),
39      offset = self.offset.as_ref().map(|v| format!("OFFSET {v}")).unwrap_or_default(),
40    ))
41  }
42}
43
44impl<I, T> traits::IsSelect for Paginate<T, I>
45where T: traits::AsStatement<I> + traits::IsSelect,
46{}
47