openbilibili/app/interface/main/space/dao/bplus.go

package dao

import (
	"context"
	"net/http"
	"net/url"
	"strconv"

	"go-common/app/interface/main/space/model"
	"go-common/library/ecode"
	"go-common/library/net/metadata"

	"github.com/pkg/errors"
)

const (
	_groupsCountURI = "/link_group/v1/member/created_groups_num"
	_dynamicListURI = "/dynamic_svr/v0/dynamic_svr/co_space_history"
	_dynamicCntURI  = "/dynamic_svr/v0/dynamic_svr/space_dy_num"
	_dynamicURI     = "/dynamic_svr/v1/dynamic_svr/get_dynamic_detail"
)

// GroupsCount .
func (d *Dao) GroupsCount(c context.Context, mid, vmid int64) (count int, err error) {
	var (
		req *http.Request
		ip  = metadata.String(c, metadata.RemoteIP)
	)
	params := url.Values{}
	params.Set("master_uid", strconv.FormatInt(vmid, 10))
	if req, err = d.httpR.NewRequest(http.MethodGet, d.groupsCountURL, ip, params); err != nil {
		return
	}
	req.Header.Set("X-BiliLive-UID", strconv.FormatInt(mid, 10))
	var res struct {
		Code int `json:"code"`
		Data *struct {
			Num int `json:"num"`
		}
	}
	if err = d.httpR.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 {
		count = res.Data.Num
	}
	return
}

// DynamicCnt dynamic count.
func (d *Dao) DynamicCnt(c context.Context, mid int64) (cnt int64, err error) {
	var (
		params = url.Values{}
		ip     = metadata.String(c, metadata.RemoteIP)
	)
	params.Set("uids", strconv.FormatInt(mid, 10))
	var res struct {
		Code int `json:"code"`
		Data struct {
			Items []*struct {
				UID int64 `json:"uid"`
				Num int64 `json:"num"`
			} `json:"items"`
		}
	}
	if err = d.httpR.Get(c, d.dynamicCntURL, ip, params, &res); err != nil {
		return
	}
	if res.Code != ecode.OK.Code() {
		err = errors.Wrap(ecode.Int(res.Code), d.dynamicCntURL+"?"+params.Encode())
		return
	}
	if len(res.Data.Items) > 0 && res.Data.Items[0].UID == mid {
		cnt = res.Data.Items[0].Num
	}
	return
}

// DynamicList .
func (d *Dao) DynamicList(c context.Context, mid, vmid, dyID int64, qn, page int) (data *model.DyList, err error) {
	ip := metadata.String(c, metadata.RemoteIP)
	params := url.Values{}
	if mid > 0 {
		params.Set("visitor_uid", strconv.FormatInt(mid, 10))
	}
	params.Set("host_uid", strconv.FormatInt(vmid, 10))
	params.Set("offset_dynamic_id", strconv.FormatInt(dyID, 10))
	params.Set("qn", strconv.Itoa(qn))
	params.Set("page", strconv.Itoa(page))
	var res struct {
		Code int           `json:"code"`
		Data *model.DyList `json:"data"`
	}
	if err = d.httpR.Get(c, d.dynamicListURL, ip, params, &res); err != nil {
		return
	}
	if res.Code != ecode.OK.Code() {
		err = errors.Wrap(ecode.Int(res.Code), d.dynamicListURL+"?"+params.Encode())
		return
	}
	data = res.Data
	return
}

// Dynamic .
func (d *Dao) Dynamic(c context.Context, mid, dynamicID int64, qn int) (data *model.DyCard, err error) {
	ip := metadata.String(c, metadata.RemoteIP)
	params := url.Values{}
	if mid > 0 {
		params.Set("uid", strconv.FormatInt(mid, 10))
	}
	params.Set("dynamic_id", strconv.FormatInt(dynamicID, 10))
	params.Set("qn", strconv.Itoa(qn))
	var res struct {
		Code int `json:"code"`
		Data struct {
			Card *model.DyCard `json:"card"`
		} `json:"data"`
	}
	if err = d.httpR.Get(c, d.dynamicURL, ip, params, &res); err != nil {
		return
	}
	if res.Code != ecode.OK.Code() {
		err = errors.Wrap(ecode.Int(res.Code), d.dynamicURL+"?"+params.Encode())
		return
	}
	data = res.Data.Card
	return
}