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 具有正确的更新值。

以上就是golang gorm 更新并返回的详细内容，更多请关注编程网其它相关文章！