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.
94 lines
2.4 KiB
94 lines
2.4 KiB
package rpcclient
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
relationTb "github.com/OpenIMSDK/Open-IM-Server/pkg/common/db/table/relation"
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/discoveryregistry"
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/errs"
|
|
sdkws "github.com/OpenIMSDK/Open-IM-Server/pkg/proto/sdkws"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
type MetaClient struct {
|
|
// contains filtered or unexported fields
|
|
client discoveryregistry.SvcDiscoveryRegistry
|
|
rpcRegisterName string
|
|
getUsersInfo func(ctx context.Context, userIDs []string) ([]CommonUser, error)
|
|
}
|
|
|
|
func NewMetaClient(client discoveryregistry.SvcDiscoveryRegistry, rpcRegisterName string, opts ...MetaClientOptions) *MetaClient {
|
|
c := &MetaClient{
|
|
client: client,
|
|
rpcRegisterName: rpcRegisterName,
|
|
}
|
|
for _, opt := range opts {
|
|
opt(c)
|
|
}
|
|
return c
|
|
}
|
|
|
|
type MetaClientOptions func(*MetaClient)
|
|
|
|
func WithDBFunc(fn func(ctx context.Context, userIDs []string) (users []*relationTb.UserModel, err error)) MetaClientOptions {
|
|
return func(s *MetaClient) {
|
|
f := func(ctx context.Context, userIDs []string) (result []CommonUser, err error) {
|
|
users, err := fn(ctx, userIDs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, user := range users {
|
|
result = append(result, user)
|
|
}
|
|
return result, nil
|
|
}
|
|
s.getUsersInfo = f
|
|
}
|
|
}
|
|
|
|
func WithRpcFunc(fn func(ctx context.Context, userIDs []string) ([]*sdkws.UserInfo, error)) MetaClientOptions {
|
|
return func(s *MetaClient) {
|
|
f := func(ctx context.Context, userIDs []string) (result []CommonUser, err error) {
|
|
users, err := fn(ctx, userIDs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, user := range users {
|
|
result = append(result, user)
|
|
}
|
|
return result, err
|
|
}
|
|
s.getUsersInfo = f
|
|
}
|
|
}
|
|
|
|
func (m *MetaClient) getFaceURLAndName(userID string) (faceURL, nickname string, err error) {
|
|
users, err := m.getUsersInfo(context.Background(), []string{userID})
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
if len(users) == 0 {
|
|
return "", "", errs.ErrRecordNotFound.Wrap(fmt.Sprintf("notification user %s not found", userID))
|
|
}
|
|
return users[0].GetFaceURL(), users[0].GetNickname(), nil
|
|
}
|
|
|
|
func (m *MetaClient) getConn(ctx context.Context) (*grpc.ClientConn, error) {
|
|
return m.client.GetConn(ctx, m.rpcRegisterName)
|
|
}
|
|
|
|
type CommonUser interface {
|
|
GetNickname() string
|
|
GetFaceURL() string
|
|
GetUserID() string
|
|
GetEx() string
|
|
}
|
|
|
|
type CommonGroup interface {
|
|
GetNickname() string
|
|
GetFaceURL() string
|
|
GetGroupID() string
|
|
GetEx() string
|
|
}
|