sql - 将查询结果从结构转换为另一个 Golang 包的字符串

Question

我在网上和 SO 中搜索了一个解决方案，但没有找到适用于返回值的解决方案。这是一个简单的 sql 查询，其中包含我要返回的多行。不包括错误处理:

func Fetch(query string) (string) {

    type User struct{
        id string
        name string
    }

    rows, err := db.Query(query)

    users := make([]*User, 0)
    for rows.Next() {
      user := new(User)
      err := rows.Scan(&user.id, &user.name)
      users = append(users, user)
    }

    return(users)
}

编译时出现这个错误:

cannot use users (type []*User) as type string in return argument

如何获取正确的返回值？

预期的输入是

JD John Doe --OR-- {id:"JD",name:"John Doe"}

Answer 1

将此添加到您的代码中:

type userSlice []*User

func (us userSlice) String() string{
    var s []string
    for _, u := range us {
        if u != nil {
            s = append(s, fmt.Sprintf("%s %s", u.id, u.name))
        }
    }
    return strings.Join(s, "\n")
}

type User struct{
  id string
  name string
}

在您的 Fetch 函数中，像这样替换最后一个 return 语句:

func Fetch(query string) (string) {
  // Note that we declare the User type outside the function.
  rows, err := db.Query(query)

  users := make([]*User, 0)
  for rows.Next() {
    user := new(User)
    err := rows.Scan(&user.id, &user.name)
    users = append(users, user)
  }

  return(userSlice(users).String()) // Replace this line in your code
}