otter-space-sdk/pkg/models/orm.go

package models

import (
	"time"

	gonanoid "github.com/matoous/go-nanoid/v2"
	"gorm.io/gorm"
)

type ID interface {
	UserID | SourceID | PostID | UserSourceID | UserFavoriteID
}

type BaseModel[T ID] struct {
	ID        T              `json:"id" gorm:"primaryKey"`
	CreatedAt time.Time      `json:"-"`
	UpdatedAt time.Time      `json:"-"`
	DeletedAt gorm.DeletedAt `json:"-" gorm:"index"`
}

func (base *BaseModel[T]) BeforeCreate(db *gorm.DB) error {
	var defaultVar T

	if base.ID == defaultVar {
		id, err := gonanoid.New(25)
		if err != nil {
			return err
		}

		base.ID = T(id)
	}

	return nil
}