bilibili-backup/app/interface/main/app-interface/dao/account/dao.go

package account

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

	"go-common/app/interface/main/app-interface/conf"
	account "go-common/app/service/main/account/model"
	accrpc "go-common/app/service/main/account/rpc/client"
	memberrpc "go-common/app/service/main/member/api/gorpc"
	blockmodel "go-common/app/service/main/member/model/block"
	"go-common/library/ecode"
	"go-common/library/log"
	bm "go-common/library/net/http/blademaster"
	"go-common/library/net/metadata"

	"github.com/pkg/errors"
)

// Dao is account dao.
type Dao struct {
	client *bm.Client
	// rpc
	accRPC    *accrpc.Service3
	memberRPC *memberrpc.Service
}

// New account dao.
func New(c *conf.Config) (d *Dao) {
	d = &Dao{
		client:    bm.NewClient(c.HTTPClient),
		accRPC:    accrpc.New3(c.AccountRPC),
		memberRPC: memberrpc.New(c.MemberRPC),
	}
	return
}

// BlockTime get user blocktime
func (d *Dao) BlockTime(c context.Context, mid int64) (blockTime int64, err error) {
	info, err := d.memberRPC.BlockInfo(c, &blockmodel.RPCArgInfo{MID: mid})
	if err != nil {
		err = errors.Wrapf(err, "%v", mid)
		return
	}
	if info.EndTime > 0 {
		blockTime = info.EndTime
	}
	return
}

// Profile3 get profile
func (d *Dao) Profile3(c context.Context, mid int64) (card *account.ProfileStat, err error) {
	arg := &account.ArgMid{Mid: mid}
	if card, err = d.accRPC.ProfileWithStat3(c, arg); err != nil {
		err = errors.Wrapf(err, "%v", arg)
	}
	return
}

// Card get card
func (d *Dao) Card(c context.Context, mid int64) (card *account.Card, err error) {
	if card, err = d.accRPC.Card3(c, &account.ArgMid{Mid: mid}); err != nil {
		err = errors.Wrapf(err, "%v", mid)
	}
	return
}

// ProfileByName3 rpc card get by name
func (d *Dao) ProfileByName3(c context.Context, name string) (card *account.ProfileStat, err error) {
	infos, err := d.accRPC.InfosByName3(c, &account.ArgNames{Names: []string{name}})
	if err != nil {
		err = errors.Wrapf(err, "%v", name)
		return
	}
	if len(infos) == 0 {
		err = ecode.NothingFound
		return
	}
	for mid := range infos {
		card, err = d.Profile3(c, mid)
		break
	}
	return
}

// Infos3 rpc info get by mids .
func (d *Dao) Infos3(c context.Context, mids []int64) (res map[int64]*account.Info, err error) {
	arg := &account.ArgMids{Mids: mids}
	if res, err = d.accRPC.Infos3(c, arg); err != nil {
		err = errors.Wrapf(err, "%v", arg)
	}
	return
}

// Relations3 relations.
func (d *Dao) Relations3(c context.Context, owners []int64, mid int64) (follows map[int64]bool) {
	if len(owners) == 0 {
		return nil
	}
	follows = make(map[int64]bool, len(owners))
	for _, owner := range owners {
		follows[owner] = false
	}
	var (
		am  map[int64]*account.Relation
		err error
		ip  = metadata.String(c, metadata.RemoteIP)
	)
	arg := &account.ArgRelations{Owners: owners, Mid: mid, RealIP: ip}
	if am, err = d.accRPC.Relations3(c, arg); err != nil {
		log.Error("d.accRPC.Relations2(%v) error(%v)", arg, err)
		return
	}
	for i, a := range am {
		if _, ok := follows[i]; ok {
			follows[i] = a.Following
		}
	}
	return
}

// RichRelations3 rich relations.
func (d *Dao) RichRelations3(c context.Context, owner, mid int64) (rel int, err error) {
	var (
		res map[int64]int
		ip  = metadata.String(c, metadata.RemoteIP)
	)
	arg := &account.ArgRichRelation{Mids: []int64{mid}, Owner: owner, RealIP: ip}
	if res, err = d.accRPC.RichRelations3(c, arg); err != nil {
		err = errors.Wrapf(err, "%v", arg)
		return
	}
	if r, ok := res[mid]; ok {
		rel = r
	}
	return
}

// Cards3 is
func (d *Dao) Cards3(c context.Context, mids []int64) (res map[int64]*account.Card, err error) {
	arg := &account.ArgMids{Mids: mids}
	if res, err = d.accRPC.Cards3(c, arg); err != nil {
		err = errors.Wrapf(err, "%v")
	}
	return
}

// UserCheck 各种入口白名单
// https://www.tapd.cn/20055921/prong/stories/view/1120055921001066980  动态互推TAPD在此！！
func (d *Dao) UserCheck(c context.Context, mid int64, checkURL string) (ok bool, err error) {
	params := url.Values{}
	params.Set("uid", strconv.FormatInt(mid, 10))
	var res struct {
		Code int `json:"code"`
		Data struct {
			Status int `json:"status"`
		} `json:"data"`
	}
	if err = d.client.Get(c, checkURL, "", params, &res); err != nil {
		return
	}
	if res.Code != ecode.OK.Code() {
		err = errors.Wrap(ecode.Int(res.Code), checkURL+"?"+params.Encode())
		return
	}
	if res.Data.Status == 1 {
		ok = true
	}
	return
}

// RedDot 我的页小红点逻辑
func (d *Dao) RedDot(c context.Context, mid int64, redDotURL string) (ok bool, err error) {
	params := url.Values{}
	params.Set("mid", strconv.FormatInt(mid, 10))
	var res struct {
		Code int `json:"code"`
		Data struct {
			RedDot bool `json:"red_dot"`
		} `json:"data"`
	}
	if err = d.client.Get(c, redDotURL, "", params, &res); err != nil {
		return
	}
	if res.Code != ecode.OK.Code() {
		err = errors.Wrap(ecode.Int(res.Code), redDotURL+"?"+params.Encode())
		return
	}
	log.Warn("reddot response mid(%d) url(%s) res(%t)", mid, redDotURL+"?"+params.Encode(), res.Data.RedDot)
	ok = res.Data.RedDot
	return
}