package/aorm
1
0
Fork 0
mirror of https://github.com/tangpanqing/aorm.git synced 2025-10-06 16:36:57 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

new test

Browse Source
This commit is contained in:
tangpanqing
2023-01-06 17:11:55 +08:00
parent 9f9a7c975f
commit e65a46e49c
3 changed files with 158 additions and 122 deletions
Download Patch File Download Diff File Expand all files Collapse all files

107
builder/crud.go
View File

@@ -94,14 +94,27 @@ func (b *Builder) Insert(dest interface{}) (int64, error) {
typeOf := reflect.TypeOf(dest) typeOf := reflect.TypeOf(dest)
valueOf := reflect.ValueOf(dest) valueOf := reflect.ValueOf(dest)
//主键名字
var primaryKey = ""
var keys []string var keys []string
var paramList []any var paramList []any
var place []string var place []string
for i := 0; i < typeOf.Elem().NumField(); i++ { for i := 0; i < typeOf.Elem().NumField(); i++ {
key := helper.UnderLine(typeOf.Elem().Field(i).Name)
//如果是Postgres数据库，寻找主键
if b.driverName == model.Postgres {
tag := typeOf.Elem().Field(i).Tag.Get("aorm")
if -1 != strings.Index(tag, "primary") {
primaryKey = key
}
}
isNotNull := valueOf.Elem().Field(i).Field(0).Field(1).Bool() isNotNull := valueOf.Elem().Field(i).Field(0).Field(1).Bool()
if isNotNull { if isNotNull {
key := helper.UnderLine(typeOf.Elem().Field(i).Name)
val := valueOf.Elem().Field(i).Field(0).Field(0).Interface() val := valueOf.Elem().Field(i).Field(0).Field(0).Interface()
keys = append(keys, key) keys = append(keys, key)
paramList = append(paramList, val) paramList = append(paramList, val)
place = append(place, "?") place = append(place, "?")
@@ -110,14 +123,11 @@ func (b *Builder) Insert(dest interface{}) (int64, error) {
sqlStr := "INSERT INTO " + b.getTableNameCommon(typeOf, valueOf) + " (" + strings.Join(keys, ",") + ") VALUES (" + strings.Join(place, ",") + ")" sqlStr := "INSERT INTO " + b.getTableNameCommon(typeOf, valueOf) + " (" + strings.Join(keys, ",") + ") VALUES (" + strings.Join(place, ",") + ")"
if b.driverName == model.Postgres {
sqlStr = convertToPostgresSql(sqlStr)
}
if b.driverName == model.Mssql { if b.driverName == model.Mssql {
return b.insertForMssqlOrPostgres(sqlStr+"; select ID = convert(bigint, SCOPE_IDENTITY())", paramList...) return b.insertForMssqlOrPostgres(sqlStr+"; SELECT SCOPE_IDENTITY()", paramList...)
} else if b.driverName == model.Postgres { } else if b.driverName == model.Postgres {
return b.insertForMssqlOrPostgres(sqlStr+" returning id", paramList...) sqlStr = convertToPostgresSql(sqlStr)
return b.insertForMssqlOrPostgres(sqlStr+" RETURNING "+primaryKey, paramList...)
} else { } else {
return b.insertForCommon(sqlStr, paramList...) return b.insertForCommon(sqlStr, paramList...)
} }
@@ -125,6 +135,11 @@ func (b *Builder) Insert(dest interface{}) (int64, error) {
//对于Mssql,Postgres类型数据库，为了获取最后插入的id，需要改写入为查询 //对于Mssql,Postgres类型数据库，为了获取最后插入的id，需要改写入为查询
func (b *Builder) insertForMssqlOrPostgres(sql string, paramList ...any) (int64, error) { func (b *Builder) insertForMssqlOrPostgres(sql string, paramList ...any) (int64, error) {
if b.isDebug {
fmt.Println(sql)
fmt.Println(paramList...)
}
rows, err := b.LinkCommon.Query(sql, paramList...) rows, err := b.LinkCommon.Query(sql, paramList...)
if err != nil { if err != nil {
return 0, err return 0, err
@@ -324,21 +339,21 @@ func (b *Builder) GetSqlAndParams() (string, []interface{}) {
groupStr, paramList := b.handleGroup(paramList) groupStr, paramList := b.handleGroup(paramList)
havingStr, paramList := b.handleHaving(paramList) havingStr, paramList := b.handleHaving(paramList)
orderStr, paramList := b.handleOrder(paramList) orderStr, paramList := b.handleOrder(paramList)
limitStr, paramList := b.handleLimit(b.offset, b.pageSize, paramList) limitStr, paramList := b.handleLimit(paramList)
lockStr := handleLockForUpdate(b.isLockForUpdate) lockStr := b.handleLockForUpdate()
sqlStr := "SELECT " + fieldStr + " FROM " + tableName + " " + b.tableAlias + joinStr + whereStr + groupStr + havingStr + orderStr + limitStr + lockStr sql := "SELECT " + fieldStr + " FROM " + tableName + " " + b.tableAlias + joinStr + whereStr + groupStr + havingStr + orderStr + limitStr + lockStr
if b.driverName == model.Postgres { if b.driverName == model.Postgres {
sqlStr = convertToPostgresSql(sqlStr) sql = convertToPostgresSql(sql)
} }
if b.isDebug { if b.isDebug {
fmt.Println(sqlStr) fmt.Println(sql)
//fmt.Println(paramList...) fmt.Println(paramList...)
} }
return sqlStr, paramList return sql, paramList
} }
// Update 更新记录 // Update 更新记录
@@ -351,10 +366,6 @@ func (b *Builder) Update(dest interface{}) (int64, error) {
whereStr, paramList := b.handleWhere(paramList) whereStr, paramList := b.handleWhere(paramList)
sqlStr := "UPDATE " + b.getTableNameCommon(typeOf, valueOf) + setStr + whereStr sqlStr := "UPDATE " + b.getTableNameCommon(typeOf, valueOf) + setStr + whereStr
if b.driverName == model.Postgres {
sqlStr = convertToPostgresSql(sqlStr)
}
return b.ExecAffected(sqlStr, paramList...) return b.ExecAffected(sqlStr, paramList...)
} }
@@ -378,18 +389,16 @@ func (b *Builder) Delete(destList ...interface{}) (int64, error) {
whereStr, paramList := b.handleWhere(paramList) whereStr, paramList := b.handleWhere(paramList)
sqlStr := "DELETE FROM " + tableName + whereStr sqlStr := "DELETE FROM " + tableName + whereStr
if b.driverName == model.Postgres {
sqlStr = convertToPostgresSql(sqlStr)
}
return b.ExecAffected(sqlStr, paramList...) return b.ExecAffected(sqlStr, paramList...)
} }
// Truncate 清空记录, sqlite3不支持此操作 // Truncate 清空记录
func (b *Builder) Truncate() (int64, error) { func (b *Builder) Truncate() (int64, error) {
sqlStr := "TRUNCATE TABLE " + getTableNameByTable(b.table) sqlStr := ""
if b.driverName == model.Sqlite3 { if b.driverName == model.Sqlite3 {
sqlStr = "DELETE FROM " + getTableNameByTable(b.table) sqlStr = "DELETE FROM " + getTableNameByTable(b.table)
} else {
sqlStr = "TRUNCATE TABLE " + getTableNameByTable(b.table)
} }
return b.ExecAffected(sqlStr) return b.ExecAffected(sqlStr)
@@ -399,7 +408,7 @@ func (b *Builder) Truncate() (int64, error) {
func (b *Builder) Exists() (bool, error) { func (b *Builder) Exists() (bool, error) {
var obj IntStruct var obj IntStruct
err := b.selectCommon("", "1 as c", nil, "").Limit(0, 1).GetOne(&obj) err := b.selectCommon("", "1 AS c", nil, "").Limit(0, 1).GetOne(&obj)
if err != nil { if err != nil {
return false, err return false, err
} }
@@ -458,8 +467,9 @@ func (b *Builder) Value(field interface{}, dest interface{}) error {
} }
// Pluck 获取某一列的值 // Pluck 获取某一列的值
func (b *Builder) Pluck(fieldName interface{}, values interface{}) error { func (b *Builder) Pluck(field interface{}, values interface{}) error {
b.Select(fieldName) b.Select(field)
fieldName := getFieldName(field)
rows, errRows := b.GetRows() rows, errRows := b.GetRows()
defer rows.Close() defer rows.Close()
@@ -506,10 +516,6 @@ func (b *Builder) Increment(field interface{}, step int) (int64, error) {
whereStr, paramList := b.handleWhere(paramList) whereStr, paramList := b.handleWhere(paramList)
sqlStr := "UPDATE " + getTableNameByTable(b.table) + " SET " + getFieldName(field) + "=" + getFieldName(field) + "+?" + whereStr sqlStr := "UPDATE " + getTableNameByTable(b.table) + " SET " + getFieldName(field) + "=" + getFieldName(field) + "+?" + whereStr
if b.driverName == model.Postgres {
sqlStr = convertToPostgresSql(sqlStr)
}
return b.ExecAffected(sqlStr, paramList...) return b.ExecAffected(sqlStr, paramList...)
} }
@@ -520,31 +526,27 @@ func (b *Builder) Decrement(field interface{}, step int) (int64, error) {
whereStr, paramList := b.handleWhere(paramList) whereStr, paramList := b.handleWhere(paramList)
sqlStr := "UPDATE " + getTableNameByTable(b.table) + " SET " + getFieldName(field) + "=" + getFieldName(field) + "-?" + whereStr sqlStr := "UPDATE " + getTableNameByTable(b.table) + " SET " + getFieldName(field) + "=" + getFieldName(field) + "-?" + whereStr
if b.driverName == model.Postgres {
sqlStr = convertToPostgresSql(sqlStr)
}
return b.ExecAffected(sqlStr, paramList...) return b.ExecAffected(sqlStr, paramList...)
} }
// Exec 通用执行-新增,更新,删除 // Exec 通用执行-新增,更新,删除
func (b *Builder) Exec(sqlStr string, args ...interface{}) (sql.Result, error) { func (b *Builder) Exec(sql string, paramList ...interface{}) (sql.Result, error) {
if b.driverName == model.Postgres { if b.driverName == model.Postgres {
sqlStr = convertToPostgresSql(sqlStr) sql = convertToPostgresSql(sql)
} }
if b.isDebug { if b.isDebug {
fmt.Println(sqlStr) fmt.Println(sql)
//fmt.Println(args...) fmt.Println(paramList...)
} }
smt, err1 := b.LinkCommon.Prepare(sqlStr) smt, err1 := b.LinkCommon.Prepare(sql)
if err1 != nil { if err1 != nil {
return nil, err1 return nil, err1
} }
defer smt.Close() defer smt.Close()
res, err2 := smt.Exec(args...) res, err2 := smt.Exec(paramList...)
if err2 != nil { if err2 != nil {
return nil, err2 return nil, err2
} }
@@ -554,8 +556,12 @@ func (b *Builder) Exec(sqlStr string, args ...interface{}) (sql.Result, error) {
} }
// ExecAffected 通用执行-更新,删除 // ExecAffected 通用执行-更新,删除
func (b *Builder) ExecAffected(sqlStr string, args ...interface{}) (int64, error) { func (b *Builder) ExecAffected(sql string, paramList ...interface{}) (int64, error) {
res, err := b.Exec(sqlStr, args...) if b.driverName == model.Postgres {
sql = convertToPostgresSql(sql)
}
res, err := b.Exec(sql, paramList...)
if err != nil { if err != nil {
return 0, err return 0, err
} }
@@ -624,14 +630,25 @@ func (b *Builder) LockForUpdate(isLockForUpdate bool) *Builder {
} }
//拼接SQL,查询与筛选通用操作 //拼接SQL,查询与筛选通用操作
func (b *Builder) whereAndHaving(where []WhereItem, paramList []any) ([]string, []any) { func (b *Builder) whereAndHaving(where []WhereItem, paramList []any, isFromHaving bool) ([]string, []any) {
var whereList []string var whereList []string
for i := 0; i < len(where); i++ { for i := 0; i < len(where); i++ {
allFieldName := "" allFieldName := ""
if where[i].Prefix != "" { if where[i].Prefix != "" {
allFieldName += where[i].Prefix + "." allFieldName += where[i].Prefix + "."
} }
allFieldName += getFieldName(where[i].Field)
//如果是mssql或者Postgres,并且来自having的话，需要特殊处理
if (b.driverName == model.Mssql || b.driverName == model.Postgres) && isFromHaving {
fieldNameCurrent := getFieldName(where[i].Field)
for m := 0; m < len(b.selectList); m++ {
if fieldNameCurrent == getFieldName(b.selectList[m].FieldNew) {
allFieldName += handleFieldWith(b.selectList[m])
}
}
} else {
allFieldName += getFieldName(where[i].Field)
}
if "**builder.Builder" == reflect.TypeOf(where[i].Val).String() { if "**builder.Builder" == reflect.TypeOf(where[i].Val).String() {
executor := *(**Builder)(unsafe.Pointer(reflect.ValueOf(where[i].Val).Pointer())) executor := *(**Builder)(unsafe.Pointer(reflect.ValueOf(where[i].Val).Pointer()))

58
builder/handle.go
View File

@@ -7,6 +7,27 @@ import (
"strings" "strings"
) )
func handleFieldWith(selectItem SelectItem) string {
str := ""
if selectItem.FuncName != "" {
str += selectItem.FuncName
str += "("
}
if selectItem.Prefix != "" {
str += selectItem.Prefix
str += "."
}
str += getFieldName(selectItem.Field)
if selectItem.FuncName != "" {
str += ")"
}
return str
}
//拼接SQL,字段相关 //拼接SQL,字段相关
func (b *Builder) handleField(paramList []any) (string, []any) { func (b *Builder) handleField(paramList []any) (string, []any) {
fieldStr := "" fieldStr := ""
@@ -25,22 +46,7 @@ func (b *Builder) handleField(paramList []any) (string, []any) {
for i := 0; i < len(b.selectList); i++ { for i := 0; i < len(b.selectList); i++ {
selectItem := b.selectList[i] selectItem := b.selectList[i]
str := "" str := handleFieldWith(selectItem)
if selectItem.FuncName != "" {
str += selectItem.FuncName
str += "("
}
if selectItem.Prefix != "" {
str += selectItem.Prefix
str += "."
}
str += getFieldName(selectItem.Field)
if selectItem.FuncName != "" {
str += ")"
}
if selectItem.FieldNew != nil { if selectItem.FieldNew != nil {
str += " AS " str += " AS "
@@ -68,7 +74,7 @@ func (b *Builder) handleWhere(paramList []any) (string, []any) {
return "", paramList return "", paramList
} }
strList, paramList := b.whereAndHaving(b.whereList, paramList) strList, paramList := b.whereAndHaving(b.whereList, paramList, false)
return " WHERE " + strings.Join(strList, " AND "), paramList return " WHERE " + strings.Join(strList, " AND "), paramList
} }
@@ -136,7 +142,7 @@ func (b *Builder) handleHaving(paramList []any) (string, []any) {
return "", paramList return "", paramList
} }
strList, paramList := b.whereAndHaving(b.havingList, paramList) strList, paramList := b.whereAndHaving(b.havingList, paramList, true)
return " Having " + strings.Join(strList, " AND "), paramList return " Having " + strings.Join(strList, " AND "), paramList
} }
@@ -156,20 +162,20 @@ func (b *Builder) handleOrder(paramList []any) (string, []any) {
} }
//拼接SQL,分页相关 Postgres数据库分页数量在前偏移在后，其他数据库偏移量在前分页数量在后，另外Mssql数据库的关键词是offset...next //拼接SQL,分页相关 Postgres数据库分页数量在前偏移在后，其他数据库偏移量在前分页数量在后，另外Mssql数据库的关键词是offset...next
func (b *Builder) handleLimit(offset int, pageSize int, paramList []any) (string, []any) { func (b *Builder) handleLimit(paramList []any) (string, []any) {
if 0 == pageSize { if 0 == b.pageSize {
return "", paramList return "", paramList
} }
str := "" str := ""
if b.driverName == model.Postgres { if b.driverName == model.Postgres {
paramList = append(paramList, pageSize) paramList = append(paramList, b.pageSize)
paramList = append(paramList, offset) paramList = append(paramList, b.offset)
str = " Limit ? offset ? " str = " Limit ? offset ? "
} else { } else {
paramList = append(paramList, offset) paramList = append(paramList, b.offset)
paramList = append(paramList, pageSize) paramList = append(paramList, b.pageSize)
str = " Limit ?,? " str = " Limit ?,? "
if b.driverName == model.Mssql { if b.driverName == model.Mssql {
@@ -181,8 +187,8 @@ func (b *Builder) handleLimit(offset int, pageSize int, paramList []any) (string
} }
//拼接SQL,锁 //拼接SQL,锁
func handleLockForUpdate(isLock bool) string { func (b *Builder) handleLockForUpdate() string {
if isLock { if b.isLockForUpdate {
return " FOR UPDATE" return " FOR UPDATE"
} }

115
test/aorm_test.go
View File

@@ -15,6 +15,11 @@ import (
"time" "time"
) )
type Student struct {
StudentId null.Int `aorm:"primary;auto_increment" json:"studentId"`
Name null.String `aorm:"size:100;not null;comment:名字" json:"name"`
}
type Article struct { type Article struct {
Id null.Int `aorm:"primary;auto_increment" json:"id"` Id null.Int `aorm:"primary;auto_increment" json:"id"`
Type null.Int `aorm:"index;comment:类型" json:"type"` Type null.Int `aorm:"index;comment:类型" json:"type"`
@@ -58,6 +63,7 @@ type PersonWithArticleCount struct {
ArticleCount null.Int `aorm:"comment:文章数量" json:"articleCount"` ArticleCount null.Int `aorm:"comment:文章数量" json:"articleCount"`
} }
var student = Student{}
var person = Person{} var person = Person{}
var article = Article{} var article = Article{}
var articleVO = ArticleVO{} var articleVO = ArticleVO{}
@@ -70,10 +76,10 @@ func TestAll(t *testing.T) {
aorm.Store(&personAge, &personWithArticleCount) aorm.Store(&personAge, &personWithArticleCount)
dbList := make([]aorm.DbContent, 0) dbList := make([]aorm.DbContent, 0)
//dbList = append(dbList, testSqlite3Connect()) dbList = append(dbList, testSqlite3Connect())
dbList = append(dbList, testMysqlConnect()) dbList = append(dbList, testMysqlConnect())
//dbList = append(dbList, testPostgresConnect()) dbList = append(dbList, testPostgresConnect())
//dbList = append(dbList, testMssqlConnect()) dbList = append(dbList, testMssqlConnect())
for i := 0; i < len(dbList); i++ { for i := 0; i < len(dbList); i++ {
dbItem := dbList[i] dbItem := dbList[i]
@@ -82,6 +88,7 @@ func TestAll(t *testing.T) {
testShowCreateTable(dbItem.DriverName, dbItem.DbLink) testShowCreateTable(dbItem.DriverName, dbItem.DbLink)
id := testInsert(dbItem.DriverName, dbItem.DbLink) id := testInsert(dbItem.DriverName, dbItem.DbLink)
testInsertBatch(dbItem.DriverName, dbItem.DbLink) testInsertBatch(dbItem.DriverName, dbItem.DbLink)
testGetOne(dbItem.DriverName, dbItem.DbLink, id) testGetOne(dbItem.DriverName, dbItem.DbLink, id)
testGetMany(dbItem.DriverName, dbItem.DbLink) testGetMany(dbItem.DriverName, dbItem.DbLink)
@@ -197,6 +204,7 @@ func testMigrate(driver string, db *sql.DB) {
//AutoMigrate //AutoMigrate
aorm.Migrator(db).Driver(driver).Opinion("ENGINE", "InnoDB").Opinion("COMMENT", "人员表").AutoMigrate(&Person{}) aorm.Migrator(db).Driver(driver).Opinion("ENGINE", "InnoDB").Opinion("COMMENT", "人员表").AutoMigrate(&Person{})
aorm.Migrator(db).Driver(driver).Opinion("ENGINE", "InnoDB").Opinion("COMMENT", "文章").AutoMigrate(&Article{}) aorm.Migrator(db).Driver(driver).Opinion("ENGINE", "InnoDB").Opinion("COMMENT", "文章").AutoMigrate(&Article{})
aorm.Migrator(db).Driver(driver).Opinion("ENGINE", "InnoDB").Opinion("COMMENT", "学生").AutoMigrate(&Student{})
//Migrate //Migrate
aorm.Migrator(db).Driver(driver).Opinion("ENGINE", "InnoDB").Opinion("COMMENT", "人员表").Migrate("person_1", &Person{}) aorm.Migrator(db).Driver(driver).Opinion("ENGINE", "InnoDB").Opinion("COMMENT", "人员表").Migrate("person_1", &Person{})
@@ -217,18 +225,18 @@ func testInsert(driver string, db *sql.DB) int64 {
Test: null.FloatFrom(2), Test: null.FloatFrom(2),
} }
id, errInsert := aorm.Use(db).Debug(true).Driver(driver).Insert(&obj) id, errInsert := aorm.Use(db).Debug(false).Driver(driver).Insert(&obj)
if errInsert != nil { if errInsert != nil {
panic(driver + " testInsert " + "found err: " + errInsert.Error()) panic(driver + " testInsert " + "found err: " + errInsert.Error())
} }
aorm.Use(db).Debug(true).Driver(driver).Insert(&Article{ aorm.Use(db).Debug(false).Driver(driver).Insert(&Article{
Type: null.IntFrom(0), Type: null.IntFrom(0),
PersonId: null.IntFrom(id), PersonId: null.IntFrom(id),
ArticleBody: null.StringFrom("文章内容"), ArticleBody: null.StringFrom("文章内容"),
}) })
var personItem Person var personItem Person
err := aorm.Use(db).Table(&person).Debug(true).Driver(driver).Table(&person).WhereEq(&person.Id, id).OrderBy(&person.Id, builder.Desc).GetOne(&personItem) err := aorm.Use(db).Table(&person).Debug(false).Driver(driver).Table(&person).WhereEq(&person.Id, id).OrderBy(&person.Id, builder.Desc).GetOne(&personItem)
if err != nil { if err != nil {
fmt.Println(err.Error()) fmt.Println(err.Error())
} }
@@ -257,6 +265,11 @@ func testInsert(driver string, db *sql.DB) int64 {
fmt.Println(driver + ",Test not match, expected: " + fmt.Sprintf("%v", obj.Test.Float64) + " ,but real is : " + fmt.Sprintf("%v", personItem.Test.Float64)) fmt.Println(driver + ",Test not match, expected: " + fmt.Sprintf("%v", obj.Test.Float64) + " ,but real is : " + fmt.Sprintf("%v", personItem.Test.Float64))
} }
//测试非id主键
aorm.Use(db).Debug(false).Driver(driver).Insert(&Student{
Name: null.StringFrom("new student"),
})
return id return id
} }
@@ -282,7 +295,7 @@ func testInsertBatch(driver string, db *sql.DB) int64 {
Test: null.FloatFrom(200.15987654321987654321), Test: null.FloatFrom(200.15987654321987654321),
}) })
count, err := aorm.Use(db).Debug(true).Driver(driver).InsertBatch(&batch) count, err := aorm.Use(db).Debug(false).Driver(driver).InsertBatch(&batch)
if err != nil { if err != nil {
panic(driver + " testInsertBatch " + "found err:" + err.Error()) panic(driver + " testInsertBatch " + "found err:" + err.Error())
} }
@@ -292,7 +305,7 @@ func testInsertBatch(driver string, db *sql.DB) int64 {
func testGetOne(driver string, db *sql.DB, id int64) { func testGetOne(driver string, db *sql.DB, id int64) {
var personItem Person var personItem Person
errFind := aorm.Use(db).Debug(true).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Id, id).GetOne(&personItem) errFind := aorm.Use(db).Debug(false).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Id, id).GetOne(&personItem)
if errFind != nil { if errFind != nil {
panic(driver + "testGetOne" + "found err") panic(driver + "testGetOne" + "found err")
} }
@@ -300,26 +313,26 @@ func testGetOne(driver string, db *sql.DB, id int64) {
func testGetMany(driver string, db *sql.DB) { func testGetMany(driver string, db *sql.DB) {
var list []Person var list []Person
errSelect := aorm.Use(db).Driver(driver).Debug(true).Table(&person).WhereEq(&person.Type, 0).GetMany(&list) errSelect := aorm.Use(db).Driver(driver).Debug(false).Table(&person).WhereEq(&person.Type, 0).GetMany(&list)
if errSelect != nil { if errSelect != nil {
panic(driver + " testGetMany " + "found err:" + errSelect.Error()) panic(driver + " testGetMany " + "found err:" + errSelect.Error())
} }
} }
func testUpdate(driver string, db *sql.DB, id int64) { func testUpdate(driver string, db *sql.DB, id int64) {
_, errUpdate := aorm.Use(db).Debug(true).Driver(driver).WhereEq(&person.Id, id).Update(&Person{Name: null.StringFrom("Bob")}) _, errUpdate := aorm.Use(db).Debug(false).Driver(driver).WhereEq(&person.Id, id).Update(&Person{Name: null.StringFrom("Bob")})
if errUpdate != nil { if errUpdate != nil {
panic(driver + "testUpdate" + "found err") panic(driver + "testUpdate" + "found err")
} }
} }
func testDelete(driver string, db *sql.DB, id int64) { func testDelete(driver string, db *sql.DB, id int64) {
_, errDelete := aorm.Use(db).Driver(driver).Debug(true).Table(&person).WhereEq(&person.Id, id).Delete() _, errDelete := aorm.Use(db).Driver(driver).Debug(false).Table(&person).WhereEq(&person.Id, id).Delete()
if errDelete != nil { if errDelete != nil {
panic(driver + "testDelete" + "found err") panic(driver + "testDelete" + "found err")
} }
_, errDelete2 := aorm.Use(db).Driver(driver).Debug(true).Delete(&Person{ _, errDelete2 := aorm.Use(db).Driver(driver).Debug(false).Delete(&Person{
Id: null.IntFrom(id), Id: null.IntFrom(id),
}) })
if errDelete2 != nil { if errDelete2 != nil {
@@ -328,7 +341,7 @@ func testDelete(driver string, db *sql.DB, id int64) {
} }
func testExists(driver string, db *sql.DB, id int64) bool { func testExists(driver string, db *sql.DB, id int64) bool {
exists, err := aorm.Use(db).Driver(driver).Debug(true).Table(&person).WhereEq(&person.Id, id).OrderBy(&person.Id, builder.Desc).Exists() exists, err := aorm.Use(db).Driver(driver).Debug(false).Table(&person).WhereEq(&person.Id, id).OrderBy(&person.Id, builder.Desc).Exists()
if err != nil { if err != nil {
panic(driver + " testExists " + "found err:" + err.Error()) panic(driver + " testExists " + "found err:" + err.Error())
} }
@@ -336,12 +349,12 @@ func testExists(driver string, db *sql.DB, id int64) bool {
} }
func testTable(driver string, db *sql.DB) { func testTable(driver string, db *sql.DB) {
_, err := aorm.Use(db).Debug(true).Driver(driver).Table("person_1").Insert(&Person{Name: null.StringFrom("Cherry")}) _, err := aorm.Use(db).Debug(false).Driver(driver).Table("person_1").Insert(&Person{Name: null.StringFrom("Cherry")})
if err != nil { if err != nil {
panic(driver + " testTable " + "found err:" + err.Error()) panic(driver + " testTable " + "found err:" + err.Error())
} }
_, err2 := aorm.Use(db).Debug(true).Driver(driver).Table(&person).Insert(&Person{Name: null.StringFrom("Cherry")}) _, err2 := aorm.Use(db).Debug(false).Driver(driver).Table(&person).Insert(&Person{Name: null.StringFrom("Cherry")})
if err2 != nil { if err2 != nil {
panic(driver + " testTable " + "found err:" + err2.Error()) panic(driver + " testTable " + "found err:" + err2.Error())
} }
@@ -349,7 +362,7 @@ func testTable(driver string, db *sql.DB) {
func testSelect(driver string, db *sql.DB) { func testSelect(driver string, db *sql.DB) {
var listByFiled []Person var listByFiled []Person
err := aorm.Use(db).Debug(true).Driver(driver).Table(&person).Select(&person.Name).Select(&person.Age).WhereEq(&person.Age, 18).GetMany(&listByFiled) err := aorm.Use(db).Debug(false).Driver(driver).Table(&person).Select(&person.Name).Select(&person.Age).WhereEq(&person.Age, 18).GetMany(&listByFiled)
if err != nil { if err != nil {
panic(driver + " testSelect " + "found err:" + err.Error()) panic(driver + " testSelect " + "found err:" + err.Error())
} }
@@ -359,7 +372,7 @@ func testSelectWithSub(driver string, db *sql.DB) {
var listByFiled []PersonWithArticleCount var listByFiled []PersonWithArticleCount
sub := aorm.Sub().Table(&article).SelectCount(&article.Id, "article_count_tem").WhereRawEq(&article.PersonId, &person.Id) sub := aorm.Sub().Table(&article).SelectCount(&article.Id, "article_count_tem").WhereRawEq(&article.PersonId, &person.Id)
err := aorm.Use(db).Debug(true). err := aorm.Use(db).Debug(false).
Driver(driver). Driver(driver).
SelectExp(&sub, &personWithArticleCount.ArticleCount). SelectExp(&sub, &personWithArticleCount.ArticleCount).
SelectAll(&person). SelectAll(&person).
@@ -374,8 +387,8 @@ func testSelectWithSub(driver string, db *sql.DB) {
func testWhereWithSub(driver string, db *sql.DB) { func testWhereWithSub(driver string, db *sql.DB) {
var listByFiled []Person var listByFiled []Person
sub := aorm.Sub().Table(&article).SelectCount(&article.PersonId, "count_person_id").GroupBy(&article.PersonId).HavingGt("count_person_id", 0) sub := aorm.Sub().Table(&article).Driver(driver).SelectCount(&article.PersonId, "count_person_id").GroupBy(&article.PersonId).HavingGt("count_person_id", 0)
err := aorm.Use(db).Debug(true). err := aorm.Use(db).Debug(false).
Table(&person). Table(&person).
Driver(driver). Driver(driver).
WhereIn(&person.Id, &sub). WhereIn(&person.Id, &sub).
@@ -387,7 +400,7 @@ func testWhereWithSub(driver string, db *sql.DB) {
func testWhere(driver string, db *sql.DB) { func testWhere(driver string, db *sql.DB) {
var listByWhere []Person var listByWhere []Person
err := aorm.Use(db).Debug(true).Driver(driver).Table(&person).WhereArr([]builder.WhereItem{ err := aorm.Use(db).Debug(false).Driver(driver).Table(&person).WhereArr([]builder.WhereItem{
{Field: &person.Type, Opt: builder.Eq, Val: 0}, {Field: &person.Type, Opt: builder.Eq, Val: 0},
{Field: &person.Age, Opt: builder.In, Val: []int{18, 20}}, {Field: &person.Age, Opt: builder.In, Val: []int{18, 20}},
{Field: &person.Money, Opt: builder.Between, Val: []float64{100.1, 200.9}}, {Field: &person.Money, Opt: builder.Between, Val: []float64{100.1, 200.9}},
@@ -401,7 +414,7 @@ func testWhere(driver string, db *sql.DB) {
func testJoin(driver string, db *sql.DB) { func testJoin(driver string, db *sql.DB) {
var list2 []ArticleVO var list2 []ArticleVO
err := aorm.Use(db).Debug(true).Driver(driver). err := aorm.Use(db).Debug(false).Driver(driver).
Table(&article). Table(&article).
LeftJoin( LeftJoin(
&person, &person,
@@ -421,7 +434,7 @@ func testJoin(driver string, db *sql.DB) {
func testJoinWithAlias(driver string, db *sql.DB) { func testJoinWithAlias(driver string, db *sql.DB) {
var list2 []ArticleVO var list2 []ArticleVO
err := aorm.Use(db).Debug(true).Driver(driver). err := aorm.Use(db).Debug(false).Driver(driver).
Table(&article, "o"). Table(&article, "o").
LeftJoin( LeftJoin(
&person, &person,
@@ -444,7 +457,7 @@ func testGroupBy(driver string, db *sql.DB) {
var personAgeItem PersonAge var personAgeItem PersonAge
var where []builder.WhereItem var where []builder.WhereItem
where = append(where, builder.WhereItem{Field: "type", Opt: builder.Eq, Val: 0}) where = append(where, builder.WhereItem{Field: "type", Opt: builder.Eq, Val: 0})
err := aorm.Use(db).Debug(true). err := aorm.Use(db).Debug(false).
Table(&person). Table(&person).
Select(&person.Age). Select(&person.Age).
SelectCount(&person.Age, &personAge.AgeCount). SelectCount(&person.Age, &personAge.AgeCount).
@@ -461,7 +474,7 @@ func testGroupBy(driver string, db *sql.DB) {
func testHaving(driver string, db *sql.DB) { func testHaving(driver string, db *sql.DB) {
var listByHaving []PersonAge var listByHaving []PersonAge
err := aorm.Use(db).Debug(true).Driver(driver). err := aorm.Use(db).Debug(false).Driver(driver).
Table(&person). Table(&person).
Select(&person.Age). Select(&person.Age).
SelectCount(&person.Age, &personAge.AgeCount). SelectCount(&person.Age, &personAge.AgeCount).
@@ -477,7 +490,7 @@ func testHaving(driver string, db *sql.DB) {
func testOrderBy(driver string, db *sql.DB) { func testOrderBy(driver string, db *sql.DB) {
var listByOrder []Person var listByOrder []Person
err := aorm.Use(db).Debug(true).Driver(driver). err := aorm.Use(db).Debug(false).Driver(driver).
Table(&person). Table(&person).
WhereEq(&person.Type, 0). WhereEq(&person.Type, 0).
OrderBy(&person.Age, builder.Desc). OrderBy(&person.Age, builder.Desc).
@@ -487,7 +500,7 @@ func testOrderBy(driver string, db *sql.DB) {
} }
var listByOrder2 []Person var listByOrder2 []Person
err2 := aorm.Use(db).Debug(true).Driver(driver). err2 := aorm.Use(db).Debug(false).Driver(driver).
Table(&person, "o"). Table(&person, "o").
WhereEq(&person.Type, 0, "o"). WhereEq(&person.Type, 0, "o").
OrderBy(&person.Age, builder.Desc, "o"). OrderBy(&person.Age, builder.Desc, "o").
@@ -499,7 +512,7 @@ func testOrderBy(driver string, db *sql.DB) {
func testLimit(driver string, db *sql.DB) { func testLimit(driver string, db *sql.DB) {
var list3 []Person var list3 []Person
err1 := aorm.Use(db).Debug(true). err1 := aorm.Use(db).Debug(false).
Table(&person). Table(&person).
WhereEq(&person.Type, 0). WhereEq(&person.Type, 0).
Limit(50, 10). Limit(50, 10).
@@ -511,7 +524,7 @@ func testLimit(driver string, db *sql.DB) {
} }
var list4 []Person var list4 []Person
err := aorm.Use(db).Debug(true). err := aorm.Use(db).Debug(false).
Driver(driver). Driver(driver).
Table(&person). Table(&person).
WhereEq(&person.Type, 0). WhereEq(&person.Type, 0).
@@ -529,7 +542,7 @@ func testLock(driver string, db *sql.DB, id int64) {
} }
var itemByLock Person var itemByLock Person
err := aorm.Use(db). err := aorm.Use(db).
Debug(true). Debug(false).
LockForUpdate(true). LockForUpdate(true).
Table(&person). Table(&person).
WhereEq(&person.Id, id). WhereEq(&person.Id, id).
@@ -542,14 +555,14 @@ func testLock(driver string, db *sql.DB, id int64) {
} }
func testIncrement(driver string, db *sql.DB, id int64) { 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) _, err := aorm.Use(db).Debug(false).Driver(driver).Table(&person).WhereEq(&person.Id, id).Increment(&person.Age, 1)
if err != nil { if err != nil {
panic(driver + " testIncrement " + "found err:" + err.Error()) panic(driver + " testIncrement " + "found err:" + err.Error())
} }
} }
func testDecrement(driver string, db *sql.DB, id int64) { func testDecrement(driver string, db *sql.DB, id int64) {
_, err := aorm.Use(db).Debug(true).Driver(driver).Table(&person).WhereEq(&person.Id, id).Decrement(&person.Age, 2) _, err := aorm.Use(db).Debug(false).Driver(driver).Table(&person).WhereEq(&person.Id, id).Decrement(&person.Age, 2)
if err != nil { if err != nil {
panic(driver + "testDecrement" + "found err") panic(driver + "testDecrement" + "found err")
} }
@@ -558,25 +571,25 @@ func testDecrement(driver string, db *sql.DB, id int64) {
func testValue(driver string, db *sql.DB, id int64) { func testValue(driver string, db *sql.DB, id int64) {
var name string var name string
errName := aorm.Use(db).Debug(true).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Id, id).Value(&person.Name, &name) errName := aorm.Use(db).Debug(false).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Id, id).Value(&person.Name, &name)
if errName != nil { if errName != nil {
panic(driver + "testValue" + "found err") panic(driver + "testValue" + "found err")
} }
var age int64 var age int64
errAge := aorm.Use(db).Debug(true).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Id, id).Value(&person.Age, &age) errAge := aorm.Use(db).Debug(false).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Id, id).Value(&person.Age, &age)
if errAge != nil { if errAge != nil {
panic(driver + "testValue" + "found err") panic(driver + "testValue" + "found err")
} }
var money float32 var money float32
errMoney := aorm.Use(db).Debug(true).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Id, id).Value(&person.Money, &money) errMoney := aorm.Use(db).Debug(false).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Id, id).Value(&person.Money, &money)
if errMoney != nil { if errMoney != nil {
panic(driver + "testValue" + "found err") panic(driver + "testValue" + "found err")
} }
var test float64 var test float64
errTest := aorm.Use(db).Debug(true).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Id, id).Value(&person.Test, &test) errTest := aorm.Use(db).Debug(false).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Id, id).Value(&person.Test, &test)
if errTest != nil { if errTest != nil {
panic(driver + "testValue" + "found err") panic(driver + "testValue" + "found err")
} }
@@ -584,60 +597,60 @@ func testValue(driver string, db *sql.DB, id int64) {
func testPluck(driver string, db *sql.DB) { func testPluck(driver string, db *sql.DB) {
var nameList []string var nameList []string
errNameList := aorm.Use(db).Debug(true).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Type, 0).Limit(0, 3).Pluck(&person.Name, &nameList) errNameList := aorm.Use(db).Debug(false).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Type, 0).Limit(0, 3).Pluck(&person.Name, &nameList)
if errNameList != nil { if errNameList != nil {
panic(driver + "testPluck" + "found err") panic(driver + "testPluck" + "found err")
} }
var ageList []int64 var ageList []int64
errAgeList := aorm.Use(db).Debug(true).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Type, 0).Limit(0, 3).Pluck(&person.Age, &ageList) errAgeList := aorm.Use(db).Debug(false).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Type, 0).Limit(0, 3).Pluck(&person.Age, &ageList)
if errAgeList != nil { if errAgeList != nil {
panic(driver + "testPluck" + "found err:" + errAgeList.Error()) panic(driver + "testPluck" + "found err:" + errAgeList.Error())
} }
var moneyList []float32 var moneyList []float32
errMoneyList := aorm.Use(db).Debug(true).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Type, 0).Limit(0, 3).Pluck(&person.Money, &moneyList) errMoneyList := aorm.Use(db).Debug(false).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Type, 0).Limit(0, 3).Pluck(&person.Money, &moneyList)
if errMoneyList != nil { if errMoneyList != nil {
panic(driver + "testPluck" + "found err") panic(driver + "testPluck" + "found err")
} }
var testList []float64 var testList []float64
errTestList := aorm.Use(db).Debug(true).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Type, 0).Limit(0, 3).Pluck(&person.Test, &testList) errTestList := aorm.Use(db).Debug(false).Driver(driver).Table(&person).OrderBy(&person.Id, builder.Desc).WhereEq(&person.Type, 0).Limit(0, 3).Pluck(&person.Test, &testList)
if errTestList != nil { if errTestList != nil {
panic(driver + "testPluck" + "found err") panic(driver + "testPluck" + "found err")
} }
} }
func testCount(driver string, db *sql.DB) { func testCount(driver string, db *sql.DB) {
_, err := aorm.Use(db).Debug(true).Table(&person).WhereEq(&person.Age, 18).Driver(driver).Count("*") _, err := aorm.Use(db).Debug(false).Table(&person).WhereEq(&person.Age, 18).Driver(driver).Count("*")
if err != nil { if err != nil {
panic(driver + "testCount" + "found err") panic(driver + "testCount" + "found err")
} }
} }
func testSum(driver string, db *sql.DB) { func testSum(driver string, db *sql.DB) {
_, err := aorm.Use(db).Debug(true).Table(&person).WhereEq(&person.Age, 18).Driver(driver).Sum(&person.Age) _, err := aorm.Use(db).Debug(false).Table(&person).WhereEq(&person.Age, 18).Driver(driver).Sum(&person.Age)
if err != nil { if err != nil {
panic(driver + "testSum" + "found err") panic(driver + "testSum" + "found err")
} }
} }
func testAvg(driver string, db *sql.DB) { func testAvg(driver string, db *sql.DB) {
_, err := aorm.Use(db).Debug(true).Table(&person).WhereEq(&person.Age, 18).Driver(driver).Avg(&person.Age) _, err := aorm.Use(db).Debug(false).Table(&person).WhereEq(&person.Age, 18).Driver(driver).Avg(&person.Age)
if err != nil { if err != nil {
panic(driver + "testAvg" + "found err") panic(driver + "testAvg" + "found err")
} }
} }
func testMin(driver string, db *sql.DB) { func testMin(driver string, db *sql.DB) {
_, err := aorm.Use(db).Debug(true).Table(&person).WhereEq(&person.Age, 18).Driver(driver).Min(&person.Age) _, err := aorm.Use(db).Debug(false).Table(&person).WhereEq(&person.Age, 18).Driver(driver).Min(&person.Age)
if err != nil { if err != nil {
panic(driver + "testMin" + "found err") panic(driver + "testMin" + "found err")
} }
} }
func testMax(driver string, db *sql.DB) { func testMax(driver string, db *sql.DB) {
_, err := aorm.Use(db).Debug(true).Table(&person).WhereEq(&person.Age, 18).Driver(driver).Max(&person.Age) _, err := aorm.Use(db).Debug(false).Table(&person).WhereEq(&person.Age, 18).Driver(driver).Max(&person.Age)
if err != nil { if err != nil {
panic(driver + "testMax" + "found err") panic(driver + "testMax" + "found err")
} }
@@ -645,14 +658,14 @@ func testMax(driver string, db *sql.DB) {
func testDistinct(driver string, db *sql.DB) { func testDistinct(driver string, db *sql.DB) {
var listByFiled []Person var listByFiled []Person
err := aorm.Use(db).Debug(true).Driver(driver).Distinct(true).Table(&person).Select(&person.Name).Select(&person.Age).WhereEq(&person.Age, 18).GetMany(&listByFiled) err := aorm.Use(db).Debug(false).Driver(driver).Distinct(true).Table(&person).Select(&person.Name).Select(&person.Age).WhereEq(&person.Age, 18).GetMany(&listByFiled)
if err != nil { if err != nil {
panic(driver + " testSelect " + "found err:" + err.Error()) panic(driver + " testSelect " + "found err:" + err.Error())
} }
} }
func testExec(driver string, db *sql.DB) { func testExec(driver string, db *sql.DB) {
_, err := aorm.Use(db).Debug(true).Driver(driver).Exec("UPDATE person SET name = ? WHERE person.id=?", "Bob", 3) _, err := aorm.Use(db).Debug(false).Driver(driver).Exec("UPDATE person SET name = ? WHERE person.id=?", "Bob", 3)
if err != nil { if err != nil {
panic(driver + "testExec" + "found err") panic(driver + "testExec" + "found err")
} }
@@ -661,7 +674,7 @@ func testExec(driver string, db *sql.DB) {
func testTransaction(driver string, db *sql.DB) { func testTransaction(driver string, db *sql.DB) {
tx, _ := db.Begin() tx, _ := db.Begin()
id, errInsert := aorm.Use(tx).Debug(true).Driver(driver).Insert(&Person{ id, errInsert := aorm.Use(tx).Debug(false).Driver(driver).Insert(&Person{
Name: null.StringFrom("Alice"), Name: null.StringFrom("Alice"),
}) })
@@ -671,7 +684,7 @@ func testTransaction(driver string, db *sql.DB) {
return return
} }
_, errCount := aorm.Use(tx).Debug(true).Driver(driver).Table(&person).WhereEq(&person.Id, id).Count("*") _, errCount := aorm.Use(tx).Debug(false).Driver(driver).Table(&person).WhereEq(&person.Id, id).Count("*")
if errCount != nil { if errCount != nil {
tx.Rollback() tx.Rollback()
panic(driver + "testTransaction" + "found err") panic(driver + "testTransaction" + "found err")
@@ -679,14 +692,14 @@ func testTransaction(driver string, db *sql.DB) {
} }
var personItem Person var personItem Person
errPerson := aorm.Use(tx).Debug(true).Driver(driver).Table(&person).WhereEq(&person.Id, id).OrderBy(&person.Id, builder.Desc).GetOne(&personItem) errPerson := aorm.Use(tx).Debug(false).Driver(driver).Table(&person).WhereEq(&person.Id, id).OrderBy(&person.Id, builder.Desc).GetOne(&personItem)
if errPerson != nil { if errPerson != nil {
tx.Rollback() tx.Rollback()
panic(driver + "testTransaction" + "found err") panic(driver + "testTransaction" + "found err")
return return
} }
_, errUpdate := aorm.Use(tx).Debug(true).Driver(driver).Where(&Person{ _, errUpdate := aorm.Use(tx).Debug(false).Driver(driver).Where(&Person{
Id: null.IntFrom(id), Id: null.IntFrom(id),
}).Update(&Person{ }).Update(&Person{
Name: null.StringFrom("Bob"), Name: null.StringFrom("Bob"),
@@ -702,7 +715,7 @@ func testTransaction(driver string, db *sql.DB) {
} }
func testTruncate(driver string, db *sql.DB) { func testTruncate(driver string, db *sql.DB) {
_, err := aorm.Use(db).Debug(true).Driver(driver).Table(&person).Truncate() _, err := aorm.Use(db).Debug(false).Driver(driver).Table(&person).Truncate()
if err != nil { if err != nil {
panic(driver + " testTruncate " + "found err") panic(driver + " testTruncate " + "found err")
} }