You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
213 lines
7.1 KiB
213 lines
7.1 KiB
package search |
|
|
|
import ( |
|
"context" |
|
"go-common/library/sync/errgroup" |
|
"go-common/library/xstr" |
|
"net/http" |
|
"net/url" |
|
"strconv" |
|
|
|
mdlSearch "go-common/app/interface/main/tv/model/search" |
|
"go-common/library/ecode" |
|
"go-common/library/log" |
|
"go-common/library/net/metadata" |
|
) |
|
|
|
// UserSearch search user . |
|
func (d *Dao) UserSearch(ctx context.Context, arg *mdlSearch.UserSearch) (user []*mdlSearch.User, err error) { |
|
var ( |
|
params = url.Values{} |
|
req *http.Request |
|
ip = metadata.String(ctx, metadata.RemoteIP) |
|
) |
|
params.Set("platform", arg.Platform) |
|
params.Set("mobi_app", arg.MobiAPP) |
|
params.Set("build", arg.Build) |
|
params.Set("keyword", arg.Keyword) |
|
params.Set("page", strconv.Itoa(arg.Page)) |
|
params.Set("pagesize", strconv.Itoa(arg.Pagesize)) |
|
params.Set("userid", strconv.FormatInt(arg.MID, 10)) |
|
params.Set("order", arg.Order) |
|
params.Set("main_ver", "v3") // 支持赛事搜索 |
|
params.Set("search_type", "bili_user") |
|
params.Set("user_type", strconv.Itoa(arg.UserType)) // 用户类型 |
|
params.Set("highlight", strconv.Itoa(arg.Highlight)) |
|
params.Set("order_sort", strconv.Itoa(arg.OrderSort)) |
|
params.Set("from_source", arg.FromSource) |
|
params.Set("bili_user_vl", strconv.Itoa(d.cfgWild.BiliUserVl)) |
|
// new request |
|
if req, err = d.client.NewRequest("GET", d.userSearch, ip, params); err != nil { |
|
log.Error("[wild.UserSearch] d.client.NewRequest url(%s) error(%v)", d.userSearch, err) |
|
return |
|
} |
|
req.Header.Set("Buvid", arg.Buvid) |
|
var res struct { |
|
Code int `json:"code"` |
|
SeID string `json:"seid"` |
|
Pages int `json:"numPages"` |
|
List []*mdlSearch.User `json:"result"` |
|
} |
|
if err = d.client.Do(ctx, req, &res); err != nil { |
|
log.Error("[wild.UserSearch] d.client.Do url(%s) error(%v)", d.userSearch, err) |
|
return |
|
} |
|
if res.Code != ecode.OK.Code() { |
|
err = ecode.Int(res.Code) |
|
log.Error("[wild.UserSearch] url(%s) error(%v)", d.userSearch, err) |
|
return |
|
} |
|
user = res.List |
|
return |
|
} |
|
|
|
// SearchAllWild wild search all . |
|
func (d *Dao) SearchAllWild(ctx context.Context, arg *mdlSearch.UserSearch) (user *mdlSearch.Search, err error) { |
|
var ( |
|
req *http.Request |
|
ip = metadata.String(ctx, metadata.RemoteIP) |
|
) |
|
params := url.Values{} |
|
user = &mdlSearch.Search{} |
|
params.Set("build", arg.Build) |
|
params.Set("keyword", arg.Keyword) |
|
params.Set("main_ver", "v3") |
|
params.Set("mobi_app", arg.MobiAPP) |
|
params.Set("device", arg.Device) |
|
params.Set("userid", strconv.FormatInt(arg.MID, 10)) |
|
params.Set("tids", strconv.Itoa(arg.RID)) |
|
params.Set("highlight", strconv.Itoa(arg.Highlight)) |
|
params.Set("page", strconv.Itoa(arg.Page)) |
|
params.Set("pagesize", strconv.Itoa(arg.Pagesize)) |
|
params.Set("bili_user_num", strconv.Itoa(d.cfgWild.BiliUserNum)) |
|
params.Set("bili_user_vl", strconv.Itoa(d.cfgWild.BiliUserVl)) |
|
params.Set("user_num", strconv.Itoa(d.cfgWild.UserNum)) |
|
params.Set("user_video_limit", strconv.Itoa(d.cfgWild.UserVideoLimit)) |
|
params.Set("platform", arg.Platform) |
|
// params.Set("duration", strconv.Itoa(arg.Duration)) // 视频时长筛选，默认是0 |
|
params.Set("order", arg.Order) |
|
params.Set("search_type", "all") |
|
params.Set("from_source", arg.FromSource) |
|
params.Set("media_bangumi_num", strconv.Itoa(arg.SeasonNum)) |
|
params.Set("movie_num", strconv.Itoa(arg.MovieNum)) |
|
params.Set("is_new_pgc", "1") // 新番剧 |
|
params.Set("media_ft_num", strconv.Itoa(arg.MovieNum)) |
|
// params.Set("flow_need", "1") // 混排 |
|
// params.Set("query_rec_need", "1") // 搜索结果推荐 |
|
// new request |
|
if req, err = d.client.NewRequest("GET", d.userSearch, ip, params); err != nil { |
|
log.Error("d.client.NewRequest URI(%s) error(%v)", d.userSearch, err) |
|
return |
|
} |
|
req.Header.Set("Buvid", arg.Buvid) |
|
if err = d.client.Do(ctx, req, user); err != nil { |
|
log.Error("[wild.SearchAllWild] d.client.Do() url(%s) error(%v)", d.userSearch, err) |
|
} |
|
return |
|
} |
|
|
|
// card bangumi card . |
|
func (d *Dao) cardInfo(c context.Context, mid int64, sids []int64) (s map[string]*mdlSearch.Card, err error) { |
|
ip := metadata.String(c, metadata.RemoteIP) |
|
params := url.Values{} |
|
params.Set("mid", strconv.FormatInt(mid, 10)) |
|
params.Set("season_ids", xstr.JoinInts(sids)) |
|
var res struct { |
|
Code int `json:"code"` |
|
Result map[string]*mdlSearch.Card `json:"result"` |
|
} |
|
if err = d.client.Get(c, d.card, ip, params, &res); err != nil { |
|
log.Error("d.client.NewRequest url(%s) error(%v)", d.resultURL, err) |
|
return |
|
} |
|
if res.Code != ecode.OK.Code() { |
|
err = ecode.Int(res.Code) |
|
log.Error("[wild] cardInfo error(%v)", err) |
|
return |
|
} |
|
s = res.Result |
|
return |
|
} |
|
|
|
// PgcSearch . |
|
func (d *Dao) PgcSearch(c context.Context, arg *mdlSearch.UserSearch) (st *mdlSearch.TypeSearch, err error) { |
|
var ( |
|
req *http.Request |
|
ip = metadata.String(c, metadata.RemoteIP) |
|
seasonIDs []int64 |
|
bangumis map[string]*mdlSearch.Card |
|
items1, items2 []*mdlSearch.Item |
|
) |
|
params := url.Values{} |
|
params.Set("build", arg.Build) |
|
params.Set("keyword", arg.Keyword) |
|
params.Set("main_ver", "v3") |
|
params.Set("mobi_app", arg.MobiAPP) |
|
params.Set("device", arg.Device) |
|
params.Set("userid", strconv.FormatInt(arg.MID, 10)) |
|
params.Set("highlight", strconv.Itoa(arg.Highlight)) |
|
params.Set("page", strconv.Itoa(arg.Page)) |
|
params.Set("pagesize", strconv.Itoa(arg.Pagesize)) |
|
params.Set("platform", arg.Platform) |
|
params.Set("order", arg.Order) |
|
params.Set("search_type", "all") |
|
params.Set("from_source", arg.FromSource) |
|
params.Set("media_bangumi_num", strconv.Itoa(arg.SeasonNum)) |
|
params.Set("media_ft_num", strconv.Itoa(arg.MovieNum)) |
|
params.Set("is_new_pgc", "1") |
|
if req, err = d.client.NewRequest("GET", d.userSearch, ip, params); err != nil { |
|
log.Error("d.client.NewRequest url(%s) error(%v)", d.userSearch, err) |
|
return |
|
} |
|
req.Header.Set("Buvid", arg.Buvid) |
|
res := &mdlSearch.Search{} |
|
if err = d.client.Do(c, req, res); err != nil { |
|
log.Error("[wild.PgcSearch] d.client.Do url(%s) error(%v)", d.userSearch, err) |
|
return |
|
} |
|
if res.Code != ecode.OK.Code() { |
|
err = ecode.Int(res.Code) |
|
log.Error("[wild.PgcSearch] code(%d) error(%v)", res.Code, err) |
|
return |
|
} |
|
for _, v := range res.Result.MediaBangumi { |
|
seasonIDs = append(seasonIDs, v.SeasonID) |
|
} |
|
for _, v := range res.Result.MediaFt { |
|
seasonIDs = append(seasonIDs, v.SeasonID) |
|
} |
|
if len(seasonIDs) > 0 { |
|
if bangumis, err = d.cardInfo(c, arg.MID, seasonIDs); err != nil { |
|
log.Error("[wild.PgcSearch] MovieByType2 %+v", err) |
|
return |
|
} |
|
} |
|
if len(bangumis) > 0 { |
|
group := new(errgroup.Group) |
|
group.Go(func() error { |
|
items1 = make([]*mdlSearch.Item, 0, len(res.Result.MediaBangumi)) |
|
for _, v := range res.Result.MediaBangumi { |
|
si := &mdlSearch.Item{} |
|
si.FromMedia(v, "", mdlSearch.GotoMovie, bangumis) |
|
items1 = append(items1, si) |
|
} |
|
return nil |
|
}) |
|
group.Go(func() error { |
|
items2 = make([]*mdlSearch.Item, 0, len(res.Result.MediaFt)) |
|
for _, v := range res.Result.MediaFt { |
|
si := &mdlSearch.Item{} |
|
si.FromMedia(v, "", mdlSearch.GotoMovie, bangumis) |
|
items2 = append(items2, si) |
|
} |
|
return nil |
|
}) |
|
if err = group.Wait(); err != nil { |
|
log.Error("[wild.PgcSearch] group.Wait() is error(%v)", err) |
|
return |
|
} |
|
} |
|
items1 = append(items1, items2...) |
|
st = &mdlSearch.TypeSearch{TrackID: res.Trackid, Pages: res.Page, Total: res.Total, Items: items1} |
|
return |
|
}
|
|
|