rel.go

package bench

import (
	"context"
	"github.com/efectn/go-orm-benchmarks/helper"
	"testing"

	"github.com/go-rel/postgres"
	relware "github.com/go-rel/rel"
	"github.com/go-rel/rel/where"
)

type Rel struct {
	helper.ORMInterface
	conn relware.Repository
	db   relware.Adapter
}

func CreateRel() helper.ORMInterface {
	return &Rel{}
}

func (rel *Rel) Name() string {
	return "rel"
}

func (rel *Rel) Init() error {
	var err error
	rel.db, err = postgres.Open(helper.OrmSource)
	if err != nil {
		return err
	}

	rel.conn = relware.New(rel.db)
	if err := rel.conn.Ping(ctx); err != nil {
		return err
	}

	// Disable debug logging
	rel.conn.Instrumentation(func(ctx context.Context, op string, message string, args ...any) func(err error) {
		return func(err error) {
			if err != nil {
				panic(err)
			}
		}
	})

	return nil
}

func (rel *Rel) Close() error {
	return rel.db.Close()
}

func (rel *Rel) Insert(b *testing.B) {
	m := NewModel3()

	b.ReportAllocs()
	b.ResetTimer()

	for i := 0; i < b.N; i++ {
		m.ID = 0
		err := rel.conn.Insert(ctx, m)
		if err != nil {
			helper.SetError(b, rel.Name(), "Insert", err.Error())
		}
	}
}

func (rel *Rel) InsertMulti(b *testing.B) {
	ms := make([]*Model3, 0, 100)
	for i := 0; i < 100; i++ {
		ms = append(ms, NewModel3())
	}

	b.ReportAllocs()
	b.ResetTimer()

	for i := 0; i < b.N; i++ {
		for _, m := range ms {
			m.ID = 0
		}
		err := rel.conn.InsertAll(ctx, &ms)
		if err != nil {
			helper.SetError(b, rel.Name(), "InsertMulti", err.Error())
		}
	}
}

func (rel *Rel) Update(b *testing.B) {
	m := NewModel3()
	m.ID = 0
	err := rel.conn.Insert(ctx, m)
	if err != nil {
		helper.SetError(b, rel.Name(), "Update", err.Error())
	}

	b.ReportAllocs()
	b.ResetTimer()

	for i := 0; i < b.N; i++ {
		err := rel.conn.Update(ctx, m)
		if err != nil {
			helper.SetError(b, rel.Name(), "Update", err.Error())
		}
	}
}

func (rel *Rel) Read(b *testing.B) {
	m := NewModel3()
	m.ID = 0
	err := rel.conn.Insert(ctx, m)
	if err != nil {
		helper.SetError(b, rel.Name(), "Read", err.Error())
	}

	b.ReportAllocs()
	b.ResetTimer()

	for i := 0; i < b.N; i++ {
		err := rel.conn.Find(ctx, m)
		if err != nil {
			helper.SetError(b, rel.Name(), "Read", err.Error())
		}
	}
}

func (rel *Rel) ReadSlice(b *testing.B) {
	m := NewModel3()
	for i := 0; i < 100; i++ {
		m.ID = 0
		err := rel.conn.Insert(ctx, m)
		if err != nil {
			helper.SetError(b, rel.Name(), "ReadSlice", err.Error())
		}
	}

	b.ReportAllocs()
	b.ResetTimer()

	for i := 0; i < b.N; i++ {
		var ms []Model3
		err := rel.conn.FindAll(ctx, &ms, where.Gt("id", 0), relware.Limit(100))
		if err != nil {
			helper.SetError(b, rel.Name(), "ReadSlice", err.Error())
		}
	}
}