|
|
|
@ -108,9 +108,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 {
|
|
|
|
@ -119,7 +120,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
|
|
|
|
@ -284,16 +298,3 @@ func (s *NotificationSender) Notification(
|
|
|
|
|
) 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
|
|
|
|
|
}
|
|
|
|
|