msg_gateway and msg_transfer modules

2026-05-21 09:19:01 +08:00 · 2021-05-26 19:24:25 +08:00
parent a7bedf1f79
commit f686fbb4ee
13 changed files with 948 additions and 0 deletions
@@ -0,0 +1,155 @@
+package gate
+
+import (
+	"Open_IM/src/common/config"
+	"Open_IM/src/common/constant"
+	"Open_IM/src/common/log"
+	pbRelay "Open_IM/src/proto/relay"
+	"Open_IM/src/utils"
+	"context"
+	"encoding/json"
+	"fmt"
+	"github.com/gorilla/websocket"
+	"github.com/skiffer-git/grpc-etcdv3/getcdv3"
+	"google.golang.org/grpc"
+	"net"
+	"strings"
+)
+
+type RPCServer struct {
+	rpcPort         int
+	rpcRegisterName string
+	etcdSchema      string
+	etcdAddr        []string
+}
+
+func (r *RPCServer) onInit(rpcPort int) {
+	r.rpcPort = rpcPort
+	r.rpcRegisterName = config.Config.RpcRegisterName.OpenImOnlineMessageRelayName
+	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.ErrorByArgs(fmt.Sprintf("fail to listening consumer, err:%v\n", err))
+		return
+	}
+	defer listener.Close()
+	srv := grpc.NewServer()
+	defer srv.GracefulStop()
+	pbRelay.RegisterOnlineMessageRelayServiceServer(srv, r)
+	err = getcdv3.RegisterEtcd4Unique(r.etcdSchema, strings.Join(r.etcdAddr, ","), ip, r.rpcPort, r.rpcRegisterName, 10)
+	if err != nil {
+		log.ErrorByKv("register push message rpc to etcd err", "", "err", err.Error())
+	}
+	err = srv.Serve(listener)
+	if err != nil {
+		log.ErrorByKv("push message rpc listening err", "", "err", err.Error())
+		return
+	}
+}
+func (r *RPCServer) MsgToUser(_ context.Context, in *pbRelay.MsgToUserReq) (*pbRelay.MsgToUserResp, error) {
+	log.InfoByKv("PushMsgToUser is arriving", in.OperationID, "args", in.String())
+	var resp []*pbRelay.SingleMsgToUser
+	var RecvID string
+	msg := make(map[string]interface{})
+	mReply := make(map[string]interface{})
+	mReply["reqIdentifier"] = constant.WSPushMsg
+	mReply["errCode"] = 0
+	mReply["errMsg"] = ""
+	msg["sendID"] = in.SendID
+	msg["recvID"] = in.RecvID
+	msg["msgFrom"] = in.MsgFrom
+	msg["contentType"] = in.ContentType
+	msg["sessionType"] = in.SessionType
+	msg["serverMsgID"] = in.ServerMsgID
+	msg["content"] = in.Content
+	msg["seq"] = in.RecvSeq
+	msg["sendTime"] = in.SendTime
+	msg["isEmphasize"] = in.IsEmphasize
+	msg["senderPlatformID"] = in.PlatformID
+	mReply["data"] = msg
+	bMsg, _ := json.Marshal(mReply)
+	switch in.GetContentType() {
+	case constant.SyncSenderMsg:
+		log.InfoByKv("come sync", in.OperationID, "args", in.String())
+		RecvID = in.GetSendID()
+		for key, conn := range ws.wsUserToConn {
+			UIDAndPID := strings.Split(key, " ")
+			if UIDAndPID[0] == RecvID && utils.PlatformIDToName(in.GetPlatformID()) != UIDAndPID[1] {
+				resultCode := sendMsgToUser(conn, bMsg, in, UIDAndPID[1], UIDAndPID[0])
+				temp := &pbRelay.SingleMsgToUser{
+					ResultCode:     resultCode,
+					RecvID:         UIDAndPID[0],
+					RecvPlatFormID: utils.PlatformNameToID(UIDAndPID[1]),
+				}
+				resp = append(resp, temp)
+			}
+
+		}
+	default:
+		log.InfoByKv("not come sync", in.OperationID, "args", in.String())
+		switch in.SessionType {
+		case constant.SingleChatType:
+			log.InfoByKv("come single", in.OperationID, "args", in.String())
+			RecvID = in.GetRecvID()
+		case constant.GroupChatType:
+			RecvID = strings.Split(in.GetRecvID(), " ")[0]
+		default:
+		}
+		log.InfoByKv("come for range", in.OperationID, "args", in.String())
+
+		for key, conn := range ws.wsUserToConn {
+			UIDAndPID := strings.Split(key, " ")
+			if UIDAndPID[0] == RecvID {
+				resultCode := sendMsgToUser(conn, bMsg, in, UIDAndPID[1], UIDAndPID[0])
+				temp := &pbRelay.SingleMsgToUser{
+					ResultCode:     resultCode,
+					RecvID:         UIDAndPID[0],
+					RecvPlatFormID: utils.PlatformNameToID(UIDAndPID[1]),
+				}
+				resp = append(resp, temp)
+			}
+		}
+	}
+	return &pbRelay.MsgToUserResp{
+		Resp: resp,
+	}, nil
+}
+
+//func (r *RPCServer) SendMsgByWS(_ context.Context, in *pbRelay.SendMsgByWSReq) (*pbRelay.MsgToUserResp, error) {
+//	log.InfoByKv("SendMsgByWS is arriving ", in.OperationID, "args", in.String())
+//	resp := new(pbRelay.MsgToUserResp)
+//	MsgId := ws.genMsgNum()
+//	pbData := pbMsg.WSToMsgSvrChatMsg{}
+//	pbData.SendID = in.SendID
+//	pbData.RecvID = in.RecvID
+//	pbData.MsgID = MsgId
+//	pbData.SessionType = in.SessionType
+//	pbData.MsgFrom = in.MsgFrom
+//	pbData.Content = in.Content
+//	pbData.ContentType = in.ContentType
+//	pbData.OperationID = in.OperationID
+//	pbData.SendTime = in.SendTime
+//	pbData.PlatformID = in.PlatformID
+//	pKafka.writeMsg(&pbData)
+//	return resp, nil
+//}
+
+func sendMsgToUser(conn *websocket.Conn, bMsg []byte, in *pbRelay.MsgToUserReq, RecvPlatForm, RecvID string) (ResultCode int64) {
+	err := ws.writeMsg(conn, websocket.TextMessage, bMsg)
+	if err != nil {
+		log.ErrorByKv("PushMsgToUser is failed By Ws", "", "Addr", conn.RemoteAddr().String(),
+			"error", err, "senderPlatform", utils.PlatformIDToName(in.PlatformID), "recvPlatform", RecvPlatForm, "args", in.String(), "recvID", RecvID)
+		ResultCode = -2
+		return ResultCode
+	} else {
+		log.InfoByKv("PushMsgToUser is success By Ws", in.OperationID, "args", in.String())
+		ResultCode = 0
+		return ResultCode
+	}
+
+}