登录
首页 >  Golang >  Go问答

使用 QueryRow 和 INSERT 在 Golang 中直接执行 SQL 查询和插入操作是否安全?

来源:stackoverflow

时间:2024-03-27 16:57:29 209浏览 收藏

今天golang学习网给大家带来了《使用 QueryRow 和 INSERT 在 Golang 中直接执行 SQL 查询和插入操作是否安全?》,其中涉及到的知识点包括等等,无论你是小白还是老手,都适合看一看哦~有好的建议也欢迎大家在评论留言,若是看完有所收获,也希望大家能多多点赞支持呀!一起加油学习~

问题内容

Go 数据库/sql Postgres 适配器不支持 LastInsertId。来自文档:

pq 不支持 Result 类型的 LastInsertId() 方法 数据库/sql。返回 INSERT(或 UPDATE 或 DELETE),将 Postgres RETURNING 子句与标准查询一起使用或 查询行调用。

但是 Go 数据库/sql 文档建议不要使用 Query 来修改数据库,因为它:

保留数据库连接直到 sql.Rows 关闭。自从 可能有未读数据(例如更多数据行),连接可以 不被使用。在上面的例子中,连接永远不会 再次释放。

文档还说“你永远不应该像这样使用 Query()。”

如果您需要结果行,大多数人建议使用 QueryRow 与 Postgres 一起执行 INSERT。我不完全理解文档给出的技术推理,但它似乎与我读到的内容相冲突。使用 QueryRow 执行 INSERT 是否安全且被视为良好实践?


解决方案


该指南(顺便说一句,它不是 database/sql 文档)没有错误,但您错过了引用段落上方的代码片段中的重要部分,请注意 // bad 注释:

_, err := db.query("delete from users") // bad

特别注意 _。这在 go 中称为“空白标识符”,它允许您丢弃不关心的值。

当调用query时,你会得到一个*sql.rows实例作为第一个返回值,这个值用完后需要关闭,如果你丢弃它,你就无法关闭它,如果你不这样做不关闭它你会泄漏连接。 *sql.rows通过调用clos​​e方法关闭。

当调用queryrow时,你会返回一个*sql.row值,这个值用完后需要关闭,如果你丢弃它就不能关闭它,如果你不关闭它你会泄漏连接。 *sql.row通过调用scan方法关闭。

“使用 queryrow 执行 insert 是否安全且被视为良好实践?”

是的,是的,只要你正确处理返回值。

做这样的事情绝对没问题:

var id int
row := db.queryrow("insert ... returning id", params...)
if err := row.scan(&id); err != nil { // scan will release the connection
    return err
}
fmt.println(id)

或者:

rows, err := db.query("insert ... values (...), (...) returning id", params...)
if err != nil {
    return err
}
defer rows.close() // make sure that the connection is released once the function returns/panics

var ids []int
for rows.next() {
    var id int
    if err := rows.scan(&id); err != nil {
        return err
    }
    ids = append(ids, id)
}
if err := rows.err(); err != nil {
    return err
}
fmt.println(ids)

可能您对数据库/sql 文档的理解过于字面意思。 重点是,在提供的示例中结果被丢弃

if _, err := db.Query(``); err != nil {

}

当您执行此操作时,很有可能有一天您会遇到错误,例如连接过多,因为您在没有强制关闭调用的情况下丢弃了结果。

所以答案是您可以安全地在插入查询中使用 queryrow。 queryrow 关闭后台所有内容后调用 scan 方法 https://golang.org/src/database/sql/sql.go?s=88812:88857#L3094

理论要掌握,实操不能落!以上关于《使用 QueryRow 和 INSERT 在 Golang 中直接执行 SQL 查询和插入操作是否安全?》的详细介绍,大家都掌握了吧!如果想要继续提升自己的能力,那么就来关注golang学习网公众号吧!

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