90 lines
3.0 KiB
Go
90 lines
3.0 KiB
Go
package dao
|
|
|
|
import (
|
|
"context"
|
|
"slogan-agent/common"
|
|
"slogan-agent/styleagent/consts"
|
|
"slogan-agent/styleagent/model/entity"
|
|
|
|
"github.com/gogf/gf/v2/database/gdb"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
)
|
|
|
|
var AvatarModel = &avatarModelDao{}
|
|
|
|
type avatarModelDao struct{}
|
|
|
|
func init() {
|
|
ctx := context.Background()
|
|
_, err := g.DB().Exec(ctx, `CREATE TABLE IF NOT EXISTS `+consts.TableNameAvatarModel+` (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
user_id INTEGER NOT NULL UNIQUE,
|
|
glb_url TEXT NOT NULL DEFAULT '',
|
|
build_status TEXT NOT NULL DEFAULT 'pending',
|
|
error TEXT NOT NULL DEFAULT '',
|
|
params_snapshot TEXT NOT NULL DEFAULT '',
|
|
created_at DATETIME DEFAULT (datetime('now','localtime')),
|
|
updated_at DATETIME DEFAULT (datetime('now','localtime'))
|
|
)`)
|
|
if err != nil {
|
|
g.Log().Warningf(ctx, "create avatar_model table failed: %v", err)
|
|
}
|
|
// 迁移：移除已废弃列（模板/肤色/纹理从未写入，frames_url 已从代码删除）
|
|
for _, col := range []string{"face_template_id", "body_template_id", "skin_tone_index", "face_texture_url", "frames_url"} {
|
|
if !avatarColumnExists(ctx, consts.TableNameAvatarModel, col) {
|
|
continue
|
|
}
|
|
if _, err := g.DB().Exec(ctx, "ALTER TABLE "+consts.TableNameAvatarModel+" DROP COLUMN "+col); err != nil {
|
|
g.Log().Warningf(ctx, "drop avatar_model column %s failed: %v", col, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func avatarColumnExists(ctx context.Context, table, column string) bool {
|
|
rows, err := g.DB().Ctx(ctx).Query(ctx, "PRAGMA table_info("+table+")")
|
|
if err != nil {
|
|
g.Log().Warningf(ctx, "query avatar_model columns failed: %v", err)
|
|
return false
|
|
}
|
|
for _, r := range rows {
|
|
if r["name"].String() == column {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (d *avatarModelDao) Insert(ctx context.Context, data *entity.AvatarModel) (int64, error) {
|
|
r, err := g.DB().Exec(ctx,
|
|
"INSERT INTO "+consts.TableNameAvatarModel+" (user_id, glb_url, build_status, error, params_snapshot, created_at, updated_at) VALUES (?, ?, ?, ?, ?, datetime('now','localtime'), datetime('now','localtime'))",
|
|
data.UserId, data.GlbUrl, data.BuildStatus, data.Error, data.ParamsSnapshot)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
common.CacheClear(ctx, g.DB(), consts.TableNameAvatarModel)
|
|
return r.LastInsertId()
|
|
}
|
|
|
|
func (d *avatarModelDao) GetByUser(ctx context.Context, userId int64) (*entity.AvatarModel, error) {
|
|
var a entity.AvatarModel
|
|
err := g.DB().Model(consts.TableNameAvatarModel).Ctx(ctx).
|
|
Cache(gdb.CacheOption{Duration: common.CacheTTL(), Name: common.CacheName(consts.TableNameAvatarModel, "GetByUser", userId)}).
|
|
Where("user_id", userId).OrderDesc("id").Scan(&a)
|
|
if err != nil && !common.IsNotFound(err) {
|
|
return nil, err
|
|
}
|
|
if a.Id == 0 {
|
|
return nil, nil
|
|
}
|
|
return &a, nil
|
|
}
|
|
|
|
func (d *avatarModelDao) Update(ctx context.Context, id int64, data map[string]any) error {
|
|
_, err := g.DB().Model(consts.TableNameAvatarModel).Ctx(ctx).Data(data).Where("id", id).Update()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
common.CacheClear(ctx, g.DB(), consts.TableNameAvatarModel)
|
|
return nil
|
|
}
|