package main

import (
	"database/sql"
	"fmt"
	"log"
	"time"

	_ "github.com/go-sql-driver/mysql" // 导入驱动，但不在代码中直接使用，所以用 _ 别名
)

// User 结构体用于映射数据库表记录
type User struct {
	ID        int
	Username  string
	Email     string
	CreatedAt time.Time
}

func main() {
	// 1. 连接数据库
	// DSN (Data Source Name) 格式: 用户名:密码@tcp(主机:端口)/数据库名?参数
	dsn := "root:your_password@tcp(127.0.0.1:3306)/testdb?parseTime=true"
	db, err := sql.Open("mysql", dsn)
	if err != nil {
		log.Fatal("连接数据库失败: ", err)
	}
	defer db.Close() // 确保程序结束时关闭数据库连接池

	// 2. 测试连接
	err = db.Ping()
	if err != nil {
		log.Fatal("Ping数据库失败: ", err)
	}
	fmt.Println("成功连接到MySQL数据库!")

	// 3. 配置连接池（可选但推荐）
	db.SetMaxOpenConns(25)                 // 最大打开连接数
	db.SetMaxIdleConns(25)                 // 最大空闲连接数
	db.SetConnMaxLifetime(5 * time.Minute) // 连接最大生命周期

	// 4. 创建表 (使用Exec)
	createTableSQL := `CREATE TABLE IF NOT EXISTS users (
		id INT AUTO_INCREMENT PRIMARY KEY,
		username VARCHAR(50) NOT NULL UNIQUE,
		email VARCHAR(100) NOT NULL,
		created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
	) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;`

	_, err = db.Exec(createTableSQL)
	if err != nil {
		log.Fatal("创建表失败: ", err)
	}
	fmt.Println("users 表创建成功或已存在。")

	// 5. 插入数据 (使用预编译语句防止SQL注入)
	insertSQL := "INSERT INTO users (username, email) VALUES (?, ?)"
	stmt, err := db.Prepare(insertSQL) // 预编译语句
	if err != nil {
		log.Fatal("准备插入语句失败: ", err)
	}
	defer stmt.Close() // 关闭预编译语句

	// 执行插入
	result, err := stmt.Exec("zhangsan", "[email protected]")
	if err != nil {
		log.Fatal("插入数据失败: ", err)
	}
	id, _ := result.LastInsertId()
	fmt.Printf("插入成功，ID: %d\n", id)

	// 批量插入另一个用户
	_, err = stmt.Exec("lisi", "[email protected]")
	if err != nil {
		log.Fatal("插入数据失败: ", err)
	}
	fmt.Println("批量插入lisi成功。")

	// 6. 查询单行数据 (QueryRow)
	var user User
	queryRowSQL := "SELECT id, username, email, created_at FROM users WHERE username = ?"
	err = db.QueryRow(queryRowSQL, "zhangsan").Scan(
		&user.ID, &user.Username, &user.Email, &user.CreatedAt,
	)
	if err != nil {
		if err == sql.ErrNoRows {
			fmt.Println("未找到用户zhangsan")
		} else {
			log.Fatal("查询单行数据失败: ", err)
		}
	} else {
		fmt.Printf("查询结果: %+v\n", user)
	}

	// 7. 查询多行数据 (Query)
	queryMultiSQL := "SELECT id, username, email FROM users ORDER BY id"
	rows, err := db.Query(queryMultiSQL)
	if err != nil {
		log.Fatal("查询多行数据失败: ", err)
	}
	defer rows.Close() // 非常重要！确保Rows关闭以释放连接

	fmt.Println("\n所有用户:")
	for rows.Next() {
		var u User
		err := rows.Scan(&u.ID, &u.Username, &u.Email)
		if err != nil {
			log.Fatal("扫描行数据失败: ", err)
		}
		fmt.Printf("ID: %d, 用户名: %s, 邮箱: %s\n", u.ID, u.Username, u.Email)
	}
	// 检查遍历过程中是否有错误
	if err = rows.Err(); err != nil {
		log.Fatal("遍历行时出错: ", err)
	}

	// 8. 更新数据 (Exec)
	updateSQL := "UPDATE users SET email = ? WHERE username = ?"
	result, err = db.Exec(updateSQL, "[email protected]", "zhangsan")
	if err != nil {
		log.Fatal("更新数据失败: ", err)
	}
	rowsAffected, _ := result.RowsAffected()
	fmt.Printf("更新成功，影响行数: %d\n", rowsAffected)

	// 9. 删除数据 (Exec)
	deleteSQL := "DELETE FROM users WHERE username = ?"
	result, err = db.Exec(deleteSQL, "lisi")
	if err != nil {
		log.Fatal("删除数据失败: ", err)
	}
	rowsAffected, _ = result.RowsAffected()
	fmt.Printf("删除成功，影响行数: %d\n", rowsAffected)

	// 10. 事务操作示例
	fmt.Println("\n开始事务示例...")
	tx, err := db.Begin() // 开始事务
	if err != nil {
		log.Fatal("开始事务失败: ", err)
	}

	// 在事务中执行多个操作
	_, err = tx.Exec("UPDATE users SET email = ? WHERE username = ?", "[email protected]", "zhangsan")
	if err != nil {
		tx.Rollback() // 如果有错误，回滚事务
		log.Fatal("事务中更新失败: ", err)
	}

	// 假设还需要另一个操作
	// _, err = tx.Exec("INSERT INTO logs ...")
	// if err != nil { tx.Rollback(); ... }

	// 提交事务
	err = tx.Commit()
	if err != nil {
		log.Fatal("提交事务失败: ", err)
	}
	fmt.Println("事务提交成功，数据已持久化。")

	fmt.Println("\n所有操作完成!")
}