119 lines
2.8 KiB
Go
119 lines
2.8 KiB
Go
package commentser
|
|
|
|
import (
|
|
"91porn-server/app/service/searcher"
|
|
"91porn-server/common"
|
|
"91porn-server/common/log"
|
|
"91porn-server/models/v/cmtmod"
|
|
"91porn-server/models/v/usermod"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
type VideoRes = searcher.VideoRes
|
|
|
|
type UserBInfo = usermod.BaseInfo
|
|
|
|
type Comment = cmtmod.Comment
|
|
|
|
type ReplyInfo struct {
|
|
RUID uint64 `json:"rUID"` //回复者UID
|
|
RName string `json:"rName"` //回复者名字
|
|
RPortrait string `json:"rPortrait"` //回复者头像
|
|
RContent string `json:"rContent"` //回复内容
|
|
RTime time.Time `json:"rTime"` //回复时间
|
|
RType cmtmod.CmtType `json:"rType"` //回复类型
|
|
Video VideoRes `json:"video"` //视屏信息
|
|
}
|
|
|
|
type ReplyPage struct {
|
|
HasNext bool `json:"hasNext"`
|
|
List []ReplyInfo `json:"list"`
|
|
}
|
|
|
|
func ReplyPages(uid uint64, skip, limit int64) (ReplyPage, error) {
|
|
limitEx := limit + 1
|
|
cmtList, err := cmtmod.ListByPublisherID(skip, limitEx, uid)
|
|
if err != nil {
|
|
return ReplyPage{}, err
|
|
}
|
|
hasNext := false
|
|
if len(cmtList) > int(limit) {
|
|
hasNext = true
|
|
cmtList = cmtList[:limit]
|
|
}
|
|
vids := make([]ObjectID, 0, len(cmtList))
|
|
uids := make([]uint64, 0, len(cmtList))
|
|
for _, v := range cmtList {
|
|
vids = append(vids, v.ObjID) //视屏ID
|
|
uids = append(uids, v.UserID)
|
|
}
|
|
var (
|
|
wg = sync.WaitGroup{}
|
|
videoResList []VideoRes
|
|
userInfos []*UserBInfo
|
|
)
|
|
wg.Add(2)
|
|
common.Go(func() {
|
|
defer wg.Done()
|
|
list, err := searcher.GetVideoResList(uid, vids)
|
|
if err != nil {
|
|
log.Error("commentser GetVideoResList faild", log.E(err))
|
|
}
|
|
videoResList = list
|
|
})
|
|
common.Go(func() {
|
|
defer wg.Done()
|
|
list, err := usermod.GetUsersBaseInfo(uids)
|
|
if err != nil {
|
|
log.Error("commentser GetUsersBaseInfo faild", log.E(err))
|
|
}
|
|
userInfos = list
|
|
})
|
|
wg.Wait()
|
|
videoResMap := make(map[ObjectID]VideoRes, len(videoResList))
|
|
for _, v := range videoResList {
|
|
vidp := v.VideoInfo.ID
|
|
if vidp != nil {
|
|
videoResMap[*vidp] = v
|
|
}
|
|
}
|
|
userInfoMap := make(map[uint64]*usermod.BaseInfo, len(userInfos))
|
|
for _, v := range userInfos {
|
|
userInfoMap[v.UID] = v
|
|
}
|
|
replyInfos := toReplyInfoList(cmtList, videoResMap, userInfoMap)
|
|
return ReplyPage{
|
|
hasNext,
|
|
replyInfos,
|
|
}, nil
|
|
}
|
|
|
|
func toReplyInfoList(cmtList []Comment,
|
|
videoResMap map[ObjectID]VideoRes,
|
|
userInfoMap map[uint64]*usermod.BaseInfo,
|
|
) []ReplyInfo {
|
|
replyInfos := make([]ReplyInfo, 0, len(cmtList))
|
|
for _, v := range cmtList {
|
|
userInfo, ok := userInfoMap[v.UserID]
|
|
if !ok {
|
|
continue
|
|
}
|
|
videoRes, ok := videoResMap[v.ObjID]
|
|
if !ok {
|
|
continue
|
|
}
|
|
replyInfo := ReplyInfo{
|
|
RUID: userInfo.UID,
|
|
RName: userInfo.Name,
|
|
RPortrait: userInfo.Portrait,
|
|
RContent: v.Content,
|
|
RType: v.Type,
|
|
RTime: v.CreatedAt,
|
|
Video: videoRes,
|
|
}
|
|
replyInfos = append(replyInfos, replyInfo)
|
|
}
|
|
return replyInfos
|
|
}
|