使用 Go 连接器连接到 Aurora DSQL 集群

Aurora DSQL Connector for Go 使用自动 IAM 身份验证封装 pgx。此连接器处理令牌生成、SSL 配置和连接管理，因此您可以专注于应用程序逻辑。

关于连接器

Aurora DSQL 要求使用基于 IAM 的身份验证以及限时令牌，而现有 Go PostgreSQL 驱动程序并不原生支持这种方法。适用于 Go 的 Aurora DSQL 连接器在 pgx 驱动程序之上添加一个身份验证层来处理 IAM 令牌生成，使您无需更改现有 pgx 工作流程即可连接到 Aurora DSQL。

什么是 Aurora DSQL 身份验证？

在 Aurora DSQL 中，身份验证包括：

适用于 Go 的 Aurora DSQL 连接器旨在满足这些要求，并在建立连接时自动生成 IAM 身份验证令牌。

适用于 Go 的 Aurora DSQL 连接器的优势

通过适用于 Go 的 Aurora DSQL 连接器，您可以继续使用现有的 pgx 工作流程，同时通过以下方式启用 IAM 身份验证：

主要 功能

先决条件

此连接器使用适用于 Go 的 AWS SDK v2 默认凭证链，该凭证链按以下顺序解析凭证：

安装

使用 Go 模块安装连接器：

go get github.com/awslabs/aurora-dsql-connectors/go/pgx/dsql

快速入门

以下示例演示如何创建连接池和执行查询：

package main

import (
    "context"
    "log"
    
    "github.com/awslabs/aurora-dsql-connectors/go/pgx/dsql"
)

func main() {
    ctx := context.Background()
    
    // Create a connection pool
    pool, err := dsql.NewPool(ctx, dsql.Config{
        Host: "your-cluster.dsql.us-east-1.on.aws",
    })
    if err != nil {
        log.Fatal(err)
    }
    defer pool.Close()
    
    // Execute a query
    var greeting string
    err = pool.QueryRow(ctx, "SELECT 'Hello, DSQL!'").Scan(&greeting)
    if err != nil {
        log.Fatal(err)
    }
    log.Println(greeting)
}

配置选项

此连接器支持以下配置选项：

连接字符串格式

该连接器支持 PostgreSQL 和 DSQL 连接字符串格式：

postgres://[user@]host[:port]/[database][?param=value&...]
dsql://[user@]host[:port]/[database][?param=value&...]

支持的查询参数：

示例：

// Full endpoint (region auto-detected)
pool, _ := dsql.NewPool(ctx, "postgres://admin@cluster.dsql.us-east-1.on.aws/postgres")

// Using dsql:// scheme (also supported)
pool, _ := dsql.NewPool(ctx, "dsql://admin@cluster.dsql.us-east-1.on.aws/postgres")

// With explicit region
pool, _ := dsql.NewPool(ctx, "postgres://admin@cluster.dsql.us-east-1.on.aws/mydb?region=us-east-1")

// With AWS profile
pool, _ := dsql.NewPool(ctx, "postgres://admin@cluster.dsql.us-east-1.on.aws/postgres?profile=dev")

高级用法

主机配置

该连接器支持两种主机格式：

完整端点（自动检测区域）：

pool, _ := dsql.NewPool(ctx, dsql.Config{
    Host: "your-cluster.dsql.us-east-1.on.aws",
})

集群 ID（需要区域）：

pool, _ := dsql.NewPool(ctx, dsql.Config{
    Host:   "your-cluster-id",
    Region: "us-east-1",
})

池配置调整

为您的工作负载配置连接池：

pool, err := dsql.NewPool(ctx, dsql.Config{
    Host:              "your-cluster.dsql.us-east-1.on.aws",
    MaxConns:          20,
    MinConns:          5,
    MaxConnLifetime:   time.Hour,
    MaxConnIdleTime:   30 * time.Minute,
    HealthCheckPeriod: time.Minute,
})

单一连接使用

对于简单的脚本或不需要连接池时：

conn, err := dsql.Connect(ctx, dsql.Config{
    Host: "your-cluster.dsql.us-east-1.on.aws",
})
if err != nil {
    log.Fatal(err)
}
defer conn.Close(ctx)

// Use the connection
rows, err := conn.Query(ctx, "SELECT * FROM users")

使用 AWS 配置文件

为凭证指定 AWS 配置文件：

pool, err := dsql.NewPool(ctx, dsql.Config{
    Host:    "your-cluster.dsql.us-east-1.on.aws",
    Profile: "production",
})

OCC 重试

Aurora DSQL 使用乐观并发控制（OCC）。当两个事务修改相同的数据时，提交的第一个事务获胜，第二个事务收到 OCC 错误。

occretry 程序包为带指数回退和抖动的自动重试提供了帮助程序。使用以下命令进行安装：

go get github.com/awslabs/aurora-dsql-connectors/go/pgx/occretry

使用 WithRetry 进行事务写入：

err := occretry.WithRetry(ctx, pool, occretry.DefaultConfig(), func(tx pgx.Tx) error {
    _, err := tx.Exec(ctx, "UPDATE accounts SET balance = balance - $1 WHERE id = $2", 100, fromID)
    if err != nil {
        return err
    }
    _, err = tx.Exec(ctx, "UPDATE accounts SET balance = balance + $1 WHERE id = $2", 100, toID)
    return err
})

对于 DDL 或单个语句，使用 ExecWithRetry：

err := occretry.ExecWithRetry(ctx, pool, occretry.DefaultConfig(),
    "CREATE TABLE IF NOT EXISTS users (id UUID PRIMARY KEY, name TEXT)")

示例

有关更全面的示例和使用案例，请参阅适用于 Go 的 Aurora DSQL 连接器示例。