From f96a5735021c8db5f427d56d59385131041c065c Mon Sep 17 00:00:00 2001
From: mi-wada <49638956+mi-wada@users.noreply.github.com>
Date: Tue, 16 Nov 2021 14:40:48 +0900
Subject: [PATCH] Changed to check case insensitively (#132)

---
 src/database/mysql.rs    | 2 +-
 src/database/postgres.rs | 2 +-
 src/database/sqlite.rs   | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

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<ExecuteResult> {
         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<ExecuteResult> {
         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<ExecuteResult> {
         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![];