如何使用 GORM 的 FirstOrCreate() 方法（或 Django 的 get_or_create）来确保只创建一个实例？

使用 GORM 的 `FirstOrCreate()` 方法（或 Django 的 `get_or_create()`) 可以确保在数据库中只创建一个实例。然而，这些方法使用两个数据库操作（查询和插入），可能导致在查询和插入之间出现竞态条件，从而创建多个实例。为了防止这种情况，建议在模型字段上添加唯一约束。在 Django 中，可以使用 `unique_together` 元类，而在 GORM 中，可以使用 `unique_index` 选项。

我正在考虑将 gorm 用于应用程序，并正在研究 firstorcreate 的工作原理，它似乎使用两个数据库操作。考虑这个示例脚本：

package main

import (
    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/sqlite"
    "github.com/sirupsen/logrus"
)

type user struct {
    gorm.model
    name string
    age  uint
}

func main() {
    db, err := gorm.open("sqlite3", "examplegorm.db")
    if err != nil {
        logrus.fatalf("open db: %v", err)
    }
    defer db.close()

    db.logmode(true)
    db.automigrate(&user{})

    var user user
    db.where(user{name: "non_existing"}).attrs(user{age: 20}).firstorcreate(&user)
}

运行此命令并检查日志后，我发现（除了自动迁移之外）它使用两个查询，一个 select 和一个 insert：

kurt@kurts-macbook-pro-13 ~/d/scratch> go run gorm_example.go

(/users/kurt/documents/scratch/gorm_example.go:23) 
[2020-01-05 09:09:10]  [1.03ms]  create table "users" ("id" integer primary key autoincrement,"created_at" datetime,"updated_at" datetime,"deleted_at" datetime,"name" varchar(255),"age" integer )  
[0 rows affected or returned ] 

(/users/kurt/documents/scratch/gorm_example.go:23) 
[2020-01-05 09:09:10]  [0.86ms]  create index idx_users_deleted_at on "users"(deleted_at)   
[0 rows affected or returned ] 

(/users/kurt/documents/scratch/gorm_example.go:26) 
[2020-01-05 09:09:10]  [0.28ms]  select * from "users"  where "users"."deleted_at" is null and (("users"."name" = 'non_existing')) order by "users"."id" asc limit 1  
[0 rows affected or returned ] 

(/users/kurt/documents/scratch/gorm_example.go:26) 
[2020-01-05 09:09:10]  [0.31ms]  insert  into "users" ("created_at","updated_at","deleted_at","name","age") values ('2020-01-05 09:09:10','2020-01-05 09:09:10',null,'non_existing',20)  
[1 rows affected or returned ]

据我了解，https://stackoverflow.com/a/16128088/995862，但是，

在 sql dbms 中，选择-测试-插入方法是一个错误：没有什么可以阻止另一个进程在 select 和 insert 语句之间插入“丢失”的行。

django 的 get_or_create() 似乎以类似的方式工作。鉴于此模型，

from django.db import models

class user(models.model):
    name = models.charfield(max_length=255)
    age = models.positiveintegerfield() 

如果我启用数据库日志记录并运行 get_or_create() 查询，我会看到

In [1]: from djangoapp.models import *                                                                                           

In [2]: User.objects.get_or_create(name="jinzhu", age=20)                                                                        
(0.000) SELECT "djangoapp_user"."id", "djangoapp_user"."name", "djangoapp_user"."age" FROM "djangoapp_user" WHERE ("djangoapp_user"."age" = 20 AND "djangoapp_user"."name" = 'jinzhu') LIMIT 21; args=(20, 'jinzhu')
(0.000) BEGIN; args=None
(0.000) INSERT INTO "djangoapp_user" ("name", "age") VALUES ('jinzhu', 20); args=['jinzhu', 20]
Out[2]: (<User: User object (1)>, True)

简而言之，如果我想确保只创建一条记录，似乎我应该避免使用 orm（例如 gorm 或 django orm）并编写自己的查询？

我的第二个问题是如何在 gorm 中获得 django 的 created 布尔值的等效值。我是否应该确定生成的 gorm.db 的 rowsaffected 是否为 1 来确定是否实际创建了行？

解决方案

您应该只在查询模型字段上添加 unique 约束，这足以使其在数据库中保持一致

对于 django，将向模型添加元类

class meta:
    unique_together = ['name', 'age']

对于 gorm

Name string `gorm:"unique_index:idx_name_age"`
Age  uint   `gorm:"unique_index:idx_name_age"`

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