351 lines
8.7 KiB
Go
351 lines
8.7 KiB
Go
|
package service
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"go-common/app/service/bbq/common"
|
|||
|
|
"go-common/library/conf/env"
|
|||
|
|
"go-common/library/ecode"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"go-common/app/interface/bbq/app-bbq/api/http/v1"
|
|||
|
|
"go-common/app/interface/bbq/app-bbq/model"
|
|||
|
|
"go-common/app/interface/bbq/app-bbq/model/grpc"
|
|||
|
|
rec "go-common/app/service/bbq/recsys/api/grpc/v1"
|
|||
|
|
user "go-common/app/service/bbq/user/api"
|
|||
|
|
video "go-common/app/service/bbq/video/api/grpc/v1"
|
|||
|
|
"go-common/library/log"
|
|||
|
|
"go-common/library/net/trace"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// SvList 短视屏推荐列表
|
|||
|
|
func (s *Service) SvList(c context.Context, pageSize int64, mid int64, base *v1.Base, deviceID string) (res []*v1.VideoResponse, err error) {
|
|||
|
|
var (
|
|||
|
|
svids []int64
|
|||
|
|
svRes map[int64]*v1.VideoResponse
|
|||
|
|
)
|
|||
|
|
res = make([]*v1.VideoResponse, 0)
|
|||
|
|
//推荐列表
|
|||
|
|
svids, err = s.dao.RawRecList(c, pageSize, mid, base.BUVID)
|
|||
|
|
if err != nil || len(svids) == 0 || env.DeployEnv == env.DeployEnvUat {
|
|||
|
|
log.Warnv(c, log.KV("log", fmt.Sprintf("s.dao.GetList err[%v]", err)))
|
|||
|
|
// 降级
|
|||
|
|
svids, _ = s.dao.GetList(c, pageSize)
|
|||
|
|
}
|
|||
|
|
svRes, err = s.svInfos(c, svids, mid, false)
|
|||
|
|
for _, id := range svids {
|
|||
|
|
if sv, ok := svRes[id]; ok {
|
|||
|
|
if common.IsRecommendSvStateAvailable(int64(sv.State)) {
|
|||
|
|
res = append(res, sv)
|
|||
|
|
} else {
|
|||
|
|
log.Warnw(c, "log", "get error svid in recommend list", "svid", id, "mid", mid, "sv", sv)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// svPlays 批量获取playurl(相对地址方法)
|
|||
|
|
func (s *Service) svPlays(c context.Context, svids []int64) map[int64]*v1.VideoPlay {
|
|||
|
|
var (
|
|||
|
|
relAddr []string
|
|||
|
|
err error
|
|||
|
|
bvcUrls map[string]*grpc.VideoKeyItem
|
|||
|
|
bvcKeys map[int64][]*model.SVBvcKey
|
|||
|
|
)
|
|||
|
|
playMap := make(map[int64]*v1.VideoPlay)
|
|||
|
|
bvcKeys, err = s.dao.RawSVBvcKey(c, svids)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("s.dao.RawSVBvcKey err[%v]", err)
|
|||
|
|
}
|
|||
|
|
for id, keys := range bvcKeys {
|
|||
|
|
playMap[id] = &v1.VideoPlay{
|
|||
|
|
SVID: id,
|
|||
|
|
}
|
|||
|
|
for k, v := range keys {
|
|||
|
|
if k == 0 {
|
|||
|
|
playMap[id].Quality = int64(v.CodeRate)
|
|||
|
|
}
|
|||
|
|
fi := &v1.FileInfo{
|
|||
|
|
TimeLength: v.Duration,
|
|||
|
|
FileSize: v.FileSize,
|
|||
|
|
Path: v.Path,
|
|||
|
|
}
|
|||
|
|
playMap[id].FileInfo = append(playMap[id].FileInfo, fi)
|
|||
|
|
playMap[id].SupportQuality = append(playMap[id].SupportQuality, int64(v.CodeRate))
|
|||
|
|
relAddr = append(relAddr, v.Path)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
bvcUrls, err = s.dao.RelPlayURLs(c, relAddr)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("s.dao.RelPlayURLs err[%v]", err)
|
|||
|
|
}
|
|||
|
|
//拼装playurl
|
|||
|
|
for _, svid := range svids {
|
|||
|
|
if play, ok := playMap[svid]; ok {
|
|||
|
|
for fk, f := range play.FileInfo {
|
|||
|
|
if urls, ok := bvcUrls[f.Path]; ok {
|
|||
|
|
playMap[svid].ExpireTime = int64(urls.Etime)
|
|||
|
|
playMap[svid].CurrentTime = time.Now().Unix()
|
|||
|
|
for _, u := range urls.URL {
|
|||
|
|
if playMap[svid].FileInfo[fk].URL == "" {
|
|||
|
|
playMap[svid].FileInfo[fk].URL = u
|
|||
|
|
if playMap[svid].URL == "" {
|
|||
|
|
playMap[svid].URL = u
|
|||
|
|
}
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
if playMap[svid].FileInfo[fk].URLBc == "" {
|
|||
|
|
playMap[svid].FileInfo[fk].URLBc = u
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
delete(playMap, svid)
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
playMap[svid] = play
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return playMap
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SvStatistics 视频统计服务
|
|||
|
|
func (s *Service) SvStatistics(c context.Context, mid int64, svids []int64) (res []*v1.SvStatRes, err error) {
|
|||
|
|
var (
|
|||
|
|
stMap map[int64]*model.SvStInfo
|
|||
|
|
ulike map[int64]bool
|
|||
|
|
upIDs []int64
|
|||
|
|
)
|
|||
|
|
svInfos, _ := s.dao.RawVideos(c, svids)
|
|||
|
|
for _, sv := range svInfos {
|
|||
|
|
upIDs = append(upIDs, sv.MID)
|
|||
|
|
}
|
|||
|
|
stMap, err = s.dao.RawVideoStatistic(c, svids)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("s.dao.RawVideoStatistic err[%v]", err)
|
|||
|
|
}
|
|||
|
|
//点赞状态
|
|||
|
|
ulike, err = s.dao.CheckUserLike(c, mid, svids)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("s.dao.CheckUserLike err[%v]", err)
|
|||
|
|
}
|
|||
|
|
uflw, _ := s.dao.BatchUserInfo(c, mid, upIDs, false, false, true)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("s.dao.IsFollow err[%v]", err)
|
|||
|
|
}
|
|||
|
|
for _, id := range svids {
|
|||
|
|
rp := &v1.SvStatRes{}
|
|||
|
|
rp.SVID = id
|
|||
|
|
if st, ok := stMap[id]; ok {
|
|||
|
|
rp.Like = st.Like
|
|||
|
|
rp.Share = st.Share
|
|||
|
|
rp.Play = st.Play
|
|||
|
|
rp.Subtitles = st.Subtitles
|
|||
|
|
rp.Reply = st.Reply
|
|||
|
|
}
|
|||
|
|
if l, ok := ulike[id]; ok {
|
|||
|
|
rp.IsLike = l
|
|||
|
|
}
|
|||
|
|
if sv, ok := svInfos[id]; ok {
|
|||
|
|
if f, ok2 := uflw[sv.MID]; ok2 {
|
|||
|
|
rp.FollowState = f.FollowState
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
res = append(res, rp)
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SvCPlays 批量拉取playurl
|
|||
|
|
func (s *Service) SvCPlays(c context.Context, svids []int64, mid int64) (res []*v1.VideoPlay, err error) {
|
|||
|
|
res = make([]*v1.VideoPlay, 0)
|
|||
|
|
|
|||
|
|
//视频列表
|
|||
|
|
svRes, err := s.dao.RawVideos(c, svids)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("s.dao.RawVideos err[%v]", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
avaliableSvids := make([]int64, 0)
|
|||
|
|
for _, v := range svRes {
|
|||
|
|
if (mid == 0 || v.MID != mid) && common.IsSvStateGuestAvailable(int64(v.State)) {
|
|||
|
|
avaliableSvids = append(avaliableSvids, v.SVID)
|
|||
|
|
} else if v.MID == mid && common.IsSvStateOwnerAvailable(int64(v.State)) {
|
|||
|
|
avaliableSvids = append(avaliableSvids, v.SVID)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
playMap := s.svPlays(c, avaliableSvids)
|
|||
|
|
for _, svid := range avaliableSvids {
|
|||
|
|
var play *v1.VideoPlay
|
|||
|
|
if p, ok := playMap[svid]; !ok {
|
|||
|
|
log.Warn("play不存在 svid[%d]", svid)
|
|||
|
|
continue
|
|||
|
|
} else {
|
|||
|
|
play = p
|
|||
|
|
}
|
|||
|
|
res = append(res, play)
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SvDetail 单条sv的视频详情，暂时只用于评论中转页
|
|||
|
|
func (s *Service) SvDetail(c context.Context, svid int64, mid int64) (res *v1.VideoResponse, err error) {
|
|||
|
|
_, err = s.dao.VideoBase(c, mid, svid)
|
|||
|
|
if err != nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
svInfos, err := s.svInfos(c, []int64{svid}, mid, true)
|
|||
|
|
if err != nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if val, exists := svInfos[svid]; exists {
|
|||
|
|
res = val
|
|||
|
|
} else {
|
|||
|
|
err = ecode.VideoUnExists
|
|||
|
|
log.Infow(c, "log", "not sv info", "svid", svid)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// svInfos 批量获取视频信息
|
|||
|
|
// @params allowState 可放出状态，传空为app整体可露出状态
|
|||
|
|
// @params needStInfo 是否需要视频统计数据
|
|||
|
|
func (s *Service) svInfos(c context.Context, ids []int64, mid int64, needStInfo bool) (res map[int64]*v1.VideoResponse, err error) {
|
|||
|
|
var (
|
|||
|
|
mids []int64
|
|||
|
|
svRes map[int64]*model.SvInfo
|
|||
|
|
ulike map[int64]bool
|
|||
|
|
stMap map[int64]*model.SvStInfo
|
|||
|
|
)
|
|||
|
|
res = make(map[int64]*v1.VideoResponse)
|
|||
|
|
stMap = make(map[int64]*model.SvStInfo)
|
|||
|
|
|
|||
|
|
//视频列表
|
|||
|
|
svRes, err = s.dao.RawVideos(c, ids)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("s.dao.RawVideos err[%v]", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
for _, v := range svRes {
|
|||
|
|
mids = append(mids, v.MID)
|
|||
|
|
}
|
|||
|
|
if mid != 0 {
|
|||
|
|
ulike, err = s.dao.CheckUserLike(c, mid, ids)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("s.dao.CheckUserLike err[%v]", err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
// query id
|
|||
|
|
tracer, _ := trace.FromContext(c)
|
|||
|
|
queryID := fmt.Sprintf("%s", tracer)
|
|||
|
|
//账号
|
|||
|
|
var userMap map[int64]*user.UserBase
|
|||
|
|
userMap, err = s.dao.JustGetUserBase(c, mids)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("s.dao.UserBase err[%v]", err)
|
|||
|
|
}
|
|||
|
|
// play信息
|
|||
|
|
playMap := s.svPlays(c, ids)
|
|||
|
|
if needStInfo {
|
|||
|
|
stMap, err = s.dao.RawVideoStatistic(c, ids)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("s.dao.RawVideoStatistic err[%v]", err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
// extension信息
|
|||
|
|
extensions, tmpErr := s.getExtension(c, ids)
|
|||
|
|
if tmpErr != nil {
|
|||
|
|
log.Warnw(c, "log", "get extension fail")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for _, v := range svRes {
|
|||
|
|
if common.IsSvStateAvailable(int64(v.State)) {
|
|||
|
|
sv := &v1.VideoResponse{}
|
|||
|
|
if acc, ok := userMap[v.MID]; ok {
|
|||
|
|
sv.UserInfo = *acc
|
|||
|
|
}
|
|||
|
|
if lk, ok := ulike[v.SVID]; ok {
|
|||
|
|
sv.IsLike = lk
|
|||
|
|
}
|
|||
|
|
sv.SVID = v.SVID
|
|||
|
|
sv.Title = v.Title
|
|||
|
|
sv.Content = v.Content
|
|||
|
|
sv.MID = v.MID
|
|||
|
|
sv.Duration = v.Duration
|
|||
|
|
sv.Pubtime = v.Pubtime
|
|||
|
|
sv.Ctime = v.Ctime
|
|||
|
|
sv.AVID = v.AVID
|
|||
|
|
sv.CID = v.CID
|
|||
|
|
sv.From = v.From
|
|||
|
|
sv.CoverURL = v.CoverURL
|
|||
|
|
sv.CoverHeight = v.CoverHeight
|
|||
|
|
sv.CoverWidth = v.CoverWidth
|
|||
|
|
sv.QueryID = queryID
|
|||
|
|
sv.State = v.State
|
|||
|
|
if play, ok := playMap[v.SVID]; ok {
|
|||
|
|
sv.Play = *play
|
|||
|
|
res[v.SVID] = sv
|
|||
|
|
} else {
|
|||
|
|
log.Warn("play不存在 svid[%d]，此条记录直接舍弃", v.SVID)
|
|||
|
|
}
|
|||
|
|
if st, ok := stMap[v.SVID]; ok {
|
|||
|
|
sv.SvStInfo = *st
|
|||
|
|
}
|
|||
|
|
if extension, exists := extensions[v.SVID]; exists {
|
|||
|
|
sv.Extension = extension.Extension
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
res[v.SVID] = sv
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SvRelRec 相关推荐服务
|
|||
|
|
func (s *Service) SvRelRec(c context.Context, data *v1.SvRelReq) (res map[string]interface{}, err error) {
|
|||
|
|
res = make(map[string]interface{})
|
|||
|
|
var svMap map[int64]*v1.VideoResponse
|
|||
|
|
list := make([]*v1.VideoResponse, 0)
|
|||
|
|
relReq := &rec.RecsysRequest{
|
|||
|
|
SVID: data.SVID,
|
|||
|
|
Offset: data.Offset,
|
|||
|
|
Limit: data.Limit,
|
|||
|
|
QueryID: data.QueryID,
|
|||
|
|
App: data.APP,
|
|||
|
|
AppVersion: data.APPVersion,
|
|||
|
|
BUVID: data.BUVID,
|
|||
|
|
MID: data.MID,
|
|||
|
|
}
|
|||
|
|
IDList, err := s.dao.RelRecList(c, relReq)
|
|||
|
|
if err != nil {
|
|||
|
|
err = nil
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
svMap, err = s.svInfos(c, IDList, data.MID, false)
|
|||
|
|
if err != nil {
|
|||
|
|
err = nil
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
for _, id := range IDList {
|
|||
|
|
if sv, ok := svMap[id]; ok {
|
|||
|
|
list = append(list, sv)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
res["list"] = list
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SvDel 视频删除
|
|||
|
|
func (s *Service) SvDel(c context.Context, in *video.VideoDeleteRequest) (interface{}, error) {
|
|||
|
|
return s.dao.SvDel(c, in)
|
|||
|
|
}
|