mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-11-04 03:13:15 +08:00
92 lines
2.3 KiB
Go
92 lines
2.3 KiB
Go
package msg
|
|
|
|
import (
|
|
"Open_IM/internal/rpc/msg/widget"
|
|
"Open_IM/pkg/common/config"
|
|
"Open_IM/pkg/common/kafka"
|
|
"Open_IM/pkg/common/log"
|
|
"Open_IM/pkg/grpc-etcdv3/getcdv3"
|
|
pbChat "Open_IM/pkg/proto/chat"
|
|
"Open_IM/pkg/utils"
|
|
"context"
|
|
"net"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/spf13/viper"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
type rpcChat struct {
|
|
rpcPort int
|
|
rpcRegisterName string
|
|
etcdSchema string
|
|
etcdAddr []string
|
|
producer *kafka.Producer
|
|
// beforeSenders are filters which will be triggered before send msg
|
|
beforeSenders []widget.BeforeSendHandler
|
|
}
|
|
|
|
func NewRpcChatServer(port int) *rpcChat {
|
|
log.NewPrivateLog("msg")
|
|
rc := rpcChat{
|
|
rpcPort: port,
|
|
rpcRegisterName: config.Config.RpcRegisterName.OpenImOfflineMessageName,
|
|
etcdSchema: config.Config.Etcd.EtcdSchema,
|
|
etcdAddr: config.Config.Etcd.EtcdAddr,
|
|
}
|
|
rc.producer = kafka.NewKafkaProducer(config.Config.Kafka.Ws2mschat.Addr, config.Config.Kafka.Ws2mschat.Topic)
|
|
return &rc
|
|
}
|
|
|
|
func (rpc *rpcChat) UseWidgetBeforSend(hs ...widget.BeforeSendHandler) {
|
|
rpc.beforeSenders = append(rpc.beforeSenders, hs...)
|
|
}
|
|
|
|
func (rpc *rpcChat) callWidgetBeforeSend(ctx context.Context, pb *pbChat.SendMsgReq) (*pbChat.SendMsgResp, bool, error) {
|
|
for _, handler := range rpc.beforeSenders {
|
|
res, ok, err := handler(ctx, pb)
|
|
if err != nil {
|
|
return nil, false, err
|
|
}
|
|
if !ok {
|
|
return res, ok, nil
|
|
}
|
|
}
|
|
|
|
return nil, true, nil
|
|
}
|
|
|
|
func (rpc *rpcChat) Run() {
|
|
log.Info("", "", "rpc get_token init...")
|
|
|
|
address := utils.ServerIP + ":" + strconv.Itoa(rpc.rpcPort)
|
|
listener, err := net.Listen("tcp", address)
|
|
if err != nil {
|
|
log.Error("", "", "listen network failed, err = %s, address = %s", err.Error(), address)
|
|
return
|
|
}
|
|
log.Info("", "", "listen network success, address = %s", address)
|
|
|
|
//grpc server
|
|
srv := grpc.NewServer()
|
|
defer srv.GracefulStop()
|
|
|
|
//service registers with etcd
|
|
|
|
pbChat.RegisterChatServer(srv, rpc)
|
|
host := viper.GetString("endpoints.rpc_msg")
|
|
err = getcdv3.RegisterEtcd(rpc.etcdSchema, strings.Join(rpc.etcdAddr, ","), host, rpc.rpcPort, rpc.rpcRegisterName, 10)
|
|
if err != nil {
|
|
log.Error("", "", "register rpc get_token to etcd failed, err = %s", err.Error())
|
|
return
|
|
}
|
|
|
|
err = srv.Serve(listener)
|
|
if err != nil {
|
|
log.Info("", "", "rpc get_token fail, err = %s", err.Error())
|
|
return
|
|
}
|
|
log.Info("", "", "rpc get_token init success")
|
|
}
|