huanghaosheng/VLoop
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: Worker 队列补充 DLX args + GetAllComments 加排序和分页限制

Browse Source
This commit is contained in:
Sisyphus
2026-04-25 17:10:55 +08:00
parent dd47b48473
commit 48a7eea0f9
2 changed files with 60 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
backend/cmd/worker/main.go
View File

@@ -9,6 +9,7 @@ import (
"feedsystem_video_go/internal/social"
"feedsystem_video_go/internal/video"
"feedsystem_video_go/internal/worker"
mqrabbit "feedsystem_video_go/internal/middleware/rabbitmq"
"log"
"os"
"os/signal"
@@ -194,7 +195,7 @@ func declareSocialTopology(ch *amqp.Channel) error {
false,
false,
false,
nil,
amqp.Table{"x-dead-letter-exchange": mqrabbit.DLXExchange},
)
if err != nil {
return err
@@ -231,7 +232,7 @@ func declarePopularityTopology(ch *amqp.Channel) error {
false,
false,
false,
nil,
amqp.Table{"x-dead-letter-exchange": mqrabbit.DLXExchange},
)
if err != nil {
return err
@@ -265,7 +266,7 @@ func declareLikeTopology(ch *amqp.Channel) error {
false,
false,
false,
nil,
amqp.Table{"x-dead-letter-exchange": mqrabbit.DLXExchange},
)
if err != nil {
return err
@@ -299,7 +300,7 @@ func declareCommentTopology(ch *amqp.Channel) error {
false,
false,
false,
nil,
amqp.Table{"x-dead-letter-exchange": mqrabbit.DLXExchange},
)
if err != nil {
return err

106
backend/internal/video/comment_repo.go
View File

@@ -1,51 +1,55 @@
package video
import (
"context"
"gorm.io/gorm"
)
type CommentRepository struct {
db *gorm.DB
}
func NewCommentRepository(db *gorm.DB) *CommentRepository {
return &CommentRepository{db: db}
}
func (r *CommentRepository) CreateComment(ctx context.Context, comment *Comment) error {
return r.db.WithContext(ctx).Create(comment).Error
}
func (r *CommentRepository) DeleteComment(ctx context.Context, comment *Comment) error {
return r.db.WithContext(ctx).Delete(comment).Error
}
func (r *CommentRepository) GetAllComments(ctx context.Context, videoID uint) ([]Comment, error) {
var comments []Comment
err := r.db.WithContext(ctx).Where("video_id = ?", videoID).Find(&comments).Error
return comments, err
}
func (r *CommentRepository) IsExist(ctx context.Context, id uint) (bool, error) {
var comment Comment
if err := r.db.WithContext(ctx).First(&comment, id).Error; err != nil {
if err == gorm.ErrRecordNotFound {
return false, nil
}
return false, err
}
return true, nil
}
func (r *CommentRepository) GetByID(ctx context.Context, id uint) (*Comment, error) {
var comment Comment
if err := r.db.WithContext(ctx).First(&comment, id).Error; err != nil {
if err == gorm.ErrRecordNotFound {
return nil, nil
}
return nil, err
}
return &comment, nil
}
package video
import (
"context"
"gorm.io/gorm"
)
type CommentRepository struct {
db *gorm.DB
}
func NewCommentRepository(db *gorm.DB) *CommentRepository {
return &CommentRepository{db: db}
}
func (r *CommentRepository) CreateComment(ctx context.Context, comment *Comment) error {
return r.db.WithContext(ctx).Create(comment).Error
}
func (r *CommentRepository) DeleteComment(ctx context.Context, comment *Comment) error {
return r.db.WithContext(ctx).Delete(comment).Error
}
func (r *CommentRepository) GetAllComments(ctx context.Context, videoID uint) ([]Comment, error) {
var comments []Comment
err := r.db.WithContext(ctx).
Where("video_id = ?", videoID).
Order("created_at asc").
Limit(200).
Find(&comments).Error
return comments, err
}
func (r *CommentRepository) IsExist(ctx context.Context, id uint) (bool, error) {
var comment Comment
if err := r.db.WithContext(ctx).First(&comment, id).Error; err != nil {
if err == gorm.ErrRecordNotFound {
return false, nil
}
return false, err
}
return true, nil
}
func (r *CommentRepository) GetByID(ctx context.Context, id uint) (*Comment, error) {
var comment Comment
if err := r.db.WithContext(ctx).First(&comment, id).Error; err != nil {
if err == gorm.ErrRecordNotFound {
return nil, nil
}
return nil, err
}
return &comment, nil
}