This commit is contained in:
150
backend_v1/app/services/posts.go
Normal file
150
backend_v1/app/services/posts.go
Normal file
@@ -0,0 +1,150 @@
|
||||
package services
|
||||
|
||||
import (
|
||||
"context"
|
||||
"quyun/v2/app/requests"
|
||||
"quyun/v2/database/models"
|
||||
"time"
|
||||
|
||||
"github.com/pkg/errors"
|
||||
"github.com/samber/lo"
|
||||
"go.ipao.vip/gen"
|
||||
)
|
||||
|
||||
// @provider
|
||||
type posts struct{}
|
||||
|
||||
// IncrViewCount
|
||||
func (m *posts) IncrViewCount(ctx context.Context, postID int64) error {
|
||||
tbl, query := models.PostQuery.QueryContext(ctx)
|
||||
|
||||
_, err := query.Where(tbl.ID.Eq(postID)).Inc(tbl.Views, 1)
|
||||
if err != nil {
|
||||
return errors.Wrapf(err, "failed to increment view count for post %d", postID)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// List
|
||||
func (m *posts) List(
|
||||
ctx context.Context,
|
||||
pagination *requests.Pagination,
|
||||
conds ...gen.Condition,
|
||||
) (*requests.Pager, error) {
|
||||
pagination.Format()
|
||||
|
||||
tbl, query := models.PostQuery.QueryContext(ctx)
|
||||
items, cnt, err := query.Where(conds...).
|
||||
Order(tbl.ID.Desc()).
|
||||
FindByPage(int(pagination.Offset()), int(pagination.Limit))
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "list post failed")
|
||||
}
|
||||
|
||||
return &requests.Pager{
|
||||
Items: items,
|
||||
Total: cnt,
|
||||
Pagination: *pagination,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// SendTo
|
||||
func (m *posts) SendTo(ctx context.Context, postID, userID int64) error {
|
||||
model := &models.UserPost{
|
||||
UserID: userID,
|
||||
PostID: postID,
|
||||
Price: -1,
|
||||
}
|
||||
return model.Create(ctx)
|
||||
}
|
||||
|
||||
// PostBoughtStatistics 获取指定文件 ID 的购买次数
|
||||
func (m *posts) BoughtStatistics(ctx context.Context, postIds []int64) (map[int64]int64, error) {
|
||||
tbl, query := models.UserPostQuery.QueryContext(ctx)
|
||||
|
||||
var items []struct {
|
||||
Count int64
|
||||
PostID int64
|
||||
}
|
||||
err := query.Select(
|
||||
tbl.UserID.Count().As("count"),
|
||||
tbl.PostID,
|
||||
).
|
||||
Where(tbl.PostID.In(postIds...)).
|
||||
Group(tbl.PostID).Scan(&items)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
result := make(map[int64]int64)
|
||||
for _, item := range items {
|
||||
result[item.PostID] = item.Count
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
|
||||
// Bought 获取用户购买记录
|
||||
func (m *posts) Bought(ctx context.Context, userId int64, pagination *requests.Pagination) (*requests.Pager, error) {
|
||||
pagination.Format()
|
||||
tbl, query := models.UserPostQuery.QueryContext(ctx)
|
||||
|
||||
items, cnt, err := query.
|
||||
Where(tbl.UserID.Eq(userId)).
|
||||
FindByPage(int(pagination.Offset()), int(pagination.Limit))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
postIds := lo.Map(items, func(item *models.UserPost, _ int) int64 { return item.PostID })
|
||||
postInfoMap := lo.KeyBy(items, func(item *models.UserPost) int64 { return item.PostID })
|
||||
|
||||
postItemMap, err := m.GetPostsMapByIDs(ctx, postIds)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
type retItem struct {
|
||||
Title string `json:"title"`
|
||||
Price int64 `json:"price"`
|
||||
BoughtAt time.Time `json:"bought_at"`
|
||||
}
|
||||
|
||||
var retItems []retItem
|
||||
for _, postID := range postIds {
|
||||
post, ok := postItemMap[postID]
|
||||
if !ok {
|
||||
continue
|
||||
}
|
||||
|
||||
postInfo := postInfoMap[postID]
|
||||
|
||||
retItems = append(retItems, retItem{
|
||||
Title: post.Title,
|
||||
Price: postInfo.Price,
|
||||
BoughtAt: postInfo.CreatedAt,
|
||||
})
|
||||
}
|
||||
|
||||
return &requests.Pager{
|
||||
Items: retItems,
|
||||
Total: cnt,
|
||||
Pagination: *pagination,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// GetPostsMapByIDs
|
||||
func (m *posts) GetPostsMapByIDs(ctx context.Context, ids []int64) (map[int64]*models.Post, error) {
|
||||
tbl, query := models.PostQuery.QueryContext(ctx)
|
||||
posts, err := query.Where(tbl.ID.In(ids...)).Find()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return lo.KeyBy(posts, func(item *models.Post) int64 { return item.ID }), nil
|
||||
}
|
||||
|
||||
// GetMediaByIds
|
||||
func (m *posts) GetMediaByIds(ctx context.Context, ids []int64) ([]*models.Media, error) {
|
||||
if len(ids) == 0 {
|
||||
return nil, nil
|
||||
}
|
||||
tbl, query := models.MediaQuery.QueryContext(ctx)
|
||||
return query.Where(tbl.ID.In(ids...)).Find()
|
||||
}
|
||||
Reference in New Issue
Block a user