|
|
|
@ -9,6 +9,7 @@ import (
|
|
|
|
|
"fmt"
|
|
|
|
|
"github.com/go-redis/redis/v7"
|
|
|
|
|
"reflect"
|
|
|
|
|
"regexp"
|
|
|
|
|
"strconv"
|
|
|
|
|
"strings"
|
|
|
|
|
"time"
|
|
|
|
@ -143,8 +144,10 @@ func batchReadRedis(ids []string, prefix string) ([]map[string]interface{}, []st
|
|
|
|
|
*/
|
|
|
|
|
func count(baseSql string, pkField string, args []interface{}) (int32, error) {
|
|
|
|
|
//替换掉分页代码,获取数据量总数
|
|
|
|
|
countSql := strings.Replace(baseSql, "SELECT "+pkField+" ", "SELECT count(*) as count ", -1)
|
|
|
|
|
fmt.Println("countSql==", countSql)
|
|
|
|
|
reg := regexp.MustCompile(`SELECT ` + pkField + ` `)
|
|
|
|
|
//countSql := strings.Replace(baseSql, "SELECT "+pkField+" ", "SELECT count(*) as count ", -1)
|
|
|
|
|
countSql := reg.ReplaceAllString(baseSql, "SELECT count(*) as count ")
|
|
|
|
|
fmt.Println("countSql==", countSql)
|
|
|
|
|
var count int32
|
|
|
|
|
//去掉最后的limit+offset
|
|
|
|
|
args = args[0 : len(args)-2]
|
|
|
|
|