在同一函数中对 Postgres 进行多次查询

哈喽！今天心血来潮给大家带来了《在同一函数中对 Postgres 进行多次查询》，想必大家应该对Golang都不陌生吧，那么阅读本文就都不会很困难，以下内容主要涉及到，若是你正在学习Golang，千万别错过这篇文章~希望能帮助到你！

我是 go 新手，很抱歉提前提出了这个愚蠢的问题！

我正在使用 gin 框架，并希望在同一处理程序中对数据库进行多个查询 (database/sql + lib/pq)

userIds := []int{}
bookIds := []int{}
var id int

/* Handling first query here */
rows, err := pgClient.Query(getUserIdsQuery)
defer rows.Close()
if err != nil {
    return
}
for rows.Next() {
    err := rows.Scan(&id)
    if err != nil {
        return
    }
    userIds = append(userIds, id)
}

/* Handling second query here */
rows, err = pgClient.Query(getBookIdsQuery)
defer rows.Close()
if err != nil {
    return
}
for rows.Next() {
    err := rows.Scan(&id)
    if err != nil {
        return
    }
    bookIds = append(bookIds, id)
}

我有几个关于此代码的问题（任何改进和最佳实践将不胜感激）

1. 在这种情况下，go 是否可以正确处理 defer rows.close() ？我的意思是我稍后在代码中重新分配了 rows 变量，那么编译器会跟踪这两者并在函数末尾正确关闭吗？

2. 是否可以重用 id 共享变量，还是应该在 rows.next() 循环中迭代时重新声明它？

3. 在一个处理程序中进行更多查询的更好方法是什么？我是否应该有某种 writer 来接受查询和切片并用检索到的 id 填充它？

谢谢。

正确答案

我从未使用过 go-pg 库，我的回答主要集中在其他内容上，这些内容是通用的，并不特定于 golang 或 go-pg。

• 1. 不管这里的 rows 在两个查询之间共享时具有相同的引用（因此一个 rows.close() 调用就足够了，除非库有一些特殊的实现），定义两个变量更干净，例如userrows 和 bookrows。
• 1. 虽然我已经说过我没有使用过 go-pg，但我相信您不需要迭代行并手动扫描所有行的 id，我相信该库已经提供了一些像这样的 api（基于快速查看文档）：

  userids := []int{}
  err := pgclient.query(&userids, "select id from users where ...", args...)

• 1. 关于你的第二个问题，这取决于你所说的“好的”是什么意思。由于您进行了一些同步迭代，我认为这不会导致错误，但就编码风格而言，我个人不会这样做。
• 1. 我认为针对您的情况最好的做法是：

  // repo layer
  func getUserIds(args whatever) ([]int, err) {...}
  // these can be exposed, based on your packaging logic
  func getBookIds(args whatever) ([]int, err) {...}
  
  // service layer, or wherever you want to aggregate both queries
  func getUserAndBookIds() ([]int, []int, err) {
      userIds, err := getUserIds(...)
      // potential error handling
      bookIds, err := getBookIds(...)
      // potential error handling
      return userIds, bookIds, nil // you have done err handling earlier
  }

  我认为这段代码更容易阅读/维护。您不会面临变量重新分配和其他问题。

您可以查看 go-pg documentations，了解有关如何改进查询的更多详细信息。

好了，本文到此结束，带大家了解了《在同一函数中对 Postgres 进行多次查询》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！