chat_dataset.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 ChatDataset struct {
	Id        int64 // 唯一标识
	CompanyId int64
	UserId    int64
	Name      string // 知识库名称
	Desc      string // 描述
	Status    int    // 1:启用 2:禁用
	CreatedAt int64
	UpdatedAt int64
	DeletedAt int64
	Version   int
	IsDel     soft_delete.DeletedAt `gorm:"softDelete:flag,DeletedAtField:DeletedAt"`
}

func (m *ChatDataset) TableName() string {
	return "chat.dataset"
}

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

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

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

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

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