Golang GORM 實作類似於軟刪除的archived_at

問題內容

我正在使用 GORM V1。我有一個要求，我們想要新增一個類似 GORM deleted_at 的 archived_at 欄位。實體可以存檔和取消存檔，預設我們希望使用 archived_at 為 NULL 來查詢記錄。

我目前的想法是利用 GORM 回呼來註冊回呼

1. 在 gorm:query 回調之前註冊回調。
2. 檢查架構/模型是否需要 archived_at 欄位。
3. 如果確實有該字段，請添加必要條件。

這聽起來不錯，但我如何有效地複製相當於存檔的 Unscoped() 。

• 取得存檔記錄的慣用方法是什麼？如何指示 archived_at 欄位是否應加入到必要的回呼中？
• 我還會遇到這樣的情況：我只想取得存檔記錄（其中 archived_at IS NOT NULL）。

編輯 - 我的用例是同時使用 deleted_at 和 archived_at 字段，而不是使用一個字段代替另一個字段。我想保留軟刪除的能力，同時新增歸檔實體的能力。用戶可能只是被存檔，然後可能被刪除（軟刪除）。

正確答案

邀請回饋。這是我目前想到的。

回呼.go

package db

import (
    "fmt"
    "reflect"

    "gorm.io/gorm"
)

const (
    gormsettingkeyunscopearchive = "unscope_archive"
    structfieldnamearchivedat    = "archivedat"
)

// archivedquerycallback - conditionally adds "where archived_at is null" if the model being queried has the following
// 1. struct field represented by structfieldnamearchivedat
// 2. gorm instance setting gormsettingkeyunscopearchive, see unscopearchive
func archivedquerycallback(db *gorm.db) {
    // check if model is a pointer and has an indirect struct type
    if db.statement.model != nil &&
        reflect.typeof(db.statement.model).kind() == reflect.ptr &&
        reflect.indirect(reflect.valueof(db.statement.model)).kind() == reflect.struct {
        stmt := &gorm.statement{db: db}
        parseerr := stmt.parse(db.statement.model)
        if parseerr != nil {
            panic(parseerr)
        }

        if _, archivedatexists := stmt.schema.fieldsbyname[structfieldnamearchivedat]; archivedatexists {
            v, ok := db.instanceget(gormsettingkeyunscopearchive)
            if ok {
                if v == true {
                    return
                }
            }

            db.where(fmt.sprintf("%s is null", stmt.schema.fieldsbyname[structfieldnamearchivedat].dbname))
        }
    }
}

範圍.go

// unscopearchive - sets a true value for the key gormsettingkeyunscopearchive
func unscopearchive(db *gorm.db) *gorm.db {
    db = db.instanceset(gormsettingkeyunscopearchive, true)

    return db
}

main.go

   type User {
       ID           string         `gorm:"primary_key" json:"id" valid:"uuidv4, optional"`
    CreatedAt    time.Time      `valid:"-" json:"created_at"`
    UpdatedAt    time.Time      `valid:"-" json:"-"`
    DeletedAt    gorm.DeletedAt `sql:"index" valid:"-" json:"-"`
    ArchivedAt   time.Time
   }

   var user []User
   ctx := context.Background()
   dbClient := InitializeGORM() //helper 
   _ := dbClient.WithContext(ctx).Find(&user).Error // SELECT * FROM users WHERE deleted_at IS NULL AND archived_at IS NULL;

   _ := dbClient.WithContext(ctx).Scopes(UnscopeArchive).Find(&user).Error // SELECT * FROM users WHERE deleted_at IS NULL;

以上是Golang GORM 實作類似於軟刪除的archived_at的詳細內容。更多資訊請關注PHP中文網其他相關文章！