100 lines
2.8 KiB
Go
100 lines
2.8 KiB
Go
package dao
|
||
|
||
import (
|
||
"context"
|
||
"go-common/app/interface/bbq/app-bbq/api/http/v1"
|
||
"go-common/app/interface/bbq/app-bbq/model"
|
||
user "go-common/app/service/bbq/user/api"
|
||
accountv1 "go-common/app/service/main/account/api"
|
||
"go-common/library/ecode"
|
||
"go-common/library/log"
|
||
)
|
||
|
||
//GetUserBProfile 获取用户全量b站信息
|
||
func (d *Dao) GetUserBProfile(c context.Context, mid int64) (res *accountv1.ProfileReply, err error) {
|
||
req := &accountv1.MidReq{
|
||
Mid: mid,
|
||
RealIp: "",
|
||
}
|
||
res, err = d.accountClient.Profile3(c, req)
|
||
return
|
||
}
|
||
|
||
// Login .
|
||
func (d *Dao) Login(c context.Context, userBase *user.UserBase) (res *user.UserBase, err error) {
|
||
res, err = d.userClient.Login(c, userBase)
|
||
if err != nil {
|
||
log.Errorv(c, log.KV("log", "login fail"))
|
||
}
|
||
return
|
||
}
|
||
|
||
// BatchUserInfo 提供批量获取UserInfo的方法
|
||
// 由于user service返回的结构和video的回包不同，因此这里进行映射，返回video-c的结构，避免外部使用方多次映射
|
||
func (d *Dao) BatchUserInfo(c context.Context, visitorMID int64, MIDs []int64, needDesc, needStat, needFollowState bool) (res map[int64]*v1.UserInfo, err error) {
|
||
res = make(map[int64]*v1.UserInfo)
|
||
if len(MIDs) == 0 {
|
||
return
|
||
}
|
||
if len(MIDs) > model.BatchUserLen {
|
||
err = ecode.BatchUserTooLong
|
||
return
|
||
}
|
||
|
||
userReq := &user.ListUserInfoReq{Mid: visitorMID, UpMid: MIDs, NeedDesc: needDesc, NeedStat: needStat, NeedFollowState: needFollowState}
|
||
reply, err := d.userClient.ListUserInfo(c, userReq)
|
||
if err != nil {
|
||
log.Errorv(c, log.KV("log", "get user info fail: req=%s"+userReq.String()))
|
||
return
|
||
}
|
||
|
||
for _, userInfo := range reply.List {
|
||
newUserInfo := &v1.UserInfo{UserBase: *userInfo.UserBase}
|
||
if userInfo.UserStat != nil {
|
||
newUserInfo.UserStat = *userInfo.UserStat
|
||
}
|
||
newUserInfo.FollowState = userInfo.FollowState
|
||
res[userInfo.UserBase.Mid] = newUserInfo
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
//JustGetUserBase 只取UserBase，不要其他
|
||
func (d *Dao) JustGetUserBase(c context.Context, mids []int64) (res map[int64]*user.UserBase, err error) {
|
||
res = make(map[int64]*user.UserBase)
|
||
userInfos, err := d.BatchUserInfo(c, 0, mids, false, false, false)
|
||
if err != nil {
|
||
log.Warnv(c, log.KV("log", "get user info fail"))
|
||
return
|
||
}
|
||
|
||
for mid, userInfo := range userInfos {
|
||
res[mid] = &userInfo.UserBase
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
// EditUserBase .
|
||
func (d *Dao) EditUserBase(c context.Context, userBase *user.UserBase) (err error) {
|
||
_, err = d.userClient.UserEdit(c, userBase)
|
||
if err != nil {
|
||
log.Warnw(c, "log", "edit user base fail: req="+userBase.String(), "err", err.Error())
|
||
return
|
||
}
|
||
return
|
||
}
|
||
|
||
// PhoneCheck .
|
||
func (d *Dao) PhoneCheck(c context.Context, mid int64) (telStatus int32, err error) {
|
||
req := &user.PhoneCheckReq{Mid: mid}
|
||
res, err := d.userClient.PhoneCheck(c, req)
|
||
if err != nil {
|
||
log.Errorw(c, "log", "call phone check fail", "err", err, "mid", mid)
|
||
return
|
||
}
|
||
telStatus = res.TelStatus
|
||
return
|
||
}
|