Go语言中利用构建约束实现App Engine与标准SQL环境的条件编译

// dbconnector.go
package mylib

import (
    "database/sql"
    "context" // 用于App Engine上下文
)

// DBConnector 定义了一个通用的数据库连接器接口
type DBConnector interface {
    // Connect 方法用于建立数据库连接。
    // 在App Engine环境下，可能需要传入appengine.Context。
    // 为保持通用性，我们在此处传入context.Context。
    Connect(ctx context.Context) (*sql.DB, error)
    // Close 方法用于关闭数据库连接。
    Close(*sql.DB) error
    // ... 更多通用的数据库操作方法可以根据需要添加
}

// NewDBConnector 返回一个适合当前环境的数据库连接器实例。
// 具体的实现将由带有构建约束的文件提供。
func NewDBConnector() DBConnector {
    return newDBConnectorImpl()
}

// newDBConnectorImpl 是一个内部函数，它的具体实现将由
// cloudsql_appengine.go 和 cloudsql_standalone.go 提供。
// 这样上层调用者只需调用 NewDBConnector。
func newDBConnectorImpl() DBConnector

// +build appengine

package mylib

import (
    "context"
    "database/sql"
    "log"

    // 注意：对于较新的GAE SDK和Go版本，通常使用google.golang.org/appengine
    // 如果你的项目使用旧版SDK，可能需要将导入路径改为 "appengine"
    "google.golang.org/appengine"
    "google.golang.org/appengine/cloudsql"
)

// appEngineDBConnector 是针对App Engine环境的实现
type appEngineDBConnector struct{}

// newDBConnectorImpl 的App Engine版本
func newDBConnectorImpl() DBConnector {
    log.Println("Initializing App Engine Cloud SQL connector.")
    return &appEngineDBConnector{}
}

// Connect 方法在App Engine环境下使用cloudsql.Dial
func (c *appEngineDBConnector) Connect(ctx context.Context) (*sql.DB, error) {
    // 确保传入的上下文是App Engine上下文
    appCtx := appengine.NewContext(ctx)

    // 替换为你的Cloud SQL实例连接名，例如 "project-id:region:instance-name"
    // 这个连接名通常在app.yaml或环境变量中配置
    const cloudSQLConnectionName = "your_cloud_sql_connection_name"

    db, err := cloudsql.Dial(appCtx, cloudSQLConnectionName)
    if err != nil {
        log.Printf("Failed to connect to Cloud SQL: %v", err)
        return nil, err
    }
    return db, nil
}

// Close 方法关闭数据库连接
func (c *appEngineDBConnector) Close(db *sql.DB) error {
    if db != nil {
        return db.Close()
    }
    return nil
}

// +build !appengine

package mylib

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

    _ "github.com/go-sql-driver/mysql" // 导入MySQL驱动，注意下划线表示只导入包进行初始化
)

// standaloneDBConnector 是针对标准环境的实现
type standaloneDBConnector struct{}

// newDBConnectorImpl 的标准Go环境版本
func newDBConnectorImpl() DBConnector {
    log.Println("Initializing standalone MySQL connector.")
    return &standaloneDBConnector{}
}

// Connect 方法在标准Go环境下使用database/sql和MySQL驱动
func (c *standaloneDBConnector) Connect(ctx context.Context) (*sql.DB, error) {
    // 从环境变量或配置文件中获取数据库连接字符串
    // 示例：DSN="user:password@tcp(127.0.0.1:3306)/dbname?parseTime=true"
    dsn := os.Getenv("DATABASE_DSN")
    if dsn == "" {
        // 提供一个默认值或报错
        log.Println("DATABASE_DSN environment variable not set. Using default local MySQL DSN.")
        dsn = "root:password@tcp(127.0.0.1:3306)/testdb?parseTime=true"
    }

    db, err := sql.Open("mysql", dsn)
    if err != nil {
        log.Printf("Failed to open database connection: %v", err)
        return nil, err
    }

    // 尝试ping数据库以确保连接有效
    if err = db.PingContext(ctx); err != nil {
        db.Close() // ping失败，关闭连接
        log.Printf("Failed to connect to database: %v", err)
        return nil, err
    }

    return db, nil
}

// Close 方法关闭数据库连接
func (c *standaloneDBConnector) Close(db *sql.DB) error {
    if db != nil {
        return db.Close()
    }
    return nil
}

package main

import (
    "context"
    "fmt"
    "log"
    "mylib" // 假设你的库名为mylib
)

func main() {
    // 创建一个上下文，在App Engine环境下它会是appengine.Context
    // 在标准环境下它就是普通的context.Background()
    ctx := context.Background()

    connector := mylib.NewDBConnector()
    db, err := connector.Connect(ctx)
    if err != nil {
        log.Fatalf("Error connecting to database: %v", err)
    }
    defer func() {
        if closeErr := connector.Close(db); closeErr != nil {
            log.Printf("Error closing database connection: %v", closeErr)
        }
    }()

    fmt.Println("Successfully connected to the database!")

    // 可以在这里执行数据库查询等操作
    // 例如：
    // var version string
    // err = db.QueryRowContext(ctx, "SELECT VERSION()").Scan(&version)
    // if err != nil {
    //     log.Fatalf("Query error: %v", err)
    // }
    // fmt.Printf("Database Version: %s\n", version)
}