This commit is contained in:
tangpanqing
2023-01-06 06:58:03 +08:00
parent 8e1910c6da
commit 0cac8714dc
6 changed files with 241 additions and 214 deletions

View File

@@ -17,6 +17,7 @@ type FieldInfo struct {
}
type WhereItem struct {
FuncName string
Prefix string
Field interface{}
Opt string

View File

@@ -12,6 +12,8 @@ import (
"unsafe"
)
const Count = "COUNT"
const Desc = "DESC"
const Asc = "ASC"
@@ -49,7 +51,7 @@ type Builder struct {
tableName string
selectList []SelectItem
selectExpList []*SelectItem
groupList []string
groupList []interface{}
whereList []WhereItem
joinList []JoinItem
havingList []WhereItem
@@ -320,8 +322,8 @@ func (ex *Builder) GetSqlAndParams() (string, []interface{}) {
fieldStr, paramList := ex.handleField(paramList)
whereStr, paramList := ex.handleWhere(paramList)
joinStr, paramList := ex.handleJoin(paramList)
groupStr := handleGroup(ex.groupList)
havingStr, paramList := ex.handleHaving(ex.havingList, paramList)
groupStr, paramList := ex.handleGroup(paramList)
havingStr, paramList := ex.handleHaving(paramList)
orderStr, paramList := ex.handleOrder(paramList)
limitStr, paramList := ex.handleLimit(ex.offset, ex.pageSize, paramList)
lockStr := handleLockForUpdate(ex.isLockForUpdate)
@@ -566,7 +568,7 @@ func (b *Builder) Table(table interface{}, alias ...string) *Builder {
}
// GroupBy 链式操作,以某字段进行分组
func (ex *Builder) GroupBy(fieldName string) *Builder {
func (ex *Builder) GroupBy(fieldName interface{}) *Builder {
ex.groupList = append(ex.groupList, fieldName)
return ex
}
@@ -606,15 +608,17 @@ func (ex *Builder) LockForUpdate(isLockForUpdate bool) *Builder {
func (ex *Builder) whereAndHaving(where []WhereItem, paramList []any) ([]string, []any) {
var whereList []string
for i := 0; i < len(where); i++ {
prefix := where[i].Prefix
fieldName := getFieldName(where[i].Field)
allFieldName := where[i].Prefix + "." + getFieldName(where[i].Field)
if where[i].FuncName != "" {
allFieldName = where[i].FuncName + "(" + allFieldName + ")"
}
if "**builder.Builder" == reflect.TypeOf(where[i].Val).String() {
executor := *(**Builder)(unsafe.Pointer(reflect.ValueOf(where[i].Val).Pointer()))
subSql, subParams := executor.GetSqlAndParams()
if where[i].Opt != Raw {
whereList = append(whereList, prefix+"."+fieldName+" "+where[i].Opt+" "+"("+subSql+")")
whereList = append(whereList, allFieldName+" "+where[i].Opt+" "+"("+subSql+")")
paramList = append(paramList, subParams...)
} else {
@@ -622,15 +626,15 @@ func (ex *Builder) whereAndHaving(where []WhereItem, paramList []any) ([]string,
} else {
if where[i].Opt == Eq || where[i].Opt == Ne || where[i].Opt == Gt || where[i].Opt == Ge || where[i].Opt == Lt || where[i].Opt == Le {
if ex.driverName == model.Sqlite3 {
whereList = append(whereList, prefix+"."+fieldName+" "+where[i].Opt+" "+"?")
whereList = append(whereList, allFieldName+" "+where[i].Opt+" "+"?")
} else {
switch where[i].Val.(type) {
case float32:
whereList = append(whereList, ex.getConcatForFloat(prefix+"."+fieldName, "''")+" "+where[i].Opt+" "+"?")
whereList = append(whereList, ex.getConcatForFloat(allFieldName, "''")+" "+where[i].Opt+" "+"?")
case float64:
whereList = append(whereList, ex.getConcatForFloat(prefix+"."+fieldName, "''")+" "+where[i].Opt+" "+"?")
whereList = append(whereList, ex.getConcatForFloat(allFieldName, "''")+" "+where[i].Opt+" "+"?")
default:
whereList = append(whereList, prefix+"."+fieldName+" "+where[i].Opt+" "+"?")
whereList = append(whereList, allFieldName+" "+where[i].Opt+" "+"?")
}
}
@@ -639,7 +643,7 @@ func (ex *Builder) whereAndHaving(where []WhereItem, paramList []any) ([]string,
if where[i].Opt == Between || where[i].Opt == NotBetween {
values := toAnyArr(where[i].Val)
whereList = append(whereList, prefix+"."+fieldName+" "+where[i].Opt+" "+"(?) AND (?)")
whereList = append(whereList, allFieldName+" "+where[i].Opt+" "+"(?) AND (?)")
paramList = append(paramList, values...)
}
@@ -657,7 +661,7 @@ func (ex *Builder) whereAndHaving(where []WhereItem, paramList []any) ([]string,
}
}
whereList = append(whereList, prefix+"."+fieldName+" "+where[i].Opt+" "+ex.getConcatForLike(valueStr...))
whereList = append(whereList, allFieldName+" "+where[i].Opt+" "+ex.getConcatForLike(valueStr...))
}
if where[i].Opt == In || where[i].Opt == NotIn {
@@ -667,12 +671,12 @@ func (ex *Builder) whereAndHaving(where []WhereItem, paramList []any) ([]string,
placeholder = append(placeholder, "?")
}
whereList = append(whereList, prefix+"."+fieldName+" "+where[i].Opt+" "+"("+strings.Join(placeholder, ",")+")")
whereList = append(whereList, allFieldName+" "+where[i].Opt+" "+"("+strings.Join(placeholder, ",")+")")
paramList = append(paramList, values...)
}
if where[i].Opt == Raw {
whereList = append(whereList, prefix+"."+fieldName+fmt.Sprintf("%v", where[i].Val))
whereList = append(whereList, allFieldName+fmt.Sprintf("%v", where[i].Val))
}
}
}

View File

@@ -59,9 +59,9 @@ func (ex *Builder) handleWhere(paramList []any) (string, []any) {
return "", paramList
}
whereList, paramList := ex.whereAndHaving(ex.whereList, paramList)
strList, paramList := ex.whereAndHaving(ex.whereList, paramList)
return " WHERE " + strings.Join(whereList, " AND "), paramList
return " WHERE " + strings.Join(strList, " AND "), paramList
}
//拼接SQL,更新信息
@@ -110,23 +110,28 @@ func (b *Builder) handleJoin(paramList []interface{}) (string, []interface{}) {
}
//拼接SQL,结果分组
func handleGroup(groupList []string) string {
if len(groupList) == 0 {
return ""
}
return " GROUP BY " + strings.Join(groupList, ",")
}
//拼接SQL,结果筛选
func (ex *Builder) handleHaving(having []WhereItem, paramList []any) (string, []any) {
if len(having) == 0 {
func (ex *Builder) handleGroup(paramList []any) (string, []any) {
if len(ex.groupList) == 0 {
return "", paramList
}
whereList, paramList := ex.whereAndHaving(having, paramList)
var groupList []string
for i := 0; i < len(ex.groupList); i++ {
groupList = append(groupList, getFieldName(ex.groupList[i]))
}
return " Having " + strings.Join(whereList, " AND "), paramList
return " GROUP BY " + strings.Join(groupList, ","), paramList
}
//拼接SQL,结果筛选
func (ex *Builder) handleHaving(paramList []any) (string, []any) {
if len(ex.havingList) == 0 {
return "", paramList
}
strList, paramList := ex.whereAndHaving(ex.havingList, paramList)
return " Having " + strings.Join(strList, " AND "), paramList
}
//拼接SQL,结果排序

View File

@@ -33,8 +33,10 @@ func (ex *Builder) HavingArr(havingList []WhereItem) *Builder {
return ex
}
func (ex *Builder) HavingEq(field string, val interface{}) *Builder {
func (ex *Builder) HavingEq(funcName string, field interface{}, val interface{}, prefix ...string) *Builder {
ex.havingList = append(ex.havingList, WhereItem{
FuncName: funcName,
Prefix: getPrefixByField(field, prefix...),
Field: field,
Opt: Eq,
Val: val,
@@ -42,8 +44,10 @@ func (ex *Builder) HavingEq(field string, val interface{}) *Builder {
return ex
}
func (ex *Builder) HavingNe(field string, val interface{}) *Builder {
func (ex *Builder) HavingNe(funcName string, field interface{}, val interface{}, prefix ...string) *Builder {
ex.havingList = append(ex.havingList, WhereItem{
FuncName: funcName,
Prefix: getPrefixByField(field, prefix...),
Field: field,
Opt: Ne,
Val: val,
@@ -51,8 +55,10 @@ func (ex *Builder) HavingNe(field string, val interface{}) *Builder {
return ex
}
func (ex *Builder) HavingGt(field string, val interface{}) *Builder {
func (ex *Builder) HavingGt(funcName string, field interface{}, val interface{}, prefix ...string) *Builder {
ex.havingList = append(ex.havingList, WhereItem{
FuncName: funcName,
Prefix: getPrefixByField(field, prefix...),
Field: field,
Opt: Gt,
Val: val,
@@ -60,8 +66,10 @@ func (ex *Builder) HavingGt(field string, val interface{}) *Builder {
return ex
}
func (ex *Builder) HavingGe(field string, val interface{}) *Builder {
func (ex *Builder) HavingGe(funcName string, field interface{}, val interface{}, prefix ...string) *Builder {
ex.havingList = append(ex.havingList, WhereItem{
FuncName: funcName,
Prefix: getPrefixByField(field, prefix...),
Field: field,
Opt: Ge,
Val: val,
@@ -69,8 +77,10 @@ func (ex *Builder) HavingGe(field string, val interface{}) *Builder {
return ex
}
func (ex *Builder) HavingLt(field string, val interface{}) *Builder {
func (ex *Builder) HavingLt(funcName string, field interface{}, val interface{}, prefix ...string) *Builder {
ex.havingList = append(ex.havingList, WhereItem{
FuncName: funcName,
Prefix: getPrefixByField(field, prefix...),
Field: field,
Opt: Lt,
Val: val,
@@ -78,8 +88,10 @@ func (ex *Builder) HavingLt(field string, val interface{}) *Builder {
return ex
}
func (ex *Builder) HavingLe(field string, val interface{}) *Builder {
func (ex *Builder) HavingLe(funcName string, field interface{}, val interface{}, prefix ...string) *Builder {
ex.havingList = append(ex.havingList, WhereItem{
FuncName: funcName,
Prefix: getPrefixByField(field, prefix...),
Field: field,
Opt: Le,
Val: val,
@@ -87,8 +99,10 @@ func (ex *Builder) HavingLe(field string, val interface{}) *Builder {
return ex
}
func (ex *Builder) HavingIn(field string, val interface{}) *Builder {
func (ex *Builder) HavingIn(funcName string, field interface{}, val interface{}, prefix ...string) *Builder {
ex.havingList = append(ex.havingList, WhereItem{
FuncName: funcName,
Prefix: getPrefixByField(field, prefix...),
Field: field,
Opt: In,
Val: val,
@@ -96,8 +110,10 @@ func (ex *Builder) HavingIn(field string, val interface{}) *Builder {
return ex
}
func (ex *Builder) HavingNotIn(field string, val interface{}) *Builder {
func (ex *Builder) HavingNotIn(funcName string, field interface{}, val interface{}, prefix ...string) *Builder {
ex.havingList = append(ex.havingList, WhereItem{
FuncName: funcName,
Prefix: getPrefixByField(field, prefix...),
Field: field,
Opt: NotIn,
Val: val,
@@ -105,8 +121,10 @@ func (ex *Builder) HavingNotIn(field string, val interface{}) *Builder {
return ex
}
func (ex *Builder) HavingBetween(field string, val interface{}) *Builder {
func (ex *Builder) HavingBetween(funcName string, field interface{}, val interface{}, prefix ...string) *Builder {
ex.havingList = append(ex.havingList, WhereItem{
FuncName: funcName,
Prefix: getPrefixByField(field, prefix...),
Field: field,
Opt: Between,
Val: val,
@@ -114,8 +132,10 @@ func (ex *Builder) HavingBetween(field string, val interface{}) *Builder {
return ex
}
func (ex *Builder) HavingNotBetween(field string, val interface{}) *Builder {
func (ex *Builder) HavingNotBetween(funcName string, field interface{}, val interface{}, prefix ...string) *Builder {
ex.havingList = append(ex.havingList, WhereItem{
FuncName: funcName,
Prefix: getPrefixByField(field, prefix...),
Field: field,
Opt: NotBetween,
Val: val,
@@ -123,8 +143,10 @@ func (ex *Builder) HavingNotBetween(field string, val interface{}) *Builder {
return ex
}
func (ex *Builder) HavingLike(field string, val interface{}) *Builder {
func (ex *Builder) HavingLike(funcName string, field interface{}, val interface{}, prefix ...string) *Builder {
ex.havingList = append(ex.havingList, WhereItem{
FuncName: funcName,
Prefix: getPrefixByField(field, prefix...),
Field: field,
Opt: Like,
Val: val,
@@ -132,8 +154,10 @@ func (ex *Builder) HavingLike(field string, val interface{}) *Builder {
return ex
}
func (ex *Builder) HavingNotLike(field string, val interface{}) *Builder {
func (ex *Builder) HavingNotLike(funcName string, field interface{}, val interface{}, prefix ...string) *Builder {
ex.havingList = append(ex.havingList, WhereItem{
FuncName: funcName,
Prefix: getPrefixByField(field, prefix...),
Field: field,
Opt: NotLike,
Val: val,
@@ -141,8 +165,10 @@ func (ex *Builder) HavingNotLike(field string, val interface{}) *Builder {
return ex
}
func (ex *Builder) HavingRaw(field string, val interface{}) *Builder {
func (ex *Builder) HavingRaw(funcName string, field interface{}, val interface{}, prefix ...string) *Builder {
ex.havingList = append(ex.havingList, WhereItem{
FuncName: funcName,
Prefix: getPrefixByField(field, prefix...),
Field: field,
Opt: Raw,
Val: val,

View File

@@ -34,66 +34,66 @@ func (b *Builder) WhereArr(whereList []WhereItem) *Builder {
}
func (b *Builder) WhereEq(field interface{}, val interface{}, prefix ...string) *Builder {
b.whereList = append(b.whereList, WhereItem{getPrefixByField(field, prefix...), field, Eq, val})
b.whereList = append(b.whereList, WhereItem{"", getPrefixByField(field, prefix...), field, Eq, val})
return b
}
func (b *Builder) WhereNe(field interface{}, val interface{}, prefix ...string) *Builder {
b.whereList = append(b.whereList, WhereItem{getPrefixByField(field, prefix...), field, Ne, val})
b.whereList = append(b.whereList, WhereItem{"", getPrefixByField(field, prefix...), field, Ne, val})
return b
}
func (b *Builder) WhereGt(field interface{}, val interface{}, prefix ...string) *Builder {
b.whereList = append(b.whereList, WhereItem{getPrefixByField(field, prefix...), field, Gt, val})
b.whereList = append(b.whereList, WhereItem{"", getPrefixByField(field, prefix...), field, Gt, val})
return b
}
func (b *Builder) WhereGe(field interface{}, val interface{}, prefix ...string) *Builder {
b.whereList = append(b.whereList, WhereItem{getPrefixByField(field, prefix...), field, Ge, val})
b.whereList = append(b.whereList, WhereItem{"", getPrefixByField(field, prefix...), field, Ge, val})
return b
}
func (b *Builder) WhereLt(field interface{}, val interface{}, prefix ...string) *Builder {
b.whereList = append(b.whereList, WhereItem{getPrefixByField(field, prefix...), field, Lt, val})
b.whereList = append(b.whereList, WhereItem{"", getPrefixByField(field, prefix...), field, Lt, val})
return b
}
func (b *Builder) WhereLe(field interface{}, val interface{}, prefix ...string) *Builder {
b.whereList = append(b.whereList, WhereItem{getPrefixByField(field, prefix...), field, Le, val})
b.whereList = append(b.whereList, WhereItem{"", getPrefixByField(field, prefix...), field, Le, val})
return b
}
func (b *Builder) WhereIn(field interface{}, val interface{}, prefix ...string) *Builder {
b.whereList = append(b.whereList, WhereItem{getPrefixByField(field, prefix...), field, In, val})
b.whereList = append(b.whereList, WhereItem{"", getPrefixByField(field, prefix...), field, In, val})
return b
}
func (b *Builder) WhereNotIn(field interface{}, val interface{}, prefix ...string) *Builder {
b.whereList = append(b.whereList, WhereItem{getPrefixByField(field, prefix...), field, NotIn, val})
b.whereList = append(b.whereList, WhereItem{"", getPrefixByField(field, prefix...), field, NotIn, val})
return b
}
func (b *Builder) WhereBetween(field interface{}, val interface{}, prefix ...string) *Builder {
b.whereList = append(b.whereList, WhereItem{getPrefixByField(field, prefix...), field, Between, val})
b.whereList = append(b.whereList, WhereItem{"", getPrefixByField(field, prefix...), field, Between, val})
return b
}
func (b *Builder) WhereNotBetween(field interface{}, val interface{}, prefix ...string) *Builder {
b.whereList = append(b.whereList, WhereItem{getPrefixByField(field, prefix...), field, NotBetween, val})
b.whereList = append(b.whereList, WhereItem{"", getPrefixByField(field, prefix...), field, NotBetween, val})
return b
}
func (b *Builder) WhereLike(field interface{}, val interface{}, prefix ...string) *Builder {
b.whereList = append(b.whereList, WhereItem{getPrefixByField(field, prefix...), field, Like, val})
b.whereList = append(b.whereList, WhereItem{"", getPrefixByField(field, prefix...), field, Like, val})
return b
}
func (b *Builder) WhereNotLike(field interface{}, val interface{}, prefix ...string) *Builder {
b.whereList = append(b.whereList, WhereItem{getPrefixByField(field, prefix...), field, NotLike, val})
b.whereList = append(b.whereList, WhereItem{"", getPrefixByField(field, prefix...), field, NotLike, val})
return b
}
func (b *Builder) WhereRaw(field interface{}, val interface{}, prefix ...string) *Builder {
b.whereList = append(b.whereList, WhereItem{getPrefixByField(field, prefix...), field, Raw, val})
b.whereList = append(b.whereList, WhereItem{"", getPrefixByField(field, prefix...), field, Raw, val})
return b
}

View File

@@ -9,6 +9,7 @@ import (
_ "github.com/mattn/go-sqlite3"
"github.com/tangpanqing/aorm"
"github.com/tangpanqing/aorm/builder"
"github.com/tangpanqing/aorm/model"
"github.com/tangpanqing/aorm/null"
"testing"
"time"
@@ -60,10 +61,12 @@ type PersonWithArticleCount struct {
var person = Person{}
var article = Article{}
var articleVO = ArticleVO{}
var personAge = PersonAge{}
func TestAll(t *testing.T) {
aorm.Store(&person, &article)
aorm.Store(&articleVO)
aorm.Store(&personAge)
dbList := make([]aorm.DbContent, 0)
//dbList = append(dbList, testSqlite3Connect())
@@ -104,12 +107,12 @@ func TestAll(t *testing.T) {
testJoin(dbItem.DriverName, dbItem.DbLink)
testJoinWithAlias(dbItem.DriverName, dbItem.DbLink)
//testGroupBy(dbItem.DriverName, dbItem.DbLink)
//testHaving(dbItem.DriverName, dbItem.DbLink)
testGroupBy(dbItem.DriverName, dbItem.DbLink)
testHaving(dbItem.DriverName, dbItem.DbLink)
return
testOrderBy(dbItem.DriverName, dbItem.DbLink)
break
//testLimit(dbItem.DriverName, dbItem.DbLink)
//testLock(dbItem.DriverName, dbItem.DbLink, id2)
testLimit(dbItem.DriverName, dbItem.DbLink)
testLock(dbItem.DriverName, dbItem.DbLink, id2)
testIncrement(dbItem.DriverName, dbItem.DbLink, id2)
testDecrement(dbItem.DriverName, dbItem.DbLink, id2)
@@ -218,7 +221,7 @@ func testInsert(driver string, db *sql.DB) int64 {
if errInsert != nil {
panic(driver + " testInsert " + "found err: " + errInsert.Error())
}
aorm.Use(db).Debug(false).Driver(driver).Insert(&Article{
aorm.Use(db).Debug(true).Driver(driver).Insert(&Article{
Type: null.IntFrom(0),
PersonId: null.IntFrom(id),
ArticleBody: null.StringFrom("文章内容"),
@@ -327,7 +330,7 @@ func testExists(driver string, db *sql.DB, id int64) bool {
//
//func testTable(driver string, db *sql.DB) {
// _, err := aorm.Use(db).Debug(false).Driver(driver).Table("person_1").Insert(&Person{Name: null.StringFrom("Cherry")})
// _, err := aorm.Use(db).Debug(true).Driver(driver).Table("person_1").Insert(&Person{Name: null.StringFrom("Cherry")})
// if err != nil {
// panic(driver + " testTable " + "found err:" + err.Error())
// }
@@ -335,7 +338,7 @@ func testExists(driver string, db *sql.DB, id int64) bool {
//
//func testSelect(driver string, db *sql.DB) {
// var listByFiled []Person
// err := aorm.Use(db).Debug(false).Driver(driver).Select("name,age").Where(&Person{Age: null.IntFrom(18)}).GetMany(&listByFiled)
// err := aorm.Use(db).Debug(true).Driver(driver).Select("name,age").Where(&Person{Age: null.IntFrom(18)}).GetMany(&listByFiled)
// if err != nil {
// panic(driver + " testSelect " + "found err:" + err.Error())
// }
@@ -345,7 +348,7 @@ func testExists(driver string, db *sql.DB, id int64) bool {
// var listByFiled []PersonWithArticleCount
//
// sub := aorm.Sub().Table("article").SelectCount("id", "article_count_tem").WhereRaw("person_id", "=person.id")
// err := aorm.Use(db).Debug(false).
// err := aorm.Use(db).Debug(true).
// Driver(driver).
// SelectExp(&sub, "article_count").
// Select("*").
@@ -362,7 +365,7 @@ func testExists(driver string, db *sql.DB, id int64) bool {
//
// sub := aorm.Sub().Table("article").Select("person_id").GroupBy("person_id").HavingGt("count(person_id)", 0)
//
// err := aorm.Use(db).Debug(false).
// err := aorm.Use(db).Debug(true).
// Table("person").
// Driver(driver).
// WhereIn("id", &sub).
@@ -383,7 +386,7 @@ func testExists(driver string, db *sql.DB, id int64) bool {
// where1 = append(where1, builder.WhereItem{Field: "money", Opt: builder.Eq, Val: 100.15})
// where1 = append(where1, builder.WhereItem{Field: "name", Opt: builder.Like, Val: []string{"%", "li", "%"}})
//
// err := aorm.Use(db).Debug(false).Driver(driver).Table("person").WhereArr(where1).GetMany(&listByWhere)
// err := aorm.Use(db).Debug(true).Driver(driver).Table("person").WhereArr(where1).GetMany(&listByWhere)
// if err != nil {
// panic(driver + "testWhere" + "found err")
// }
@@ -430,48 +433,41 @@ func testJoinWithAlias(driver string, db *sql.DB) {
}
}
//func testGroupBy(driver string, db *sql.DB) {
// var personAge PersonAge
// var where []builder.WhereItem
// where = append(where, builder.WhereItem{Field: "type", Opt: builder.Eq, Val: 0})
// err := aorm.Use(db).Debug(false).
// Table("person").
// Select("age").
// Select("count(age) as age_count").
// GroupBy("age").
// WhereArr(where).
// Driver(driver).
// OrderBy("age", "DESC").
// GetOne(&personAge)
// if err != nil {
// panic(driver + "testGroupBy" + "found err")
// }
//}
//
//func testHaving(driver string, db *sql.DB) {
// var listByHaving []PersonAge
//
// var where3 []builder.WhereItem
// where3 = append(where3, builder.WhereItem{Field: "type", Opt: builder.Eq, Val: 0})
//
// var having []builder.WhereItem
// having = append(having, builder.WhereItem{Field: "count(age)", Opt: builder.Gt, Val: 4})
//
// err := aorm.Use(db).Debug(false).
// Table("person").
// Select("age").
// Select("count(age) as age_count").
// GroupBy("age").
// WhereArr(where3).
// Driver(driver).
// OrderBy("age", "DESC").
// HavingArr(having).
// GetMany(&listByHaving)
// if err != nil {
// panic(driver + " testHaving " + "found err")
// }
//}
//
func testGroupBy(driver string, db *sql.DB) {
var personAgeItem PersonAge
var where []builder.WhereItem
where = append(where, builder.WhereItem{Field: "type", Opt: builder.Eq, Val: 0})
err := aorm.Use(db).Debug(true).
Table(&person).
Select(&person.Age).
SelectCount(&person.Age, &personAge.AgeCount).
GroupBy(&person.Age).
WhereEq(&person.Type, 0).
Driver(driver).
OrderBy(&person.Age, builder.Desc).
GetOne(&personAgeItem)
if err != nil {
panic(driver + "testGroupBy" + "found err")
}
}
func testHaving(driver string, db *sql.DB) {
var listByHaving []PersonAge
err := aorm.Use(db).Debug(true).Driver(driver).
Table(&person).
Select(&person.Age).
SelectCount(&person.Age, &personAge.AgeCount).
GroupBy(&person.Age).
WhereEq(&person.Type, 0).
OrderBy(&person.Age, builder.Desc).
HavingGt(builder.Count, &person.Age, 4).
GetMany(&listByHaving)
if err != nil {
panic(driver + " testHaving " + "found err")
}
}
func testOrderBy(driver string, db *sql.DB) {
var listByOrder []Person
err := aorm.Use(db).Debug(true).Driver(driver).
@@ -494,55 +490,50 @@ func testOrderBy(driver string, db *sql.DB) {
}
}
//
//func testLimit(driver string, db *sql.DB) {
// var list3 []Person
// var where1 []builder.WhereItem
// where1 = append(where1, builder.WhereItem{Field: "type", Opt: builder.Eq, Val: 0})
// err1 := aorm.Use(db).Debug(false).
// Table("person").
// WhereArr(where1).
// Limit(50, 10).
// Driver(driver).
// OrderBy("id", "DESC").
// GetMany(&list3)
// if err1 != nil {
// panic(driver + "testLimit" + "found err")
// }
//
// var list4 []Person
// var where2 []builder.WhereItem
// where2 = append(where2, builder.WhereItem{Field: "type", Opt: builder.Eq, Val: 0})
// err := aorm.Use(db).Debug(false).
// Table("person").
// WhereArr(where2).
// Page(3, 10).
// Driver(driver).
// OrderBy("id", "DESC").
// GetMany(&list4)
// if err != nil {
// panic(driver + "testPage" + "found err")
// }
//}
func testLimit(driver string, db *sql.DB) {
var list3 []Person
err1 := aorm.Use(db).Debug(true).
Table(&person).
WhereEq(&person.Type, 0).
Limit(50, 10).
Driver(driver).
OrderBy(&person.Id, builder.Desc).
GetMany(&list3)
if err1 != nil {
panic(driver + "testLimit" + "found err")
}
var list4 []Person
err := aorm.Use(db).Debug(true).
Driver(driver).
Table(&person).
WhereEq(&person.Type, 0).
Page(3, 10).
OrderBy(&person.Id, builder.Desc).
GetMany(&list4)
if err != nil {
panic(driver + "testPage" + "found err")
}
}
func testLock(driver string, db *sql.DB, id int64) {
if driver == model.Sqlite3 || driver == model.Mssql {
return
}
var itemByLock Person
err := aorm.Use(db).
Debug(true).
LockForUpdate(true).
Table(&person).
WhereEq(&person.Id, id).
Driver(driver).
OrderBy(&person.Id, builder.Desc).
GetOne(&itemByLock)
if err != nil {
panic(driver + "testLock" + "found err")
}
}
//func testLock(driver string, db *sql.DB, id int64) {
// if driver == model.Sqlite3 || driver == model.Mssql {
// return
// }
//
// var itemByLock Person
// err := aorm.Use(db).
// Debug(false).
// LockForUpdate(true).
// Where(&Person{Id: null.IntFrom(id)}).
// Driver(driver).
// OrderBy("id", "DESC").
// GetOne(&itemByLock)
// if err != nil {
// panic(driver + "testLock" + "found err")
// }
//}
//
func testIncrement(driver string, db *sql.DB, id int64) {
_, err := aorm.Use(db).Debug(true).Driver(driver).Table(&person).WhereEq(&person.Id, id).Increment(&person.Age, 1)
if err != nil {
@@ -560,25 +551,25 @@ func testDecrement(driver string, db *sql.DB, id int64) {
func testValue(driver string, db *sql.DB, id int64) {
var name string
errName := aorm.Use(db).Debug(false).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Id, id).Value(&person.Name, &name)
errName := aorm.Use(db).Debug(true).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Id, id).Value(&person.Name, &name)
if errName != nil {
panic(driver + "testValue" + "found err")
}
var age int64
errAge := aorm.Use(db).Debug(false).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Id, id).Value(&person.Age, &age)
errAge := aorm.Use(db).Debug(true).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Id, id).Value(&person.Age, &age)
if errAge != nil {
panic(driver + "testValue" + "found err")
}
var money float32
errMoney := aorm.Use(db).Debug(false).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Id, id).Value(&person.Money, &money)
errMoney := aorm.Use(db).Debug(true).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Id, id).Value(&person.Money, &money)
if errMoney != nil {
panic(driver + "testValue" + "found err")
}
var test float64
errTest := aorm.Use(db).Debug(false).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Id, id).Value(&person.Test, &test)
errTest := aorm.Use(db).Debug(true).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Id, id).Value(&person.Test, &test)
if errTest != nil {
panic(driver + "testValue" + "found err")
}
@@ -620,7 +611,7 @@ func testCount(driver string, db *sql.DB) {
//
func testSum(driver string, db *sql.DB) {
_, err := aorm.Use(db).Debug(false).Table(&person).WhereEq(&person.Age, 18).Driver(driver).Sum(&person.Age)
_, err := aorm.Use(db).Debug(true).Table(&person).WhereEq(&person.Age, 18).Driver(driver).Sum(&person.Age)
if err != nil {
panic(driver + "testSum" + "found err")
}
@@ -634,14 +625,14 @@ func testAvg(driver string, db *sql.DB) {
}
func testMin(driver string, db *sql.DB) {
_, err := aorm.Use(db).Debug(false).Table(&person).WhereEq(&person.Age, 18).Driver(driver).Min(&person.Age)
_, err := aorm.Use(db).Debug(true).Table(&person).WhereEq(&person.Age, 18).Driver(driver).Min(&person.Age)
if err != nil {
panic(driver + "testMin" + "found err")
}
}
func testMax(driver string, db *sql.DB) {
_, err := aorm.Use(db).Debug(false).Table(&person).WhereEq(&person.Age, 18).Driver(driver).Max(&person.Age)
_, err := aorm.Use(db).Debug(true).Table(&person).WhereEq(&person.Age, 18).Driver(driver).Max(&person.Age)
if err != nil {
panic(driver + "testMax" + "found err")
}
@@ -649,7 +640,7 @@ func testMax(driver string, db *sql.DB) {
//
//func testExec(driver string, db *sql.DB) {
// _, err := aorm.Use(db).Debug(false).Driver(driver).Exec("UPDATE person SET name = ? WHERE id=?", "Bob", 3)
// _, err := aorm.Use(db).Debug(true).Driver(driver).Exec("UPDATE person SET name = ? WHERE id=?", "Bob", 3)
// if err != nil {
// panic(driver + "testExec" + "found err")
// }
@@ -658,7 +649,7 @@ func testMax(driver string, db *sql.DB) {
//func testTransaction(driver string, db *sql.DB) {
// tx, _ := db.Begin()
//
// id, errInsert := aorm.Use(tx).Debug(false).Driver(driver).Insert(&Person{
// id, errInsert := aorm.Use(tx).Debug(true).Driver(driver).Insert(&Person{
// Name: null.StringFrom("Alice"),
// })
//
@@ -668,7 +659,7 @@ func testMax(driver string, db *sql.DB) {
// return
// }
//
// _, errCount := aorm.Use(tx).Debug(false).Driver(driver).Where(&Person{
// _, errCount := aorm.Use(tx).Debug(true).Driver(driver).Where(&Person{
// Id: null.IntFrom(id),
// }).Count("*")
// if errCount != nil {
@@ -678,7 +669,7 @@ func testMax(driver string, db *sql.DB) {
// }
//
// var person Person
// errPerson := aorm.Use(tx).Debug(false).Where(&Person{
// errPerson := aorm.Use(tx).Debug(true).Where(&Person{
// Id: null.IntFrom(id),
// }).Driver(driver).OrderBy("id", "DESC").GetOne(&person)
// if errPerson != nil {
@@ -687,7 +678,7 @@ func testMax(driver string, db *sql.DB) {
// return
// }
//
// _, errUpdate := aorm.Use(tx).Debug(false).Driver(driver).Where(&Person{
// _, errUpdate := aorm.Use(tx).Debug(true).Driver(driver).Where(&Person{
// Id: null.IntFrom(id),
// }).Update(&Person{
// Name: null.StringFrom("Bob"),
@@ -703,7 +694,7 @@ func testMax(driver string, db *sql.DB) {
//}
//
//func testTruncate(driver string, db *sql.DB) {
// _, err := aorm.Use(db).Debug(false).Driver(driver).Table("person").Truncate()
// _, err := aorm.Use(db).Debug(true).Driver(driver).Table("person").Truncate()
// if err != nil {
// panic(driver + " testTruncate " + "found err")
// }
@@ -714,7 +705,7 @@ func testMax(driver string, db *sql.DB) {
// var where2 []builder.WhereItem
// where2 = append(where2, builder.WhereItem{Field: "o.type", Opt: builder.Eq, Val: 0})
// where2 = append(where2, builder.WhereItem{Field: "p.age", Opt: builder.In, Val: []int{18, 20}})
// err := aorm.Use(db).Debug(false).
// err := aorm.Use(db).Debug(true).
// Table("article o").
// LeftJoin("person p", helper.Ul("p.id=o.personId")).
// Select("o.*").