为接口类型切片提供填充数据的方法（例如，用于数据库/sql 实现的 ScanAll）

本篇文章给大家分享《为接口类型切片提供填充数据的方法（例如，用于数据库/sql 实现的 ScanAll）》，覆盖了Golang的常见基础知识，其实一个语言的全部知识点一篇文章是不可能说完的，但希望通过这些问题，让读者对自己的掌握程度有一定的认识(B 数)，从而弥补自己的不足，更好的掌握它。

我们如何实现一个函数，返回 sql 查询产生的所有行并将它们转换为接口数组 dest（可能无法像 scan 一样工作）？

我假设目标数组必须作为函数的参数给出。但是，我仍然不知道我应该如何完成实现：

func GetAll(query string, dest interface{}) error {

  rows, err := s.db.Query(query)
  if err != nil {
    return err
  }
  defer rows.Close()

  for rows.Next() {
    var destRow ??? /* do not have a type. using reflect.TypeOf(dest).Elem()? */
    err := rows.Scan(&destRow)
    if err != nil {
      return err
    }
    dest = append(dest, destRow) /* would even compile? */
  }
  return nil
}

它看起来与 json.unmarshal 实际要做的没有太大不同...

解决方案

这是使用 reflect package 的解决方案。使用指向切片的指针调用该函数。

func getall(query string, dest interface{}) error {

    // return error if dest is not a pointer to a slice.
    slice := reflect.valueof(dest)
    if slice.kind() != reflect.ptr {
        return errors.new("dest must be pointer")
    }
    slice = slice.elem()
    if slice.kind() != reflect.slice {
        return errors.new("dest must be pointer to struct")
    }

    rows, err := db.query(query)
    if err != nil {
        return err
    }
    defer rows.close()

    for rows.next() {
        // create a new slice element. the variable elementp holds a 
        // pointer to the new element.
        elementp := reflect.new(slice.type().elem())

        err := rows.scan(elementp.interface())
        if err != nil {
            return err
        }

        // append the element to the slice.
        slice.set(reflect.append(slice, elementp.elem()))
    }
    return nil
}

当查询结果具有单个列并且 dest 是指向适合该列的类型的切片的指针时，上述代码适用于标准数据库/sql 包。这是一个例子：

var names []string
if err := GetAll(db, "select name from people", &names); err != nil {
    // handle error
}

此代码还应该与支持将多个列扫描到结构的数据库包一起使用。

到这里，我们也就讲完了《为接口类型切片提供填充数据的方法（例如，用于数据库/sql 实现的 ScanAll）》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！