Claude Opus 5
+89
@@ -0,0 +1,89 @@
|
||||
package aiplazadata
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"91porn-server/common/cachev2"
|
||||
"91porn-server/common/db"
|
||||
"91porn-server/models"
|
||||
"91porn-server/models/v/aiplazamod"
|
||||
|
||||
"go.mongodb.org/mongo-driver/bson"
|
||||
"go.mongodb.org/mongo-driver/bson/primitive"
|
||||
)
|
||||
|
||||
type ListRes struct {
|
||||
List []aiplazamod.AiPlaza
|
||||
Count int64
|
||||
HasNext bool
|
||||
}
|
||||
|
||||
// GetListFromCache 从缓存获取列表数据
|
||||
func GetListFromCache(cond bson.M, skip, limit int64, sort bson.D) ([]aiplazamod.AiPlaza, int64, bool, error) {
|
||||
var data ListRes
|
||||
_, err := cachev2.Classes().CacheTime(10*time.Minute).AutoListKey(models.AiPlaza).ResBind(&data).Cache(GetListData, cond, skip, limit, sort)
|
||||
return data.List, data.Count, data.HasNext, err
|
||||
}
|
||||
|
||||
// GetListData 获取列表数据并组装适合缓存格式返回
|
||||
func GetListData(cond bson.M, skip, limit int64, sort bson.D) (res ListRes, err error) {
|
||||
res.List, res.Count, res.HasNext, err = aiplazamod.GetList(cond, skip, limit, sort)
|
||||
return
|
||||
}
|
||||
|
||||
// GetInfoFromCache 获取数据详情
|
||||
func GetInfoFromCache(id primitive.ObjectID) (res aiplazamod.AiPlaza, err error) {
|
||||
_, err = cachev2.Classes().CacheTime(30*time.Minute).AutoInfoKey(models.AiPlaza, id.Hex()).ResBind(&res).Cache(aiplazamod.GetInfo, id)
|
||||
return
|
||||
}
|
||||
|
||||
// GetAllFromCache 获取所有数据
|
||||
func GetAllFromCache(cond bson.M, sort bson.D) (res []aiplazamod.AiPlaza, err error) {
|
||||
_, err = cachev2.Classes().CacheTime(30*time.Minute).AutoListKey(models.AiPlaza+"-all").ResBind(&res).Cache(aiplazamod.GetAll, cond, sort)
|
||||
return
|
||||
}
|
||||
|
||||
// InsertData 插入数据
|
||||
func InsertData(t *db.MongoTool, data aiplazamod.AiPlaza) (primitive.ObjectID, error) {
|
||||
newID, err := aiplazamod.Insert(t, data)
|
||||
if err != nil {
|
||||
return newID, err
|
||||
}
|
||||
// 清除缓存数据
|
||||
_, err = cachev2.Classes().Table(models.AiPlaza).AutoClear(true, nil)
|
||||
if err != nil {
|
||||
return newID, err
|
||||
}
|
||||
return newID, nil
|
||||
}
|
||||
|
||||
// UpdateData 更新数据
|
||||
func UpdateData(t *db.MongoTool, id string, data map[string]interface{}) (int64, error) {
|
||||
objID, err := primitive.ObjectIDFromHex(id)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
affect, err := aiplazamod.UpdateByID(t, objID, data)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
// 清理缓存
|
||||
_, _ = cachev2.Classes().Table(models.AiPlaza).AutoClear(true, &id)
|
||||
|
||||
return affect, err
|
||||
}
|
||||
|
||||
// DeleteData 删除数据
|
||||
func DeleteData(t *db.MongoTool, id ...primitive.ObjectID) (err error) {
|
||||
err = aiplazamod.DeleteByID(t, id...)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
// 清理缓存
|
||||
all := "*"
|
||||
_, _ = cachev2.Classes().Table(models.AiPlaza).AutoClear(true, &all)
|
||||
return
|
||||
}
|
||||
+153
@@ -0,0 +1,153 @@
|
||||
package aiplazadata
|
||||
|
||||
import (
|
||||
"91porn-server/app/service/vidhelpser"
|
||||
"91porn-server/common"
|
||||
"91porn-server/models/v/aiplazamod"
|
||||
"91porn-server/models/v/collectmod"
|
||||
"91porn-server/models/v/likemod"
|
||||
"91porn-server/models/v/vidmod"
|
||||
"go.mongodb.org/mongo-driver/bson/primitive"
|
||||
"sync"
|
||||
)
|
||||
|
||||
// FormatAppDataList 格式化app列表数据
|
||||
func FormatAppDataList(origin []aiplazamod.AiPlaza) (res []*aiplazamod.AiPlazaInfo) {
|
||||
res = make([]*aiplazamod.AiPlazaInfo, len(origin))
|
||||
if len(origin) == 0 {
|
||||
return
|
||||
}
|
||||
// 获取用户数据
|
||||
uids := []uint64{}
|
||||
for _, item := range origin {
|
||||
uids = append(uids, item.Uid)
|
||||
}
|
||||
userMap := vidhelpser.GetSimpleUserMapNoUID(uids)
|
||||
// 组装返回数据
|
||||
for i, item := range origin {
|
||||
res[i] = FormatAppSimpleData(item, userMap)
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
// FormatAppSimpleData 格式化app数据
|
||||
func FormatAppSimpleData(item aiplazamod.AiPlaza, userMap map[uint64]*vidmod.Publisher) (res *aiplazamod.AiPlazaInfo) {
|
||||
if item.ID.IsZero() {
|
||||
return
|
||||
}
|
||||
|
||||
// 组装返回数据
|
||||
res = &aiplazamod.AiPlazaInfo{
|
||||
ID: item.ID,
|
||||
Type: item.Type,
|
||||
Template: item.Template,
|
||||
Uid: item.Uid,
|
||||
Title: item.Title,
|
||||
OriginalImage: item.OriginalImage,
|
||||
OriginalVideo: item.OriginalVideo,
|
||||
OriginalVideoCover: item.OriginalVideoCover,
|
||||
OriginContent: item.OriginContent,
|
||||
GenerateImage: item.GenerateImage,
|
||||
GenerateVideo: item.GenerateVideo,
|
||||
GenerateVideoCover: item.GenerateVideoCover,
|
||||
Status: item.Status,
|
||||
Reason: item.Reason,
|
||||
SortCode: item.SortCode,
|
||||
LikeCount: item.LikeCount,
|
||||
CollectCount: item.CollectCount,
|
||||
CommentCount: item.CommentCount,
|
||||
ViewCount: item.ViewCount,
|
||||
ReviewAt: item.ReviewAt,
|
||||
CreatedAt: item.CreatedAt,
|
||||
UpdatedAt: item.UpdatedAt,
|
||||
}
|
||||
if userMap != nil {
|
||||
user, ok := userMap[item.Uid]
|
||||
if ok && user != nil {
|
||||
res.Name = user.Name
|
||||
res.Gender = user.Gender
|
||||
res.Portrait = user.Portrait
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// FormatAppData 格式化app列表数据
|
||||
func FormatAppData(uid uint64, item aiplazamod.AiPlaza) (res *aiplazamod.AiPlazaInfo) {
|
||||
if item.ID.IsZero() {
|
||||
return
|
||||
}
|
||||
|
||||
userMap := vidhelpser.GetSimpleUserMapNoUID([]uint64{item.Uid})
|
||||
// 获取是否点赞，收藏数据
|
||||
dataStatus := GetDataStatus(uid, []primitive.ObjectID{item.ID})
|
||||
// 组装返回数据
|
||||
res = &aiplazamod.AiPlazaInfo{
|
||||
ID: item.ID,
|
||||
Type: item.Type,
|
||||
Template: item.Template,
|
||||
Uid: item.Uid,
|
||||
Title: item.Title,
|
||||
OriginalImage: item.OriginalImage,
|
||||
OriginalVideo: item.OriginalVideo,
|
||||
OriginalVideoCover: item.OriginalVideoCover,
|
||||
OriginContent: item.OriginContent,
|
||||
GenerateImage: item.GenerateImage,
|
||||
GenerateVideo: item.GenerateVideo,
|
||||
GenerateVideoCover: item.GenerateVideoCover,
|
||||
Status: item.Status,
|
||||
Reason: item.Reason,
|
||||
SortCode: item.SortCode,
|
||||
ReviewAt: item.ReviewAt,
|
||||
CreatedAt: item.CreatedAt,
|
||||
UpdatedAt: item.UpdatedAt,
|
||||
}
|
||||
if userMap != nil {
|
||||
user, ok := userMap[item.Uid]
|
||||
if ok && user != nil {
|
||||
res.Name = user.Name
|
||||
res.Gender = user.Gender
|
||||
res.Portrait = user.Portrait
|
||||
}
|
||||
}
|
||||
if dataStatus != nil {
|
||||
res.HasLiked = dataStatus[item.ID].HasLiked
|
||||
res.HasCollected = dataStatus[item.ID].HasCollected
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// DataStatus 数据状态
|
||||
type DataStatus struct {
|
||||
//已点赞
|
||||
HasLiked bool `json:"hasLiked" bson:"hasLiked"`
|
||||
//已收藏
|
||||
HasCollected bool `json:"hasCollected" bson:"hasCollected"`
|
||||
}
|
||||
|
||||
// GetDataStatus 获取数据状态信息
|
||||
func GetDataStatus(uid uint64, objIds []primitive.ObjectID) map[primitive.ObjectID]DataStatus {
|
||||
mLike := make(map[primitive.ObjectID]bool)
|
||||
mCollect := make(map[primitive.ObjectID]bool)
|
||||
var wg sync.WaitGroup
|
||||
wg.Add(2)
|
||||
common.Go(func() {
|
||||
defer wg.Done()
|
||||
mLike, _ = likemod.IsLikeVideos(uid, objIds)
|
||||
})
|
||||
common.Go(func() {
|
||||
defer wg.Done()
|
||||
mCollect, _ = collectmod.IsCollectVideos(uid, objIds)
|
||||
})
|
||||
wg.Wait()
|
||||
m := make(map[primitive.ObjectID]DataStatus)
|
||||
for _, v := range objIds {
|
||||
info := DataStatus{
|
||||
HasLiked: mLike[v],
|
||||
HasCollected: mCollect[v],
|
||||
}
|
||||
m[v] = info
|
||||
}
|
||||
return m
|
||||
}
|
||||
Reference in New Issue
Block a user