登录
首页 >  Golang >  Go问答

当我在 db(这里是 postgres)中的结构未知时,如何检索 GoLang 数据库/sql 中的值?

来源:stackoverflow

时间:2024-04-14 13:54:30 372浏览 收藏

哈喽!大家好,很高兴又见面了,我是golang学习网的一名作者,今天由我给大家带来一篇《当我在 db(这里是 postgres)中的结构未知时,如何检索 GoLang 数据库/sql 中的值?》,本文主要会讲到等等知识点,希望大家一起学习进步,也欢迎大家关注、点赞、收藏、转发! 下面就一起来看看吧!

问题内容

我通过 github.com/lib/pq 将 go 与 postgresql 结合使用,并且当我的结构已知时能够成功获取记录。 现在我的查询是当我的结构动态变化时如何获取记录?

通过rows.columns,我可以获取列名称,但是您可以帮助我获取所有行的这些列的值吗?我引用了@luke 回答的这个链接,不过,这里的人已经定义了一个结构。 是否可以使用 golang 数据库/sql 按名称检索列值

type Person struct {
    Id int
    Name string
}

同时,我没有固定的结构,所以我将如何遍历所有列,以及所有行。我的方法是首先使用一个指针循环遍历所有列,然后使用另一个指针循环到下一行。 仍然无法对此进行编码,您能帮我解决这个问题吗,例如如何继续并获取值。


解决方案


由于您事先不知道结构,因此可以将行作为空接口的二维切片返回。但是,为了使行扫描正常工作,您需要将值预先分配给适当的类型,为此,您可以使用 ColumnTypes 方法和 reflect 包。请记住,并非每个驱动程序都提供对列类型的访问,因此请确保您使用的驱动程序提供对列类型的访问。

rows, err := db.query("select * from foobar")
if err != nil {
    return err
}
defer rows.close()

// get column type info
columntypes, err := rows.columntypes()
if err != nil {
    return err
}

// used for allocation & dereferencing
rowvalues := make([]reflect.value, len(columntypes))
for i := 0; i < len(columntypes); i++ {
    // allocate reflect.value representing a **t value
    rowvalues[i] = reflect.new(reflect.ptrto(columntypes[i].scantype()))
}

resultlist := [][]interface{}{}
for rows.next() {
    // initially will hold pointers for scan, after scanning the
    // pointers will be dereferenced so that the slice holds actual values
    rowresult := make([]interface{}, len(columntypes))
    for i := 0; i < len(columntypes); i++ {
        // get the **t value from the reflect.value
        rowresult[i] = rowvalues[i].interface()
    }

    // scan each column value into the corresponding **t value
    if err := rows.scan(rowresult...); err != nil {
        return err
    }

    // dereference pointers
    for i := 0; i < len(rowvalues); i++ {
        // first pointer deref to get reflect.value representing a *t value,
        // if rv.isnil it means column value was null
        if rv := rowvalues[i].elem(); rv.isnil() {
            rowresult[i] = nil
        } else {
            // second deref to get reflect.value representing the t value
            // and call interface to get t value from the reflect.value
            rowresult[i] = rv.elem().interface()
        }
    }

    resultlist = append(resultlist, rowresult)

}
if err := rows.err(); err != nil {
    return err
}

fmt.println(resultlist)

此函数在不了解有关列类型和计数的任何信息的情况下打印查询结果。它是 previous answer 的变体,未使用 reflect 软件包。

func printQueryResult(db *sql.DB, query string) error {
    rows, err := db.Query(query)
    if err != nil {
        return fmt.Errorf("canot run query %s: %w", query, err)
    }
    defer rows.Close()

    cols, _ := rows.Columns()
    row := make([]interface{}, len(cols))
    rowPtr := make([]interface{}, len(cols))
    for i := range row {
        rowPtr[i] = &row[i]
    }
    fmt.Println(cols)
    for rows.Next() {
        err = rows.Scan(rowPtr...)
        if err != nil {
            fmt.Println("cannot scan row:", err)
        }
        fmt.Println(row...)
    }
    return rows.Err()
}

诀窍在于 rows.scan 可以将值扫描到 *interface{} 中,但您必须将其包装在 interface{} 中才能使用 ... 将其传递给 scan

好了,本文到此结束,带大家了解了《当我在 db(这里是 postgres)中的结构未知时,如何检索 GoLang 数据库/sql 中的值?》,希望本文对你有所帮助!关注golang学习网公众号,给大家分享更多Golang知识!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>