mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-10-31 08:29:33 +08:00
e3f609b3ae
* pb * fix: Modifying other fields while setting IsPrivateChat does not take effect * fix: quote message error revoke * refactoring scheduled tasks * refactoring scheduled tasks * refactoring scheduled tasks * refactoring scheduled tasks * refactoring scheduled tasks * refactoring scheduled tasks * rpc client * rpc client * rpc client * rpc client * rpc client * rpc client * rpc client * rpc client
96 lines
2.9 KiB
Go
96 lines
2.9 KiB
Go
package rpcli
|
|
|
|
import (
|
|
"context"
|
|
"github.com/openimsdk/protocol/sdkws"
|
|
"github.com/openimsdk/protocol/user"
|
|
"github.com/openimsdk/tools/errs"
|
|
"github.com/openimsdk/tools/utils/datautil"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
func NewUserClient(cc grpc.ClientConnInterface) *UserClient {
|
|
return &UserClient{user.NewUserClient(cc)}
|
|
}
|
|
|
|
type UserClient struct {
|
|
user.UserClient
|
|
}
|
|
|
|
func (x *UserClient) GetUsersInfo(ctx context.Context, userIDs []string) ([]*sdkws.UserInfo, error) {
|
|
if len(userIDs) == 0 {
|
|
return nil, nil
|
|
}
|
|
req := &user.GetDesignateUsersReq{UserIDs: userIDs}
|
|
return extractField(ctx, x.UserClient.GetDesignateUsers, req, (*user.GetDesignateUsersResp).GetUsersInfo)
|
|
}
|
|
|
|
func (x *UserClient) GetUserInfo(ctx context.Context, userID string) (*sdkws.UserInfo, error) {
|
|
return firstValue(x.GetUsersInfo(ctx, []string{userID}))
|
|
}
|
|
|
|
func (x *UserClient) CheckUser(ctx context.Context, userIDs []string) error {
|
|
if len(userIDs) == 0 {
|
|
return nil
|
|
}
|
|
users, err := x.GetUsersInfo(ctx, userIDs)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if len(users) != len(userIDs) {
|
|
return errs.ErrRecordNotFound.WrapMsg("user not found")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (x *UserClient) GetUsersInfoMap(ctx context.Context, userIDs []string) (map[string]*sdkws.UserInfo, error) {
|
|
users, err := x.GetUsersInfo(ctx, userIDs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return datautil.SliceToMap(users, func(e *sdkws.UserInfo) string {
|
|
return e.UserID
|
|
}), nil
|
|
}
|
|
|
|
func (x *UserClient) GetAllOnlineUsers(ctx context.Context, cursor uint64) (*user.GetAllOnlineUsersResp, error) {
|
|
req := &user.GetAllOnlineUsersReq{Cursor: cursor}
|
|
return x.UserClient.GetAllOnlineUsers(ctx, req)
|
|
}
|
|
|
|
func (x *UserClient) GetUsersOnlinePlatform(ctx context.Context, userIDs []string) ([]*user.OnlineStatus, error) {
|
|
if len(userIDs) == 0 {
|
|
return nil, nil
|
|
}
|
|
req := &user.GetUserStatusReq{UserIDs: userIDs}
|
|
return extractField(ctx, x.UserClient.GetUserStatus, req, (*user.GetUserStatusResp).GetStatusList)
|
|
|
|
}
|
|
|
|
func (x *UserClient) GetUserOnlinePlatform(ctx context.Context, userID string) ([]int32, error) {
|
|
status, err := x.GetUsersOnlinePlatform(ctx, []string{userID})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(status) == 0 {
|
|
return nil, nil
|
|
}
|
|
return status[0].PlatformIDs, nil
|
|
}
|
|
|
|
func (x *UserClient) SetUserOnlineStatus(ctx context.Context, req *user.SetUserOnlineStatusReq) error {
|
|
if len(req.Status) == 0 {
|
|
return nil
|
|
}
|
|
return ignoreResp(x.UserClient.SetUserOnlineStatus(ctx, req))
|
|
}
|
|
|
|
func (x *UserClient) GetNotificationByID(ctx context.Context, userID string) error {
|
|
return ignoreResp(x.UserClient.GetNotificationAccount(ctx, &user.GetNotificationAccountReq{UserID: userID}))
|
|
}
|
|
|
|
func (x *UserClient) GetAllUserIDs(ctx context.Context, pageNumber, showNumber int32) ([]string, error) {
|
|
req := &user.GetAllUserIDReq{Pagination: &sdkws.RequestPagination{PageNumber: pageNumber, ShowNumber: showNumber}}
|
|
return extractField(ctx, x.UserClient.GetAllUserID, req, (*user.GetAllUserIDResp).GetUserIDs)
|
|
}
|