在 Golang 中设置 Postgres 数据库的架构指定

最近发现不少小伙伴都对Golang很感兴趣，所以今天继续给大家介绍Golang相关的知识，本文《在 Golang 中设置 Postgres 数据库的架构指定》主要内容涉及到等等知识点，希望能帮到你！当然如果阅读本文时存在不同想法，可以在评论中表达，但是请勿使用过激的措辞~

我有一个数据库，以不同的模式（每个租户的模式）存储客户数据。我需要能够在执行查询时动态设置架构名称。我的问题是我使用 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查询。

解决方案

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

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

关于使用 database/sql 事务的好文章是 Go Blog article "Executing transactions"。

示例事务块可能如下所示（请注意，没有错误检查，或检查 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)
}

可以获取更完整的示例 here。

今天关于《在 Golang 中设置 Postgres 数据库的架构指定》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！