117 lines
2.9 KiB
Go
Executable File
117 lines
2.9 KiB
Go
Executable File
package mediabookshelfdata
|
|
|
|
import (
|
|
"91porn-server/models/v/mediabookshelfmod"
|
|
"91porn-server/models/v/mediamod"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
"time"
|
|
)
|
|
|
|
// AppData 需要组装返回的数据
|
|
type AppData struct {
|
|
BaseInfo *mediabookshelfmod.MediaBookshelfInfo `json:"baseInfo"` // 基础信息
|
|
}
|
|
|
|
// FormatAppDataList 格式化app列表数据
|
|
func FormatAppDataList(origin []mediabookshelfmod.MediaBookshelf) (res []AppData) {
|
|
res = make([]AppData, len(origin))
|
|
if len(origin) == 0 {
|
|
return
|
|
}
|
|
|
|
// 组装返回数据
|
|
for i, item := range origin {
|
|
res[i] = FormatAppData(item)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// FormatAppData 格式化app数据
|
|
func FormatAppData(item mediabookshelfmod.MediaBookshelf) (res AppData) {
|
|
if item.ID.IsZero() {
|
|
return
|
|
}
|
|
|
|
// 组装返回数据
|
|
res.BaseInfo = &mediabookshelfmod.MediaBookshelfInfo{
|
|
ID: item.ID,
|
|
Name: item.Name,
|
|
UID: item.UID,
|
|
ReadAt: item.ReadAt,
|
|
CreatedAt: item.CreatedAt,
|
|
UpdateTime: item.UpdateTime,
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// AppSimpleData 仅返回基础数据
|
|
type AppSimpleData *mediabookshelfmod.MediaBookshelfInfo
|
|
|
|
// FormatAppSimpleDataList 格式化app列表数据
|
|
func FormatAppSimpleDataList(origin []mediabookshelfmod.MediaBookshelf, skip, limit int64, sort bson.D) (res []AppSimpleData) {
|
|
res = make([]AppSimpleData, len(origin))
|
|
if len(origin) == 0 {
|
|
return
|
|
}
|
|
|
|
var mids []primitive.ObjectID
|
|
for i := range origin {
|
|
mids = append(mids, origin[i].MID)
|
|
}
|
|
|
|
filter := bson.M{"_id": bson.M{"$in": mids}}
|
|
opts := options.Find().SetSkip(skip).SetLimit(limit)
|
|
if len(sort) > 0 {
|
|
opts = opts.SetSort(sort)
|
|
}
|
|
media, err := mediamod.QueryMediaByCond(filter, opts)
|
|
if err != nil {
|
|
return
|
|
}
|
|
m := make(map[primitive.ObjectID]*mediamod.Media)
|
|
for i := range media {
|
|
m[media[i].ID] = media[i]
|
|
}
|
|
// 组装返回数据
|
|
for i, item := range origin {
|
|
res[i] = FormatAppSimpleData(item, m[item.MID])
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// FormatAppSimpleData 格式化app数据
|
|
func FormatAppSimpleData(item mediabookshelfmod.MediaBookshelf, media *mediamod.Media) (res AppSimpleData) {
|
|
if item.ID.IsZero() {
|
|
return
|
|
}
|
|
|
|
// 组装返回数据
|
|
res = &mediabookshelfmod.MediaBookshelfInfo{
|
|
ID: item.MID,
|
|
//MID: item.MID,
|
|
Name: media.Title,
|
|
VerticalCover: media.VerticalCover,
|
|
TotalEpisode: media.TotalEpisode,
|
|
CurrentEpisode: media.CurrentEpisode,
|
|
MediaType: media.MediaType,
|
|
MediaSubType: int(media.MediaSubType),
|
|
UpdateStatus: media.UpdateStatus,
|
|
CountBrowse: media.CountBrowse,
|
|
Permission: media.Permission,
|
|
Price: media.Price,
|
|
UID: item.UID,
|
|
ReadAt: item.ReadAt,
|
|
CreatedAt: item.CreatedAt,
|
|
UpdateTime: item.UpdateTime,
|
|
}
|
|
if time.Now().Add(-time.Hour*24*7).Unix() < media.ContentUpdateTime.Unix() {
|
|
res.IsNew = true
|
|
}
|
|
return
|
|
}
|