77 lines
2.2 KiB
Go
77 lines
2.2 KiB
Go
package dao
|
||
|
||
import (
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
type Character struct {
|
||
ID int32 `gorm:"primaryKey;type:int;not null;comment:'角色ID'"`
|
||
VoiceId int `gorm:"type:int;default null;comment:'声音ID'"`
|
||
AvatarId string `gorm:"type:varchar(32);not null;comment:'角色头像的uuid'"`
|
||
BackgroundIds string `gorm:"type:varchar(255);not null;comment:'聊天背景图集合，逗号分隔，元素是背景图的uuid'"`
|
||
Name string `gorm:"type:varchar(32);not null;comment:'角色名称'"`
|
||
WakeupWords string `gorm:"type:varchar(255);not null;"`
|
||
WorldScenario string `gorm:"type:mediumtext;not null;"`
|
||
Description string `gorm:"type:mediumtext;not null;"`
|
||
Emojis string `gorm:"type:json;not null;"`
|
||
Dialogues string `gorm:"type:mediumtext;not null;"`
|
||
}
|
||
|
||
func (Character) TableName() string { return "character" }
|
||
|
||
func Character_Insert(data *Character) (int32, error) {
|
||
if err := s.db.Create(data).Error; err != nil {
|
||
return 0, err
|
||
}
|
||
return data.ID, nil
|
||
}
|
||
|
||
func Character_Query(characterId int32) (*Character, error) {
|
||
var data Character
|
||
err := s.db.Where("id = ?", characterId).First(&data).Error
|
||
if err != nil {
|
||
if err == gorm.ErrRecordNotFound {
|
||
return nil, nil
|
||
}
|
||
return nil, err
|
||
}
|
||
return &data, nil
|
||
}
|
||
|
||
func Character_QueryAllCharacter() ([]*Character, error) {
|
||
var list []*Character
|
||
err := s.db.Find(&list).Error
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return list, nil
|
||
}
|
||
|
||
// 根据角色ID列表，查询信息
|
||
func Character_QueryList(characterIds []int32) ([]*Character, error) {
|
||
var list []*Character
|
||
if err := s.db.Find(&list, characterIds).Error; err != nil {
|
||
return nil, err
|
||
}
|
||
return list, nil
|
||
}
|
||
|
||
func Character_Delete(characterId int32) error {
|
||
return s.db.Delete(&Character{}, characterId).Error
|
||
}
|
||
|
||
type CharacterBaseInfo struct {
|
||
ID int32 `gorm:"id"`
|
||
AvatarId string `gorm:"avatar_id"`
|
||
Name string `gorm:"name"`
|
||
}
|
||
|
||
// 根据头像的ID列表，查询头像和名称信息
|
||
func Character_QueryList_BaseInfo(characterIds []int32) ([]*CharacterBaseInfo, error) {
|
||
var list []*CharacterBaseInfo
|
||
if err := s.db.Model(&Character{}).Where("id in (?)", characterIds).Select("id, avatar_id, name").Scan(&list).Error; err != nil {
|
||
return nil, err
|
||
}
|
||
return list, nil
|
||
}
|