fix: 修复表名大小写问题

APIJSON · Jan 1, 2023 · ce62b79 · ce62b79
1 parent c282590
commit ce62b79
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/db/db.go b/db/db.go
@@ -50,6 +50,7 @@ func Init(database string, dataSource string) {
 		for rows.Next() {
 			var name string
 			rows.Scan(&name)
+			name = strings.ToLower(name)
 			AllTable[name] = TableMeta{Name: name, Columns: loadColumnMeta(name)}
 		}
 		logger.Infof("LoadTableMeta END, Table size: %d", len(AllTable))

diff --git a/handler/get.go b/handler/get.go
@@ -151,6 +151,7 @@ func (n *QueryNode) Result() interface{} {
 	}
 	return nil
 }
+
 func (n *QueryNode) doQueryData() {
 	if n.completed {
 		return
@@ -257,7 +258,7 @@ func (c *QueryContext) parseByKey(key string) {
 func NewQueryNode(c *QueryContext, path, key string, queryMap map[string]interface{}) *QueryNode {
 	n := &QueryNode{
 		ctx:         c,
-		Key:         key,
+		Key:         strings.ToLower(key),
 		Path:        path,
 		RequestMap:  queryMap,
 		start:       time.Now().UnixNano(),

diff --git a/handler/handler.go b/handler/handler.go
@@ -4,7 +4,7 @@ import (
 	"bytes"
 	"encoding/json"
 	"github.com/j2go/apijson/logger"
-	"io/ioutil"
+	"io"
 	"net/http"
 )
 
@@ -15,14 +15,14 @@ func commonHandle(w http.ResponseWriter, r *http.Request, bodyHandler func(map[s
 		w.WriteHeader(http.StatusOK)
 		return
 	}
-	if data, err := ioutil.ReadAll(r.Body); err != nil {
+	if requestBody, err := io.ReadAll(r.Body); err != nil {
 		logger.Error("请求参数有问题: " + err.Error())
 		w.WriteHeader(http.StatusBadRequest)
 		return
 	} else {
-		logger.Infof("request: %s", string(data))
+		logger.Infof("request: %s", string(requestBody))
 		var bodyMap map[string]interface{}
-		if err = json.Unmarshal(data, &bodyMap); err != nil {
+		if err = json.Unmarshal(requestBody, &bodyMap); err != nil {
 			logger.Error("请求体 JSON 格式有问题: " + err.Error())
 			w.WriteHeader(http.StatusBadRequest)
 			return