Go语言连接Google Cloud SQL：最新实践与驱动选择

package main

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

    // 导入MySQL驱动。通常推荐go-sql-driver/mysql
    // 注意：这里使用下划线_导入，表示只导入其副作用（注册驱动），而不直接使用其导出的任何标识符。
    _ "github.com/go-sql-driver/mysql"

    // 如果在App Engine标准环境，可能需要导入并注册cloudsql驱动
    // import "google.golang.org/appengine/cloudsql"
    // cloudsql.RegisterDriver()
)

// 适用于通过Cloud SQL Proxy连接
// 假设Cloud SQL Proxy在本地监听3306端口
dbUser := "your_db_user"         // 替换为你的数据库用户名
dbPassword := "your_db_password" // 替换为你的数据库密码
dbHost := "127.0.0.1"            // Cloud SQL Proxy默认监听地址
dbPort := "3306"                 // Cloud SQL Proxy默认监听端口
dbName := "your_database_name"   // 替换为你的数据库名

// 注意：`parseTime=true` 参数很重要，它会将MySQL的DATE、DATETIME类型自动转换为Go的time.Time类型。
connectionStringTCP := fmt.Sprintf(
    "%s:%s@tcp(%s:%s)/%s?parseTime=true",
    dbUser, dbPassword, dbHost, dbPort, dbName,
)

// 适用于App Engine标准环境的Unix socket连接
// 格式：user:password@unix(/cloudsql/PROJECT_ID:REGION:INSTANCE_NAME)/dbname
cloudSQLConnectionName := "your-project-id:your-region:your-instance-name" // 替换为你的Cloud SQL连接名
dbUser := "your_db_user"
dbPassword := "your_db_password"
dbName := "your_database_name"

connectionStringUnix := fmt.Sprintf(
    "%s:%s@unix(/cloudsql/%s)/%s?parseTime=true",
    dbUser, dbPassword, cloudSQLConnectionName, dbName,
)

func main() {
    // 根据部署环境选择合适的连接字符串
    // 生产环境通常从环境变量获取敏感信息
    var connectionString string

    // 示例：根据环境变量判断使用哪种连接方式
    if os.Getenv("GAE_APPLICATION") != "" {
        // 在App Engine环境，使用Unix socket
        cloudSQLConnectionName := os.Getenv("CLOUD_SQL_CONNECTION_NAME") // 例如: "project-id:region:instance-name"
        dbUser := os.Getenv("DB_USER")
        dbPassword := os.Getenv("DB_PASSWORD")
        dbName := os.Getenv("DB_NAME")
        if cloudSQLConnectionName == "" || dbUser == "" || dbPassword == "" || dbName == "" {
            log.Fatal("App Engine Cloud SQL connection environment variables not set.")
        }
        connectionString = fmt.Sprintf(
            "%s:%s@unix(/cloudsql/%s)/%s?parseTime=true",
            dbUser, dbPassword, cloudSQLConnectionName, dbName,
        )
        fmt.Println("Using App Engine Unix socket connection.")
    } else {
        // 本地开发或非App Engine环境，使用Cloud SQL Proxy (TCP)
        dbUser := os.Getenv("DB_USER")         // 建议从环境变量获取
        dbPassword := os.Getenv("DB_PASSWORD") // 建议从环境变量获取
        dbHost := os.Getenv("DB_HOST")         // 例如: "127.0.0.1"
        dbPort := os.Getenv("DB_PORT")         // 例如: "3306"
        dbName := os.Getenv("DB_NAME")         // 建议从环境变量获取

        // 提供默认值，方便本地测试
        if dbUser == "" { dbUser = "root" }
        if dbPassword == "" { dbPassword = "your_strong_password" } // **请务必替换为你的实际密码或从环境变量获取**
        if dbHost == "" { dbHost = "127.0.0.1" }
        if dbPort == "" { dbPort = "3306" }
        if dbName == "" { dbName = "testdb" }

        connectionString = fmt.Sprintf(
            "%s:%s@tcp(%s:%s)/%s?parseTime=true",
            dbUser, dbPassword, dbHost, dbPort, dbName,
        )
        fmt.Println("Using TCP connection (likely via Cloud SQL Proxy).")
    }

    // 打开数据库连接
    db, err := sql.Open("mysql", connectionString)
    if err != nil {
        log.Fatalf("无法打开数据库连接: %v", err)
    }
    defer db.Close() // 确保在函数结束时关闭数据库连接

    // 验证连接是否成功
    err = db.Ping()
    if err != nil {
        log.Fatalf("无法连接到数据库: %v", err)
    }
    fmt.Println("成功连接到Google Cloud SQL！")

    // 示例：执行一个简单的查询
    var version string
    err = db.QueryRow("SELECT VERSION()").Scan(&version)
    if err != nil {
        log.Fatalf("查询失败: %v", err)
    }
    fmt.Printf("MySQL 版本: %s\n", version)

    // 更多数据库操作示例：
    // 创建表
    _, 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,
            created_at DATETIME DEFAULT CURRENT_TIMESTAMP
        );
    `)
    if err != nil {
        log.Fatalf("创建表失败: %v", err)
    }
    fmt.Println("表 'users' 检查或创建成功。")

    // 插入数据
    result, err := db.Exec("INSERT INTO users(name, email) VALUES(?, ?)", "Alice", "alice@example.com")
    if err != nil {
        log.Printf("插入数据失败 (可能已存在): %v", err)
    } else {
        id, _ := result.LastInsertId()
        fmt.Printf("插入用户 Alice, ID: %d\n", id)
    }

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

    fmt.Println("\n所有用户:")
    for rows.Next() {
        var id int
        var name, email string
        var createdAt sql.NullTime // 使用sql.NullTime处理可空时间字段
        if err := rows.Scan(&id, &name, &email, &createdAt); err != nil {
            log.Fatalf("扫描行失败: %v", err)
        }
        fmt.Printf("ID: %d, Name: %s, Email: %s, CreatedAt: %v\n", id, name, email, createdAt.Time)
    }
    if err := rows.Err(); err != nil {
        log.Fatalf("遍历行错误: %v", err)
    }
}