import (
	"database/sql"
	"errors"
	"fmt"

	_ "modernc.org/sqlite"
)

func openCourseDB(path string) *sql.DB {
	db, err := sql.Open("sqlite", path)
	if err != nil {
		panic(err)
	}
	db.SetMaxOpenConns(1)
	return db
}

const dbPath = "file:_02_5_transactions_indexes_solutions.db"

%%
// users テーブルをシードする（練習問題が使うテーブルのみ再掲。本編のaccountsは解答には不要）。
db := openCourseDB(dbPath)
defer db.Close()

db.Exec(`DROP TABLE IF EXISTS users`)
db.Exec(`CREATE TABLE users (id INTEGER PRIMARY KEY, email TEXT)`)

tx, err := db.Begin()
if err != nil {
	panic(err)
}
for i := 0; i < 20000; i++ {
	if _, err := tx.Exec(`INSERT INTO users (email) VALUES (?)`, fmt.Sprintf("user%d@example.com", i)); err != nil {
		panic(err)
	}
}
if err := tx.Commit(); err != nil {
	panic(err)
}
db.Exec(`CREATE INDEX IF NOT EXISTS idx_users_email ON users(email)`)

var count int
db.QueryRow(`SELECT COUNT(*) FROM users`).Scan(&count)
fmt.Println("✅ users をシードしました。行数:", count)

✅ users をシードしました。行数: 20000

func countUsersByEmailDomain(db *sql.DB, domain string) (int, error) {
	var n int
	err := db.QueryRow(`SELECT COUNT(*) FROM users WHERE email LIKE ?`, "%@"+domain).Scan(&n)
	if err != nil {
		return 0, err
	}
	return n, nil
}

import "reflect"

func mustEqual(got, want any, name string) {
	if reflect.DeepEqual(got, want) {
		fmt.Printf("✅ Passed: %s\n", name)
		return
	}
	panic(fmt.Sprintf("❌ %s\n  got  = %v (%T)\n  want = %v (%T)", name, got, got, want, want))
}

%%
// このチェックは SELECT のみで DB を変更しないため、何度実行しても同じ結果になる。
db := openCourseDB(dbPath)
defer db.Close()

n1, err1 := countUsersByEmailDomain(db, "example.com")
mustEqual(err1, nil, "example.com はエラーなし")
mustEqual(n1, 20000, "全ユーザーが example.com ドメイン")

n2, err2 := countUsersByEmailDomain(db, "not-exist.example")
mustEqual(err2, nil, "存在しないドメインはエラーではない")
mustEqual(n2, 0, "存在しないドメインは0件")

fmt.Println("🎉 すべてのチェックが通りました")

✅ Passed: example.com はエラーなし
✅ Passed: 全ユーザーが example.com ドメイン

✅ Passed: 存在しないドメインはエラーではない
✅ Passed: 存在しないドメインは0件
🎉 すべてのチェックが通りました

解答 02.5 — トランザクションとインデックス（countUsersByEmailDomain）¶

前提コード（問題ノートブックと同じ定義・シード）¶

解答: countUsersByEmailDomain¶

チェック（問題ノートブックと同じ期待値）¶

解説¶