卓越飞翔博客卓越飞翔博客

卓越飞翔 - 您值得收藏的技术分享站
技术文章34051本站已运行391

golang gorm 更新并返回

golang gorm 更新并返回

问题内容

在我的 golang 项目中,我使用 gorm,我需要使用返回子句进行 upsert 查询,以从查询中获取修改后的值。我可以进行 upsert,但我不知道如何将返回子句连接到它。表名是counters,代码如下:

te := struct {
        name string //key column
        counter int
    }{
        name: "name_to_update",
        counter: 2,
    }

db.
    //model(te).
    clauses(
        //clause.returning{columns: []clause.column{{name: "counter"}}},
        clause.onconflict{
            columns: []clause.column{{name: "name"}}, // key column
            doupdates: clause.assignments(map[string]interface{}{
                "counter": gorm.expr("counters.counter + ?", 1),
            }),
        },
    ).create(&te)

生成的 sql 查询是:

INSERT INTO "counters" ("counter", "name") VALUES (0, "name_to_update") ON CONFLICT ("name") 
DO UPDATE SET "counter"=counters.counter + 1 RETURNING "name" //I need updated counter value here, not name

因此计数器已更新,这没问题,但当我需要计数器的更新值时,它会返回键列(在返回中)。有什么想法如何修复它吗?谢谢


正确答案


我不确定匿名结构是否会导致问题。

此外,从您的代码中还不清楚表名称 - “counters” - 来自何处。

我已经尝试过你的解决方案 - 但使用了模型的专用结构 - 并且效果很好。

type counter struct {
    name    string `gorm:"primarykey"`
    counter int
}

...

counter := counter{name: "name_to_update", counter: 2}

    db.
        clauses(
            clause.returning{columns: []clause.column{{name: "counter"}}},
            clause.onconflict{
                columns: []clause.column{{name: "name"}},
                doupdates: clause.assignments(map[string]interface{}{
                    "counter": gorm.expr("counters.counter + ?", 1),
                }),
            },
        ).create(&counter)

    fmt.println(counter.counter)

上面的代码生成以下 sql

INSERT INTO "counters" ("name","counter") VALUES ('name_to_update',10) ON CONFLICT ("name") DO UPDATE SET "counter"=counters.counter + 1 RETURNING "counter"

并且 counter.counter 具有正确的更新值。

卓越飞翔博客
上一篇: 从“map”返回值并打印它们
下一篇: 返回列表
留言与评论(共有 0 条评论)
   
验证码:
隐藏边栏