Victor_Jay
b9d90fe7ec
- Post 模型新增 ViewsCount 冗余计数字段 - 新建 PostReadLog 表（user_id + post_id 唯一索引），已登录用户每篇文章只计一次 - 新建 PostGuestReadLog 表（visitor_id + post_id 唯一索引），访客基于 Cookie 标识去重 - Repository 层新增 RecordRead / RecordGuestRead / IncrementViewsCount - Service 层编排去重-计数逻辑，计数失败不影响页面渲染 - Controller 层 ShowPage + ShowAPI 触发计数，仅 approved 状态生效 - 双层防刷：唯一索引去重 + 2 秒冷却间隔 - 访客 Cookie (visitor_id) 基于 crypto/rand 生成，30 天有效期
244 lines
7.8 KiB
Go
244 lines
7.8 KiB
Go
package repository
|
||
|
||
import (
|
||
"time"
|
||
|
||
"metazone.cc/metalab/internal/model"
|
||
|
||
"gorm.io/gorm"
|
||
"gorm.io/gorm/clause"
|
||
)
|
||
|
||
// PostRepo 帖子数据访问
|
||
type PostRepo struct {
|
||
db *gorm.DB
|
||
}
|
||
|
||
// NewPostRepo 构造函数
|
||
func NewPostRepo(db *gorm.DB) *PostRepo {
|
||
return &PostRepo{db: db}
|
||
}
|
||
|
||
// Create 创建帖子
|
||
func (r *PostRepo) Create(post *model.Post) error {
|
||
return r.db.Create(post).Error
|
||
}
|
||
|
||
// FindByID 按 ID 查找帖子
|
||
func (r *PostRepo) FindByID(id uint) (*model.Post, error) {
|
||
var post model.Post
|
||
err := r.db.First(&post, id).Error
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return &post, nil
|
||
}
|
||
|
||
// FindByIDWithAuthor 按 ID 查找帖子（含作者用户名）
|
||
func (r *PostRepo) FindByIDWithAuthor(id uint) (*model.Post, error) {
|
||
var post model.Post
|
||
err := r.db.Table("posts").
|
||
Select("posts.*, users.username as author_name").
|
||
Joins("LEFT JOIN users ON users.uid = posts.user_id").
|
||
Where("posts.id = ?", id).
|
||
First(&post).Error
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return &post, nil
|
||
}
|
||
|
||
// buildQuery 构建帖子查询公共部分（联表 + 未删除 + 关键词 + 可选状态过滤）
|
||
func (r *PostRepo) buildQuery(keyword, status string) *gorm.DB {
|
||
query := r.db.Table("posts").
|
||
Select("posts.*, users.username as author_name").
|
||
Joins("LEFT JOIN users ON users.uid = posts.user_id").
|
||
Where("posts.deleted_at IS NULL")
|
||
|
||
if keyword != "" {
|
||
like := "%" + keyword + "%"
|
||
query = query.Where("posts.title LIKE ?", like)
|
||
}
|
||
if status != "" {
|
||
query = query.Where("posts.status = ?", status)
|
||
}
|
||
return query
|
||
}
|
||
|
||
// FindPageable 分页查询帖子列表（仅 approved 状态，关键词模糊搜索标题）
|
||
func (r *PostRepo) FindPageable(keyword string, offset, limit int) ([]model.Post, int64, error) {
|
||
query := r.buildQuery(keyword, model.PostStatusApproved)
|
||
return r.pageResults(query, offset, limit)
|
||
}
|
||
|
||
// FindAdminPageable 管理后台分页查询（全状态筛选，待审置顶）
|
||
func (r *PostRepo) FindAdminPageable(keyword, status string, offset, limit int) ([]model.Post, int64, error) {
|
||
query := r.buildQuery(keyword, status)
|
||
return r.pageResultsAdmin(query, offset, limit)
|
||
}
|
||
|
||
// FindByUserID 分页查询某用户发布的帖子（仅 approved，含作者用户名）
|
||
func (r *PostRepo) FindByUserID(userID uint, offset, limit int) ([]model.Post, int64, error) {
|
||
query := r.buildQuery("", model.PostStatusApproved).
|
||
Where("posts.user_id = ?", userID)
|
||
return r.pageResults(query, offset, limit)
|
||
}
|
||
|
||
// CountUserPosts 统计某用户的已发布文章数
|
||
func (r *PostRepo) CountUserPosts(userID uint) (int64, error) {
|
||
var count int64
|
||
err := r.db.Model(&model.Post{}).
|
||
Where("user_id = ? AND status = ? AND deleted_at IS NULL", userID, model.PostStatusApproved).
|
||
Count(&count).Error
|
||
return count, err
|
||
}
|
||
|
||
// GetUserStats 获取用户的文章统计数据（总获赞/总收藏）
|
||
func (r *PostRepo) GetUserStats(userID uint) (likes, favorites int64, err error) {
|
||
type stats struct {
|
||
TotalLikes int64
|
||
TotalFavorites int64
|
||
}
|
||
var s stats
|
||
err = r.db.Model(&model.Post{}).
|
||
Select("COALESCE(SUM(likes_count), 0) AS total_likes, COALESCE(SUM(favorites_count), 0) AS total_favorites").
|
||
Where("user_id = ? AND status = ? AND deleted_at IS NULL", userID, model.PostStatusApproved).
|
||
Scan(&s).Error
|
||
return s.TotalLikes, s.TotalFavorites, err
|
||
}
|
||
|
||
// FindByUserIDAndStatus 分页查询某用户指定状态的帖子（空 status=全状态）
|
||
func (r *PostRepo) FindByUserIDAndStatus(userID uint, status string, offset, limit int) ([]model.Post, int64, error) {
|
||
query := r.buildQuery("", status).
|
||
Where("posts.user_id = ?", userID)
|
||
return r.pageResults(query, offset, limit)
|
||
}
|
||
|
||
// FindPendingRevisions 查询有待审修订的帖子（approved 且 pending_body 不为空）
|
||
func (r *PostRepo) FindPendingRevisions(keyword string, offset, limit int) ([]model.Post, int64, error) {
|
||
query := r.buildQuery(keyword, model.PostStatusApproved).
|
||
Where("posts.pending_body != ''")
|
||
return r.pageResults(query, offset, limit)
|
||
}
|
||
|
||
// pageResults 执行 Count + Offset/Limit + ORDER BY
|
||
func (r *PostRepo) pageResults(query *gorm.DB, offset, limit int) ([]model.Post, int64, error) {
|
||
var total int64
|
||
if err := query.Count(&total).Error; err != nil {
|
||
return nil, 0, err
|
||
}
|
||
|
||
var posts []model.Post
|
||
err := query.Order("posts.created_at DESC").
|
||
Offset(offset).Limit(limit).Find(&posts).Error
|
||
if err != nil {
|
||
return nil, 0, err
|
||
}
|
||
return posts, total, nil
|
||
}
|
||
|
||
// pageResultsAdmin 管理后台分页：待审稿件（pending 或 修订待审）置顶优先
|
||
func (r *PostRepo) pageResultsAdmin(query *gorm.DB, offset, limit int) ([]model.Post, int64, error) {
|
||
var total int64
|
||
if err := query.Count(&total).Error; err != nil {
|
||
return nil, 0, err
|
||
}
|
||
|
||
var posts []model.Post
|
||
err := query.Order(`CASE WHEN posts.status = 'pending' OR (posts.status = 'approved' AND posts.pending_body != '') THEN 0 ELSE 1 END ASC, posts.created_at DESC`).
|
||
Offset(offset).Limit(limit).Find(&posts).Error
|
||
if err != nil {
|
||
return nil, 0, err
|
||
}
|
||
return posts, total, nil
|
||
}
|
||
|
||
// Update 更新帖子
|
||
func (r *PostRepo) Update(post *model.Post) error {
|
||
return r.db.Save(post).Error
|
||
}
|
||
|
||
// SoftDelete 软删除帖子
|
||
func (r *PostRepo) SoftDelete(id uint) error {
|
||
return r.db.Delete(&model.Post{}, id).Error
|
||
}
|
||
|
||
// CountPostsByStatus 统计帖子数量（status 为空则统计全部未删除帖子）
|
||
func (r *PostRepo) CountPostsByStatus(status string) (int64, error) {
|
||
query := r.buildQuery("", status)
|
||
var total int64
|
||
err := query.Count(&total).Error
|
||
return total, err
|
||
}
|
||
|
||
// CountPending 统计待审核帖子数（pending 状态 + approved 但有待审修订）
|
||
func (r *PostRepo) CountPending() (int64, error) {
|
||
var total int64
|
||
err := r.db.Table("posts").
|
||
Where("deleted_at IS NULL").
|
||
Where("status = ? OR (status = ? AND pending_body != '')", model.PostStatusPending, model.PostStatusApproved).
|
||
Count(&total).Error
|
||
return total, err
|
||
}
|
||
|
||
// Restore 恢复软删除
|
||
func (r *PostRepo) Restore(id uint) error {
|
||
result := r.db.Unscoped().Model(&model.Post{}).Where("id = ?", id).Update("deleted_at", nil)
|
||
if result.Error != nil {
|
||
return result.Error
|
||
}
|
||
if result.RowsAffected == 0 {
|
||
return gorm.ErrRecordNotFound
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// readCooldown 两次阅读记录最小间隔
|
||
const readCooldown = 2 * time.Second
|
||
|
||
// RecordRead 记录已登录用户阅读文章（防刷量：取去重+冷却）
|
||
// 返回 true 表示首次阅读（新记录），false 表示重复或冷却中
|
||
func (r *PostRepo) RecordRead(userID, postID uint) (bool, error) {
|
||
// 冷却检查：同用户两次阅读之间至少间隔 2 秒
|
||
var lastRead model.PostReadLog
|
||
if err := r.db.Where("user_id = ?", userID).
|
||
Order("read_at DESC").First(&lastRead).Error; err == nil {
|
||
if time.Since(lastRead.ReadAt) < readCooldown {
|
||
return false, nil
|
||
}
|
||
}
|
||
|
||
result := r.db.Clauses(clause.OnConflict{DoNothing: true}).
|
||
Create(&model.PostReadLog{UserID: userID, PostID: postID})
|
||
if result.Error != nil {
|
||
return false, result.Error
|
||
}
|
||
return result.RowsAffected > 0, nil
|
||
}
|
||
|
||
// RecordGuestRead 记录访客阅读文章（基于 Cookie visitor_id 去重+冷却）
|
||
// 返回 true 表示首次阅读（新记录），false 表示重复或冷却中
|
||
func (r *PostRepo) RecordGuestRead(visitorID string, postID uint) (bool, error) {
|
||
// 冷却检查：同访客两次阅读之间至少间隔 2 秒
|
||
var lastRead model.PostGuestReadLog
|
||
if err := r.db.Where("visitor_id = ?", visitorID).
|
||
Order("read_at DESC").First(&lastRead).Error; err == nil {
|
||
if time.Since(lastRead.ReadAt) < readCooldown {
|
||
return false, nil
|
||
}
|
||
}
|
||
|
||
result := r.db.Clauses(clause.OnConflict{DoNothing: true}).
|
||
Create(&model.PostGuestReadLog{VisitorID: visitorID, PostID: postID})
|
||
if result.Error != nil {
|
||
return false, result.Error
|
||
}
|
||
return result.RowsAffected > 0, nil
|
||
}
|
||
|
||
// IncrementViewsCount 阅读数 +1
|
||
func (r *PostRepo) IncrementViewsCount(postID uint) error {
|
||
return r.db.Model(&model.Post{}).Where("id = ?", postID).
|
||
Update("views_count", gorm.Expr("views_count + 1")).Error
|
||
}
|