chat_document.go 1.3 KB
package models

import (
	"fmt"
	"gitlab.fjmaimaimai.com/allied-creation/su-micro/cmd/ep/chat/internal/pkg/domain"
	"gorm.io/gorm"
	"gorm.io/plugin/soft_delete"
)

type ChatDocument struct {
	Id        int64 // 唯一标识
	Name      string
	FileType  string
	Status    int
	Metadata  domain.DocumentMetadata `gorm:"type:jsonb;serializer:json"`
	CreatedAt int64
	UpdatedAt int64
	DeletedAt int64
	Version   int
	IsDel     soft_delete.DeletedAt `gorm:"softDelete:flag,DeletedAtField:DeletedAt"`
}

func (m *ChatDocument) TableName() string {
	return "chat.document"
}

func (m *ChatDocument) BeforeCreate(tx *gorm.DB) (err error) {
	// m.CreatedAt = time.Now().Unix()
	// m.UpdatedAt = time.Now().Unix()
	return
}

func (m *ChatDocument) BeforeUpdate(tx *gorm.DB) (err error) {
	// m.UpdatedAt = time.Now().Unix()
	return
}

func (m *ChatDocument) CacheKeyFunc() string {
	if m.Id == 0 {
		return ""
	}
	return fmt.Sprintf("%v:cache:%v:id:%v", domain.ProjectName, m.TableName(), m.Id)
}

func (m *ChatDocument) CacheKeyFuncByObject(obj interface{}) string {
	if v, ok := obj.(*ChatDocument); ok {
		return v.CacheKeyFunc()
	}
	return ""
}

func (m *ChatDocument) CachePrimaryKeyFunc() string {
	if len("") == 0 {
		return ""
	}
	return fmt.Sprintf("%v:cache:%v:primarykey:%v", domain.ProjectName, m.TableName(), "key")
}