mybatis_sql/
limit.rs

1use crate::template::TEMPLATE;
2use mybatis_core::db::DriverType;
3use mybatis_core::Error;
4use mybatis_core::Result;
5
6pub trait PageLimit {
7    /// return  sql
8    fn page_limit_sql(&self, offset: u64, size: u64) -> Result<String>;
9}
10
11impl PageLimit for DriverType {
12    fn page_limit_sql(&self, offset: u64, size: u64) -> Result<String> {
13        return match self {
14            DriverType::Mysql => Ok(format!(" {} {},{}", TEMPLATE.limit.value, offset, size)),
15            DriverType::Postgres => Ok(format!(
16                " {} {} {} {}",
17                TEMPLATE.limit.value, size, TEMPLATE.offset.value, offset
18            )),
19            DriverType::Sqlite => Ok(format!(
20                " {} {} {} {}",
21                TEMPLATE.limit.value, size, TEMPLATE.offset.value, offset
22            )),
23            DriverType::Mssql => {
24                //sqlserver
25                Ok(format!(
26                    " {} {} {} {} {}",
27                    TEMPLATE.offset.value,
28                    offset,
29                    TEMPLATE.rows_fetch_next.value,
30                    size,
31                    TEMPLATE.rows_only.value
32                ))
33            }
34            DriverType::None => Err(Error::from(format!(
35                "[mybatis] not support now for DriverType:{:?}",
36                DriverType::None
37            ))),
38        };
39    }
40}
41
42#[test]
43pub fn test_create_limit() {
44    let mysql_limit = DriverType::Mysql.page_limit_sql(1, 20).unwrap();
45    println!("{}", mysql_limit);
46    let pg_limit = DriverType::Postgres.page_limit_sql(1, 20).unwrap();
47    println!("{}", pg_limit);
48    let sqlite_limit = DriverType::Sqlite.page_limit_sql(1, 20).unwrap();
49    println!("{}", sqlite_limit);
50}
mybatis_sql/limit.rs

mybatis_sql/
limit.rs
