feat: conditional.go query, production friendly
This commit is contained in:
Aitimate
parent
aeb298635b
commit
4eb10914ab
319
conditional/conditional.go
Normal file
319
conditional/conditional.go
Normal file
@ -0,0 +1,319 @@
|
||||
package conditional
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"gorm.io/gorm"
|
||||
"reflect"
|
||||
"regexp"
|
||||
"strings"
|
||||
"unicode"
|
||||
"unsafe"
|
||||
)
|
||||
|
||||
const GeneralSumKey = "#sum"
|
||||
|
||||
var isInjectionReg = regexp.MustCompile("[^a-zA-Z_]+")
|
||||
|
||||
type BaseCondition struct {
|
||||
Page *int `json:"page" form:"page"` // zh: 页码
|
||||
Pagesize *int `json:"pagesize" form:"pagesize"` // zh: 每页大小
|
||||
OrderKey *string `json:"orderKey" form:"orderKey"` // zh: 排序字段名
|
||||
Desc *bool `json:"desc" form:"desc"` // zh: 是否降序
|
||||
}
|
||||
|
||||
type GeneralResult struct {
|
||||
Total int64 `json:"total"`
|
||||
List []map[string]interface{} `json:"list"`
|
||||
Sum map[string]interface{} `json:"sum"`
|
||||
}
|
||||
|
||||
type wrapDB struct {
|
||||
Db *gorm.DB
|
||||
maxPagesize int
|
||||
includeEmptyString int
|
||||
Page *int `json:"page" form:"page"` // zh: 页码
|
||||
Pagesize *int `json:"pagesize" form:"pagesize"` // zh: 每页大小
|
||||
OrderKey *string `json:"orderKey" form:"orderKey"` // zh: 排序字段名
|
||||
Desc *bool `json:"desc" form:"desc"` // zh: 是否降序
|
||||
}
|
||||
|
||||
func camelCaseToUnderscore(s string) string {
|
||||
var output []rune
|
||||
output = append(output, unicode.ToLower(rune(s[0])))
|
||||
for i := 1; i < len(s); i++ {
|
||||
if unicode.IsUpper(rune(s[i])) {
|
||||
output = append(output, '_')
|
||||
}
|
||||
output = append(output, unicode.ToLower(rune(s[i])))
|
||||
}
|
||||
return string(output)
|
||||
}
|
||||
|
||||
func underscoreToUpperCamelCase(s string) string {
|
||||
var output []rune
|
||||
for i, f := 0, false; i < len(s); i++ {
|
||||
if s[i] == '_' {
|
||||
f = true
|
||||
continue
|
||||
}
|
||||
if f {
|
||||
f = false
|
||||
output = append(output, unicode.ToUpper(rune(s[i])))
|
||||
} else {
|
||||
output = append(output, rune(s[i]))
|
||||
}
|
||||
}
|
||||
return string(output)
|
||||
}
|
||||
|
||||
func (w *wrapDB) doOrder() *wrapDB {
|
||||
if w.OrderKey != nil {
|
||||
if orderKey := *w.OrderKey; orderKey != "" && !isInjectionReg.MatchString(orderKey) {
|
||||
if w.Desc != nil && *w.Desc {
|
||||
orderKey += " desc"
|
||||
}
|
||||
w.Db.Order(orderKey)
|
||||
}
|
||||
}
|
||||
return w
|
||||
}
|
||||
|
||||
func (w *wrapDB) doPage() *wrapDB {
|
||||
if w.Pagesize == nil || *w.Pagesize > w.maxPagesize {
|
||||
w.Pagesize = &w.maxPagesize
|
||||
}
|
||||
if *w.Pagesize > 0 {
|
||||
w.Db.Limit(*w.Pagesize)
|
||||
}
|
||||
if w.Page != nil {
|
||||
w.Db.Offset(*w.Pagesize * (*w.Page - 1))
|
||||
}
|
||||
return w
|
||||
}
|
||||
|
||||
func (w *wrapDB) doWhere(key string, val interface{}) *wrapDB {
|
||||
if len(key) == 0 || strings.HasPrefix(key, "#") {
|
||||
return w
|
||||
}
|
||||
if w.includeEmptyString < 1 {
|
||||
if ref := reflect.ValueOf(val); ref.Kind() == reflect.String && ref.String() == "" {
|
||||
return w
|
||||
}
|
||||
}
|
||||
db := w.Db
|
||||
if key = camelCaseToUnderscore(key); len(key) > 3 {
|
||||
pre := key[:3]
|
||||
switch pre {
|
||||
case "neq":
|
||||
if key[3] == '_' {
|
||||
db.Where(fmt.Sprintf("`%s` <> ?", key[4:]), val)
|
||||
}
|
||||
case "gt_":
|
||||
db.Where(fmt.Sprintf("`%s` >= ?", key[3:]), val)
|
||||
case "lt_":
|
||||
db.Where(fmt.Sprintf("`%s` <= ?", key[3:]), val)
|
||||
case "in_":
|
||||
db.Where(fmt.Sprintf("`%s` in ?", key[3:]), val)
|
||||
case "nin":
|
||||
if key[3] == '_' {
|
||||
db.Where(fmt.Sprintf("`%s` not in ?", key[4:]), val)
|
||||
}
|
||||
case "lik":
|
||||
if strings.HasPrefix(key, "like_") {
|
||||
db.Where(fmt.Sprintf("`%s` like ?", key[5:]), val)
|
||||
}
|
||||
case "nli":
|
||||
if strings.HasPrefix(key, "nlike_") {
|
||||
db.Where(fmt.Sprintf("`%s` not like ?", key[6:]), val)
|
||||
}
|
||||
case "pag":
|
||||
if key == "page" {
|
||||
var page int
|
||||
ref := reflect.ValueOf(val)
|
||||
if ref.CanFloat() {
|
||||
page = int(ref.Float())
|
||||
} else if ref.CanInt() {
|
||||
page = int(ref.Int())
|
||||
} else {
|
||||
page = int(ref.Uint())
|
||||
}
|
||||
w.Page = &page
|
||||
} else if key == "pagesize" || key == "page_size" {
|
||||
var pagesize int
|
||||
ref := reflect.ValueOf(val)
|
||||
if ref.CanFloat() {
|
||||
pagesize = int(ref.Float())
|
||||
} else if ref.CanInt() {
|
||||
pagesize = int(ref.Int())
|
||||
} else {
|
||||
pagesize = int(ref.Uint())
|
||||
}
|
||||
w.Pagesize = &pagesize
|
||||
}
|
||||
case "ord":
|
||||
if key == "order_key" {
|
||||
v := camelCaseToUnderscore(val.(string))
|
||||
if strings.HasPrefix(v, "desc_") {
|
||||
n, b := v[5:], true
|
||||
w.OrderKey, w.Desc = &n, &b
|
||||
} else {
|
||||
var n string
|
||||
if strings.HasPrefix(v, "asc_") {
|
||||
n = v[4:]
|
||||
} else {
|
||||
n = v
|
||||
}
|
||||
w.OrderKey = &n
|
||||
}
|
||||
}
|
||||
case "eq_":
|
||||
db.Where(fmt.Sprintf("`%s` = ?", key[3:]), val)
|
||||
default:
|
||||
db.Where(fmt.Sprintf("`%s` = ?", key), val)
|
||||
}
|
||||
} else {
|
||||
db.Where(fmt.Sprintf("`%s` = ?", key), val)
|
||||
}
|
||||
return w
|
||||
}
|
||||
|
||||
func (w *wrapDB) doDeepWhere(k string, v reflect.Value) *wrapDB {
|
||||
kind := v.Kind()
|
||||
switch kind {
|
||||
case reflect.Pointer, reflect.UnsafePointer:
|
||||
if !v.IsNil() {
|
||||
w.doDeepWhere(k, v.Elem())
|
||||
}
|
||||
case reflect.Struct:
|
||||
t, n := v.Type(), v.NumField()
|
||||
for i := 0; i < n; i++ {
|
||||
ki, vi := t.Field(i).Name, v.Field(i)
|
||||
kind = vi.Kind()
|
||||
switch kind {
|
||||
case reflect.Pointer, reflect.UnsafePointer:
|
||||
w.doDeepWhere(ki, vi.Elem())
|
||||
case reflect.Struct, reflect.Map:
|
||||
w.doDeepWhere("", vi)
|
||||
default:
|
||||
w.doWhere(ki, vi.Interface())
|
||||
}
|
||||
}
|
||||
case reflect.Map:
|
||||
if keys := v.MapKeys(); len(keys) > 0 && keys[0].Kind() == reflect.String {
|
||||
for _, key := range keys {
|
||||
w.doWhere(key.String(), v.MapIndex(key).Interface())
|
||||
}
|
||||
}
|
||||
default:
|
||||
w.doWhere(k, v.Interface())
|
||||
}
|
||||
return w
|
||||
}
|
||||
|
||||
func QueryGeneralConditional(db *gorm.DB, search map[string]interface{}, maxPagesize, includeEmptyString int) (gr GeneralResult, err error) {
|
||||
gr = GeneralResult{List: make([]map[string]interface{}, 0), Sum: make(map[string]interface{}), Total: 0}
|
||||
wdb := &wrapDB{Db: db, maxPagesize: maxPagesize, includeEmptyString: includeEmptyString}
|
||||
for key, val := range search {
|
||||
wdb.doWhere(key, val)
|
||||
}
|
||||
if err = wdb.Db.Count(&gr.Total).Error; err != nil {
|
||||
return
|
||||
}
|
||||
if gr.Total > 0 {
|
||||
wdb.doOrder().doPage()
|
||||
if err = wdb.Db.Scan(&gr.List).Error; err != nil {
|
||||
return
|
||||
}
|
||||
// [Optional] Underscore field name to UpperCamelCase
|
||||
if len(gr.List) > 0 {
|
||||
list := make([]map[string]interface{}, 0)
|
||||
for _, m := range gr.List {
|
||||
nm := make(map[string]interface{})
|
||||
for k, v := range m {
|
||||
nm[underscoreToUpperCamelCase(k)] = v
|
||||
}
|
||||
list = append(list, nm)
|
||||
}
|
||||
gr.List = list
|
||||
}
|
||||
// page-1 can do sum only
|
||||
if wdb.Page != nil && *wdb.Page == 1 {
|
||||
var sumFields []string
|
||||
if sumKeys := search[GeneralSumKey]; sumKeys != nil {
|
||||
for _, sumKey := range sumKeys.([]string) {
|
||||
if isInjectionReg.MatchString(sumKey) {
|
||||
err = fmt.Errorf("Ilegal sumKey: %s ", sumKey)
|
||||
return
|
||||
}
|
||||
sumFields = append(sumFields, camelCaseToUnderscore(sumKey))
|
||||
}
|
||||
}
|
||||
if len(sumFields) > 0 {
|
||||
var sb strings.Builder
|
||||
for _, field := range sumFields {
|
||||
sb.Write([]byte(fmt.Sprintf("sum(`%s`) as `%s`, ", field, field)))
|
||||
}
|
||||
if sb.Len() > 16 {
|
||||
if err = db.Select(sb.String()[:sb.Len()-2]).Scan(gr.Sum).Error; err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func QueryStructConditional(db *gorm.DB, search interface{}, list interface{}, sum interface{}, total *int64, maxPagesize, includeEmptyString int) (err error) {
|
||||
wdb := &wrapDB{Db: db, maxPagesize: maxPagesize, includeEmptyString: includeEmptyString}
|
||||
if search != nil {
|
||||
wdb.doDeepWhere("", reflect.ValueOf(search))
|
||||
}
|
||||
if total != nil {
|
||||
if err = wdb.Db.Count(total).Error; err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
if *total > 0 {
|
||||
wdb.doOrder().doPage()
|
||||
if list != nil {
|
||||
if err = wdb.Db.Scan(list).Error; err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
// page-1 can do sum only
|
||||
if wdb.Page != nil && *wdb.Page == 1 && sum != nil {
|
||||
sv := reflect.ValueOf(sum)
|
||||
sk := sv.Kind()
|
||||
switch sk {
|
||||
case reflect.Pointer, reflect.UnsafePointer:
|
||||
sv = reflect.ValueOf(sv.Elem().Interface())
|
||||
sk = sv.Kind()
|
||||
if sk == reflect.Struct {
|
||||
if t, n := sv.Type(), sv.NumField(); n > 0 {
|
||||
var sb strings.Builder
|
||||
for i := 0; i < n; i++ {
|
||||
ns := camelCaseToUnderscore(t.Field(i).Name)
|
||||
sb.Write([]byte(fmt.Sprintf("sum(`%s`) as `%s`, ", ns, ns)))
|
||||
}
|
||||
if sb.Len() > 0 {
|
||||
db.Select(sb.String()[:sb.Len()-2]).Scan(sum)
|
||||
}
|
||||
}
|
||||
}
|
||||
default:
|
||||
return fmt.Errorf("[SUM] Unknow: %s , only pointer allowed", sum)
|
||||
}
|
||||
}
|
||||
} else {
|
||||
if list != nil {
|
||||
// set empty array
|
||||
s := (*reflect.SliceHeader)(reflect.ValueOf(list).UnsafePointer())
|
||||
if s.Data == 0 {
|
||||
e := make([]interface{}, 0)
|
||||
s.Data = (uintptr)(unsafe.Pointer(&e))
|
||||
}
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
200
conditional/conditional_gen_test.go
Normal file
200
conditional/conditional_gen_test.go
Normal file
@ -0,0 +1,200 @@
|
||||
package conditional
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"gorm.io/driver/mysql"
|
||||
"gorm.io/gorm"
|
||||
"gorm.io/gorm/logger"
|
||||
"log"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func initDB() *gorm.DB {
|
||||
db, err := gorm.Open(mysql.Open("root:root@tcp(127.0.0.1:3306)/gorm_test?charset=utf8mb4&parseTime=True&loc=Local"), &gorm.Config{Logger: logger.Default.LogMode(logger.Info)})
|
||||
if err != nil {
|
||||
panic("failed to connect database")
|
||||
}
|
||||
// `id` INT unsigned NOT NULL AUTO_INCREMENT COMMENT 'user ID',
|
||||
// `name` VARCHAR(64) NOT NULL COMMENT '钱包地址',
|
||||
// `level` INT unsigned NOT NULL COMMENT '用户等级',
|
||||
// `status` int unsigned NOT NULL DEFAULT '0' COMMENT '结算状态 0: 正常 20禁用',
|
||||
// `created_at` bigint NOT NULL COMMENT '创建时间 毫秒',
|
||||
// `updated_at` bigint NOT NULL COMMENT '更新时间 毫秒',
|
||||
db = db.Table("user")
|
||||
return db
|
||||
}
|
||||
|
||||
func TestQueryGeneralConditionalNeq(t *testing.T) {
|
||||
search := make(map[string]interface{})
|
||||
search["neqId"] = uint(1)
|
||||
search["neq_id"] = uint(1)
|
||||
gr, err := QueryGeneralConditional(initDB(), search, 10, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
marshal, _ := json.Marshal(gr)
|
||||
log.Println(string(marshal))
|
||||
}
|
||||
|
||||
func TestQueryGeneralConditionalEq(t *testing.T) {
|
||||
search := make(map[string]interface{})
|
||||
search["id"] = uint(1)
|
||||
search["eq_id"] = uint(1)
|
||||
search["eqId"] = uint(1)
|
||||
gr, err := QueryGeneralConditional(initDB(), search, 10, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
marshal, _ := json.Marshal(gr)
|
||||
log.Println(string(marshal))
|
||||
}
|
||||
|
||||
func TestQueryGeneralConditionalLt(t *testing.T) {
|
||||
search := make(map[string]interface{})
|
||||
search["ltId"] = uint(2)
|
||||
gr, err := QueryGeneralConditional(initDB(), search, 10, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
marshal, _ := json.Marshal(gr)
|
||||
log.Println(string(marshal))
|
||||
}
|
||||
|
||||
func TestQueryGeneralConditionalGt(t *testing.T) {
|
||||
search := make(map[string]interface{})
|
||||
search["gtId"] = uint(2)
|
||||
gr, err := QueryGeneralConditional(initDB(), search, 10, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
marshal, _ := json.Marshal(gr)
|
||||
log.Println(string(marshal))
|
||||
}
|
||||
|
||||
func TestQueryGeneralConditionalIn(t *testing.T) {
|
||||
search := make(map[string]interface{})
|
||||
in := make([]uint, 0)
|
||||
in = append(in, 1)
|
||||
in = append(in, 2)
|
||||
in = append(in, 3)
|
||||
search["inId"] = in
|
||||
gr, err := QueryGeneralConditional(initDB(), search, 10, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
marshal, _ := json.Marshal(gr)
|
||||
log.Println(string(marshal))
|
||||
}
|
||||
|
||||
func TestQueryGeneralConditionalNin(t *testing.T) {
|
||||
search := make(map[string]interface{})
|
||||
in := make([]uint, 0)
|
||||
in = append(in, 1)
|
||||
in = append(in, 2)
|
||||
in = append(in, 3)
|
||||
search["ninId"] = in
|
||||
gr, err := QueryGeneralConditional(initDB(), search, 10, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
marshal, _ := json.Marshal(gr)
|
||||
log.Println(string(marshal))
|
||||
}
|
||||
|
||||
func TestQueryGeneralConditionalLike(t *testing.T) {
|
||||
search := make(map[string]interface{})
|
||||
search["likeName"] = "_oo"
|
||||
gr, err := QueryGeneralConditional(initDB(), search, 10, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
marshal, _ := json.Marshal(gr)
|
||||
log.Println(string(marshal))
|
||||
}
|
||||
|
||||
func TestQueryGeneralConditionalNlike(t *testing.T) {
|
||||
search := make(map[string]interface{})
|
||||
search["likeName"] = "f%"
|
||||
gr, err := QueryGeneralConditional(initDB(), search, 10, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
marshal, _ := json.Marshal(gr)
|
||||
log.Println(string(marshal))
|
||||
}
|
||||
|
||||
func TestQueryGeneralConditionalPage(t *testing.T) {
|
||||
search := make(map[string]interface{})
|
||||
search["page"] = 2
|
||||
search["pagesize"] = 2
|
||||
gr, err := QueryGeneralConditional(initDB(), search, 10, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
marshal, _ := json.Marshal(gr)
|
||||
log.Println(string(marshal))
|
||||
}
|
||||
|
||||
func TestQueryGeneralConditionalOrder(t *testing.T) {
|
||||
search := make(map[string]interface{})
|
||||
//search["orderKey"] = "descId"
|
||||
search["orderKey"] = "Id"
|
||||
search["orderKey"] = "ascId"
|
||||
//search["orderKey"] = "descId or 1"
|
||||
gr, err := QueryGeneralConditional(initDB(), search, 2, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
marshal, _ := json.Marshal(gr)
|
||||
log.Println(string(marshal))
|
||||
}
|
||||
|
||||
func TestQueryGeneralConditionalPage1Sum(t *testing.T) {
|
||||
search := make(map[string]interface{})
|
||||
search["page"] = 1
|
||||
search[GeneralSumKey] = []string{"level"}
|
||||
gr, err := QueryGeneralConditional(initDB(), search, 2, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
marshal, _ := json.Marshal(gr)
|
||||
log.Println(string(marshal))
|
||||
}
|
||||
|
||||
func TestQueryGeneralConditionalNotAllowEmptyString(t *testing.T) {
|
||||
search := make(map[string]interface{})
|
||||
search["likeName"] = ""
|
||||
gr, err := QueryGeneralConditional(initDB(), search, 2, 0)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
marshal, _ := json.Marshal(gr)
|
||||
log.Println(string(marshal))
|
||||
search["likeName"] = "f%"
|
||||
gr, err = QueryGeneralConditional(initDB(), search, 2, 0)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
marshal, _ = json.Marshal(gr)
|
||||
log.Println(string(marshal))
|
||||
}
|
||||
|
||||
func TestQueryGeneralConditionalMaxCount(t *testing.T) {
|
||||
// [unsafe] Unrestricted mode : <=0
|
||||
search := make(map[string]interface{})
|
||||
gr, err := QueryGeneralConditional(initDB(), search, -1, 0)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
marshal, _ := json.Marshal(gr)
|
||||
log.Println(string(marshal))
|
||||
|
||||
// [safe] > 0
|
||||
search = make(map[string]interface{})
|
||||
gr, err = QueryGeneralConditional(initDB(), search, 1, 0)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
marshal, _ = json.Marshal(gr)
|
||||
log.Println(string(marshal))
|
||||
}
|
||||
178
conditional/conditional_struct_test.go
Normal file
178
conditional/conditional_struct_test.go
Normal file
@ -0,0 +1,178 @@
|
||||
package conditional
|
||||
|
||||
import (
|
||||
"log"
|
||||
"testing"
|
||||
)
|
||||
|
||||
type User struct {
|
||||
Id uint `json:"id"`
|
||||
Name string `json:"name"`
|
||||
Level uint `json:"level"`
|
||||
Status uint `json:"status"`
|
||||
CreatedAt uint `json:"createdAt"`
|
||||
UpdatedAt uint `json:"updatedAt"`
|
||||
}
|
||||
|
||||
func TestQueryStructConditionalNeq(t *testing.T) {
|
||||
search := struct {
|
||||
NeqId uint
|
||||
}{NeqId: 1}
|
||||
list, total := new([]User), new(int64)
|
||||
err := QueryStructConditional(initDB(), search, list, nil, total, 10, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
log.Println(list, *total)
|
||||
}
|
||||
|
||||
func TestQueryStructConditionalEq(t *testing.T) {
|
||||
//search := struct {
|
||||
// Id uint
|
||||
//}{Id: 1}
|
||||
search := struct {
|
||||
EqId uint
|
||||
}{EqId: 1}
|
||||
list, total := new([]User), new(int64)
|
||||
err := QueryStructConditional(initDB(), search, list, nil, total, 10, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
log.Println(list, *total)
|
||||
}
|
||||
|
||||
func TestQueryStructConditionalLt(t *testing.T) {
|
||||
search := struct {
|
||||
LtId uint
|
||||
}{LtId: 1}
|
||||
list, total := new([]User), new(int64)
|
||||
err := QueryStructConditional(initDB(), search, list, nil, total, 10, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
log.Println(list, *total)
|
||||
}
|
||||
|
||||
func TestQueryStructConditionalGt(t *testing.T) {
|
||||
search := struct {
|
||||
GtId uint
|
||||
}{GtId: 1}
|
||||
list, total := new([]User), new(int64)
|
||||
err := QueryStructConditional(initDB(), search, list, nil, total, 10, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
log.Println(list, *total)
|
||||
}
|
||||
|
||||
func TestQueryStructConditionalIn(t *testing.T) {
|
||||
search := struct {
|
||||
InId []uint
|
||||
}{InId: []uint{1, 2}}
|
||||
list, total := new([]User), new(int64)
|
||||
err := QueryStructConditional(initDB(), search, list, nil, total, 10, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
log.Println(list, *total)
|
||||
}
|
||||
|
||||
func TestQueryStructConditionalNin(t *testing.T) {
|
||||
search := struct {
|
||||
NinId []uint
|
||||
}{NinId: []uint{1, 2}}
|
||||
list, total := new([]User), new(int64)
|
||||
err := QueryStructConditional(initDB(), search, list, nil, total, 10, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
log.Println(list, *total)
|
||||
}
|
||||
|
||||
func TestQueryStructConditionalLike(t *testing.T) {
|
||||
search := struct {
|
||||
LikeName string
|
||||
}{LikeName: "f%"}
|
||||
list, total := new([]User), new(int64)
|
||||
err := QueryStructConditional(initDB(), search, list, nil, total, 10, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
log.Println(list, *total)
|
||||
}
|
||||
|
||||
func TestQueryStructConditionalNlike(t *testing.T) {
|
||||
search := struct {
|
||||
NlikeName string
|
||||
}{NlikeName: "f%"}
|
||||
list, total := new([]User), new(int64)
|
||||
err := QueryStructConditional(initDB(), search, list, nil, total, 10, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
log.Println(list, *total)
|
||||
}
|
||||
|
||||
func TestQueryStructConditionalPage(t *testing.T) {
|
||||
search := struct {
|
||||
Page int
|
||||
Pagesize int
|
||||
}{Page: 2, Pagesize: 2}
|
||||
list, total := new([]User), new(int64)
|
||||
err := QueryStructConditional(initDB(), search, list, nil, total, 10, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
log.Println(list, *total)
|
||||
}
|
||||
|
||||
func TestQueryStructConditionalOrder(t *testing.T) {
|
||||
search := struct {
|
||||
OrderKey string
|
||||
}{OrderKey: "descId"}
|
||||
list, total := new([]User), new(int64)
|
||||
err := QueryStructConditional(initDB(), search, list, nil, total, 10, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
log.Println(list, *total)
|
||||
}
|
||||
|
||||
func TestQueryStructConditionalPage1Sum(t *testing.T) {
|
||||
search := struct {
|
||||
Page int
|
||||
}{Page: 1}
|
||||
sum := new(struct {
|
||||
Level uint
|
||||
})
|
||||
list, total := new([]User), new(int64)
|
||||
err := QueryStructConditional(initDB(), search, list, sum, total, 10, 1)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
log.Println(sum, list, *total)
|
||||
}
|
||||
|
||||
func TestQueryStructConditionalNotAllowEmptyString(t *testing.T) {
|
||||
search := struct {
|
||||
Name string
|
||||
}{Name: ""}
|
||||
//search := struct {
|
||||
// Name string
|
||||
//}{Name: "foo"}
|
||||
list, total := new([]User), new(int64)
|
||||
err := QueryStructConditional(initDB(), search, list, nil, total, 10, 0)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
log.Println(list, *total)
|
||||
}
|
||||
|
||||
func TestQueryStructConditionalMaxCount(t *testing.T) {
|
||||
list, total := new([]User), new(int64)
|
||||
err := QueryStructConditional(initDB(), nil, list, nil, total, 2, 0)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
log.Println(list, *total)
|
||||
}
|
||||
25
conditional/example.sql
Normal file
25
conditional/example.sql
Normal file
@ -0,0 +1,25 @@
|
||||
CREATE SCHEMA IF NOT EXISTS gorm_test;
|
||||
USE gorm_test;
|
||||
|
||||
DROP TABLE IF EXISTS `user`;
|
||||
CREATE TABLE IF NOT EXISTS `user`
|
||||
(
|
||||
`id` INT unsigned NOT NULL AUTO_INCREMENT COMMENT 'user ID',
|
||||
`name` VARCHAR(64) NOT NULL COMMENT '钱包地址',
|
||||
`level` INT unsigned NOT NULL COMMENT '用户等级',
|
||||
`status` int unsigned NOT NULL DEFAULT '0' COMMENT '结算状态 0: 正常 20禁用',
|
||||
`created_at` bigint NOT NULL COMMENT '创建时间 毫秒',
|
||||
`updated_at` bigint NOT NULL COMMENT '更新时间 毫秒',
|
||||
PRIMARY KEY (`id`),
|
||||
key key_created_at (created_at)
|
||||
) ENGINE = InnoDB
|
||||
DEFAULT CHARSET = utf8mb3 COMMENT ='user';
|
||||
|
||||
INSERT INTO gorm_test.user (name, level, status, created_at, updated_at)
|
||||
VALUES
|
||||
('Boo', 1, 1, 1682597017126, 1682597017126),
|
||||
('Foo', 2, 2, 1682697017126, 1682697017126),
|
||||
('Hoo', 3, 3, 1682797017126, 1682797017126),
|
||||
('Ioo', 4, 4, 1682897017126, 1682897017126),
|
||||
('Joo', 5, 5, 1682997017126, 1682997017126),
|
||||
('Koo', 6, 6, 1683097017126, 1683097017126);
|
||||
Loading…
x
Reference in New Issue
Block a user