我正在学习 Golang，想了解解决这个问题的“Go 之道”。

具体来说，我正在使用 sql 包，我发现我的代码中有一些冗余功能，我想将它们提取到一个函数中。

我的原始代码

我有一个用户结构，1):

type User struct {
  ID        int
  FirstName string
  LastName  string
}

2) 从数据库(Postgresql)中通过 ID 获取一个用户的函数:

func GetUserById(id int) (user User) {
  sql := `
    SELECT id, first_name, last_name
    FROM users
    WHERE id = $1
  `
  row := db.QueryRow(sql, id)
  err := row.Scan(&user.ID, &user.FirstName, &user.LastName)
  if err != nil {
    panic(err)
  }
  return
}

和 3) 从数据库中获取所有用户的函数:

func GetUsers() (users []User) {
  sql := `
    SELECT id, first_name, last_name
    FROM users
    ORDER BY last_name
  `
  rows, err := db.Query(sql)
  if err != nil {
    panic(err)
  }
  for rows.Next() {
    user := User{}
    err := rows.Scan(&user.ID, &user.FirstName, &user.LastName)
    if err != nil {
      panic(err)
    }
    users = append(users, user)
  }
  rows.Close()
  return
}

期望的重构

用户记录中只有 3 个字段，这是一个简单的示例。但是，如果有更多的字段，两个数据访问函数的 rows.Scan(...) 部分将很好地移动到一个都可以调用的函数:

func ScanUserFromRow(row *sql.Row) (user User) {
  err := row.Scan(&user.ID, &user.FirstName, &user.LastName)
  if err != nil {
    panic(err)
  }
  return
}

然后更新的数据库访问函数看起来像这样:

func GetUserById(id int) (user User) {
  sql := `
    SELECT id, first_name, last_name
    FROM users
    WHERE id = $1
  `
  row := db.QueryRow(sql, id)
  user = ScanUserFromRow(row)
  return
}

func GetUsers() (users []User) {
  sql := `
    SELECT id, first_name, last_name
    FROM users
    ORDER BY last_name
  `
  rows, err := db.Query(sql)
  if err != nil {
    panic(err)
  }
  for rows.Next() {
    user := ScanUserFromRow(rows)
    users = append(users, user)
  }
  rows.Close()
  return
}

但是，对于 GetUserById 函数，我处理的是 *sql.Row 结构指针。对于 GetUsers 函数，我正在处理一个 *sql.Rows 结构指针。两者不同……显然，但相似，因为它们都有一个Scan 方法。

似乎类型系统不允许我创建一个接受一个或另一个的方法。有没有办法利用 interface{} 来做到这一点，或者是否有其他一些更惯用的 Go 解决方案？

更新/解决方案

对于这个问题，我想说 sql.Row 和 sql.Rows 都是用 Scan 发出“嘎嘎声”的鸭子。如何使用允许两者的函数参数？

@seh provided an answer below通过使参数成为自定义接口(interface)，可以实现我希望的那种鸭子类型。这是结果代码:

type rowScanner interface {
  Scan(dest ...interface{}) error
}

func ScanPlayerFromRow(rs rowScanner) (u User) {
  err := rs.Scan(&u.ID, &u.FirstName, &u.LastName)
  if err != nil {
    panic(err)
  }
  return
}

...or, as @Kaveh pointed out below ，接口(interface)的定义可以内联在函数参数中:

func ScanPlayerFromRow(rs interface {
  Scan(des ...interface{}) error
}) (u User) {
  err := rs.Scan(&u.ID, &u.FirstName, &u.LastName)
  if err != nil {
    panic(err)
  }
  return
}

最佳答案

两者都是sql.Rows和 sql.Row有a Scan method .标准库中没有包含该方法的接口(interface)，但可以自己定义它:

type rowScanner interface {
    Scan(dest ...interface{}) error
}

然后您可以编写一个在 rowScanner 而不是 *sql.Row 或 *sql.Rows 上运行的函数:

 import "database/sql"

 type rowScanner interface {
    Scan(dest ...interface{}) error
 }

 func handleRow(scanner rowScanner) error {
    var i int
    return scanner.Scan(&i)
 }

 func main() {
    var row *sql.Row
    handleRow(row) // Crashes due to calling on a nil pointer.
    var rows *sql.Rows
    handleRow(rows)  // Crashes due to calling on a nil pointer.
 }

我没有使用真正的 *sql.Row 或 *sql.Rows 进行模拟，但这应该能让您明白。您所需的 ScanUserFromRow 函数需要 rowScanner 而不是 *sql.Row。

关于go - 创建一个接受两个具有相同方法的不同对象的函数，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/37504735/