Create & Init Project...
This commit is contained in:
228
app/interface/main/app-interface/dao/live/dao.go
Normal file
228
app/interface/main/app-interface/dao/live/dao.go
Normal file
@ -0,0 +1,228 @@
|
||||
package live
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"strconv"
|
||||
|
||||
"go-common/app/interface/main/app-interface/conf"
|
||||
"go-common/app/interface/main/app-interface/model/live"
|
||||
"go-common/library/ecode"
|
||||
httpx "go-common/library/net/http/blademaster"
|
||||
"go-common/library/net/metadata"
|
||||
"go-common/library/xstr"
|
||||
|
||||
"github.com/pkg/errors"
|
||||
)
|
||||
|
||||
const (
|
||||
_live = "/AppRoom/getRoomInfo"
|
||||
_medalStatus = "/fans_medal/v1/medal/get_medal_opened"
|
||||
_appMRoom = "/room/v1/Room/rooms_for_app_index"
|
||||
_statusInfo = "/room/v1/Room/get_status_info_by_uids"
|
||||
_visibleInfo = "/rc/v1/Glory/get_visible"
|
||||
_usersInfo = "/user/v3/User/getMultiple"
|
||||
_LiveByRID = "/room/v2/Room/get_by_ids"
|
||||
)
|
||||
|
||||
// Dao is space dao
|
||||
type Dao struct {
|
||||
client *httpx.Client
|
||||
live string
|
||||
medalStatus string
|
||||
appMRoom string
|
||||
statusInfo string
|
||||
visibleInfo string
|
||||
userInfo string
|
||||
liveByRID string
|
||||
}
|
||||
|
||||
// New initial space dao
|
||||
func New(c *conf.Config) (d *Dao) {
|
||||
d = &Dao{
|
||||
client: httpx.NewClient(c.HTTPLive),
|
||||
live: c.Host.APILiveCo + _live,
|
||||
medalStatus: c.Host.APILiveCo + _medalStatus,
|
||||
appMRoom: c.Host.APILiveCo + _appMRoom,
|
||||
statusInfo: c.Host.APILiveCo + _statusInfo,
|
||||
visibleInfo: c.Host.APILiveCo + _visibleInfo,
|
||||
userInfo: c.Host.APILiveCo + _usersInfo,
|
||||
liveByRID: c.Host.APILiveCo + _LiveByRID,
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// Live is space live data.
|
||||
func (d *Dao) Live(c context.Context, mid int64, platform string) (live json.RawMessage, err error) {
|
||||
ip := metadata.String(c, metadata.RemoteIP)
|
||||
params := url.Values{}
|
||||
params.Set("mid", strconv.FormatInt(mid, 10))
|
||||
params.Set("platform", platform)
|
||||
var res struct {
|
||||
Code int `json:"code"`
|
||||
Data json.RawMessage `json:"data"`
|
||||
}
|
||||
if err = d.client.Get(c, d.live, ip, params, &res); err != nil {
|
||||
return
|
||||
}
|
||||
if res.Code != ecode.OK.Code() {
|
||||
err = errors.Wrap(ecode.Int(res.Code), d.live+"?"+params.Encode())
|
||||
return
|
||||
}
|
||||
live = res.Data
|
||||
return
|
||||
}
|
||||
|
||||
// MedalStatus for live
|
||||
func (d *Dao) MedalStatus(c context.Context, mid int64) (status int, err error) {
|
||||
var (
|
||||
req *http.Request
|
||||
res struct {
|
||||
Code int `json:"code"`
|
||||
Data *struct {
|
||||
MasterStatus int `json:"master_status"`
|
||||
} `json:"data"`
|
||||
}
|
||||
ip = metadata.String(c, metadata.RemoteIP)
|
||||
)
|
||||
if req, err = d.client.NewRequest("GET", d.medalStatus, ip, nil); err != nil {
|
||||
return
|
||||
}
|
||||
req.Header.Set("X-BILILIVE-UID", strconv.FormatInt(mid, 10))
|
||||
if err = d.client.Do(c, req, &res); err != nil {
|
||||
err = errors.Wrapf(err, "url(%s) header(X-BiliLive-UID:%s)", req.URL.String(), req.Header.Get("X-BiliLive-UID"))
|
||||
return
|
||||
}
|
||||
if res.Code != ecode.OK.Code() {
|
||||
err = errors.Wrapf(ecode.Int(res.Code), "url(%s) header(X-BiliLive-UID:%s)", req.URL.String(), req.Header.Get("X-BiliLive-UID"))
|
||||
return
|
||||
}
|
||||
if res.Data != nil {
|
||||
status = res.Data.MasterStatus
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// AppMRoom for live
|
||||
func (d *Dao) AppMRoom(c context.Context, roomids []int64) (rs map[int64]*live.Room, err error) {
|
||||
ip := metadata.String(c, metadata.RemoteIP)
|
||||
params := url.Values{}
|
||||
params.Set("room_ids", xstr.JoinInts(roomids))
|
||||
var res struct {
|
||||
Code int `json:"code"`
|
||||
Data []*live.Room `json:"data"`
|
||||
}
|
||||
if err = d.client.Get(c, d.appMRoom, ip, params, &res); err != nil {
|
||||
return
|
||||
}
|
||||
if res.Code != ecode.OK.Code() {
|
||||
err = errors.Wrap(err, d.appMRoom+"?"+params.Encode())
|
||||
return
|
||||
}
|
||||
rs = make(map[int64]*live.Room, len(res.Data))
|
||||
for _, r := range res.Data {
|
||||
rs[r.RoomID] = r
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// StatusInfo for live
|
||||
func (d *Dao) StatusInfo(c context.Context, mids []int64) (status map[int64]*live.Status, err error) {
|
||||
ip := metadata.String(c, metadata.RemoteIP)
|
||||
params := url.Values{}
|
||||
for _, mid := range mids {
|
||||
params.Add("uids[]", strconv.FormatInt(mid, 10))
|
||||
}
|
||||
params.Set("filter_offline", "1")
|
||||
var res struct {
|
||||
Code int `json:"code"`
|
||||
Data map[int64]*live.Status `json:"data"`
|
||||
}
|
||||
if err = d.client.Get(c, d.statusInfo, ip, params, &res); err != nil {
|
||||
return
|
||||
}
|
||||
if res.Code != ecode.OK.Code() {
|
||||
err = errors.Wrap(err, d.statusInfo+"?"+params.Encode())
|
||||
return
|
||||
}
|
||||
status = res.Data
|
||||
return
|
||||
}
|
||||
|
||||
// Glory for live search
|
||||
func (d *Dao) Glory(c context.Context, uid int64) (glory []*live.Glory, err error) {
|
||||
ip := metadata.String(c, metadata.RemoteIP)
|
||||
params := url.Values{}
|
||||
params.Set("uid", strconv.FormatInt(uid, 10))
|
||||
var res struct {
|
||||
Code int `json:"code"`
|
||||
Data []*live.Glory `json:"data"`
|
||||
}
|
||||
if err = d.client.Get(c, d.visibleInfo, ip, params, &res); err != nil {
|
||||
return
|
||||
}
|
||||
if res.Code != ecode.OK.Code() {
|
||||
err = errors.Wrap(err, d.visibleInfo+"?"+params.Encode())
|
||||
return
|
||||
}
|
||||
glory = res.Data
|
||||
return
|
||||
}
|
||||
|
||||
// UserInfo for live search
|
||||
func (d *Dao) UserInfo(c context.Context, uids []int64) (userInfo map[int64]map[string]*live.Exp, err error) {
|
||||
ip := metadata.String(c, metadata.RemoteIP)
|
||||
params := url.Values{}
|
||||
for _, uid := range uids {
|
||||
params.Set("uids[]", strconv.FormatInt(uid, 10))
|
||||
}
|
||||
params.Set("attributes[]", "exp")
|
||||
var res struct {
|
||||
Code int `json:"code"`
|
||||
Data map[int64]map[string]*live.Exp `json:"data"`
|
||||
}
|
||||
if err = d.client.Get(c, d.userInfo, ip, params, &res); err != nil {
|
||||
return
|
||||
}
|
||||
if res.Code != ecode.OK.Code() {
|
||||
err = errors.Wrap(err, d.userInfo+"?"+params.Encode())
|
||||
return
|
||||
}
|
||||
userInfo = res.Data
|
||||
return
|
||||
}
|
||||
|
||||
// LiveByRIDs get live info by room_ids.
|
||||
func (d *Dao) LiveByRIDs(c context.Context, roomIDs []int64) (info map[int64]*live.RoomInfo, err error) {
|
||||
ip := metadata.String(c, metadata.RemoteIP)
|
||||
params := url.Values{}
|
||||
for _, id := range roomIDs {
|
||||
params.Add("ids[]", strconv.FormatInt(id, 10))
|
||||
}
|
||||
params.Add("fields[]", "roomid")
|
||||
params.Add("fields[]", "title")
|
||||
params.Add("fields[]", "cover")
|
||||
params.Add("fields[]", "user_cover")
|
||||
params.Add("fields[]", "uid")
|
||||
params.Add("fields[]", "uname")
|
||||
params.Add("fields[]", "area_v2_name")
|
||||
params.Add("fields[]", "live_status")
|
||||
params.Add("fields[]", "broadcast_type")
|
||||
params.Add("fields[]", "short_id")
|
||||
params.Add("need_broadcast_type", "1")
|
||||
var res struct {
|
||||
Code int `json:"code"`
|
||||
Data map[int64]*live.RoomInfo `json:"data"`
|
||||
}
|
||||
if err = d.client.Get(c, d.liveByRID, ip, params, &res); err != nil {
|
||||
return
|
||||
}
|
||||
if res.Code != ecode.OK.Code() {
|
||||
err = errors.Wrap(ecode.Int(res.Code), d.liveByRID+"?"+params.Encode())
|
||||
return
|
||||
}
|
||||
info = res.Data
|
||||
return
|
||||
}
|
||||
Reference in New Issue
Block a user