feat: limit query

Author: l1xnan

src-tauri/src/cmd.rs

@@ -29,6 +29,17 @@ pub struct DialectPayload {
   pub cwd: Option<String>,
 }
 
+#[allow(clippy::unused_async)]
+pub fn get_ast_dialect(dialect: &str) -> Box<dyn sqlparser::dialect::Dialect> {
+  match dialect {
+    "folder" | "file" | "duckdb" => Box::new(sqlparser::dialect::DuckDbDialect {}),
+    "clickhouse" => Box::new(sqlparser::dialect::ClickHouseDialect {}),
+    "mysql" => Box::new(sqlparser::dialect::MySqlDialect {}),
+    "postgres" => Box::new(sqlparser::dialect::PostgreSqlDialect {}),
+    _ => Box::new(sqlparser::dialect::GenericDialect {}),
+  }
+}
+
 #[allow(clippy::unused_async)]
 pub async fn get_dialect(
   DialectPayload {

src-tauri/src/dialect/mod.rs

@@ -198,12 +198,13 @@ pub async fn _paging_query(
   if let Some(ref _stmt) = stmt {
     sql = ast::limit_stmt(dialect, _stmt, limit, offset).unwrap_or(sql);
   }
-  let mut res = d.query_all(&sql).await?;
+  let mut res = d.query(&sql, 0, 0).await?;
 
   // get total row count
   if let Some(ref _stmt) = stmt {
     if let Some(count_sql) = ast::count_stmt(dialect, _stmt) {
-      if let Ok(count) = d._sql_row_count(&count_sql).await {
+      log::info!("count_sql: {count_sql}");
+      if let Ok(count) = d.query_count(&count_sql).await {
         res.total = count;
       };
     }

src-tauri/src/dialect/sqlite.rs

@@ -2,7 +2,6 @@ use std::convert::From;
 use std::sync::Arc;
 
 use arrow::array::*;
-use arrow::datatypes::*;
 use arrow::datatypes::{DataType, Field, Schema};
 use async_trait::async_trait;
 use rusqlite::types::Value;