Files
slogan/styleagent/dao/avatar_model_dao.go
T
adminandClaude Opus 4.7 ef22f7672a feat: slogan-agent MVP 服务端完整实现
- 用户域:注册/登录(JWT)/修改密码/个人资料
- 照片/衣橱/身形/化身:上传存储 + 3D 化身模板匹配
- 穿搭生成:天气(高德+和风+缓存) → 规则预筛 → LLM 规划(1次调用)
  → 规则评分(5维100分制) → 全低分触发 LLM 兜底创作 → 异步任务状态机
- 效果图:选主方案后异步生成 3 视角(mock/wanx 供应商 + 内容 hash 缓存 + 每日限额)
- 商业化:合作门店列表(seed 4 家)
- 冒烟:全链路端到端验证通过(mock LLM/天气),23 个 API 端点

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
2026-07-31 12:15:11 +08:00

61 lines
2.2 KiB
Go

package dao
import (
"context"
"slogan-agent/styleagent/consts"
"slogan-agent/styleagent/model/entity"
"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,
face_template_id INTEGER NOT NULL DEFAULT 0,
body_template_id INTEGER NOT NULL DEFAULT 0,
skin_tone_index INTEGER NOT NULL DEFAULT 0,
face_texture_url TEXT NOT NULL DEFAULT '',
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)
}
}
func (d *avatarModelDao) Insert(ctx context.Context, data *entity.AvatarModel) (int64, error) {
r, err := g.DB().Exec(ctx,
"INSERT INTO "+consts.TableNameAvatarModel+" (user_id, face_template_id, body_template_id, skin_tone_index, face_texture_url, glb_url, build_status, error, params_snapshot, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, datetime('now','localtime'), datetime('now','localtime'))",
data.UserId, data.FaceTemplateId, data.BodyTemplateId, data.SkinToneIndex,
data.FaceTextureUrl, data.GlbUrl, data.BuildStatus, data.Error, data.ParamsSnapshot)
if err != nil {
return 0, err
}
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).
Where("user_id", userId).OrderDesc("id").Scan(&a)
if err != nil || a.Id == 0 {
return nil, err
}
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()
return err
}