master
wangshuai 5 years ago
parent 20b487cc37
commit a05c425061

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

Loading…
Cancel
Save