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.
74 lines
2.2 KiB
74 lines
2.2 KiB
package logic
|
|
|
|
import (
|
|
"Open_IM/src/common/config"
|
|
"Open_IM/src/common/log"
|
|
"Open_IM/src/grpc-etcdv3/getcdv3"
|
|
"Open_IM/src/proto/push"
|
|
pbRelay "Open_IM/src/proto/relay"
|
|
"Open_IM/src/utils"
|
|
"context"
|
|
"google.golang.org/grpc"
|
|
"net"
|
|
"strings"
|
|
)
|
|
|
|
type RPCServer struct {
|
|
rpcPort int
|
|
rpcRegisterName string
|
|
etcdSchema string
|
|
etcdAddr []string
|
|
}
|
|
|
|
func (r *RPCServer) Init(rpcPort int) {
|
|
r.rpcPort = rpcPort
|
|
r.rpcRegisterName = config.Config.RpcRegisterName.OpenImPushName
|
|
r.etcdSchema = config.Config.Etcd.EtcdSchema
|
|
r.etcdAddr = config.Config.Etcd.EtcdAddr
|
|
}
|
|
func (r *RPCServer) run() {
|
|
ip := utils.ServerIP
|
|
registerAddress := ip + ":" + utils.IntToString(r.rpcPort)
|
|
listener, err := net.Listen("tcp", registerAddress)
|
|
if err != nil {
|
|
log.ErrorByKv("push module rpc listening port err", "", "err", err.Error())
|
|
return
|
|
}
|
|
defer listener.Close()
|
|
srv := grpc.NewServer()
|
|
defer srv.GracefulStop()
|
|
pbPush.RegisterPushMsgServiceServer(srv, r)
|
|
err = getcdv3.RegisterEtcd(r.etcdSchema, strings.Join(r.etcdAddr, ","), ip, r.rpcPort, r.rpcRegisterName, 10)
|
|
if err != nil {
|
|
log.ErrorByKv("register push module rpc to etcd err", "", "err", err.Error())
|
|
}
|
|
err = srv.Serve(listener)
|
|
if err != nil {
|
|
log.ErrorByKv("push module rpc start err", "", "err", err.Error())
|
|
return
|
|
}
|
|
}
|
|
func (r *RPCServer) PushMsg(_ context.Context, pbData *pbPush.PushMsgReq) (*pbPush.PushMsgResp, error) {
|
|
sendPbData := pbRelay.MsgToUserReq{}
|
|
sendPbData.SendTime = pbData.SendTime
|
|
sendPbData.OperationID = pbData.OperationID
|
|
sendPbData.ServerMsgID = pbData.MsgID
|
|
sendPbData.MsgFrom = pbData.MsgFrom
|
|
sendPbData.ContentType = pbData.ContentType
|
|
sendPbData.SenderNickName = pbData.SenderNickName
|
|
sendPbData.SenderFaceURL = pbData.SenderFaceURL
|
|
sendPbData.ClientMsgID = pbData.ClientMsgID
|
|
sendPbData.SessionType = pbData.SessionType
|
|
sendPbData.RecvID = pbData.RecvID
|
|
sendPbData.Content = pbData.Content
|
|
sendPbData.SendID = pbData.SendID
|
|
sendPbData.PlatformID = pbData.PlatformID
|
|
sendPbData.RecvSeq = pbData.RecvSeq
|
|
//Call push module to send message to the user
|
|
MsgToUser(&sendPbData, pbData.OfflineInfo, pbData.Options)
|
|
return &pbPush.PushMsgResp{
|
|
ResultCode: 0,
|
|
}, nil
|
|
|
|
}
|