This commit is contained in:
@@ -2,9 +2,10 @@ package services
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
|
||||
"quyun/v2/app/requests"
|
||||
"quyun/v2/database/models"
|
||||
"time"
|
||||
|
||||
"github.com/pkg/errors"
|
||||
"github.com/samber/lo"
|
||||
@@ -141,10 +142,28 @@ func (m *posts) GetPostsMapByIDs(ctx context.Context, ids []int64) (map[int64]*m
|
||||
}
|
||||
|
||||
// GetMediaByIds
|
||||
func (m *posts) GetMediaByIds(ctx context.Context, ids []int64) ([]*models.Media, error) {
|
||||
func (m *posts) GetMediasByIds(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()
|
||||
}
|
||||
|
||||
// FindByID
|
||||
func (m *posts) FindByID(ctx context.Context, id int64, conds ...gen.Condition) (*models.Post, error) {
|
||||
tbl, query := models.PostQuery.QueryContext(ctx)
|
||||
if len(conds) > 0 {
|
||||
query = query.Where(conds...)
|
||||
}
|
||||
return query.Where(tbl.ID.Eq(id)).First()
|
||||
}
|
||||
|
||||
// Count
|
||||
func (m *posts) Count(ctx context.Context, conds ...gen.Condition) (int64, error) {
|
||||
_, query := models.PostQuery.QueryContext(ctx)
|
||||
if len(conds) > 0 {
|
||||
query = query.Where(conds...)
|
||||
}
|
||||
return query.Count()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user