时间戳

首页 > 解决方案 > gorm包的Create方法中如何去掉RETURNING子句？

sql - gorm包的Create方法中如何去掉RETURNING子句？

问题描述

在为gorm包创建记录时，我对默认行为有点困惑。

city := models.City

if err := databases.DBGORM.Set("gorm:insert_option", "RETURNING *").Create(&city).Error; err != nil {
    fmt.Println(err.Error())
}

在日志中我看到这样的 SQL 查询：

INSERT INTO "my_scheme"."city" ("created_at","updated_at","deleted_at","name","country") VALUES ('2020-05-19 23:45:18','2020-05-19 23:45:18',NULL,'New York','USA') RETURNING * RETURNING "my_scheme"."city"."id"

正如您从查询中看到的那样，我有一个RETURNING不正确的双重子句并引发错误。

id在 SQL 查询末尾添加一个似乎是该Create方法的默认行为。我怎样才能改变这种行为？

模型.go：

package models

import (
    "my_app/proto"
    "time"
)

type City struct {
    Id uint64
    CreatedAt time.Time
    UpdatedAt time.Time
    DeletedAt *time.Time
    proto.City
}

func (City) TableName() string {
    return "my_scheme.city"
}

标签： sqlgogo-gorm

解决方案

不，没有办法改变这种行为。

但是如果您想在调用 Create 函数后获取 ID 或时间戳（CreatedAt 和 UpdatedAt），它们将在您的模型中通过指针传递自动更新。

如果您有另一个具有默认值的字段，请将default标签添加到模型中的该字段。调用后，gorm 也会自动更新该字段Create。

type City struct {
    Id        uint64
    CreatedAt time.Time
    UpdatedAt time.Time
    DeletedAt *time.Time

    SomeField *string `gorm:"default:test"`
}

// ...

city := models.City{}

if err := databases.DBGORM.Create(&city).Error; err != nil {
    fmt.Println(err.Error())
}

fmt.Printf("%+v", city)


[2021-04-13 21:39:44]  [1.06ms]  INSERT INTO "cities" ("created_at","updated_at","deleted_at") VALUES ('2021-04-13 21:39:44','2021-04-13 21:39:44',NULL) RETURNING "cities"."id"  
[1 rows affected or returned ] 

[2021-04-13 21:39:44]  [0.59ms]  SELECT "some_field" FROM "cities"  WHERE (id = 26)  
[1 rows affected or returned ] 

{
  "Id": 26,
  "CreatedAt": "2021-04-13T21:39:44.809605473+07:00",
  "UpdatedAt": "2021-04-13T21:39:44.809605473+07:00",
  "DeletedAt": null,
  "SomeField": "test"
}

如果您根本不想更新模型，请Create按值而不是指针将其传递给方法，并忽略gorm.ErrUnaddressable错误。

推荐阅读