Go语言跨环境数据库连接：利用构建约束优雅处理App Engine与标准SQL

cloud.go:20:2: cannot find package "appengine/cloudsql" in any of:
    /usr/local/Cellar/go/1.1.2/src/pkg/appengine/cloudsql (from $GOROOT)
    /Users/lameduck/myGo/src/appengine/cloudsql (from $GOPATH)

// +build appengine

package dbconn

import (
    "database/sql"
    // 导入appengine/cloudsql包，它在GAE环境中可用
    // 注意：实际使用时，通常会通过sql.Open("cloudsql", instanceName) 来连接
    // 此处直接导入是为了满足包查找的需求，但其内部实现细节可能依赖于GAE的上下文
    _ "appengine/cloudsql" 
)

// GetDBConnection 返回一个适合App Engine环境的数据库连接。
// instanceName 通常是Cloud SQL实例的连接名称，例如 "project-id:region:instance-name"。
func GetDBConnection(instanceName string) (*sql.DB, error) {
    // 在App Engine环境中，使用"cloudsql"驱动
    // 详细的连接字符串格式请参考Google Cloud SQL文档
    db, err := sql.Open("cloudsql", instanceName)
    if err != nil {
        return nil, err
    }
    // 可选：设置连接池参数
    // db.SetMaxOpenConns(maxOpenConns)
    // db.SetMaxIdleConns(maxIdleConns)
    // db.SetConnMaxLifetime(connMaxLifetime)
    return db, nil
}

// +build !appengine

package dbconn

import (
    "database/sql"
    // 导入标准的MySQL驱动，例如 go-sql-driver/mysql
    _ "github.com/go-sql-driver/mysql" 
)

// GetDBConnection 返回一个适合标准Go环境的数据库连接。
// dataSourceName 是标准的DSN（Data Source Name），例如 "user:password@tcp(127.0.0.1:3306)/dbname"。
func GetDBConnection(dataSourceName string) (*sql.DB, error) {
    // 在标准Go环境中，使用"mysql"驱动
    db, err := sql.Open("mysql", dataSourceName)
    if err != nil {
        return nil, err
    }
    // 可选：设置连接池参数
    // db.SetMaxOpenConns(maxOpenConns)
    // db.SetMaxIdleConns(maxIdleConns)
    // db.SetConnMaxLifetime(connMaxLifetime)
    return db, nil
}

package main

import (
    "log"
    "myproject/dbconn" // 假设你的dbconn包路径是 myproject/dbconn
    "os"
)

func main() {
    var dbIdentifier string
    // 根据环境变量或其他配置判断是GAE还是标准环境，并提供相应的连接标识符
    // 在实际应用中，这通常通过配置服务或环境变量来管理
    if os.Getenv("GAE_APPLICATION") != "" { // 简单判断是否在GAE环境
        // GAE环境下，使用Cloud SQL实例连接名称
        dbIdentifier = "your-project-id:your-region:your-instance-name" 
        log.Println("Detected App Engine environment. Using Cloud SQL instance name.")
    } else {
        // 标准环境下，使用DSN
        dbIdentifier = "user:password@tcp(127.0.0.1:3306)/dbname?parseTime=true"
        log.Println("Detected standard environment. Using standard DSN.")
    }

    db, err := dbconn.GetDBConnection(dbIdentifier)
    if err != nil {
        log.Fatalf("Failed to connect to database: %v", err)
    }
    defer db.Close()

    log.Println("Successfully connected to the database.")

    // 执行数据库操作示例
    // var version string
    // err = db.QueryRow("SELECT VERSION()").Scan(&version)
    // if err != nil {
    //  log.Fatalf("Failed to query database version: %v", err)
    // }
    // log.Printf("Database version: %s", version)
}