add cmd/open_im_push

internal/push/logic/push_rpc_server.go

@@ -0,0 +1,73 @@
+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
+
+}