如何在 Golang 中指定 Postgres 的架构？

Question

我有一个数据库，将客户数据存储在不同的模式（每个租户的模式）中。我需要能够在执行查询时动态设置架构名称。我的问题是我使用 gin 根据网络请求运行这些查询，并且我有多个线程同时运行。所以基本上这样做

_, err = db.Exec(fmt.Sprintf("set search_path to %s", schemaName))
if err != nil {
    panic(err)
}

// GAP

var queryStr = "select * from customer_data"

query, err := db.Query(queryStr)
if err != nil {
    panic(err)
}
defer query.Close()

在单线程环境中执行时工作正常，但从 Web 服务器的角度来看，这太不可靠了，因为我最终可以从客户 1 请求数据并从客户 2 获取数据，因为在代码中指定的 GAP 期间，另一个线程可以来电set search_path查询。

Answer 1

正如 @rsp 评论，您可以通过其架构（例如schema1.customer_data）来限定表名称或使用事务。

有很多原因需要将数据库分成模式，并且有很多很多使用事务的原因。

关于使用事务的一篇很好的文章database/sql是Go 博客文章“执行事务”。

示例事务块可能如下所示（请注意，没有错误检查或sql.ErrNoRows情况检查）：

// transact runs a set of SQL statements in a transaction
func transact(db *sql.DB, schema string) {

    fmt.Println("transacting for schema:", schema)

    // start a transaction
    ctx := context.Background()
    tx, err := db.BeginTx(ctx, nil)
    panicOnErr("tx setup", err)

    // defer a rollback in case anything fails
    defer tx.Rollback()

    schemaSet(tx, schema) // set the schema
    inserter(tx, "user1", 5) // insert some data
    inserterQuery(tx, "user2", 7) // insert and return data
    lister(tx) // list data

    // explicit rollback
    err = tx.Rollback()
    panicOnErr("rollback err", err)
}

此处提供了更完整的示例。