import (
	"database/sql"
	"fmt"

	_ "modernc.org/sqlite"
)

// openCourseDB はこのノートブック用の SQLite 接続を開く。
// GoNB のセルは1つの関数として実行されるため、db は各セル内で開いて同じセル内で閉じる。
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_2_sql_crud_solutions.db"

// seedInventory はテーブルを消して作り直し、初期データを入れる（問題ノートブックと同じ内容）。
// 何度実行しても同じ初期状態に戻る（冪等）。
func seedInventory() {
	db := openCourseDB(dbPath)
	defer db.Close()

	db.Exec("DROP TABLE IF EXISTS inventory")
	db.Exec(`CREATE TABLE inventory (
		id INTEGER PRIMARY KEY,
		item TEXT UNIQUE,
		stock INTEGER
	)`)
	db.Exec(`INSERT INTO inventory (item, stock) VALUES
		('apple', 10), ('banana', 5), ('cherry', 20)`)
}

// decreaseStock は item の在庫を amount だけ減らす。
// 在庫が amount 未満ならエラーを返し、在庫は変更しない。
// 成功したら更新後の在庫数を返す。
func decreaseStock(db *sql.DB, item string, amount int) (int, error) {
	var stock int
	if err := db.QueryRow("SELECT stock FROM inventory WHERE item = ?", item).Scan(&stock); err != nil {
		return 0, err
	}
	if stock < amount {
		return 0, fmt.Errorf("insufficient stock: item=%s, stock=%d, amount=%d", item, stock, amount)
	}
	newStock := stock - amount
	if _, err := db.Exec("UPDATE inventory SET stock = ? WHERE item = ?", newStock, item); err != nil {
		return 0, err
	}
	return newStock, 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))
}

%%
// チェックの前に必ず初期状態へ再シードする（このセルを何度実行しても同じ結果になるように）。
seedInventory()

db := openCourseDB(dbPath)
defer db.Close()

n1, err1 := decreaseStock(db, "banana", 3)
mustEqual(err1, nil, "decreaseStock(banana, 3) はエラーなし")
mustEqual(n1, 2, "banana の在庫が 5 → 2")

_, err2 := decreaseStock(db, "banana", 100)
mustEqual(err2 != nil, true, "在庫不足(100)はエラー")

var stockAfterErr int
db.QueryRow("SELECT stock FROM inventory WHERE item = 'banana'").Scan(&stockAfterErr)
mustEqual(stockAfterErr, 2, "エラー時は在庫が変わらない（2のまま）")

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

✅ Passed: decreaseStock(banana, 3) はエラーなし
✅ Passed: banana の在庫が 5 → 2

✅ Passed: 在庫不足(100)はエラー
✅ Passed: エラー時は在庫が変わらない（2のまま）
🎉 すべてのチェックが通りました

解答 02.2 — SQL の CRUD操作（decreaseStock）¶

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

解答: decreaseStock¶

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

解説¶