Как метод FirstOrCreate () GORM (или get_or_create Django) может обеспечить создание только одной строки? - PullRequest
Новая
       31

Как метод FirstOrCreate () GORM (или get_or_create Django) может обеспечить создание только одной строки?

0 голосов
/ 05 января 2020

Я рассматриваю возможность использования 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 ]

Как я понимаю из { ссылка }, однако,

В СУБД SQL подход select-test-insert является ошибкой: ничто не мешает другому процессу вставить «пропущенную» строку между вашими операторами select и insert.

Похоже, что Django s 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, и написать свой собственный запрос?

Второй вопрос, который у меня есть, - как получить эквивалент Django created Boolean в GORM. Должен ли я определить, равен ли RowsAffected результирующего gorm.DB 1, чтобы определить, была ли строка действительно создана или нет?

1 Ответ

1 голос
/ 06 января 2020

Вы должны просто добавить UNIQUE-ограничение на поля модели запроса, и этого будет достаточно, чтобы поддерживать его в db

для Django, что добавило бы мета-класс к модели 

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

для ГОРМ 

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