Golang连接MySQL数据库 database/sql使用指南

database/sql

database/sql

github.com/go-sql-driver/mysql

package main

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

    _ "github.com/go-sql-driver/mysql" // 引入MySQL驱动，注意前面的下划线
)

func main() {
    // DSN (Data Source Name) 连接字符串
    // 格式：user:password@tcp(host:port)/dbname?charset=utf8mb4&parseTime=True&loc=Local
    dsn := "root:password@tcp(127.0.0.1:3306)/testdb?charset=utf8mb4&parseTime=True&loc=Local"

    // 打开数据库连接
    db, err := sql.Open("mysql", dsn)
    if err != nil {
        // sql.Open 不会立即建立连接，只是验证DSN格式
        log.Fatalf("打开数据库连接失败: %v", err)
    }
    defer db.Close() // 确保在程序退出前关闭数据库连接

    // 尝试ping数据库，验证连接是否真正建立
    err = db.Ping()
    if err != nil {
        log.Fatalf("无法连接到数据库: %v", err)
    }
    fmt.Println("成功连接到MySQL数据库！")

    // 设置连接池参数，这非常重要！
    // 最大打开的连接数
    db.SetMaxOpenConns(100)
    // 最大空闲连接数
    db.SetMaxIdleConns(10)
    // 连接可重用的最长时间
    db.SetConnMaxLifetime(5 * time.Minute)

    // 接下来就可以执行SQL操作了，比如查询
    var version string
    err = db.QueryRow("SELECT VERSION()").Scan(&version)
    if err != nil {
        log.Fatalf("查询数据库版本失败: %v", err)
    }
    fmt.Printf("MySQL 版本: %s\n", version)

    // 插入数据示例
    // 这里演示了如何使用预处理语句插入数据，这是推荐的做法，可以防止SQL注入
    stmt, err := db.Prepare("INSERT INTO users(name, email) VALUES(?, ?)")
    if err != nil {
        log.Fatalf("准备插入语句失败: %v", err)
    }
    defer stmt.Close() // 确保关闭预处理语句

    res, err := stmt.Exec("张三", "zhangsan@example.com")
    if err != nil {
        log.Fatalf("插入数据失败: %v", err)
    }
    lastId, _ := res.LastInsertId()
    rowsAffected, _ := res.RowsAffected()
    fmt.Printf("插入成功，ID: %d，影响行数: %d\n", lastId, rowsAffected)

    // 查询数据示例
    rows, err := db.Query("SELECT id, name, email FROM users")
    if err != nil {
        log.Fatalf("查询数据失败: %v", err)
    }
    defer rows.Close() // 确保关闭结果集

    for rows.Next() {
        var id int
        var name, email string
        if err := rows.Scan(&id, &name, &email); err != nil {
            log.Printf("扫描行失败: %v", err)
            continue
        }
        fmt.Printf("ID: %d, Name: %s, Email: %s\n", id, name, email)
    }
    if err = rows.Err(); err != nil {
        log.Fatalf("遍历结果集出错: %v", err)
    }
}

// 假设的 users 表结构 (MySQL):
/*
CREATE TABLE `users` (
  `id` int NOT NULL AUTO_INCREMENT,
  `name` varchar(255) NOT NULL,
  `email` varchar(255) NOT NULL UNIQUE,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;
*/

user:password@tcp(host:port)/dbname?param1=value1&param2=value2

timeout

readTimeout

writeTimeout

database/sql

sql.Open

db.Close

database/sql

db.Query

db.Exec

database/sql

INSERT INTO users(name, email) VALUES(?, ?)

// 这是一个危险的例子，切勿在生产环境使用！
name := "张三"
email := "zhangsan@example.com'; DROP TABLE users; --" // 恶意输入
sql := fmt.Sprintf("INSERT INTO users(name, email) VALUES('%s', '%s')", name, email)
// 此时的sql会变成：INSERT INTO users(name, email) VALUES('张三', 'zhangsan@example.com'; DROP TABLE users; --')
// 数据库执行时，会把'; DROP TABLE users; --'当作SQL语句的一部分执行！

stmt, err := db.Prepare("INSERT INTO users(name, email) VALUES(?, ?)")
// ... 错误处理
defer stmt.Close()

// 这里的 email 变量即使包含恶意的SQL代码，也会被当作普通字符串数据处理
res, err := stmt.Exec("张三", "zhangsan@example.com'; DROP TABLE users; --")
// ... 错误处理

?

zhangsan@example.com'; DROP TABLE users; --