123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678 |
- // Copyright 2019 The Xorm Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style
- // license that can be found in the LICENSE file.
- package core
- import (
- "errors"
- "flag"
- "os"
- "testing"
- "time"
- _ "github.com/go-sql-driver/mysql"
- _ "github.com/mattn/go-sqlite3"
- )
- var (
- dbtype = flag.String("dbtype", "mysql", "database type")
- dbConn = flag.String("dbConn", "root:@/core_test?charset=utf8", "database connect string")
- createTableSql string
- )
- type User struct {
- Id int64
- Name string
- Title string
- Age float32
- Alias string
- NickName string
- Created NullTime
- }
- func init() {
- flag.Parse()
- switch *dbtype {
- case "sqlite3":
- createTableSql = "CREATE TABLE IF NOT EXISTS `user` (`id` INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL, `name` TEXT NULL, " +
- "`title` TEXT NULL, `age` FLOAT NULL, `alias` TEXT NULL, `nick_name` TEXT NULL, `created` datetime);"
- case "mysql":
- createTableSql = "CREATE TABLE IF NOT EXISTS `user` (`id` INTEGER PRIMARY KEY AUTO_INCREMENT NOT NULL, `name` TEXT NULL, " +
- "`title` TEXT NULL, `age` FLOAT NULL, `alias` TEXT NULL, `nick_name` TEXT NULL, `created` datetime);"
- default:
- panic("no db type")
- }
- }
- func testOpen() (*DB, error) {
- switch *dbtype {
- case "sqlite3":
- os.Remove("./test.db")
- return Open("sqlite3", "./test.db")
- case "mysql":
- return Open("mysql", *dbConn)
- default:
- panic("no db type")
- }
- }
- func BenchmarkOriQuery(b *testing.B) {
- b.StopTimer()
- db, err := testOpen()
- if err != nil {
- b.Error(err)
- }
- defer db.Close()
- _, err = db.Exec(createTableSql)
- if err != nil {
- b.Error(err)
- }
- for i := 0; i < 50; i++ {
- _, err = db.Exec("insert into user (`name`, title, age, alias, nick_name, created) values (?,?,?,?,?, ?)",
- "xlw", "tester", 1.2, "lunny", "lunny xiao", time.Now())
- if err != nil {
- b.Error(err)
- }
- }
- b.StartTimer()
- for i := 0; i < b.N; i++ {
- rows, err := db.Query("select * from user")
- if err != nil {
- b.Error(err)
- }
- for rows.Next() {
- var Id int64
- var Name, Title, Alias, NickName string
- var Age float32
- var Created NullTime
- err = rows.Scan(&Id, &Name, &Title, &Age, &Alias, &NickName, &Created)
- if err != nil {
- b.Error(err)
- }
- //fmt.Println(Id, Name, Title, Age, Alias, NickName)
- }
- rows.Close()
- }
- }
- func BenchmarkStructQuery(b *testing.B) {
- b.StopTimer()
- db, err := testOpen()
- if err != nil {
- b.Error(err)
- }
- defer db.Close()
- _, err = db.Exec(createTableSql)
- if err != nil {
- b.Error(err)
- }
- for i := 0; i < 50; i++ {
- _, err = db.Exec("insert into user (`name`, title, age, alias, nick_name, created) values (?,?,?,?,?, ?)",
- "xlw", "tester", 1.2, "lunny", "lunny xiao", time.Now())
- if err != nil {
- b.Error(err)
- }
- }
- b.StartTimer()
- for i := 0; i < b.N; i++ {
- rows, err := db.Query("select * from user")
- if err != nil {
- b.Error(err)
- }
- for rows.Next() {
- var user User
- err = rows.ScanStructByIndex(&user)
- if err != nil {
- b.Error(err)
- }
- if user.Name != "xlw" {
- b.Log(user)
- b.Error(errors.New("name should be xlw"))
- }
- }
- rows.Close()
- }
- }
- func BenchmarkStruct2Query(b *testing.B) {
- b.StopTimer()
- db, err := testOpen()
- if err != nil {
- b.Error(err)
- }
- defer db.Close()
- _, err = db.Exec(createTableSql)
- if err != nil {
- b.Error(err)
- }
- for i := 0; i < 50; i++ {
- _, err = db.Exec("insert into user (`name`, title, age, alias, nick_name, created) values (?,?,?,?,?,?)",
- "xlw", "tester", 1.2, "lunny", "lunny xiao", time.Now())
- if err != nil {
- b.Error(err)
- }
- }
- db.Mapper = NewCacheMapper(&SnakeMapper{})
- b.StartTimer()
- for i := 0; i < b.N; i++ {
- rows, err := db.Query("select * from user")
- if err != nil {
- b.Error(err)
- }
- for rows.Next() {
- var user User
- err = rows.ScanStructByName(&user)
- if err != nil {
- b.Error(err)
- }
- if user.Name != "xlw" {
- b.Log(user)
- b.Error(errors.New("name should be xlw"))
- }
- }
- rows.Close()
- }
- }
- func BenchmarkSliceInterfaceQuery(b *testing.B) {
- b.StopTimer()
- db, err := testOpen()
- if err != nil {
- b.Error(err)
- }
- defer db.Close()
- _, err = db.Exec(createTableSql)
- if err != nil {
- b.Error(err)
- }
- for i := 0; i < 50; i++ {
- _, err = db.Exec("insert into user (`name`, title, age, alias, nick_name,created) values (?,?,?,?,?,?)",
- "xlw", "tester", 1.2, "lunny", "lunny xiao", time.Now())
- if err != nil {
- b.Error(err)
- }
- }
- b.StartTimer()
- for i := 0; i < b.N; i++ {
- rows, err := db.Query("select * from user")
- if err != nil {
- b.Error(err)
- }
- cols, err := rows.Columns()
- if err != nil {
- b.Error(err)
- }
- for rows.Next() {
- slice := make([]interface{}, len(cols))
- err = rows.ScanSlice(&slice)
- if err != nil {
- b.Error(err)
- }
- b.Log(slice)
- switch slice[1].(type) {
- case *string:
- if *slice[1].(*string) != "xlw" {
- b.Error(errors.New("name should be xlw"))
- }
- case []byte:
- if string(slice[1].([]byte)) != "xlw" {
- b.Error(errors.New("name should be xlw"))
- }
- }
- }
- rows.Close()
- }
- }
- /*func BenchmarkSliceBytesQuery(b *testing.B) {
- b.StopTimer()
- os.Remove("./test.db")
- db, err := Open("sqlite3", "./test.db")
- if err != nil {
- b.Error(err)
- }
- defer db.Close()
- _, err = db.Exec(createTableSql)
- if err != nil {
- b.Error(err)
- }
- for i := 0; i < 50; i++ {
- _, err = db.Exec("insert into user (name, title, age, alias, nick_name,created) values (?,?,?,?,?,?)",
- "xlw", "tester", 1.2, "lunny", "lunny xiao", time.Now())
- if err != nil {
- b.Error(err)
- }
- }
- b.StartTimer()
- for i := 0; i < b.N; i++ {
- rows, err := db.Query("select * from user")
- if err != nil {
- b.Error(err)
- }
- cols, err := rows.Columns()
- if err != nil {
- b.Error(err)
- }
- for rows.Next() {
- slice := make([][]byte, len(cols))
- err = rows.ScanSlice(&slice)
- if err != nil {
- b.Error(err)
- }
- if string(slice[1]) != "xlw" {
- fmt.Println(slice)
- b.Error(errors.New("name should be xlw"))
- }
- }
- rows.Close()
- }
- }
- */
- func BenchmarkSliceStringQuery(b *testing.B) {
- b.StopTimer()
- db, err := testOpen()
- if err != nil {
- b.Error(err)
- }
- defer db.Close()
- _, err = db.Exec(createTableSql)
- if err != nil {
- b.Error(err)
- }
- for i := 0; i < 50; i++ {
- _, err = db.Exec("insert into user (name, title, age, alias, nick_name, created) values (?,?,?,?,?,?)",
- "xlw", "tester", 1.2, "lunny", "lunny xiao", time.Now())
- if err != nil {
- b.Error(err)
- }
- }
- b.StartTimer()
- for i := 0; i < b.N; i++ {
- rows, err := db.Query("select * from user")
- if err != nil {
- b.Error(err)
- }
- cols, err := rows.Columns()
- if err != nil {
- b.Error(err)
- }
- for rows.Next() {
- slice := make([]*string, len(cols))
- err = rows.ScanSlice(&slice)
- if err != nil {
- b.Error(err)
- }
- if (*slice[1]) != "xlw" {
- b.Log(slice)
- b.Error(errors.New("name should be xlw"))
- }
- }
- rows.Close()
- }
- }
- func BenchmarkMapInterfaceQuery(b *testing.B) {
- b.StopTimer()
- db, err := testOpen()
- if err != nil {
- b.Error(err)
- }
- defer db.Close()
- _, err = db.Exec(createTableSql)
- if err != nil {
- b.Error(err)
- }
- for i := 0; i < 50; i++ {
- _, err = db.Exec("insert into user (name, title, age, alias, nick_name,created) values (?,?,?,?,?,?)",
- "xlw", "tester", 1.2, "lunny", "lunny xiao", time.Now())
- if err != nil {
- b.Error(err)
- }
- }
- b.StartTimer()
- for i := 0; i < b.N; i++ {
- rows, err := db.Query("select * from user")
- if err != nil {
- b.Error(err)
- }
- for rows.Next() {
- m := make(map[string]interface{})
- err = rows.ScanMap(&m)
- if err != nil {
- b.Error(err)
- }
- switch m["name"].(type) {
- case string:
- if m["name"].(string) != "xlw" {
- b.Log(m)
- b.Error(errors.New("name should be xlw"))
- }
- case []byte:
- if string(m["name"].([]byte)) != "xlw" {
- b.Log(m)
- b.Error(errors.New("name should be xlw"))
- }
- }
- }
- rows.Close()
- }
- }
- /*func BenchmarkMapBytesQuery(b *testing.B) {
- b.StopTimer()
- os.Remove("./test.db")
- db, err := Open("sqlite3", "./test.db")
- if err != nil {
- b.Error(err)
- }
- defer db.Close()
- _, err = db.Exec(createTableSql)
- if err != nil {
- b.Error(err)
- }
- for i := 0; i < 50; i++ {
- _, err = db.Exec("insert into user (name, title, age, alias, nick_name,created) values (?,?,?,?,?,?)",
- "xlw", "tester", 1.2, "lunny", "lunny xiao", time.Now())
- if err != nil {
- b.Error(err)
- }
- }
- b.StartTimer()
- for i := 0; i < b.N; i++ {
- rows, err := db.Query("select * from user")
- if err != nil {
- b.Error(err)
- }
- for rows.Next() {
- m := make(map[string][]byte)
- err = rows.ScanMap(&m)
- if err != nil {
- b.Error(err)
- }
- if string(m["name"]) != "xlw" {
- fmt.Println(m)
- b.Error(errors.New("name should be xlw"))
- }
- }
- rows.Close()
- }
- }
- */
- /*
- func BenchmarkMapStringQuery(b *testing.B) {
- b.StopTimer()
- os.Remove("./test.db")
- db, err := Open("sqlite3", "./test.db")
- if err != nil {
- b.Error(err)
- }
- defer db.Close()
- _, err = db.Exec(createTableSql)
- if err != nil {
- b.Error(err)
- }
- for i := 0; i < 50; i++ {
- _, err = db.Exec("insert into user (name, title, age, alias, nick_name,created) values (?,?,?,?,?,?)",
- "xlw", "tester", 1.2, "lunny", "lunny xiao", time.Now())
- if err != nil {
- b.Error(err)
- }
- }
- b.StartTimer()
- for i := 0; i < b.N; i++ {
- rows, err := db.Query("select * from user")
- if err != nil {
- b.Error(err)
- }
- for rows.Next() {
- m := make(map[string]string)
- err = rows.ScanMap(&m)
- if err != nil {
- b.Error(err)
- }
- if m["name"] != "xlw" {
- fmt.Println(m)
- b.Error(errors.New("name should be xlw"))
- }
- }
- rows.Close()
- }
- }*/
- func BenchmarkExec(b *testing.B) {
- b.StopTimer()
- db, err := testOpen()
- if err != nil {
- b.Error(err)
- }
- defer db.Close()
- _, err = db.Exec(createTableSql)
- if err != nil {
- b.Error(err)
- }
- b.StartTimer()
- for i := 0; i < b.N; i++ {
- _, err = db.Exec("insert into user (`name`, title, age, alias, nick_name,created) values (?,?,?,?,?,?)",
- "xlw", "tester", 1.2, "lunny", "lunny xiao", time.Now())
- if err != nil {
- b.Error(err)
- }
- }
- }
- func BenchmarkExecMap(b *testing.B) {
- b.StopTimer()
- db, err := testOpen()
- if err != nil {
- b.Error(err)
- }
- defer db.Close()
- _, err = db.Exec(createTableSql)
- if err != nil {
- b.Error(err)
- }
- b.StartTimer()
- mp := map[string]interface{}{
- "name": "xlw",
- "title": "tester",
- "age": 1.2,
- "alias": "lunny",
- "nick_name": "lunny xiao",
- "created": time.Now(),
- }
- for i := 0; i < b.N; i++ {
- _, err = db.ExecMap("insert into user (`name`, title, age, alias, nick_name, created) "+
- "values (?name,?title,?age,?alias,?nick_name,?created)",
- &mp)
- if err != nil {
- b.Error(err)
- }
- }
- }
- func TestExecMap(t *testing.T) {
- db, err := testOpen()
- if err != nil {
- t.Error(err)
- }
- defer db.Close()
- _, err = db.Exec(createTableSql)
- if err != nil {
- t.Error(err)
- }
- mp := map[string]interface{}{
- "name": "xlw",
- "title": "tester",
- "age": 1.2,
- "alias": "lunny",
- "nick_name": "lunny xiao",
- "created": time.Now(),
- }
- _, err = db.ExecMap("insert into user (`name`, title, age, alias, nick_name,created) "+
- "values (?name,?title,?age,?alias,?nick_name,?created)",
- &mp)
- if err != nil {
- t.Error(err)
- }
- rows, err := db.Query("select * from user")
- if err != nil {
- t.Error(err)
- }
- for rows.Next() {
- var user User
- err = rows.ScanStructByName(&user)
- if err != nil {
- t.Error(err)
- }
- t.Log("--", user)
- }
- }
- func TestExecStruct(t *testing.T) {
- db, err := testOpen()
- if err != nil {
- t.Error(err)
- }
- defer db.Close()
- _, err = db.Exec(createTableSql)
- if err != nil {
- t.Error(err)
- }
- user := User{Name: "xlw",
- Title: "tester",
- Age: 1.2,
- Alias: "lunny",
- NickName: "lunny xiao",
- Created: NullTime(time.Now()),
- }
- _, err = db.ExecStruct("insert into user (`name`, title, age, alias, nick_name,created) "+
- "values (?Name,?Title,?Age,?Alias,?NickName,?Created)",
- &user)
- if err != nil {
- t.Error(err)
- }
- rows, err := db.QueryStruct("select * from user where `name` = ?Name", &user)
- if err != nil {
- t.Error(err)
- }
- for rows.Next() {
- var user User
- err = rows.ScanStructByName(&user)
- if err != nil {
- t.Error(err)
- }
- t.Log("1--", user)
- }
- }
- func BenchmarkExecStruct(b *testing.B) {
- b.StopTimer()
- db, err := testOpen()
- if err != nil {
- b.Error(err)
- }
- defer db.Close()
- _, err = db.Exec(createTableSql)
- if err != nil {
- b.Error(err)
- }
- b.StartTimer()
- user := User{Name: "xlw",
- Title: "tester",
- Age: 1.2,
- Alias: "lunny",
- NickName: "lunny xiao",
- Created: NullTime(time.Now()),
- }
- for i := 0; i < b.N; i++ {
- _, err = db.ExecStruct("insert into user (`name`, title, age, alias, nick_name,created) "+
- "values (?Name,?Title,?Age,?Alias,?NickName,?Created)",
- &user)
- if err != nil {
- b.Error(err)
- }
- }
- }
|