openbilibili/app/interface/main/app-wall/dao/live/live.go

package live

import (
	"bytes"
	"context"
	"encoding/json"
	"net/http"
	"net/url"
	"strconv"

	"go-common/app/interface/main/app-wall/conf"
	"go-common/library/ecode"
	"go-common/library/log"
	httpx "go-common/library/net/http/blademaster"
)

const (
	_liveURL   = "/gift/simGift"
	_addVipURL = "/user/v0/Vip/addVip"
)

// Dao is live dao
type Dao struct {
	client    *httpx.Client
	liveURL   string
	addVipURL string
}

// New live dao.
func New(c *conf.Config) (d *Dao) {
	d = &Dao{
		client:    httpx.NewClient(c.HTTPClient),
		liveURL:   c.Host.Live + _liveURL,
		addVipURL: c.Host.APILive + _addVipURL,
	}
	return
}

// Pack
func (d *Dao) Pack(c context.Context, mid int64, cardType int) (err error) {
	var res struct {
		Code int `json:"code"`
	}
	upack := map[string]map[string]interface{}{
		"header": map[string]interface{}{},
		"body": map[string]interface{}{
			"uid":       mid,
			"card_type": cardType,
		},
	}
	bytesData, err := json.Marshal(upack)
	if err != nil {
		log.Error("json.Marshal error(%v)", err)
		return
	}
	req, err := http.NewRequest("POST", d.liveURL, bytes.NewReader(bytesData))
	if err != nil {
		log.Error("http.NewRequest error(%v)", err)
		return
	}
	req.Header.Set("Content-Type", "application/json;charset=UTF-8")
	req.Header.Set("X-BACKEND-BILI-REAL-IP", "")
	log.Info("unicom pack mid(%d) card_type(%d)", mid, cardType)
	if err = d.client.Do(c, req, &res); err != nil || res.Code != 0 {
		err = ecode.Int(res.Code)
		log.Error("unicom pack d.client.Do(%s) mid(%d) card_type(%d) error(%v)", d.liveURL, mid, cardType, err)
		return
	}
	return
}

// AddVip add live vip
func (d *Dao) AddVip(c context.Context, mid int64, day int) (msg string, err error) {
	params := url.Values{}
	params.Set("vip_type", "1")
	params.Set("day", strconv.Itoa(day))
	params.Set("uid", strconv.FormatInt(mid, 10))
	params.Set("platform", "main")
	var res struct {
		Code int    `json:"code"`
		Msg  string `json:"message"`
	}
	if err = d.client.Post(c, d.addVipURL, "", params, &res); err != nil {
		log.Error("live add vip url(%v) error(%v)", d.addVipURL+"?"+params.Encode(), err)
		return
	}
	msg = res.Msg
	if res.Code != 0 {
		err = ecode.Int(res.Code)
		log.Error("live add vip url(%v) res code(%d)", d.addVipURL+"?"+params.Encode(), res.Code)
		return
	}
	return
}