使用 gorm 从数据库获取刚刚创建的记录

知识点掌握了，还需要不断练习才能熟练运用。下面golang学习网给大家带来一个Golang开发实战，手把手教大家学习《使用 gorm 从数据库获取刚刚创建的记录》，在实现功能的过程中也带大家重新温习相关知识点，温故而知新，回头看看说不定又有不一样的感悟！

我有一个创建新用户的函数，但是获取用户值的推荐方法不包括数据库创建的自动生成的值（id，created_at）

type user struct {
    id         string `json:"id" gorm:"primarykey"`
    email      string `json:"email"`
    password   string `json:"password"`
    created_at string `json:"created_at"`
    updated_at string `json:"updated_at"`
    deleted_at string `json:"deleted_at"`
}

type userrequest struct {
    email    string `json:"email"`
    password string `json:"password"`
}

func createuserrepo(newuser userrequest) (*user, error) {

    user := user{email: newuser.email, password: newuser.password}
    fmt.println(user)
    result := services.postgressqldb.select("email", "password").create(&user)
    fmt.println(result.rowsaffected)

    if result.error != nil {
        modules.logger.error(result.error)
        return nil, result.error
    }

    return &user, nil

}

您可以看到我的 id 和created_at 都是空字符串，尽管这些值是由我的数据库自动生成的。

{
    "id": "",
    "email": "[email protected]",
    "password": "password",
    "created_at": "",
    "updated_at": "",
    "deleted_at": ""
}

测试

在某种程度上，我认为这可以归结为编写 gorm 期望的“正确”类型。

1. 如果我将 gorm.model 添加到我的结构中，id 似乎会返回 uuid，但它也会返回一堆看起来像我不想要的重复字段（id、createdat、updatedat、deletedat）。删除 gorm.model，id 不再返回 postgres 生成的 uuid

type user struct {
        gorm.model
        id        uuid.uuid      `json:"id"`
        email     string         `json:"email"`
        password  string         `json:"password"`
        createdat time.time      `json:"created_at"`
        updatedat time.time      `json:"updated_at"`
        deletedat gorm.deletedat `json:"deleted_at"`
    }

{
    "id": 0,
    "createdat": "0001-01-01t00:00:00z",
    "updatedat": "0001-01-01t00:00:00z",
    "deletedat": null,
    "id": "b4dea226-3be2-4ee7-8548-67ccbbbcbcca",
    "email": "[email protected]",
    "password": "password",
    "created_at": "2022-06-25t20:59:27.872198797+01:00",
    "updated_at": "2022-06-25t20:59:27.872198797+01:00",
    "deleted_at": null
}

数据库 postgres

-- CreateTable
CREATE TABLE "users" (
    "id" UUID NOT NULL DEFAULT gen_random_uuid(),
    "email" TEXT NOT NULL,
    "password" TEXT,
    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
    "updated_at" TIMESTAMP(3) NOT NULL,
    "deleted_at" TIMESTAMP(3),

    CONSTRAINT "users_pkey" PRIMARY KEY ("id")
);

正确答案

我终于在没有包含的情况下找到了答案。

添加默认返回子句，返回所有新创建的值。 https://gorm.io/docs/update.html#Returning-Data-From-Modified-Rows

clauses.(clause.returning{})

model.go

type user struct {
    id        uuid.uuid      `json:"id" gorm:"primarykey:type:uuid"`
    email     string         `json:"email"`
    password  string         `json:"password"`
    createdat time.time      `json:"created_at"`
    updatedat time.time      `json:"updated_at"`
    deletedat gorm.deletedat `json:"deleted_at"`
}

存储库.go

func createuserrepo(newuser userrequest) (*user, error) {

    hash, errhash := modules.hashpassword(newuser.password)

    if errhash != nil {
        return nil, errhash
    }

    user := user{email: newuser.email, password: *hash}
    result := services.postgressqldb.clauses(clause.returning{}).select("email", "password").create(&user)
    fmt.println(result.rowsaffected)
    fmt.println(&user)

    if result.error != nil {
        modules.logger.error(result.error)
        return nil, result.error
    }

    return &user, nil

}

感谢 @thefool 对此发现的评论。

据我所知，gorm 不支持主键整数自动递增的任何其他内容。因此，要么将其保留在 postgresql 端（就像您所做的那样），要么在创建对象之前在 go 中生成自己的 uuid（使用 beforecreate 挂钩）。

此外，gorm.model 定义如下：

type model struct {
  id        uint           `gorm:"primarykey"`
  createdat time.time
  updatedat time.time
  deletedat gorm.deletedat `gorm:"index"`
}

您不必使用它，但如果不使用它，请将 gorm:"primarykey" 添加到您的主键字段。默认情况下，主字段具有自动增量，对于您的情况，我建议使用 autoincrement:false 禁用它。

type user struct {
    id        uuid.uuid      `gorm:"primarykey;autoincrement:false" json:"id"`
    email     string         `json:"email"`
    password  string         `json:"password"`
}

当您让 postgresql 处理默认值（而不是 gorm）时，您必须再次查询该对象才能访问您的 id uuid 和 created_at timestamp。

另请注意，您可以使用 gormvaluerinterface 在创建时使用 sql expr。但您仍然需要再次查询您的记录。 (https://gorm.io/docs/data_types.html#GormValuerInterface)

如果您有兴趣在 go 端处理所有内容，这里有一个包含和不包含 gorm.model 的示例。

package main

import (
    "fmt"
    "time"

    "github.com/google/uuid"
    "gorm.io/driver/sqlite"
    "gorm.io/gorm"
)

var db *gorm.db

type user struct {
    uuid      uuid.uuid `gorm:"primarykey;autoincrement:false"`
    name      string
    createdat time.time
}

type userwithgormmodel struct {
    gorm.model
    uuid uuid.uuid `gorm:"primarykey;autoincrement:false"`
    name string
}

func (u *user) beforecreate(tx *gorm.db) (err error) {

    // this is an example, refer to https://pkg.go.dev/github.com/google/uuid for good usage
    u.uuid = uuid.new()

    u.createdat = time.now()

    return
}

func (u *userwithgormmodel) beforecreate(tx *gorm.db) (err error) {

    // this is an example, refer to https://pkg.go.dev/github.com/google/uuid for good usage
    u.uuid = uuid.new()

    return
}

func connectdatabase() {

    database, err := gorm.open(sqlite.open("test.db"), &gorm.config{})
    if err != nil {
        panic("failed to connect database")
    }

    if err != nil {
        panic("failed to connect to database!")
    }

    database.automigrate(&user{}, &userwithgormmodel{})

    db = database
}

func main() {
    connectdatabase()

    user := user{name: "bob"}
    db.create(&user)
    fmt.printf("user{uuid: %s, user.name: %s, createdat: %s}\n", user.uuid, user.name, user.createdat)

    user2 := userwithgormmodel{name: "john"}
    db.create(&user2)
    fmt.printf("userwithgormmodel{uuid: %s, name: %s, createdat: %s}\n", user2.uuid, user2.name, user2.createdat)
}

输出：

User{UUID: 8551636a-540f-4733-8179-3f0cc45daf4f, User.Name: Bob, CreatedAt: 2022-06-28 11:40:10.9225724 +0200 CEST}
UserWithGormModel{UUID: 2a6f94bd-a42b-4316-90be-0e93ea5091a6, Name: John, CreatedAt: 2022-06-28 11:40:10.9318004 +0200 CEST}

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。