diff --git a/src/database/mysql.rs b/src/database/mysql.rs index a5e3971..160306f 100644 --- a/src/database/mysql.rs +++ b/src/database/mysql.rs @@ -149,7 +149,7 @@ impl Pool for MySqlPool { async fn execute(&self, query: &String) -> anyhow::Result { let query = query.trim(); - if query.starts_with("SELECT") || query.starts_with("select") { + if query.to_uppercase().starts_with("SELECT") { let mut rows = sqlx::query(query).fetch(&self.pool); let mut headers = vec![]; let mut records = vec![]; diff --git a/src/database/postgres.rs b/src/database/postgres.rs index b4e4535..a2a27bd 100644 --- a/src/database/postgres.rs +++ b/src/database/postgres.rs @@ -149,7 +149,7 @@ impl TableRow for Index { impl Pool for PostgresPool { async fn execute(&self, query: &String) -> anyhow::Result { let query = query.trim(); - if query.starts_with("SELECT") || query.starts_with("select") { + if query.to_uppercase().starts_with("SELECT") { let mut rows = sqlx::query(query).fetch(&self.pool); let mut headers = vec![]; let mut records = vec![]; diff --git a/src/database/sqlite.rs b/src/database/sqlite.rs index 295358f..83063c7 100644 --- a/src/database/sqlite.rs +++ b/src/database/sqlite.rs @@ -152,7 +152,7 @@ impl TableRow for Index { impl Pool for SqlitePool { async fn execute(&self, query: &String) -> anyhow::Result { let query = query.trim(); - if query.starts_with("SELECT") || query.starts_with("select") { + if query.to_uppercase().starts_with("SELECT") { let mut rows = sqlx::query(query).fetch(&self.pool); let mut headers = vec![]; let mut records = vec![];