331 lines
8.0 KiB
Go
331 lines
8.0 KiB
Go
package service
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"go-common/app/interface/bbq/app-bbq/api/http/v1"
|
||
"go-common/app/interface/bbq/app-bbq/model"
|
||
"go-common/app/service/bbq/common"
|
||
"go-common/library/ecode"
|
||
"go-common/library/log"
|
||
|
||
"github.com/json-iterator/go"
|
||
)
|
||
|
||
const (
|
||
searchTypeVideo = "bbq_video"
|
||
searchTypeUser = "bbq_user"
|
||
suggestType = "bbq"
|
||
suggestVer = "v4"
|
||
searchLen = 20
|
||
)
|
||
|
||
// SearchCursor .
|
||
type SearchCursor struct {
|
||
Offset int64 `json:"offset"` // 默认为0，数据从1开始
|
||
}
|
||
|
||
func parseSearchCursor(c context.Context, cursorPrev, cursorNext string) (cursor *SearchCursor, directionNext bool, err error) {
|
||
cursor = new(SearchCursor)
|
||
// 判断是向前还是向后查询
|
||
directionNext = true
|
||
cursorStr := cursorNext
|
||
if len(cursorNext) == 0 && len(cursorPrev) > 0 {
|
||
directionNext = false
|
||
cursorStr = cursorPrev
|
||
}
|
||
// 解析
|
||
if len(cursorStr) != 0 {
|
||
var cursorData = []byte(cursorStr)
|
||
err = json.Unmarshal(cursorData, &cursor)
|
||
if err != nil {
|
||
err = ecode.ReqParamErr
|
||
return
|
||
}
|
||
}
|
||
return
|
||
}
|
||
|
||
//HotWord .
|
||
func (s *Service) HotWord(c context.Context, req *v1.HotWordRequest) (res *v1.HotWordResponse, err error) {
|
||
res = new(v1.HotWordResponse)
|
||
res.List = append(res.List, "小姐姐")
|
||
res.List = append(res.List, "舞蹈")
|
||
res.List = append(res.List, "MMD")
|
||
return
|
||
}
|
||
|
||
// VideoSearch 视频搜索
|
||
func (s *Service) VideoSearch(c context.Context, arg *v1.BaseSearchReq) (res *v1.VideoSearchRes, err error) {
|
||
res = new(v1.VideoSearchRes)
|
||
var (
|
||
sres []*model.VideoSearchResult
|
||
searchRes *model.RawSearchRes
|
||
relids []int32
|
||
svids []int64
|
||
svidMap map[int64]int64
|
||
)
|
||
|
||
// 0. check
|
||
|
||
// 1. 生成请求，会根据请求参数
|
||
// v2接口
|
||
reqPage := int64(1)
|
||
firstItemOffset := int64(0)
|
||
var cursor *SearchCursor
|
||
var directionNext bool
|
||
if arg.Page == 0 {
|
||
cursor, directionNext, err = parseSearchCursor(c, arg.CursorPrev, arg.CursorNext)
|
||
if err != nil {
|
||
log.Warnw(c, "log", "parse search cursor fail", "prev", arg.CursorPrev, "next", arg.CursorNext)
|
||
return
|
||
}
|
||
firstItemOffset = cursor.Offset - 1
|
||
if directionNext {
|
||
firstItemOffset = cursor.Offset + 1
|
||
}
|
||
reqPage = (firstItemOffset-1)/searchLen + 1
|
||
log.V(1).Infow(c, "log", "v2 request", "next", arg.CursorNext, "prev", arg.CursorPrev)
|
||
} else {
|
||
reqPage = arg.Page
|
||
}
|
||
|
||
var json = jsoniter.ConfigCompatibleWithStandardLibrary
|
||
sp := &model.SearchBaseReq{
|
||
KeyWord: arg.Key,
|
||
Type: searchTypeVideo,
|
||
Page: reqPage,
|
||
PageSize: searchLen,
|
||
Highlight: int64(arg.Highlight),
|
||
}
|
||
// 相信搜索结果返回的就是searchLen
|
||
searchRes, err = s.dao.SearchBBQ(c, sp)
|
||
if searchRes.Res == nil {
|
||
return
|
||
}
|
||
if err = json.Unmarshal(searchRes.Res, &sres); err != nil {
|
||
return
|
||
}
|
||
for _, v := range sres {
|
||
relids = append(relids, v.ID)
|
||
}
|
||
if len(relids) == 0 {
|
||
log.Infov(c, log.KV("log", "search ret empty"))
|
||
return
|
||
}
|
||
ids := s.dao.ParseRel2ID(relids)
|
||
svidMap, err = s.dao.ConvID2SVID(c, ids)
|
||
if err != nil {
|
||
log.Error("ConvID2SVID err [%v]", err)
|
||
err = nil
|
||
return
|
||
}
|
||
for _, svid := range svidMap {
|
||
svids = append(svids, svid)
|
||
}
|
||
svMap, _ := s.svInfos(c, svids, 0, false)
|
||
|
||
for index, sinfo := range sres {
|
||
id := s.dao.ParseRel2ID([]int32{sinfo.ID})
|
||
if id[0] == 0 {
|
||
continue
|
||
}
|
||
if svid, ok := svidMap[id[0]]; ok {
|
||
if sv, ok := svMap[svid]; ok {
|
||
if common.IsSearchSvStateAvailable(int64(sv.State)) {
|
||
r := new(v1.VideoSearchList)
|
||
r.VideoResponse = *sv
|
||
r.SVID = svid
|
||
r.HitColumns = sinfo.HitColumns
|
||
r.TitleHighlight = sinfo.Title
|
||
r.Offset = (reqPage-1)*int64(searchLen) + int64(index) + 1
|
||
res.List = append(res.List, r)
|
||
} else {
|
||
log.Warnw(c, "log", "get error svid in search list", "svid", svid, "sv", sv)
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
res.NumPage = searchRes.PageNum
|
||
res.Page = searchRes.Page
|
||
|
||
res.HasMore = false
|
||
if arg.Page == 0 {
|
||
if (directionNext && searchRes.PageNum > searchRes.Page) || (!directionNext && searchRes.Page > 1) {
|
||
res.HasMore = true
|
||
}
|
||
|
||
var list []*v1.VideoSearchList
|
||
if directionNext {
|
||
for _, item := range res.List {
|
||
if item.Offset < firstItemOffset {
|
||
continue
|
||
}
|
||
list = append(list, item)
|
||
}
|
||
} else {
|
||
if firstItemOffset >= int64(len(res.List)) {
|
||
firstItemOffset = int64(len(res.List)) - 1
|
||
}
|
||
for i := firstItemOffset - 1; i >= 0; i-- {
|
||
list = append(list, res.List[i])
|
||
}
|
||
}
|
||
res.List = list
|
||
}
|
||
|
||
var cursorValue SearchCursor
|
||
for _, item := range res.List {
|
||
cursorValue.Offset = item.Offset
|
||
data, _ := json.Marshal(cursorValue)
|
||
item.CursorValue = string(data)
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
// UserSearch 用户搜索
|
||
func (s *Service) UserSearch(c context.Context, mid int64, arg *v1.BaseSearchReq) (res *v1.UserSearchRes, err error) {
|
||
res = new(v1.UserSearchRes)
|
||
var (
|
||
sres []*model.UserSearchResult
|
||
searchRes *model.RawSearchRes
|
||
mids []int64
|
||
)
|
||
|
||
// 1. 生成请求，会根据请求参数
|
||
// v2接口
|
||
reqPage := int64(1)
|
||
firstItemOffset := int64(0)
|
||
var cursor *SearchCursor
|
||
var directionNext bool
|
||
if arg.Page == 0 {
|
||
cursor, directionNext, err = parseSearchCursor(c, arg.CursorPrev, arg.CursorNext)
|
||
if err != nil {
|
||
log.Warnw(c, "log", "parse search cursor fail", "prev", arg.CursorPrev, "next", arg.CursorNext)
|
||
return
|
||
}
|
||
firstItemOffset = cursor.Offset - 1
|
||
if directionNext {
|
||
firstItemOffset = cursor.Offset + 1
|
||
}
|
||
reqPage = (firstItemOffset-1)/searchLen + 1
|
||
log.V(1).Infow(c, "log", "v2 request", "next", arg.CursorNext, "prev", arg.CursorPrev)
|
||
} else {
|
||
reqPage = arg.Page
|
||
}
|
||
|
||
sp := &model.SearchBaseReq{
|
||
KeyWord: arg.Key,
|
||
Type: searchTypeUser,
|
||
Page: reqPage,
|
||
PageSize: searchLen,
|
||
Highlight: int64(arg.Highlight),
|
||
}
|
||
searchRes, err = s.dao.SearchBBQ(c, sp)
|
||
if searchRes.Res == nil {
|
||
return
|
||
}
|
||
if err = json.Unmarshal(searchRes.Res, &sres); err != nil {
|
||
return
|
||
}
|
||
for _, v := range sres {
|
||
mids = append(mids, v.ID)
|
||
}
|
||
if len(mids) == 0 {
|
||
log.Infov(c, log.KV("log", "search ret empty"))
|
||
return
|
||
}
|
||
uMap, _ := s.dao.BatchUserInfo(c, mid, mids, false, true, true)
|
||
for index, sinfo := range sres {
|
||
if u, ok := uMap[sinfo.ID]; ok {
|
||
r := &v1.UserSearchList{
|
||
UserInfo: *u,
|
||
}
|
||
st := new(v1.UserStatic)
|
||
st.Fan = u.Fan
|
||
st.Follow = u.Follow
|
||
st.Like = u.Like
|
||
st.Liked = u.Liked
|
||
st.FollowState = u.FollowState
|
||
r.UserStatic = st
|
||
r.HitColumns = sinfo.HitColumns
|
||
r.UnameHighlight = sinfo.Uname
|
||
r.Offset = (reqPage-1)*int64(searchLen) + int64(index) + 1
|
||
res.List = append(res.List, r)
|
||
}
|
||
}
|
||
res.NumPage = searchRes.PageNum
|
||
res.Page = searchRes.Page
|
||
|
||
res.HasMore = false
|
||
if arg.Page == 0 {
|
||
if (directionNext && searchRes.PageNum > searchRes.Page) || (!directionNext && searchRes.Page > 1) {
|
||
res.HasMore = true
|
||
}
|
||
|
||
var list []*v1.UserSearchList
|
||
if directionNext {
|
||
for _, item := range res.List {
|
||
if item.Offset < firstItemOffset {
|
||
continue
|
||
}
|
||
list = append(list, item)
|
||
}
|
||
} else {
|
||
if firstItemOffset >= int64(len(res.List)) {
|
||
firstItemOffset = int64(len(res.List)) - 1
|
||
}
|
||
for i := firstItemOffset - 1; i >= 0; i-- {
|
||
list = append(list, res.List[i])
|
||
}
|
||
}
|
||
res.List = list
|
||
}
|
||
|
||
var cursorValue SearchCursor
|
||
for _, item := range res.List {
|
||
cursorValue.Offset = item.Offset
|
||
data, _ := json.Marshal(cursorValue)
|
||
item.CursorValue = string(data)
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
// BBQSug BBQSUG服务
|
||
func (s *Service) BBQSug(c context.Context, arg *v1.SugReq) (res []*v1.SugTag, err error) {
|
||
var (
|
||
sres []*model.RawSugTag
|
||
data json.RawMessage
|
||
)
|
||
sp := &model.SugBaseReq{
|
||
Term: arg.KeyWord,
|
||
SuggestType: suggestType,
|
||
MainVer: suggestVer,
|
||
SugNum: arg.PageSize,
|
||
Highlight: int64(arg.Highlight),
|
||
}
|
||
data, err = s.dao.SugBBQ(c, sp)
|
||
if err != nil {
|
||
log.Error("s.dao.SugBBQ err[%v] data [%s]", err, string(data))
|
||
return
|
||
}
|
||
var json = jsoniter.ConfigCompatibleWithStandardLibrary
|
||
if err = json.Unmarshal(data, &sres); err != nil {
|
||
log.Error("json.Unmarshal err[%v] data [%s]", err, string(data))
|
||
return
|
||
}
|
||
for _, sinfo := range sres {
|
||
r := new(v1.SugTag)
|
||
name := sinfo.Name
|
||
r.Name = name
|
||
r.Value = sinfo.Value
|
||
r.Type = sinfo.Type
|
||
r.Ref = sinfo.Ref
|
||
res = append(res, r)
|
||
}
|
||
return
|
||
}
|