|
|
|
@ -121,9 +121,10 @@ func newSessionTypeConf() map[int32]int32 {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
type Message struct {
|
|
|
|
|
conn grpc.ClientConnInterface
|
|
|
|
|
Client msg.MsgClient
|
|
|
|
|
discov discoveryregistry.SvcDiscoveryRegistry
|
|
|
|
|
conn grpc.ClientConnInterface
|
|
|
|
|
Client msg.MsgClient
|
|
|
|
|
discov discoveryregistry.SvcDiscoveryRegistry
|
|
|
|
|
userClient user.UserClient
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func NewMessage(discov discoveryregistry.SvcDiscoveryRegistry) *Message {
|
|
|
|
@ -132,7 +133,20 @@ func NewMessage(discov discoveryregistry.SvcDiscoveryRegistry) *Message {
|
|
|
|
|
panic(err)
|
|
|
|
|
}
|
|
|
|
|
client := msg.NewMsgClient(conn)
|
|
|
|
|
return &Message{discov: discov, conn: conn, Client: client}
|
|
|
|
|
conn, err = discov.GetConn(context.Background(), config.Config.RpcRegisterName.OpenImUserName)
|
|
|
|
|
if err != nil {
|
|
|
|
|
panic(err)
|
|
|
|
|
}
|
|
|
|
|
userClient := user.NewUserClient(conn)
|
|
|
|
|
return &Message{discov: discov, conn: conn, Client: client, userClient: userClient}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (m *Message) GetAllUserID(ctx context.Context, req *user.GetAllUserIDReq) (*user.GetAllUserIDResp, error) {
|
|
|
|
|
resp, err := m.userClient.GetAllUserID(ctx, req)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
return resp, nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
type MessageRpcClient Message
|
|
|
|
@ -265,16 +279,3 @@ func (s *NotificationSender) NotificationWithSesstionType(ctx context.Context, s
|
|
|
|
|
func (s *NotificationSender) Notification(ctx context.Context, sendID, recvID string, contentType int32, m proto.Message, opts ...NotificationOptions) error {
|
|
|
|
|
return s.NotificationWithSesstionType(ctx, sendID, recvID, contentType, s.sessionTypeConf[contentType], m, opts...)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (m *Message) GetAllUserID(ctx context.Context, req *user.GetAllUserIDReq) (*user.GetAllUserIDResp, error) {
|
|
|
|
|
conn, err := m.discov.GetConn(context.Background(), config.Config.RpcRegisterName.OpenImMsgName)
|
|
|
|
|
if err != nil {
|
|
|
|
|
panic(err)
|
|
|
|
|
}
|
|
|
|
|
client := user.NewUserClient(conn)
|
|
|
|
|
resp, err := client.GetAllUserID(ctx, req)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
return resp, nil
|
|
|
|
|
}
|
|
|
|
|