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.
201 lines
7.2 KiB
201 lines
7.2 KiB
package user
|
|
|
|
import (
|
|
"Open_IM/pkg/cms_api_struct"
|
|
"Open_IM/pkg/common/config"
|
|
"Open_IM/pkg/common/constant"
|
|
openIMHttp "Open_IM/pkg/common/http"
|
|
"Open_IM/pkg/common/log"
|
|
"Open_IM/pkg/grpc-etcdv3/getcdv3"
|
|
commonPb "Open_IM/pkg/proto/sdk_ws"
|
|
pb "Open_IM/pkg/proto/user"
|
|
"Open_IM/pkg/utils"
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func AddUser(c *gin.Context) {
|
|
var (
|
|
req cms_api_struct.AddUserRequest
|
|
reqPb pb.AddUserReq
|
|
)
|
|
if err := c.BindJSON(&req); err != nil {
|
|
log.NewError(reqPb.OperationID, utils.GetSelfFuncName(), "BindJSON failed ", err.Error())
|
|
openIMHttp.RespHttp200(c, constant.ErrArgs, nil)
|
|
return
|
|
}
|
|
reqPb.OperationID = utils.OperationIDGenerator()
|
|
log.NewInfo(reqPb.OperationID, utils.GetSelfFuncName(), "req: ", req)
|
|
utils.CopyStructFields(&reqPb, &req)
|
|
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.GetConn == nil"
|
|
log.NewError(reqPb.OperationID, errMsg)
|
|
c.JSON(http.StatusInternalServerError, gin.H{"errCode": 500, "errMsg": errMsg})
|
|
return
|
|
}
|
|
client := pb.NewUserClient(etcdConn)
|
|
_, err := client.AddUser(context.Background(), &reqPb)
|
|
if err != nil {
|
|
openIMHttp.RespHttp200(c, err, nil)
|
|
return
|
|
}
|
|
openIMHttp.RespHttp200(c, constant.OK, nil)
|
|
}
|
|
|
|
func BlockUser(c *gin.Context) {
|
|
var (
|
|
req cms_api_struct.BlockUserRequest
|
|
resp cms_api_struct.BlockUserResponse
|
|
reqPb pb.BlockUserReq
|
|
)
|
|
if err := c.BindJSON(&req); err != nil {
|
|
log.NewError("0", "BindJSON failed ", err.Error())
|
|
openIMHttp.RespHttp200(c, constant.ErrArgs, resp)
|
|
return
|
|
}
|
|
reqPb.OperationID = utils.OperationIDGenerator()
|
|
log.NewInfo(reqPb.OperationID, utils.GetSelfFuncName(), "req: ", req)
|
|
utils.CopyStructFields(&reqPb, &req)
|
|
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 := pb.NewUserClient(etcdConn)
|
|
fmt.Println(reqPb)
|
|
_, err := client.BlockUser(context.Background(), &reqPb)
|
|
if err != nil {
|
|
openIMHttp.RespHttp200(c, err, resp)
|
|
return
|
|
}
|
|
openIMHttp.RespHttp200(c, constant.OK, resp)
|
|
}
|
|
|
|
func UnblockUser(c *gin.Context) {
|
|
var (
|
|
req cms_api_struct.UnblockUserRequest
|
|
resp cms_api_struct.UnBlockUserResponse
|
|
reqPb pb.UnBlockUserReq
|
|
)
|
|
if err := c.ShouldBind(&req); err != nil {
|
|
log.NewError("0", "BindJSON failed ", err.Error())
|
|
openIMHttp.RespHttp200(c, constant.ErrArgs, resp)
|
|
return
|
|
}
|
|
reqPb.OperationID = utils.OperationIDGenerator()
|
|
log.NewInfo(reqPb.OperationID, utils.GetSelfFuncName(), "req: ", req)
|
|
utils.CopyStructFields(&reqPb, &req)
|
|
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 := pb.NewUserClient(etcdConn)
|
|
_, err := client.UnBlockUser(context.Background(), &reqPb)
|
|
if err != nil {
|
|
openIMHttp.RespHttp200(c, err, resp)
|
|
return
|
|
}
|
|
log.NewInfo(reqPb.OperationID, utils.GetSelfFuncName(), "resp: ", resp)
|
|
openIMHttp.RespHttp200(c, constant.OK, resp)
|
|
}
|
|
|
|
func GetBlockUsers(c *gin.Context) {
|
|
var (
|
|
req cms_api_struct.GetBlockUsersRequest
|
|
resp cms_api_struct.GetBlockUsersResponse
|
|
reqPb pb.GetBlockUsersReq
|
|
respPb *pb.GetBlockUsersResp
|
|
)
|
|
reqPb.Pagination = &commonPb.RequestPagination{}
|
|
if err := c.ShouldBindQuery(&req); err != nil {
|
|
log.NewError(reqPb.OperationID, utils.GetSelfFuncName(), "ShouldBindQuery failed ", err.Error())
|
|
openIMHttp.RespHttp200(c, constant.ErrArgs, resp)
|
|
return
|
|
}
|
|
reqPb.OperationID = utils.OperationIDGenerator()
|
|
log.NewInfo(reqPb.OperationID, utils.GetSelfFuncName(), "req: ", req)
|
|
utils.CopyStructFields(&reqPb.Pagination, &req)
|
|
log.NewInfo(reqPb.OperationID, utils.GetSelfFuncName(), "blockUsers", reqPb.Pagination, req)
|
|
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 := pb.NewUserClient(etcdConn)
|
|
respPb, err := client.GetBlockUsers(context.Background(), &reqPb)
|
|
if err != nil {
|
|
log.NewError(reqPb.OperationID, utils.GetSelfFuncName(), "GetBlockUsers rpc", err.Error())
|
|
openIMHttp.RespHttp200(c, err, resp)
|
|
return
|
|
}
|
|
for _, v := range respPb.BlockUsers {
|
|
resp.BlockUsers = append(resp.BlockUsers, cms_api_struct.BlockUser{
|
|
UserResponse: cms_api_struct.UserResponse{
|
|
UserId: v.User.UserId,
|
|
ProfilePhoto: v.User.ProfilePhoto,
|
|
Nickname: v.User.Nickname,
|
|
IsBlock: v.User.IsBlock,
|
|
Birth: v.User.Birth,
|
|
PhoneNumber: v.User.PhoneNumber,
|
|
Email: v.User.Email,
|
|
Gender: int(v.User.Gender),
|
|
CreateTime: v.User.CreateTime,
|
|
},
|
|
BeginDisableTime: v.BeginDisableTime,
|
|
EndDisableTime: v.EndDisableTime,
|
|
})
|
|
}
|
|
resp.ShowNumber = int(respPb.Pagination.ShowNumber)
|
|
resp.CurrentPage = int(respPb.Pagination.CurrentPage)
|
|
resp.UserNums = respPb.UserNums
|
|
log.NewInfo(reqPb.OperationID, utils.GetSelfFuncName(), "req: ", resp)
|
|
openIMHttp.RespHttp200(c, constant.OK, resp)
|
|
}
|
|
|
|
func GetBlockUserById(c *gin.Context) {
|
|
var (
|
|
req cms_api_struct.GetBlockUserRequest
|
|
resp cms_api_struct.GetBlockUserResponse
|
|
reqPb pb.GetBlockUserByIdReq
|
|
)
|
|
if err := c.ShouldBindQuery(&req); err != nil {
|
|
log.NewError("0", "BindJSON failed ", err.Error())
|
|
openIMHttp.RespHttp200(c, constant.ErrArgs, nil)
|
|
return
|
|
}
|
|
reqPb.OperationID = utils.OperationIDGenerator()
|
|
log.NewInfo(reqPb.OperationID, utils.GetSelfFuncName(), "req: ", req)
|
|
reqPb.UserId = req.UserId
|
|
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 := pb.NewUserClient(etcdConn)
|
|
respPb, err := client.GetBlockUserById(context.Background(), &reqPb)
|
|
if err != nil {
|
|
log.NewError(reqPb.OperationID, "GetBlockUserById rpc failed ", err.Error())
|
|
openIMHttp.RespHttp200(c, err, nil)
|
|
return
|
|
}
|
|
resp.EndDisableTime = respPb.BlockUser.EndDisableTime
|
|
resp.BeginDisableTime = respPb.BlockUser.BeginDisableTime
|
|
utils.CopyStructFields(&resp, respPb.BlockUser.User)
|
|
log.NewInfo(reqPb.OperationID, utils.GetSelfFuncName(), "resp: ", resp)
|
|
openIMHttp.RespHttp200(c, constant.OK, resp)
|
|
}
|