惯性聚合 高效追踪和阅读你感兴趣的博客、新闻、科技资讯
阅读原文 在惯性聚合中打开

推荐订阅源

Martin Fowler
Martin Fowler
D
DataBreaches.Net
F
Fortinet All Blogs
阮一峰的网络日志
阮一峰的网络日志
博客园_首页
Apple Machine Learning Research
Apple Machine Learning Research
H
Help Net Security
M
MIT News - Artificial intelligence
美团技术团队
人人都是产品经理
人人都是产品经理
freeCodeCamp Programming Tutorials: Python, JavaScript, Git & More
Cyber Security Advisories - MS-ISAC
Cyber Security Advisories - MS-ISAC
The Cloudflare Blog
有赞技术团队
有赞技术团队
L
LangChain Blog
博客园 - Franky
钛媒体:引领未来商业与生活新知
钛媒体:引领未来商业与生活新知
博客园 - 【当耐特】
S
SegmentFault 最新的问题
V
Visual Studio Blog
Blog — PlanetScale
Blog — PlanetScale
Hugging Face - Blog
Hugging Face - Blog
B
Blog
I
InfoQ

Jiajun的技术笔记

你好,2026! TiDB 源码阅读(六):TiDB Coprocessor 源码解析 性能优化的核心思想 TiDB 源码阅读(五):索引 TiDB 源码阅读(四):AST、逻辑计划、物理计划 CockroachDB Serverless Architecture podman 无故退出 Cursor Control-L (CTRL-L) Keyboard Shortcuts in Terminal Replace docker with podman Using xmonad with xfce4 A RC script for freebsd frpc 自己动手写一个k8s controller AI 会取代你的(编程)岗位吗? 自建DERP服务器提升Tailscale连接速度(使用Nginx转发) 自动升级Docker容器 再读《程序员修炼之道-从小工到专家》 让浏览器下载文件 再读《软件随想录》/《黑客与画家》/《软技能》 HTTP 压力测试中的 Coordinated Omission 2的补码 编程语言中的 context 是什么? flutter macOS 构建出错 Flatpak 使用小记 Golang CAS 操作是怎么实现的 PostgreSQL 当MQ来使用 Clash 结合 工作VPN 的网络设计 使用 PostgreSQL 搭建 JuiceFS PostgreSQL 配置优化和日志分析 有GitHub Copilot?那就可以搭建你的ChatGPT4服务 窗口函数的使用(以PG为例)
Go使用闭包简化数据库操作代码
Jiajun Huang · 2020-10-17 · via Jiajun的技术笔记

在日常工程中,我们可能要开事务来完成一些操作,因此就会有如下代码:

// UpdateTaskRemark 更新任务备注
func UpdateTaskRemark(taskID uint32, remark string) error {
        sql, args, err := sq.Update("tasks").Set("remark", remark).Set("updated_at", getNowTS()).Where("id = ?", taskID).ToSql()
        if err != nil {
                return err
        }

        tx, err := db.Beginx()
        if err != nil {
                return err
        }

        _, err = tx.Exec(sql, args...)
        if err != nil {
                return err
        }

        err = tx.Commit()
        if err != nil {
                logrus.Errorf("failed to commit transaction: %s, rollback it", err)
                return tx.Rollback()
        }

        return nil
}

如果只有一个,似乎没有什么问题,但是,一旦操作多了之后,要在每个model操作函数中都写上这一堆重复的代码,那就不太妙了。 因此得想一个办法来进行抽象,把开事务、提交和回滚的操作封装起来。借助闭包我们可以做到这一点:

// 工具函数用于在事务中执行,自动提交和回滚
type TxFunc func(tx *sqlx.Tx) error

// ExecInTx 传入一个闭包函数,签名类型为TxFunc。在ExecInTx中先开事务,
// 然后将事务传入闭包函数,如果闭包函数没有返回错误就提交,否则回滚。
func ExecInTx(f TxFunc) error {
        tx, err := db.Beginx()
        if err != nil {
                return err
        }

        err = f(tx)
        if err != nil {
                return err
        }

        err = tx.Commit()
        if err != nil {
                logrus.Errorf("failed to commit transaction: %s, rollback it", err)
                return tx.Rollback()
        }

        return nil
}

由此,model操作函数就简化成了:

// UpdateTaskRemark 更新任务备注
func UpdateTaskRemark(taskID uint32, remark string) error {
        sql, args, err := sq.Update("tasks").Set("remark", remark).Set("updated_at", getNowTS()).Where("id = ?", taskID).ToSql()
        if err != nil {
                return err
        }

        return ExecInTx(func(tx *sqlx.Tx) error {
                _, err := tx.Exec(sql, args...)
                return err
        })
}

如上。