将查询结果从结构转换为另一个 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)
}

编译时出现这个错误：

不能在返回参数中使用用户（类型 []*User）作为类型字符串

如何才能得到正确的返回值？

预期的输入是

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 函数中替换最后一个返回语句，如下所示：

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
}

Answer 2

如果您必须返回一个字符串，您可以使用 encoding/json 包来序列化您的用户对象，但您必须使用以大写字母开头的字段才能导出。查看完整示例：

import (
    "encoding/json"
)

func Fetch(query string) (string) {

    type User struct{
        Id string  // <-- CHANGED THIS LINE
        Name string // <-- CHANGED THIS LINE
    }

    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(ToJSON(users)) // <-- CHANGED THIS LINE
}

func ToJSON(obj interface{}) (string) {
    res, err := json.Marshal(obj)
    if err != nil {
      panic("error with json serialization " + err.Error())
    }
    return string(res)
}