数据库URL查询参数定义totalSQL和rowsSQL

This commit is contained in:
TsMask
2023-09-06 17:17:41 +08:00
parent ccc0df1ae8
commit 4d72bfd832

View File

@@ -109,27 +109,34 @@ func InitDbClient(dbType, dbUser, dbPassword, dbHost, dbPort, dbName string) err
} }
func GetUriSQLArray(r *http.Request) []string { 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 var sa []string
for _, r := range s { vars := r.URL.Query()
if r != "" {
sa = append(sa, r) // 默认SQL
if s, ok := vars["SQL"]; ok {
for _, r := range s {
if r != "" {
sa = append(sa, r)
}
} }
} }
// 查询总数 // 查询总数
if totalSQL, ok := vars["totalSQL"]; ok { 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 { 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) log.Debug("SQL array:", sa)