You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
165 lines
6.1 KiB
165 lines
6.1 KiB
3 years ago
|
package user
|
||
|
|
|
||
|
|
import (
|
||
|
|
"Open_IM/pkg/common/config"
|
||
|
|
"Open_IM/pkg/common/log"
|
||
|
|
"Open_IM/pkg/common/token_verify"
|
||
|
|
"Open_IM/pkg/grpc-etcdv3/getcdv3"
|
||
|
|
"Open_IM/pkg/utils"
|
||
|
|
|
||
|
|
// rpc "Open_IM/pkg/proto/relay"
|
||
|
|
api "Open_IM/pkg/base_info"
|
||
|
|
rpc "Open_IM/pkg/proto/user"
|
||
|
|
"context"
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
"net/http"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
//
|
||
|
|
//func GetUsersOnlineStatus(c *gin.Context) {
|
||
|
|
// params := api.GetUsersOnlineStatusReq{}
|
||
|
|
// if err := c.BindJSON(¶ms); err != nil {
|
||
|
|
// log.NewError(params.OperationID, "bind json failed ", err.Error(), c)
|
||
|
|
// c.JSON(http.StatusBadRequest, gin.H{"errCode": 400, "errMsg": err.Error()})
|
||
|
|
// return
|
||
|
|
// }
|
||
|
|
//
|
||
|
|
// if params.Secret != config.Config.Secret {
|
||
|
|
// log.NewError(params.OperationID, "parse token failed ", params.Secret, config.Config.Secret)
|
||
|
|
// c.JSON(http.StatusBadRequest, gin.H{"errCode": 401, "errMsg": "secret failed"})
|
||
|
|
// return
|
||
|
|
// }
|
||
|
|
//
|
||
|
|
// req := &pbRelay.GetUsersOnlineStatusReq{
|
||
|
|
// OperationID: params.OperationID,
|
||
|
|
// UserIDList: params.UserIDList,
|
||
|
|
// }
|
||
|
|
// var wsResult []*rpc.GetUsersOnlineStatusResp_SuccessResult
|
||
|
|
// var respResult []*rpc.GetUsersOnlineStatusResp_SuccessResult
|
||
|
|
// flag := false
|
||
|
|
// log.NewDebug(params.OperationID, "GetUsersOnlineStatus req come here", params.UserIDList)
|
||
|
|
//
|
||
|
|
// grpcCons := getcdv3.GetConn4Unique(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), config.Config.RpcRegisterName.OpenImOnlineMessageRelayName)
|
||
|
|
// for _, v := range grpcCons {
|
||
|
|
// client := rpc.NewOnlineMessageRelayServiceClient(v)
|
||
|
|
// reply, err := client.GetUsersOnlineStatus(context.Background(), req)
|
||
|
|
// if err != nil {
|
||
|
|
// log.NewError(params.OperationID, "GetUsersOnlineStatus rpc err", req.String(), err.Error())
|
||
|
|
// continue
|
||
|
|
// } else {
|
||
|
|
// if reply.ErrCode == 0 {
|
||
|
|
// wsResult = append(wsResult, reply.SuccessResult...)
|
||
|
|
// }
|
||
|
|
// }
|
||
|
|
// }
|
||
|
|
// log.NewDebug(params.OperationID, "call GetUsersOnlineStatus rpc server is success", wsResult)
|
||
|
|
// //Online data merge of each node
|
||
|
|
// for _, v1 := range params.UserIDList {
|
||
|
|
// flag = false
|
||
|
|
// temp := new(pbRelay.GetUsersOnlineStatusResp_SuccessResult)
|
||
|
|
// for _, v2 := range wsResult {
|
||
|
|
// if v2.UserID == v1 {
|
||
|
|
// flag = true
|
||
|
|
// temp.UserID = v1
|
||
|
|
// temp.Status = constant.OnlineStatus
|
||
|
|
// temp.DetailPlatformStatus = append(temp.DetailPlatformStatus, v2.DetailPlatformStatus...)
|
||
|
|
// }
|
||
|
|
// }
|
||
|
|
// if !flag {
|
||
|
|
// temp.UserID = v1
|
||
|
|
// temp.Status = constant.OfflineStatus
|
||
|
|
// }
|
||
|
|
// respResult = append(respResult, temp)
|
||
|
|
// }
|
||
|
|
// log.NewDebug(params.OperationID, "Finished merged data", respResult)
|
||
|
|
// resp := gin.H{"errCode": 0, "errMsg": "", "data": respResult}
|
||
|
|
//
|
||
|
|
// c.JSON(http.StatusOK, resp)
|
||
|
|
//}
|
||
|
|
//params := api.AddBlacklistReq{}
|
||
|
|
// if err := c.BindJSON(¶ms); err != nil {
|
||
|
|
// c.JSON(http.StatusBadRequest, gin.H{"errCode": 400, "errMsg": err.Error()})
|
||
|
|
// log.NewError("0", "BindJSON failed ", err.Error())
|
||
|
|
// return
|
||
|
|
// }
|
||
|
|
// req := &rpc.AddBlacklistReq{}
|
||
|
|
// utils.CopyStructFields(req.CommID, params)
|
||
|
|
// var ok bool
|
||
|
|
// ok, req.CommID.OpUserID = token_verify.GetUserIDFromToken(c.Request.Header.Get("token"))
|
||
|
|
// if !ok {
|
||
|
|
// log.NewError(req.CommID.OperationID, "GetUserIDFromToken false ", c.Request.Header.Get("token"))
|
||
|
|
// c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": "GetUserIDFromToken failed"})
|
||
|
|
// return
|
||
|
|
// }
|
||
|
|
// log.NewInfo(params.OperationID, "AddBlacklist args ", req.String())
|
||
|
|
|
||
|
|
func GetUserInfo(c *gin.Context) {
|
||
|
|
params := api.GetUserInfoReq{}
|
||
|
|
if err := c.BindJSON(¶ms); err != nil {
|
||
|
|
c.JSON(http.StatusBadRequest, gin.H{"errCode": http.StatusBadRequest, "errMsg": err.Error()})
|
||
|
|
return
|
||
|
|
}
|
||
|
|
req := &rpc.GetUserInfoReq{}
|
||
|
|
utils.CopyStructFields(&req, params)
|
||
|
|
var ok bool
|
||
|
|
ok, req.OpUserID = token_verify.GetUserIDFromToken(c.Request.Header.Get("token"))
|
||
|
|
if !ok {
|
||
|
|
log.NewError(req.OperationID, "GetUserIDFromToken false ", c.Request.Header.Get("token"))
|
||
|
|
c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": "GetUserIDFromToken failed"})
|
||
|
|
return
|
||
|
|
}
|
||
|
|
log.NewInfo(params.OperationID, "GetUserInfo args ", req.String())
|
||
|
|
|
||
|
|
etcdConn := getcdv3.GetConn(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), config.Config.RpcRegisterName.OpenImUserName)
|
||
|
|
client := rpc.NewUserClient(etcdConn)
|
||
|
|
RpcResp, err := client.GetUserInfo(context.Background(), req)
|
||
|
|
if err != nil {
|
||
|
|
log.NewError(req.OperationID, "GetUserInfo failed ", err.Error(), req.String())
|
||
|
|
c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": "call rpc server failed"})
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
resp := api.GetUserInfoResp{CommResp: api.CommResp{ErrCode: RpcResp.CommonResp.ErrCode, ErrMsg: RpcResp.CommonResp.ErrMsg}}
|
||
|
|
resp.UserInfoList = RpcResp.UserInfoList
|
||
|
|
c.JSON(http.StatusOK, resp)
|
||
|
|
|
||
|
|
log.NewInfo(req.OperationID, "GetUserInfo api return ", resp)
|
||
|
|
}
|
||
|
|
|
||
|
|
func UpdateUserInfo(c *gin.Context) {
|
||
|
|
log.InfoByKv("api update userinfo init...", "")
|
||
|
|
|
||
|
|
etcdConn := getcdv3.GetConn(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), config.Config.RpcRegisterName.OpenImUserName)
|
||
|
|
client := pbUser.NewUserClient(etcdConn)
|
||
|
|
//defer etcdConn.Close()
|
||
|
|
|
||
|
|
params := paramsStruct{}
|
||
|
|
if err := c.BindJSON(¶ms); err != nil {
|
||
|
|
c.JSON(http.StatusBadRequest, gin.H{"errCode": 400, "errMsg": err.Error()})
|
||
|
|
return
|
||
|
|
}
|
||
|
|
req := &pbUser.UpdateUserInfoReq{
|
||
|
|
OperationID: params.OperationID,
|
||
|
|
Token: c.Request.Header.Get("token"),
|
||
|
|
Name: params.Name,
|
||
|
|
Icon: params.Icon,
|
||
|
|
Gender: params.Gender,
|
||
|
|
Mobile: params.Mobile,
|
||
|
|
Birth: params.Birth,
|
||
|
|
Email: params.Email,
|
||
|
|
Ex: params.Ex,
|
||
|
|
Uid: params.Uid,
|
||
|
|
}
|
||
|
|
log.InfoByKv("api update user info is server", req.OperationID, req.Token)
|
||
|
|
RpcResp, err := client.UpdateUserInfo(context.Background(), req)
|
||
|
|
if err != nil {
|
||
|
|
log.Error(req.Token, req.OperationID, "err=%s,call get user info rpc server failed", err)
|
||
|
|
c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": "call rpc server failed"})
|
||
|
|
return
|
||
|
|
}
|
||
|
|
log.InfoByKv("call update user info rpc server success", params.OperationID)
|
||
|
|
c.JSON(http.StatusOK, gin.H{"errCode": RpcResp.ErrorCode, "errMsg": RpcResp.ErrorMsg})
|
||
|
|
log.InfoByKv("api update user info return success", params.OperationID, "args=%s", RpcResp.String())
|
||
|
|
}
|