Go语言在App Engine中连接Google Cloud SQL的实践指南

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

    // 导入 go-sql-driver/mysql 驱动。
    // 注意：下划线表示我们只导入包以执行其 init() 函数，而不直接使用其导出的任何标识符。
    _ "github.com/go-sql-driver/mysql"

    // 导入 appengine/cloudsql 包。
    // 在 App Engine 环境中，这会确保驱动能够通过 Unix socket 连接。
    // 如果在本地开发，此包可能不需要，但为了兼容性，通常会包含。
    _ "google.golang.org/appengine/cloudsql"
)

func getDSN() string {
    // 生产环境（App Engine）DSN
    if os.Getenv("GAE_APPLICATION") != "" {
        // 从环境变量获取 Cloud SQL 连接信息
        dbUser := os.Getenv("DB_USER")
        dbPass := os.Getenv("DB_PASS")
        dbName := os.Getenv("DB_NAME")
        instanceConnectionName := os.Getenv("INSTANCE_CONNECTION_NAME") // 格式: PROJECT_ID:REGION:INSTANCE_NAME

        if dbUser == "" || dbPass == "" || dbName == "" || instanceConnectionName == "" {
            log.Fatalf("Missing Cloud SQL environment variables for App Engine.")
        }
        // 使用 Unix socket 连接
        return fmt.Sprintf("%s:%s@unix(/cloudsql/%s)/%s?parseTime=true", dbUser, dbPass, instanceConnectionName, dbName)
    }

    // 本地开发环境DSN (例如，通过 Cloud SQL Proxy 或本地 MySQL)
    // 请根据你的本地设置进行调整
    dbUser := os.Getenv("LOCAL_DB_USER")
    dbPass := os.Getenv("LOCAL_DB_PASS")
    dbHost := os.Getenv("LOCAL_DB_HOST") // 通常是 127.0.0.1
    dbPort := os.Getenv("LOCAL_DB_PORT") // 通常是 3306
    dbName := os.Getenv("LOCAL_DB_NAME")

    if dbUser == "" || dbPass == "" || dbHost == "" || dbPort == "" || dbName == "" {
        log.Fatalf("Missing local database environment variables.")
    }
    return fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?parseTime=true", dbUser, dbPass, dbHost, dbPort, dbName)
}

func connectToCloudSQL() (*sql.DB, error) {
    dsn := getDSN()
    db, err := sql.Open("mysql", dsn)
    if err != nil {
        return nil, fmt.Errorf("无法打开数据库连接: %w", err)
    }

    // 尝试 Ping 数据库以验证连接
    if err = db.Ping(); err != nil {
        db.Close() // 如果 Ping 失败，关闭连接
        return nil, fmt.Errorf("无法连接到数据库: %w", err)
    }

    // 配置连接池
    db.SetMaxOpenConns(10) // 最大打开连接数
    db.SetMaxIdleConns(5)  // 最大空闲连接数
    // db.SetConnMaxLifetime(5 * time.Minute) // 连接最大生命周期

    log.Println("成功连接到 Cloud SQL 数据库！")
    return db, nil
}

type User struct {
    ID   int
    Name string
    Email string
}

func main() {
    db, err := connectToCloudSQL()
    if err != nil {
        log.Fatalf("连接数据库失败: %v", err)
    }
    defer db.Close() // 确保在函数结束时关闭数据库连接

    // 示例：创建表 (如果不存在)
    _, err = db.Exec(`
        CREATE TABLE IF NOT EXISTS users (
            id INT AUTO_INCREMENT PRIMARY KEY,
            name VARCHAR(255) NOT NULL,
            email VARCHAR(255) UNIQUE NOT NULL
        );
    `)
    if err != nil {
        log.Printf("创建表失败: %v", err)
        // 在生产环境中，表通常由迁移工具创建，这里只是示例
    } else {
        log.Println("表 'users' 检查或创建成功。")
    }

    // 示例：插入数据
    res, err := db.Exec("INSERT INTO users (name, email) VALUES (?, ?)", "Alice", "alice@example.com")
    if err != nil {
        log.Printf("插入数据失败: %v", err)
    } else {
        id, _ := res.LastInsertId()
        log.Printf("插入用户 Alice 成功，ID: %d", id)
    }

    // 示例：查询数据
    rows, err := db.Query("SELECT id, name, email FROM users WHERE name = ?", "Alice")
    if err != nil {
        log.Fatalf("查询数据失败: %v", err)
    }
    defer rows.Close()

    var users []User
    for rows.Next() {
        var u User
        if err := rows.Scan(&u.ID, &u.Name, &u.Email); err != nil {
            log.Fatalf("扫描行失败: %v", err)
        }
        users = append(users, u)
    }
    if err = rows.Err(); err != nil {
        log.Fatalf("遍历行错误: %v", err)
    }

    for _, u := range users {
        log.Printf("查询到用户: ID=%d, Name=%s, Email=%s", u.ID, u.Name, u.Email)
    }
}