Files
open-im-server/internal/api/user/user.go
T

515 lines
22 KiB
Go
Raw Normal View History

2021-12-27 16:48:05 +08:00
package user
import (
2021-12-31 09:25:11 +08:00
jsonData "Open_IM/internal/utils"
2021-12-27 18:22:32 +08:00
api "Open_IM/pkg/base_info"
2021-12-27 16:48:05 +08:00
"Open_IM/pkg/common/config"
2022-04-18 14:27:12 +08:00
"Open_IM/pkg/common/constant"
2021-12-27 16:48:05 +08:00
"Open_IM/pkg/common/log"
"Open_IM/pkg/common/token_verify"
"Open_IM/pkg/grpc-etcdv3/getcdv3"
2022-04-28 10:46:21 +08:00
cacheRpc "Open_IM/pkg/proto/cache"
2022-04-18 14:27:12 +08:00
pbRelay "Open_IM/pkg/proto/relay"
2021-12-28 19:28:33 +08:00
open_im_sdk "Open_IM/pkg/proto/sdk_ws"
2021-12-27 16:48:05 +08:00
rpc "Open_IM/pkg/proto/user"
2021-12-27 18:22:32 +08:00
"Open_IM/pkg/utils"
2021-12-27 16:48:05 +08:00
"context"
"net/http"
"strings"
2022-08-17 14:57:57 +08:00
"github.com/gin-gonic/gin"
2021-12-27 16:48:05 +08:00
)
2022-04-27 14:43:46 +08:00
func GetUsersInfoFromCache(c *gin.Context) {
params := api.GetUsersInfoReq{}
if err := c.BindJSON(&params); err != nil {
log.NewError("0", "BindJSON failed ", err.Error())
c.JSON(http.StatusBadRequest, gin.H{"errCode": http.StatusBadRequest, "errMsg": err.Error()})
return
}
2022-05-10 10:44:43 +08:00
log.NewInfo(params.OperationID, "GetUsersInfoFromCache req: ", params)
2022-04-28 15:13:39 +08:00
req := &rpc.GetUserInfoReq{}
utils.CopyStructFields(req, &params)
2022-04-27 14:43:46 +08:00
var ok bool
2022-05-07 11:55:19 +08:00
var errInfo string
ok, req.OpUserID, errInfo = token_verify.GetUserIDFromToken(c.Request.Header.Get("token"), req.OperationID)
2022-04-27 14:43:46 +08:00
if !ok {
2022-05-10 10:44:43 +08:00
errMsg := "GetUserIDFromToken failed " + errInfo + " token:" + c.Request.Header.Get("token")
2022-05-07 11:55:19 +08:00
log.NewError(req.OperationID, errMsg)
2022-06-28 10:29:14 +08:00
c.JSON(http.StatusBadRequest, gin.H{"errCode": 500, "errMsg": errMsg})
2022-04-27 14:43:46 +08:00
return
}
2022-08-17 12:12:54 +08:00
etcdConn := getcdv3.GetDefaultConn(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), config.Config.RpcRegisterName.OpenImUserName, req.OperationID)
2022-06-16 14:35:24 +08:00
if etcdConn == nil {
2022-08-17 12:12:54 +08:00
errMsg := req.OperationID + "getcdv3.GetDefaultConn == nil"
2022-06-16 14:35:24 +08:00
log.NewError(req.OperationID, errMsg)
c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": errMsg})
return
}
2022-04-28 15:13:39 +08:00
client := rpc.NewUserClient(etcdConn)
RpcResp, err := client.GetUserInfo(context.Background(), req)
2022-04-28 10:46:21 +08:00
if err != nil {
2022-04-28 15:13:39 +08:00
log.NewError(req.OperationID, "GetUserInfo failed ", err.Error(), req.String())
c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": "call rpc server failed"})
2022-04-28 10:46:21 +08:00
return
}
2022-04-28 15:13:39 +08:00
var publicUserInfoList []*open_im_sdk.PublicUserInfo
for _, v := range RpcResp.UserInfoList {
publicUserInfoList = append(publicUserInfoList,
&open_im_sdk.PublicUserInfo{UserID: v.UserID, Nickname: v.Nickname, FaceURL: v.FaceURL, Gender: v.Gender, Ex: v.Ex})
2022-04-28 10:46:21 +08:00
}
2022-04-28 15:13:39 +08:00
resp := api.GetUsersInfoResp{CommResp: api.CommResp{ErrCode: RpcResp.CommonResp.ErrCode, ErrMsg: RpcResp.CommonResp.ErrMsg}, UserInfoList: publicUserInfoList}
resp.Data = jsonData.JsonDataList(resp.UserInfoList)
log.NewInfo(req.OperationID, "GetUserInfo api return ", resp)
c.JSON(http.StatusOK, resp)
}
func GetFriendIDListFromCache(c *gin.Context) {
var (
req api.GetFriendIDListFromCacheReq
resp api.GetFriendIDListFromCacheResp
reqPb cacheRpc.GetFriendIDListFromCacheReq
respPb *cacheRpc.GetFriendIDListFromCacheResp
)
if err := c.BindJSON(&req); err != nil {
log.NewError(req.OperationID, "BindJSON failed ", err.Error())
c.JSON(http.StatusBadRequest, gin.H{"errCode": http.StatusBadRequest, "errMsg": err.Error()})
return
2022-04-28 10:46:21 +08:00
}
2022-05-10 10:44:43 +08:00
log.NewInfo(req.OperationID, utils.GetSelfFuncName(), req)
2022-04-28 15:13:39 +08:00
reqPb.OperationID = req.OperationID
var ok bool
2022-05-07 11:55:19 +08:00
var errInfo string
ok, reqPb.UserID, errInfo = token_verify.GetUserIDFromToken(c.Request.Header.Get("token"), req.OperationID)
2022-04-28 15:13:39 +08:00
if !ok {
2022-05-10 10:44:43 +08:00
errMsg := "GetUserIDFromToken failed " + errInfo + " token:" + c.Request.Header.Get("token")
2022-05-07 11:55:19 +08:00
log.NewError(req.OperationID, errMsg)
c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": errMsg})
2022-04-28 10:46:21 +08:00
return
}
2022-08-17 12:12:54 +08:00
etcdConn := getcdv3.GetDefaultConn(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), config.Config.RpcRegisterName.OpenImCacheName, req.OperationID)
2022-06-16 14:35:24 +08:00
if etcdConn == nil {
2022-08-17 12:12:54 +08:00
errMsg := req.OperationID + "getcdv3.GetDefaultConn == nil"
2022-06-16 14:35:24 +08:00
log.NewError(req.OperationID, errMsg)
c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": errMsg})
return
}
2022-04-28 15:13:39 +08:00
client := cacheRpc.NewCacheClient(etcdConn)
respPb, err := client.GetFriendIDListFromCache(context.Background(), &reqPb)
2022-04-27 14:43:46 +08:00
if err != nil {
2022-04-28 15:13:39 +08:00
log.NewError(req.OperationID, utils.GetSelfFuncName(), "GetFriendIDListFromCache", err.Error())
c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": "call rpc server failed:" + err.Error()})
2022-04-27 14:43:46 +08:00
return
}
2022-04-28 15:13:39 +08:00
resp.UserIDList = respPb.UserIDList
resp.CommResp = api.CommResp{ErrMsg: respPb.CommonResp.ErrMsg, ErrCode: respPb.CommonResp.ErrCode}
c.JSON(http.StatusOK, resp)
}
func GetBlackIDListFromCache(c *gin.Context) {
var (
req api.GetBlackIDListFromCacheReq
resp api.GetBlackIDListFromCacheResp
reqPb cacheRpc.GetBlackIDListFromCacheReq
respPb *cacheRpc.GetBlackIDListFromCacheResp
)
if err := c.BindJSON(&req); err != nil {
log.NewError(req.OperationID, "BindJSON failed ", err.Error())
c.JSON(http.StatusBadRequest, gin.H{"errCode": http.StatusBadRequest, "errMsg": err.Error()})
2022-04-28 10:46:21 +08:00
return
}
2022-05-10 10:44:43 +08:00
log.NewInfo(req.OperationID, utils.GetSelfFuncName(), "req: ", req)
2022-04-28 15:13:39 +08:00
reqPb.OperationID = req.OperationID
var ok bool
2022-05-07 11:55:19 +08:00
var errInfo string
ok, reqPb.UserID, errInfo = token_verify.GetUserIDFromToken(c.Request.Header.Get("token"), req.OperationID)
2022-04-28 15:13:39 +08:00
if !ok {
2022-05-07 11:55:19 +08:00
errMsg := req.OperationID + " " + "GetUserIDFromToken failed " + errInfo + " token:" + c.Request.Header.Get("token")
log.NewError(req.OperationID, errMsg)
c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": errMsg})
2022-04-28 15:13:39 +08:00
return
2022-04-28 10:46:21 +08:00
}
2022-08-17 12:12:54 +08:00
etcdConn := getcdv3.GetDefaultConn(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), config.Config.RpcRegisterName.OpenImCacheName, req.OperationID)
2022-06-16 14:35:24 +08:00
if etcdConn == nil {
2022-08-17 12:12:54 +08:00
errMsg := req.OperationID + "getcdv3.GetDefaultConn == nil"
2022-06-16 14:35:24 +08:00
log.NewError(req.OperationID, errMsg)
c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": errMsg})
return
}
2022-04-28 15:13:39 +08:00
client := cacheRpc.NewCacheClient(etcdConn)
respPb, err := client.GetBlackIDListFromCache(context.Background(), &reqPb)
2022-04-28 10:46:21 +08:00
if err != nil {
2022-04-28 15:13:39 +08:00
log.NewError(req.OperationID, utils.GetSelfFuncName(), "GetFriendIDListFromCache", err.Error())
2022-04-28 10:46:21 +08:00
c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": "call rpc server failed:" + err.Error()})
return
}
2022-04-28 15:13:39 +08:00
resp.UserIDList = respPb.UserIDList
resp.CommResp = api.CommResp{ErrMsg: respPb.CommonResp.ErrMsg, ErrCode: respPb.CommonResp.ErrCode}
2022-04-27 14:43:46 +08:00
c.JSON(http.StatusOK, resp)
}
2022-06-28 10:29:14 +08:00
// @Summary 获取用户信息
// @Description 根据用户列表批量获取用户信息
// @Tags 用户相关
// @ID GetUsersInfo
// @Accept json
// @Param token header string true "im token"
// @Param req body api.GetUsersInfoReq true "请求体"
// @Produce json
2022-06-29 18:48:17 +08:00
// @Success 0 {object} api.GetUsersInfoResp{Data=[]open_im_sdk.PublicUserInfo}
// @Failure 500 {object} api.Swagger500Resp "errCode为500 一般为服务器内部错误"
// @Failure 400 {object} api.Swagger400Resp "errCode为400 一般为参数输入错误, token未带上等"
2022-06-28 10:29:14 +08:00
// @Router /user/get_users_info [post]
func GetUsersPublicInfo(c *gin.Context) {
2022-01-13 11:10:29 +08:00
params := api.GetUsersInfoReq{}
2021-12-27 16:48:05 +08:00
if err := c.BindJSON(&params); err != nil {
2021-12-28 18:18:35 +08:00
log.NewError("0", "BindJSON failed ", err.Error())
2021-12-27 16:48:05 +08:00
c.JSON(http.StatusBadRequest, gin.H{"errCode": http.StatusBadRequest, "errMsg": err.Error()})
return
}
2022-04-26 10:33:49 +08:00
req := &rpc.GetUserInfoReq{}
utils.CopyStructFields(req, &params)
2022-05-07 11:55:19 +08:00
2021-12-27 16:48:05 +08:00
var ok bool
2022-05-07 11:55:19 +08:00
var errInfo string
ok, req.OpUserID, errInfo = token_verify.GetUserIDFromToken(c.Request.Header.Get("token"), req.OperationID)
2021-12-27 16:48:05 +08:00
if !ok {
2022-05-07 11:55:19 +08:00
errMsg := req.OperationID + " " + "GetUserIDFromToken failed " + errInfo + " token:" + c.Request.Header.Get("token")
log.NewError(req.OperationID, errMsg)
2022-06-28 10:29:14 +08:00
c.JSON(http.StatusBadRequest, gin.H{"errCode": 500, "errMsg": errMsg})
2021-12-27 16:48:05 +08:00
return
}
2022-05-07 11:55:19 +08:00
2022-04-26 10:33:49 +08:00
log.NewInfo(params.OperationID, "GetUserInfo args ", req.String())
2021-12-27 16:48:05 +08:00
2022-08-17 12:12:54 +08:00
etcdConn := getcdv3.GetDefaultConn(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), config.Config.RpcRegisterName.OpenImUserName, req.OperationID)
2022-06-16 14:35:24 +08:00
if etcdConn == nil {
2022-08-17 12:12:54 +08:00
errMsg := req.OperationID + "getcdv3.GetDefaultConn == nil"
2022-06-16 14:35:24 +08:00
log.NewError(req.OperationID, errMsg)
c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": errMsg})
return
}
2021-12-27 16:48:05 +08:00
client := rpc.NewUserClient(etcdConn)
2022-04-26 10:33:49 +08:00
RpcResp, err := client.GetUserInfo(context.Background(), req)
2022-04-25 20:05:21 +08:00
if err != nil {
2022-04-26 10:33:49 +08:00
log.NewError(req.OperationID, "GetUserInfo failed ", err.Error(), req.String())
2021-12-27 16:48:05 +08:00
c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": "call rpc server failed"})
return
}
2022-04-26 10:33:49 +08:00
var publicUserInfoList []*open_im_sdk.PublicUserInfo
for _, v := range RpcResp.UserInfoList {
2022-01-13 11:10:29 +08:00
publicUserInfoList = append(publicUserInfoList,
2022-03-08 18:48:36 +08:00
&open_im_sdk.PublicUserInfo{UserID: v.UserID, Nickname: v.Nickname, FaceURL: v.FaceURL, Gender: v.Gender, Ex: v.Ex})
2022-01-13 11:10:29 +08:00
}
2022-04-26 10:33:49 +08:00
resp := api.GetUsersInfoResp{CommResp: api.CommResp{ErrCode: RpcResp.CommonResp.ErrCode, ErrMsg: RpcResp.CommonResp.ErrMsg}, UserInfoList: publicUserInfoList}
2021-12-31 09:25:11 +08:00
resp.Data = jsonData.JsonDataList(resp.UserInfoList)
2022-04-26 10:33:49 +08:00
log.NewInfo(req.OperationID, "GetUserInfo api return ", resp)
2021-12-28 18:18:35 +08:00
c.JSON(http.StatusOK, resp)
2021-12-27 16:48:05 +08:00
}
2022-06-24 18:33:52 +08:00
// @Summary 修改用户信息
2022-06-24 20:29:53 +08:00
// @Description 修改用户信息 userID faceURL等
2022-06-28 10:29:14 +08:00
// @Tags 用户相关
2022-06-24 20:29:53 +08:00
// @ID UpdateUserInfo
2022-06-24 17:45:33 +08:00
// @Accept json
2022-06-24 20:29:53 +08:00
// @Param token header string true "im token"
2022-06-28 10:29:14 +08:00
// @Param req body api.UpdateSelfUserInfoReq true "请求体"
2022-06-24 20:29:53 +08:00
// @Produce json
// @Success 0 {object} api.UpdateUserInfoResp
2022-06-29 18:48:17 +08:00
// @Failure 500 {object} api.Swagger500Resp "errCode为500 一般为服务器内部错误"
// @Failure 400 {object} api.Swagger400Resp "errCode为400 一般为参数输入错误, token未带上等"
2022-06-24 20:29:53 +08:00
// @Router /user/update_user_info [post]
2021-12-27 16:48:05 +08:00
func UpdateUserInfo(c *gin.Context) {
2022-01-13 11:10:29 +08:00
params := api.UpdateSelfUserInfoReq{}
2021-12-27 16:48:05 +08:00
if err := c.BindJSON(&params); err != nil {
2021-12-28 18:18:35 +08:00
log.NewError("0", "BindJSON failed ", err.Error())
2021-12-27 16:48:05 +08:00
c.JSON(http.StatusBadRequest, gin.H{"errCode": 400, "errMsg": err.Error()})
return
}
2021-12-28 19:28:33 +08:00
req := &rpc.UpdateUserInfoReq{UserInfo: &open_im_sdk.UserInfo{}}
utils.CopyStructFields(req.UserInfo, &params)
2021-12-28 20:03:20 +08:00
req.OperationID = params.OperationID
2021-12-27 18:22:32 +08:00
var ok bool
2022-05-07 11:55:19 +08:00
var errInfo string
ok, req.OpUserID, errInfo = token_verify.GetUserIDFromToken(c.Request.Header.Get("token"), req.OperationID)
2021-12-27 18:22:32 +08:00
if !ok {
2022-05-07 11:55:19 +08:00
errMsg := req.OperationID + " " + "GetUserIDFromToken failed " + errInfo + " token:" + c.Request.Header.Get("token")
log.NewError(req.OperationID, errMsg)
2022-06-28 10:29:14 +08:00
c.JSON(http.StatusBadRequest, gin.H{"errCode": 500, "errMsg": errMsg})
2021-12-27 18:22:32 +08:00
return
2021-12-27 16:48:05 +08:00
}
2021-12-27 18:22:32 +08:00
log.NewInfo(params.OperationID, "UpdateUserInfo args ", req.String())
2022-08-17 12:12:54 +08:00
etcdConn := getcdv3.GetDefaultConn(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), config.Config.RpcRegisterName.OpenImUserName, req.OperationID)
2022-06-16 14:35:24 +08:00
if etcdConn == nil {
2022-08-17 12:12:54 +08:00
errMsg := req.OperationID + "getcdv3.GetDefaultConn == nil"
2022-06-16 14:35:24 +08:00
log.NewError(req.OperationID, errMsg)
c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": errMsg})
return
}
2022-04-26 10:33:49 +08:00
client := rpc.NewUserClient(etcdConn)
RpcResp, err := client.UpdateUserInfo(context.Background(), req)
2021-12-27 16:48:05 +08:00
if err != nil {
2021-12-27 18:22:32 +08:00
log.NewError(req.OperationID, "UpdateUserInfo failed ", err.Error(), req.String())
2021-12-27 16:48:05 +08:00
c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": "call rpc server failed"})
return
}
2022-04-28 15:13:39 +08:00
resp := api.UpdateUserInfoResp{CommResp: api.CommResp{ErrCode: RpcResp.CommonResp.ErrCode, ErrMsg: RpcResp.CommonResp.ErrMsg}}
2022-04-26 10:33:49 +08:00
log.NewInfo(req.OperationID, "UpdateUserInfo api return ", resp)
c.JSON(http.StatusOK, resp)
2021-12-27 16:48:05 +08:00
}
2022-06-24 17:45:33 +08:00
2022-06-28 10:29:14 +08:00
// @Summary 设置全局免打扰
// @Description 设置全局免打扰
// @Tags 用户相关
// @ID SetGlobalRecvMessageOpt
// @Accept json
// @Param token header string true "im token"
2022-06-29 18:48:17 +08:00
// @Param req body api.SetGlobalRecvMessageOptReq true "globalRecvMsgOpt为全局免打扰设置0为关闭 1为开启"
2022-06-28 10:29:14 +08:00
// @Produce json
// @Success 0 {object} api.SetGlobalRecvMessageOptResp
2022-06-29 18:48:17 +08:00
// @Failure 500 {object} api.Swagger500Resp "errCode为500 一般为服务器内部错误"
// @Failure 400 {object} api.Swagger400Resp "errCode为400 一般为参数输入错误, token未带上等"
2022-06-28 10:29:14 +08:00
// @Router /user/set_global_msg_recv_opt [post]
2022-06-16 19:33:35 +08:00
func SetGlobalRecvMessageOpt(c *gin.Context) {
params := api.SetGlobalRecvMessageOptReq{}
if err := c.BindJSON(&params); err != nil {
log.NewError("0", "BindJSON failed ", err.Error())
c.JSON(http.StatusBadRequest, gin.H{"errCode": 400, "errMsg": err.Error()})
return
}
req := &rpc.SetGlobalRecvMessageOptReq{}
utils.CopyStructFields(req, &params)
req.OperationID = params.OperationID
var ok bool
var errInfo string
ok, req.UserID, errInfo = token_verify.GetUserIDFromToken(c.Request.Header.Get("token"), req.OperationID)
if !ok {
errMsg := req.OperationID + " " + "GetUserIDFromToken failed " + errInfo + " token:" + c.Request.Header.Get("token")
log.NewError(req.OperationID, errMsg)
2022-06-28 10:29:14 +08:00
c.JSON(http.StatusBadRequest, gin.H{"errCode": 500, "errMsg": errMsg})
2022-06-16 19:33:35 +08:00
return
}
log.NewInfo(params.OperationID, "SetGlobalRecvMessageOpt args ", req.String())
2022-08-17 12:12:54 +08:00
etcdConn := getcdv3.GetDefaultConn(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), config.Config.RpcRegisterName.OpenImUserName, req.OperationID)
2022-06-16 19:33:35 +08:00
if etcdConn == nil {
2022-08-17 12:12:54 +08:00
errMsg := req.OperationID + "getcdv3.GetDefaultConn == nil"
2022-06-16 19:33:35 +08:00
log.NewError(req.OperationID, errMsg)
c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": errMsg})
return
}
client := rpc.NewUserClient(etcdConn)
RpcResp, err := client.SetGlobalRecvMessageOpt(context.Background(), req)
if err != nil {
log.NewError(req.OperationID, "SetGlobalRecvMessageOpt failed ", err.Error(), req.String())
c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": "call rpc server failed"})
return
}
resp := api.UpdateUserInfoResp{CommResp: api.CommResp{ErrCode: RpcResp.CommonResp.ErrCode, ErrMsg: RpcResp.CommonResp.ErrMsg}}
log.NewInfo(req.OperationID, "SetGlobalRecvMessageOpt api return ", resp)
c.JSON(http.StatusOK, resp)
}
2022-01-13 11:10:29 +08:00
2022-06-28 10:29:14 +08:00
// @Summary 获取自己的信息
// @Description 传入ID获取自己的信息
// @Tags 用户相关
// @ID GetSelfUserInfo
// @Accept json
// @Param token header string true "im token"
// @Param req body api.GetSelfUserInfoReq true "请求体"
// @Produce json
2022-06-29 11:18:19 +08:00
// @Success 0 {object} api.GetSelfUserInfoResp{data=open_im_sdk.UserInfo}
2022-06-29 18:48:17 +08:00
// @Failure 500 {object} api.Swagger500Resp "errCode为500 一般为服务器内部错误"
// @Failure 400 {object} api.Swagger400Resp "errCode为400 一般为参数输入错误, token未带上等"
2022-06-28 10:29:14 +08:00
// @Router /user/get_self_user_info [post]
2022-01-13 11:10:29 +08:00
func GetSelfUserInfo(c *gin.Context) {
params := api.GetSelfUserInfoReq{}
if err := c.BindJSON(&params); err != nil {
errMsg := " BindJSON failed " + err.Error()
log.NewError("0", errMsg)
c.JSON(http.StatusOK, gin.H{"errCode": 1001, "errMsg": errMsg})
2022-01-13 11:10:29 +08:00
return
}
req := &rpc.GetUserInfoReq{}
2022-01-13 11:16:55 +08:00
2022-01-13 11:10:29 +08:00
utils.CopyStructFields(req, &params)
2022-05-07 11:55:19 +08:00
2022-01-13 11:10:29 +08:00
var ok bool
2022-05-07 11:55:19 +08:00
var errInfo string
ok, req.OpUserID, errInfo = token_verify.GetUserIDFromToken(c.Request.Header.Get("token"), req.OperationID)
2022-01-13 11:10:29 +08:00
if !ok {
2022-05-07 11:55:19 +08:00
errMsg := req.OperationID + " " + "GetUserIDFromToken failed " + errInfo + " token:" + c.Request.Header.Get("token")
log.NewError(params.OperationID, errMsg)
c.JSON(http.StatusOK, gin.H{"errCode": 1001, "errMsg": errMsg})
2022-01-13 11:10:29 +08:00
return
}
2022-05-07 11:55:19 +08:00
2022-08-08 20:03:08 +08:00
req.UserIDList = append(req.UserIDList, params.UserID)
2022-01-13 11:10:29 +08:00
log.NewInfo(params.OperationID, "GetUserInfo args ", req.String())
2022-08-17 12:12:54 +08:00
etcdConn := getcdv3.GetDefaultConn(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), config.Config.RpcRegisterName.OpenImUserName, req.OperationID)
2022-06-16 14:35:24 +08:00
if etcdConn == nil {
2022-08-17 12:12:54 +08:00
errMsg := req.OperationID + "getcdv3.GetDefaultConn == nil"
2022-06-16 14:35:24 +08:00
log.NewError(req.OperationID, errMsg)
c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": errMsg})
return
}
2022-01-13 11:10:29 +08:00
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
}
if len(RpcResp.UserInfoList) == 1 {
2022-04-26 10:33:49 +08:00
resp := api.GetSelfUserInfoResp{CommResp: api.CommResp{ErrCode: RpcResp.CommonResp.ErrCode, ErrMsg: RpcResp.CommonResp.ErrMsg}, UserInfo: RpcResp.UserInfoList[0]}
resp.Data = jsonData.JsonDataOne(resp.UserInfo)
log.NewInfo(req.OperationID, "GetUserInfo api return ", resp)
c.JSON(http.StatusOK, resp)
2022-01-13 11:10:29 +08:00
} else {
resp := api.GetSelfUserInfoResp{CommResp: api.CommResp{ErrCode: constant.ErrDB.ErrCode, ErrMsg: constant.ErrDB.ErrMsg}}
2022-01-13 11:10:29 +08:00
log.NewInfo(req.OperationID, "GetUserInfo api return ", resp)
c.JSON(http.StatusOK, resp)
}
}
2022-04-18 14:27:12 +08:00
2022-06-28 10:29:14 +08:00
// @Summary 获取用户在线状态
// @Description 获取用户在线状态
// @Tags 用户相关
// @ID GetUsersOnlineStatus
// @Accept json
// @Param token header string true "im token"
// @Param req body api.GetUsersOnlineStatusReq true "请求体"
// @Produce json
// @Success 0 {object} api.GetUsersOnlineStatusResp
2022-06-29 18:48:17 +08:00
// @Failure 500 {object} api.Swagger500Resp "errCode为500 一般为服务器内部错误"
// @Failure 400 {object} api.Swagger400Resp "errCode为400 一般为参数输入错误, token未带上等"
2022-06-28 10:29:14 +08:00
// @Router /user/get_users_online_status [post]
2022-04-18 14:27:12 +08:00
func GetUsersOnlineStatus(c *gin.Context) {
params := api.GetUsersOnlineStatusReq{}
if err := c.BindJSON(&params); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"errCode": 400, "errMsg": err.Error()})
return
}
req := &pbRelay.GetUsersOnlineStatusReq{}
utils.CopyStructFields(req, &params)
2022-05-07 11:55:19 +08:00
2022-04-18 14:27:12 +08:00
var ok bool
2022-05-07 11:55:19 +08:00
var errInfo string
ok, req.OpUserID, errInfo = token_verify.GetUserIDFromToken(c.Request.Header.Get("token"), req.OperationID)
2022-04-18 14:27:12 +08:00
if !ok {
2022-05-07 11:55:19 +08:00
errMsg := req.OperationID + " " + "GetUserIDFromToken failed " + errInfo + " token:" + c.Request.Header.Get("token")
log.NewError(req.OperationID, errMsg)
2022-06-28 10:29:14 +08:00
c.JSON(http.StatusBadRequest, gin.H{"errCode": 500, "errMsg": errMsg})
2022-04-18 14:27:12 +08:00
return
}
2022-05-07 11:55:19 +08:00
2022-04-20 09:07:35 +08:00
if len(config.Config.Manager.AppManagerUid) == 0 {
log.NewError(req.OperationID, "Manager == 0")
c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": "Manager == 0"})
return
}
req.OpUserID = config.Config.Manager.AppManagerUid[0]
2022-04-18 14:27:12 +08:00
log.NewInfo(params.OperationID, "GetUsersOnlineStatus args ", req.String())
var wsResult []*pbRelay.GetUsersOnlineStatusResp_SuccessResult
var respResult []*pbRelay.GetUsersOnlineStatusResp_SuccessResult
flag := false
2022-08-17 14:57:57 +08:00
grpcCons := getcdv3.GetDefaultGatewayConn4Unique(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), params.OperationID)
2022-04-18 14:27:12 +08:00
for _, v := range grpcCons {
2022-07-27 12:04:15 +08:00
log.Debug(params.OperationID, "get node ", *v, v.Target())
2022-07-20 21:30:37 +08:00
client := pbRelay.NewRelayClient(v)
2022-04-18 14:27:12 +08:00
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.NewInfo(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)
}
resp := api.GetUsersOnlineStatusResp{CommResp: api.CommResp{ErrCode: 0, ErrMsg: ""}, SuccessResult: respResult}
if len(respResult) == 0 {
resp.SuccessResult = []*pbRelay.GetUsersOnlineStatusResp_SuccessResult{}
}
log.NewInfo(req.OperationID, "GetUsersOnlineStatus api return", resp)
c.JSON(http.StatusOK, resp)
}
2022-08-26 17:51:01 +08:00
func GetUsers(c *gin.Context) {
var (
req api.GetUsersReq
resp api.GetUsersResp
reqPb rpc.GetUsersReq
)
2022-08-30 01:38:23 +08:00
if err := c.BindJSON(&req); err != nil {
2022-08-26 17:51:01 +08:00
log.NewError(req.OperationID, "Bind failed ", err.Error(), req)
c.JSON(http.StatusBadRequest, gin.H{"errCode": 400, "errMsg": err.Error()})
return
}
var ok bool
var errInfo string
ok, _, errInfo = token_verify.GetUserIDFromToken(c.Request.Header.Get("token"), req.OperationID)
if !ok {
errMsg := req.OperationID + " " + "GetUserIDFromToken failed " + errInfo + " token:" + c.Request.Header.Get("token")
log.NewError(req.OperationID, errMsg)
c.JSON(http.StatusBadRequest, gin.H{"errCode": 500, "errMsg": errMsg})
return
}
log.NewInfo(reqPb.OperationID, utils.GetSelfFuncName(), "req: ", req)
reqPb.OperationID = req.OperationID
reqPb.UserID = req.UserID
reqPb.UserName = req.UserName
2022-09-06 20:41:46 +08:00
reqPb.Content = req.Content
2022-08-29 16:34:13 +08:00
reqPb.Pagination = &open_im_sdk.RequestPagination{ShowNumber: req.ShowNumber, PageNumber: req.PageNumber}
2022-08-26 17:51:01 +08:00
etcdConn := getcdv3.GetDefaultConn(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), config.Config.RpcRegisterName.OpenImUserName, reqPb.OperationID)
if etcdConn == nil {
errMsg := reqPb.OperationID + "getcdv3.GetDefaultConn == nil"
log.NewError(reqPb.OperationID, errMsg)
c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": errMsg})
return
}
client := rpc.NewUserClient(etcdConn)
respPb, err := client.GetUsers(context.Background(), &reqPb)
if err != nil {
log.NewError(req.OperationID, utils.GetSelfFuncName(), err.Error(), reqPb.String())
c.JSON(http.StatusBadRequest, gin.H{"errCode": 500, "errMsg": err.Error()})
return
}
2022-08-29 17:16:54 +08:00
for _, v := range respPb.UserList {
2022-09-02 11:26:25 +08:00
user := api.CMSUser{}
utils.CopyStructFields(&user, v.User)
user.IsBlock = v.IsBlock
resp.Data.UserList = append(resp.Data.UserList, &user)
2022-08-29 17:16:54 +08:00
}
2022-08-26 17:51:01 +08:00
resp.CommResp.ErrCode = respPb.CommonResp.ErrCode
resp.CommResp.ErrMsg = respPb.CommonResp.ErrMsg
resp.Data.TotalNum = respPb.TotalNums
resp.Data.CurrentPage = respPb.Pagination.CurrentPage
resp.Data.ShowNumber = respPb.Pagination.ShowNumber
log.NewInfo(req.OperationID, utils.GetSelfFuncName(), resp)
c.JSON(http.StatusOK, resp)
return
}