gorm_cockroach_test.go.txt

package gorm_vs_korm

import (
	"testing"

	"github.com/kokizzu/gotro/S"
	"github.com/sourcegraph/conc/pool"
	"github.com/zeebo/assert"
	"gorm.io/gorm"
)

var gormCockroach *gorm.DB

func BenchmarkInsertS_Cockroach_Gorm(b *testing.B) {
	if done() {
		b.SkipNow()
		return
	}
	defer timing()()
	err := gormCockroach.Exec(`TRUNCATE TABLE ` + gormTestTableName).Error
	assert.Nil(b, err)
	b.N = total
	p := pool.New().WithMaxGoroutines(cores)
	for z := uint64(1); z <= total; z++ {
		z := z
		p.Go(func() {
			err := gormCockroach.Create(&GormTestTable{
				ID:      z,
				Content: S.EncodeCB63(z, 0),
			}).Error
			assert.Nil(b, err)
		})
	}
	p.Wait()
}

func BenchmarkGetAllS_Cockroach_Gorm(b *testing.B) {
	p := pool.New().WithMaxGoroutines(cores)
	for i := uint64(1); i <= uint64(b.N); i++ {
		p.Go(func() {
			a := []GormTestTable{}
			err := gormCockroach.Limit(limit).Find(&a).Error
			assert.Nil(b, err)
		})
	}
	p.Wait()
}

func BenchmarkGetAllM_Cockroach_Gorm(b *testing.B) {
	a := []map[string]any{}
	p := pool.New().WithMaxGoroutines(cores)
	for i := uint64(1); i <= uint64(b.N); i++ {
		p.Go(func() {
			err := gormCockroach.Limit(limit).Find(&GormTestTable{}).Scan(&a).Error
			assert.Nil(b, err)
		})
	}
	p.Wait()
}

func BenchmarkGetRowS_Cockroach_Gorm(b *testing.B) {
	p := pool.New().WithMaxGoroutines(cores)
	for i := uint64(1); i <= uint64(b.N); i++ {
		i := i
		p.Go(func() {
			u := GormTestTable{}
			err := gormCockroach.Where(&GormTestTable{
				Content: S.EncodeCB63(1+uint64(i)%total, 0),
			}).First(&u).Error
			assert.Nil(b, err)
		})
	}
	p.Wait()
}

func BenchmarkGetRowM_Cockroach_Gorm(b *testing.B) {
	p := pool.New().WithMaxGoroutines(cores)
	for i := uint64(1); i <= uint64(b.N); i++ {
		i := i
		p.Go(func() {
			u := map[string]any{}
			err := gormCockroach.Model(&GormTestTable{}).Where(&GormTestTable{
				Content: S.EncodeCB63(1+uint64(i)%total, 0),
			}).First(&u).Error
			assert.Nil(b, err)
		})
	}
	p.Wait()
}