rony-chat-bot/bench/bench_modernc_test.go

package bench

import (
	"context"
	"database/sql"
	"path/filepath"
	"testing"

	_ "modernc.org/sqlite"
)

const dataPath = "../data/projects"

func openModernc(b *testing.B) *sql.DB {
	b.Helper()
	tmpDir := b.TempDir()
	dsn := filepath.Join(tmpDir, "bench.db")
	db, err := sql.Open("sqlite", dsn+"?_pragma=journal_mode(WAL)&_pragma=synchronous(NORMAL)")
	if err != nil {
		b.Fatal(err)
	}
	db.SetMaxOpenConns(1) // fair comparison: serialize writes
	if _, err := db.ExecContext(context.Background(), schema); err != nil {
		b.Fatal(err)
	}
	b.Cleanup(func() { _ = db.Close() })
	return db
}

func BenchmarkInsert_Modernc(b *testing.B) {
	rows := loadChunks(b, dataPath)
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		db := openModernc(b)
		if err := insertAll(context.Background(), db, rows); err != nil {
			b.Fatal(err)
		}
	}
}

func BenchmarkQuery_Modernc(b *testing.B) {
	db := openModernc(b)
	rows := loadChunks(b, dataPath)
	if err := insertAll(context.Background(), db, rows); err != nil {
		b.Fatal(err)
	}
	ctx := context.Background()
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		for _, q := range queries {
			if _, _, err := runQuery(ctx, db, q); err != nil {
				b.Fatal(err)
			}
		}
	}
}

func BenchmarkRoundTrip_Modernc(b *testing.B) {
	rows := loadChunks(b, dataPath)
	ctx := context.Background()
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		db := openModernc(b)
		if err := insertAll(ctx, db, rows); err != nil {
			b.Fatal(err)
		}
		for _, q := range queries {
			if _, _, err := runQuery(ctx, db, q); err != nil {
				b.Fatal(err)
			}
		}
	}
}