From 4d72bfd832a89871894e987ba9838dfb6281ead4 Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Wed, 6 Sep 2023 17:17:41 +0800
Subject: [PATCH] =?UTF-8?q?=E6=95=B0=E6=8D=AE=E5=BA=93URL=E6=9F=A5?=
 =?UTF-8?q?=E8=AF=A2=E5=8F=82=E6=95=B0=E5=AE=9A=E4=B9=89totalSQL=E5=92=8Cr?=
 =?UTF-8?q?owsSQL?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 features/dbrest/dbrest.go | 31 +++++++++++++++++++------------
 1 file changed, 19 insertions(+), 12 deletions(-)

diff --git a/features/dbrest/dbrest.go b/features/dbrest/dbrest.go
index 758474e2..d547817e 100644
--- a/features/dbrest/dbrest.go
+++ b/features/dbrest/dbrest.go
@@ -109,27 +109,34 @@ func InitDbClient(dbType, dbUser, dbPassword, dbHost, dbPort, dbName string) err
 }
 
 func GetUriSQLArray(r *http.Request) []string {
-	vars := r.URL.Query()
-	s, ok := vars["SQL"]
-	if !ok {
-		log.Info("SQL is not exist")
-		return nil
-	}
-
 	var sa []string
-	for _, r := range s {
-		if r != "" {
-			sa = append(sa, r)
+	vars := r.URL.Query()
+
+	// 默认SQL
+	if s, ok := vars["SQL"]; ok {
+		for _, r := range s {
+			if r != "" {
+				sa = append(sa, r)
+			}
 		}
 	}
 
 	// 查询总数
 	if totalSQL, ok := vars["totalSQL"]; ok {
-		sa = append(sa, totalSQL[0])
+		if totalSQL[0] != "" {
+			sa = append(sa, totalSQL[0])
+		}
 	}
 	// 查询列表
 	if rowsSQL, ok := vars["rowsSQL"]; ok {
-		sa = append(sa, rowsSQL[0])
+		if rowsSQL[0] != "" {
+			sa = append(sa, rowsSQL[0])
+		}
+	}
+
+	if len(sa) == 0 {
+		log.Info("SQL is not exist")
+		return nil
 	}
 
 	log.Debug("SQL array:", sa)