mayfly-go/server/pkg/model/model.go

package model

import (
	"time"
)

type IdGenType int

const (
	IdColumn         = "id"
	DeletedColumn    = "is_deleted" // 删除字段
	DeleteTimeColumn = "delete_time"

	ModelDeleted   int8 = 1
	ModelUndeleted int8 = 0

	IdGenTypeNone      IdGenType = 0 // 数据库处理
	IdGenTypeTimestamp IdGenType = 1 // 当前时间戳
)

// 实体接口
type ModelI interface {

	// id生成策略
	// IdGenType() IdGenType

	// 使用当前登录账号信息设置实体结构体的基础信息
	//
	// 如创建时间，修改时间，创建者，修改者信息
	SetBaseInfo(idGenType IdGenType, account *LoginAccount)
}

type IdModel struct {
	Id uint64 `json:"id"`
}

// func (m *IdModel) IdGenType() IdGenType {
// 	// 默认由数据库自行生成
// 	return IdGenTypeNone
// }

func (m *IdModel) SetBaseInfo(idGenType IdGenType, account *LoginAccount) {
	// 存在id，则赋值
	if m.Id != 0 {
		return
	}
	m.Id = GetIdByGenType(idGenType)
}

// 含有删除字段模型
type DeletedModel struct {
	IdModel
	IsDeleted  int8       `json:"-" gorm:"column:is_deleted;default:0"`
	DeleteTime *time.Time `json:"-"`
}

func (m *DeletedModel) SetBaseInfo(idGenType IdGenType, account *LoginAccount) {
	if m.Id == 0 {
		m.IdModel.SetBaseInfo(idGenType, account)
		m.IsDeleted = ModelUndeleted
	}
}

// 含有删除、创建字段模型
type CreateModel struct {
	DeletedModel
	CreateTime *time.Time `json:"createTime"`
	CreatorId  uint64     `json:"creatorId"`
	Creator    string     `json:"creator"`
}

func (m *CreateModel) SetBaseInfo(idGenType IdGenType, account *LoginAccount) {
	if m.Id != 0 {
		return
	}

	m.DeletedModel.SetBaseInfo(idGenType, account)
	nowTime := time.Now()
	m.CreateTime = &nowTime
	if account != nil {
		m.CreatorId = account.Id
		m.Creator = account.Username
	}
}

// 基础实体模型，数据表最基础字段，尽量每张表都包含这些字段
type Model struct {
	DeletedModel

	CreateTime *time.Time `json:"createTime"`
	CreatorId  uint64     `json:"creatorId"`
	Creator    string     `json:"creator"`
	UpdateTime *time.Time `json:"updateTime"`
	ModifierId uint64     `json:"modifierId"`
	Modifier   string     `json:"modifier"`
}

// 设置基础信息. 如创建时间，修改时间，创建者，修改者信息
func (m *Model) SetBaseInfo(idGenType IdGenType, account *LoginAccount) {
	nowTime := time.Now()
	isCreate := m.Id == 0
	if isCreate {
		m.IsDeleted = ModelUndeleted
		m.CreateTime = &nowTime
		m.IdModel.SetBaseInfo(idGenType, account)
	}
	m.UpdateTime = &nowTime

	if account == nil {
		return
	}
	id := account.Id
	name := account.Username
	if isCreate {
		m.CreatorId = id
		m.Creator = name
	}
	m.Modifier = name
	m.ModifierId = id
}

// 根据id生成类型，生成id
func GetIdByGenType(genType IdGenType) uint64 {
	if genType == IdGenTypeTimestamp {
		return uint64(time.Now().Unix())
	}
	return 0
}