adminandClaude Opus 4.7
ef22f7672a
- 用户域：注册/登录(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>
77 lines
2.9 KiB
Go
77 lines
2.9 KiB
Go
package dao
|
|
|
|
import (
|
|
"context"
|
|
"slogan-agent/styleagent/consts"
|
|
"slogan-agent/styleagent/model/entity"
|
|
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
)
|
|
|
|
var OutfitGenTask = &outfitGenTaskDao{}
|
|
|
|
type outfitGenTaskDao struct{}
|
|
|
|
func init() {
|
|
ctx := context.Background()
|
|
_, err := g.DB().Exec(ctx, `CREATE TABLE IF NOT EXISTS `+consts.TableNameOutfitGenTask+` (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
user_id INTEGER NOT NULL,
|
|
start_date TEXT NOT NULL DEFAULT '',
|
|
end_date TEXT NOT NULL DEFAULT '',
|
|
location TEXT NOT NULL DEFAULT '',
|
|
weather_snapshot TEXT NOT NULL DEFAULT '',
|
|
status TEXT NOT NULL DEFAULT 'pending',
|
|
error TEXT NOT NULL DEFAULT '',
|
|
model_name 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 outfit_generation_task table failed: %v", err)
|
|
}
|
|
_, _ = g.DB().Exec(ctx, "CREATE INDEX IF NOT EXISTS idx_slogan_gen_task_user ON "+consts.TableNameOutfitGenTask+"(user_id, created_at)")
|
|
}
|
|
|
|
func (d *outfitGenTaskDao) Insert(ctx context.Context, data *entity.OutfitGenerationTask) (int64, error) {
|
|
r, err := g.DB().Exec(ctx,
|
|
"INSERT INTO "+consts.TableNameOutfitGenTask+" (user_id, start_date, end_date, location, weather_snapshot, status, error, model_name, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?, datetime('now','localtime'), datetime('now','localtime'))",
|
|
data.UserId, data.StartDate, data.EndDate, data.Location, data.WeatherSnapshot, data.Status, data.Error, data.ModelName)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return r.LastInsertId()
|
|
}
|
|
|
|
func (d *outfitGenTaskDao) GetOne(ctx context.Context, id, userId int64) (*entity.OutfitGenerationTask, error) {
|
|
var t entity.OutfitGenerationTask
|
|
err := g.DB().Model(consts.TableNameOutfitGenTask).Ctx(ctx).
|
|
Where("id", id).Where("user_id", userId).Scan(&t)
|
|
if err != nil || t.Id == 0 {
|
|
return nil, err
|
|
}
|
|
return &t, nil
|
|
}
|
|
|
|
func (d *outfitGenTaskDao) Update(ctx context.Context, id int64, data g.Map) error {
|
|
_, err := g.DB().Model(consts.TableNameOutfitGenTask).Ctx(ctx).
|
|
Data(data).Where("id", id).Update()
|
|
return err
|
|
}
|
|
|
|
func (d *outfitGenTaskDao) UpdateStatus(ctx context.Context, id int64, status, errMsg string) error {
|
|
_, err := g.DB().Model(consts.TableNameOutfitGenTask).Ctx(ctx).Data(g.Map{
|
|
"status": status, "error": errMsg, "updated_at": "datetime('now','localtime')",
|
|
}).Where("id", id).Update()
|
|
return err
|
|
}
|
|
|
|
// ListUnfinished 返回未完成的任务（重启恢复用）
|
|
func (d *outfitGenTaskDao) ListUnfinished(ctx context.Context) ([]*entity.OutfitGenerationTask, error) {
|
|
var list []*entity.OutfitGenerationTask
|
|
err := g.DB().Model(consts.TableNameOutfitGenTask).Ctx(ctx).
|
|
Where("status NOT IN (?)", g.Slice{consts.TaskStatusDone, consts.TaskStatusFailed}).
|
|
OrderAsc("id").Limit(50).Scan(&list)
|
|
return list, err
|
|
}
|