使用 GORM 和 Postgresql 如何设置搜索路径？

Question

如何从 gorm 获取当前模式？通常它是“SHOW search_path”并且是一个 exec。与通常设置当前 search_path 相同“将 search_path 设置为 newschema”。其中新模式可以是一系列名称中的单个模式名称。我尝试了 db.Exec("show search_path").Find(&result)，其中结果是一个 []string，我得到一个空数组。

Answer 1

search_path通过执行查询来设置set search_path to newschema是不正确的设置方式，因为这只会对用于执行查询的连接进行更改。

要将其设置为您的应用程序将使用的完整连接集，您需要将其search_path作为参数提供给gorm.Open("postgres", "dbname=YourDB search_path=YourSearchPath")您将要使用的所有其他连接参数，例如dbnameor port,user等等。

Answer 2

在 PluralSite 的 GORM 类中找到了如何通过实验和有关 Raw 和 Scan 的提示来显示当前模式。

var currentSchema string
rows, err := db.Debug().Raw("show search_path").Rows()
rows.Next()  // to get the first and only result
rows.Scan(&currentSchema)
fmt.Printf("Search Path: %v\n", currentSchema)

返回：搜索路径：“$user”，public

考虑到@PavloStrokov 关于事务（基本上是线程和事务）的观点，我能够让模式保留在当前进程中。在处理程序中，我使用现有的数据库连接并创建一个事务。它被传递给每个被调用的接触数据库的方法。我将对 gorm 方法的调用更改为传递的 tx 而不是存储并可供任何需要它的方法使用的普通 db 变量。

我为启动这一切所做的呼吁是：

tx := db.开始()

tx.Exec("将 search_path 设置为 schema_name")

生成字符串以使用正确的 Facility 模式名称。

tx 是所有接触数据库的方法签名中的最后一个参数。例如 func methiodUsingDBSchema(val1 string, val2 string, tx *gorm.DB){}

然后我做的最后一件事是 tx.Commit() 即使这些都是读取。

我已经运行了所有测试并且运行良好。

我使用 Postgres 架构来分隔我的应用程序中的设施，并且需要根据登录的用户进行更改。所有用户和设施信息都是公开的。