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

package main

import (
    "database/sql"
    "fmt"
    "log"
    "net/http"
    "os" // 用于获取环境变量

    // 引入Go App Engine的上下文包
    "google.golang.org/appengine"
    // 引入Go App Engine的Cloud SQL包，它提供了Unix域套接字的支持
    _ "google.golang.org/appengine/cloudsql" 
    // 引入MySQL驱动，注意下划线导入表示仅为副作用（注册驱动）
    _ "github.com/go-sql-driver/mysql" 
)

// 在App Engine环境中，通常通过环境变量获取数据库配置
func getDSN(r *http.Request) string {
    dbUser := os.Getenv("DB_USER")
    dbPass := os.Getenv("DB_PASS")
    dbName := os.Getenv("DB_NAME")

    // 从App Engine上下文获取项目ID、区域和实例名称
    // 注意：在App Engine标准环境中，实例连接名通常是 PROJECT_ID:REGION:INSTANCE_NAME
    // 并且可以直接通过 /cloudsql/ 路径访问
    instanceConnectionName := os.Getenv("CLOUD_SQL_CONNECTION_NAME") // 推荐通过环境变量设置

    if dbUser == "" || dbPass == "" || dbName == "" || instanceConnectionName == "" {
        log.Printf("Warning: Missing database credentials or instance name in environment variables.")
        // 提供一个示例DSN，但实际应用中应避免硬编码
        return fmt.Sprintf("root:password@unix(/cloudsql/%s)/mydb?charset=utf8", "your-project-id:your-region:your-instance-name")
    }

    return fmt.Sprintf("%s:%s@unix(/cloudsql/%s)/%s?charset=utf8",
        dbUser, dbPass, instanceConnectionName, dbName)
}

env_variables:
  DB_USER: "your_db_user"
  DB_PASS: "your_db_password"
  DB_NAME: "your_database_name"
  CLOUD_SQL_CONNECTION_NAME: "your-project-id:your-region:your-instance-name"

// db 变量应在包级别声明，以便在整个应用生命周期中重用连接池
var db *sql.DB

func init() {
    // 在init函数中初始化db连接，但由于App Engine的请求上下文，
    // 更好的做法是在每次请求处理时获取上下文，并在处理函数内部初始化或管理连接。
    // 对于App Engine标准环境，通常在请求处理函数中打开连接，或者使用单例模式管理。
    // 这里为了演示，我们假设在请求处理函数中完成。
    http.HandleFunc("/", handler)
}

func handler(w http.ResponseWriter, r *http.Request) {
    ctx := appengine.NewContext(r) // 获取App Engine上下文

    // 如果db连接池尚未初始化，或者需要重新初始化
    if db == nil {
        dsn := getDSN(r)
        var err error
        db, err = sql.Open("mysql", dsn)
        if err != nil {
            log.Printf(ctx, "Error opening database connection: %v", err)
            http.Error(w, fmt.Sprintf("Error opening database connection: %v", err), http.StatusInternalServerError)
            return
        }

        // 配置连接池参数 (可选，但推荐)
        db.SetMaxOpenConns(10) // 最大打开连接数
        db.SetMaxIdleConns(5)  // 最大空闲连接数
    }

    // 验证连接是否有效
    if err := db.Ping(); err != nil {
        log.Printf(ctx, "Error pinging database: %v", err)
        http.Error(w, fmt.Sprintf("Error pinging database: %v", err), http.StatusInternalServerError)
        return
    }

    // 数据库操作示例：查询当前时间
    var currentTime string
    err := db.QueryRow("SELECT NOW()").Scan(&currentTime)
    if err != nil {
        log.Printf(ctx, "Error querying database: %v", err)
        http.Error(w, fmt.Sprintf("Error querying database: %v", err), http.StatusInternalServerError)
        return
    }

    fmt.Fprintf(w, "Successfully connected to Cloud SQL! Current time: %s", currentTime)
}

// 插入数据示例
func insertData(db *sql.DB, name string, age int) error {
    stmt, err := db.Prepare("INSERT INTO users (name, age) VALUES (?, ?)")
    if err != nil {
        return fmt.Errorf("prepare statement failed: %w", err)
    }
    defer stmt.Close()

    _, err = stmt.Exec(name, age)
    if err != nil {
        return fmt.Errorf("execute statement failed: %w", err)
    }
    return nil
}

// 查询数据示例
type User struct {
    ID   int
    Name string
    Age  int
}

func queryUsers(db *sql.DB) ([]User, error) {
    rows, err := db.Query("SELECT id, name, age FROM users")
    if err != nil {
        return nil, fmt.Errorf("query failed: %w", err)
    }
    defer rows.Close()

    var users []User
    for rows.Next() {
        var u User
        if err := rows.Scan(&u.ID, &u.Name, &u.Age); err != nil {
            return nil, fmt.Errorf("scan row failed: %w", err)
        }
        users = append(users, u)
    }
    if err := rows.Err(); err != nil {
        return nil, fmt.Errorf("rows iteration error: %w", err)
    }
    return users, nil
}