bench_test.go

// SPDX-FileCopyrightText: 2014-2024 caixw
//
// SPDX-License-Identifier: MIT

package orm_test

import (
	"database/sql"
	"testing"
	"time"

	"github.com/issue9/assert/v4"

	"github.com/issue9/orm/v6/internal/test"
	"github.com/issue9/orm/v6/sqlbuilder"
)

func BenchmarkDB_Insert(b *testing.B) {
	a := assert.New(b, false)

	m := &Group{
		Name:    "name",
		Created: time.Now(),
		Any:     5,
	}

	suite := test.NewSuite(a, "")

	suite.Run(func(t *test.Driver) {
		t.NotError(t.DB.Create(&Group{}))
		defer func() {
			t.NotError(t.DB.Drop(&Group{}))
		}()

		for i := 0; i < b.N; i++ {
			_, err := t.DB.Insert(m)
			t.NotError(err)
		}
	})
}

func BenchmarkDB_Update(b *testing.B) {
	a := assert.New(b, false)

	m := &Group{
		Name:    "name",
		Created: time.Now(),
		Any:     5,
	}

	suite := test.NewSuite(a, "")

	suite.Run(func(t *test.Driver) {
		t.NotError(t.DB.Create(&Group{}))
		defer func() {
			t.NotError(t.DB.Drop(&Group{}))
		}()

		// 构造数据
		for i := 0; i < 10000; i++ {
			_, err := t.DB.Insert(m)
			t.NotError(err)
		}

		m.ID = sql.NullInt64{Int64: 1, Valid: true} // 自增，从 1 开始
		for i := 0; i < b.N; i++ {
			_, err := t.DB.Update(m)
			t.NotError(err)
		}
	})
}

func BenchmarkDB_Select(b *testing.B) {
	a := assert.New(b, false)

	m := &Group{
		Name:    "name",
		Created: time.Now(),
		Any:     5,
	}

	suite := test.NewSuite(a, "")

	suite.Run(func(t *test.Driver) {
		t.NotError(t.DB.Create(&Group{}))
		defer func() {
			t.NotError(t.DB.Drop(&Group{}))
		}()

		_, err := t.DB.Insert(m)
		t.NotError(err)

		m.ID = sql.NullInt64{Int64: 1, Valid: true}
		for i := 0; i < b.N; i++ {
			found, err := t.DB.Select(m)
			t.NotError(err).True(found)
		}
	})
}

func BenchmarkDB_WhereUpdate(b *testing.B) {
	a := assert.New(b, false)

	m := &Group{
		Name:    "name",
		Created: time.Now(),
		Any:     5,
	}

	suite := test.NewSuite(a, "")

	suite.Run(func(t *test.Driver) {
		t.NotError(t.DB.Create(&Group{}))
		defer func() {
			t.NotError(t.DB.Drop(&Group{}))
		}()

		// 构造数据
		for i := 0; i < 10000; i++ {
			_, err := t.DB.Insert(m)
			t.NotError(err)
		}

		for i := 0; i < b.N; i++ {
			_, err := sqlbuilder.
				Update(t.DB).Table("{groups}").
				Set("name", "n1").
				Increase("created", 1).
				Where("{id}=?", i+1).
				Exec()
			t.NotError(err)
		}
	})
}