Victor_Jay
e1d7c318ab
- 新增域能完整分层：EnergyLog/PostEnergizeLog/DailyExpSummary 模型、EnergyRepo、EnergyService、EnergyController/AdminEnergyController - 支持签到获取域能(+1)、帖子赋能(-1/-2)/被赋能(+1/+2)、改名扣能(-6)/退款(+6)、删帖扣能(-2) - 赋能单人单帖上限20域能，每日赋能获得经验上限50 - 管理员支持批量调整域能(owner)和查询域能日志(admin) - LV0用户(exp<1)禁止签到和发帖，余额为负可签到/被赋能/删帖但不能赋能/改名 - 首次改名免费(通过HasCompletedTask判重)，改名扣能在提交审核时执行 - 移除导航栏手动签到按钮，签到融入域能系统(自动签到) - 新增个人中心域能TAB展示余额和流水日志(近7天) - 新增帖子详情页赋能按钮(轻赋/重赋) - 管理后台新增域能管理和域能日志页面
76 lines
1.9 KiB
Go
76 lines
1.9 KiB
Go
package repository
|
|
|
|
import (
|
|
"time"
|
|
|
|
"metazone.cc/metalab/internal/model"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// LevelRepo 积分/等级/签到/任务数据访问
|
|
type LevelRepo struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewLevelRepo 构造函数
|
|
func NewLevelRepo(db *gorm.DB) *LevelRepo {
|
|
return &LevelRepo{db: db}
|
|
}
|
|
|
|
// FindByID 按 UID 查找用户（实现 userLevelStore 接口）
|
|
func (r *LevelRepo) FindByID(id uint) (*model.User, error) {
|
|
var user model.User
|
|
err := r.db.First(&user, id).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &user, nil
|
|
}
|
|
|
|
// AddExp 原子增加用户经验值
|
|
func (r *LevelRepo) AddExp(userID uint, delta int) error {
|
|
return r.db.Model(&model.User{}).
|
|
Where("uid = ?", userID).
|
|
UpdateColumn("exp", gorm.Expr("exp + ?", delta)).Error
|
|
}
|
|
|
|
// FindCheckIn 查询指定自然日的签到记录
|
|
func (r *LevelRepo) FindCheckIn(userID uint, date time.Time) (*model.UserCheckIn, error) {
|
|
var ci model.UserCheckIn
|
|
err := r.db.Where("user_id = ? AND date = ?", userID, date).First(&ci).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &ci, nil
|
|
}
|
|
|
|
// CreateCheckIn 创建签到记录
|
|
func (r *LevelRepo) CreateCheckIn(ci *model.UserCheckIn) error {
|
|
return r.db.Create(ci).Error
|
|
}
|
|
|
|
// FindTask 查询用户是否已完成指定任务
|
|
func (r *LevelRepo) FindTask(userID uint, taskType string) (*model.UserTask, error) {
|
|
var t model.UserTask
|
|
err := r.db.Where("user_id = ? AND task_type = ?", userID, taskType).First(&t).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &t, nil
|
|
}
|
|
|
|
// CreateTask 创建任务完成记录
|
|
func (r *LevelRepo) CreateTask(task *model.UserTask) error {
|
|
return r.db.Create(task).Error
|
|
}
|
|
|
|
// HasTask 检查用户是否已完成过指定任务（无副作用查询）
|
|
func (r *LevelRepo) HasTask(userID uint, taskType string) (bool, error) {
|
|
var count int64
|
|
err := r.db.Model(&model.UserTask{}).
|
|
Where("user_id = ? AND task_type = ?", userID, taskType).
|
|
Count(&count).Error
|
|
return count > 0, err
|
|
}
|