package logic
import (
"Open_IM/pkg/common/config"
"Open_IM/pkg/common/constant"
"Open_IM/pkg/common/db"
kfk "Open_IM/pkg/common/kafka"
"Open_IM/pkg/common/log"
"Open_IM/pkg/grpc-etcdv3/getcdv3"
pbMsg "Open_IM/pkg/proto/chat"
pbPush "Open_IM/pkg/proto/push"
server_api_params "Open_IM/pkg/proto/sdk_ws"
"Open_IM/pkg/utils"
"context"
"errors"
"github.com/Shopify/sarama"
"github.com/golang/protobuf/proto"
"hash/crc32"
"strings"
"sync"
"time"
)
type MsgChannelValue struct {
userID string
triggerID string
msgList [ ] * pbMsg . MsgDataToMQ
lastSeq uint64
}
type TriggerChannelValue struct {
triggerID string
cmsgList [ ] * sarama . ConsumerMessage
}
type fcb func ( cMsg * sarama . ConsumerMessage , msgKey string , sess sarama . ConsumerGroupSession )
type Cmd2Value struct {
Cmd int
Value interface { }
}
type OnlineHistoryConsumerHandler struct {
msgHandle map [ string ] fcb
historyConsumerGroup * kfk . MConsumerGroup
cmdCh chan Cmd2Value
chArrays [ ChannelNum ] chan Cmd2Value
chMongoArrays [ ChannelNum ] chan Cmd2Value
msgDistributionCh chan Cmd2Value
}
func ( och * OnlineHistoryConsumerHandler ) Init ( cmdCh chan Cmd2Value ) {
och . msgHandle = make ( map [ string ] fcb )
och . msgDistributionCh = make ( chan Cmd2Value ) //no buffer channel
go och . MessagesDistributionHandle ( )
och . cmdCh = cmdCh
for i := 0 ; i < ChannelNum ; i ++ {
och . chArrays [ i ] = make ( chan Cmd2Value , 50 )
go och . Run ( i )
}
for i := 0 ; i < ChannelNum ; i ++ {
och . chMongoArrays [ i ] = make ( chan Cmd2Value , 10000 )
go och . MongoMessageRun ( i )
}
if config . Config . ReliableStorage {
och . msgHandle [ config . Config . Kafka . Ws2mschat . Topic ] = och . handleChatWs2Mongo
} else {
och . msgHandle [ config . Config . Kafka . Ws2mschat . Topic ] = och . handleChatWs2MongoLowReliability
}
och . historyConsumerGroup = kfk . NewMConsumerGroup ( & kfk . MConsumerGroupConfig { KafkaVersion : sarama . V2_0_0_0 ,
OffsetsInitial : sarama . OffsetNewest , IsReturnErr : false } , [ ] string { config . Config . Kafka . Ws2mschat . Topic } ,
config . Config . Kafka . Ws2mschat . Addr , config . Config . Kafka . ConsumerGroupID . MsgToMongo )
}
func ( och * OnlineHistoryConsumerHandler ) TriggerCmd ( status int ) {
operationID := utils . OperationIDGenerator ( )
err := sendCmd ( och . cmdCh , Cmd2Value { Cmd : status , Value : "" } , 1 )
if err != nil {
log . Error ( operationID , "TriggerCmd failed " , err . Error ( ) , status )
return
}
log . Debug ( operationID , "TriggerCmd success" , status )
}
func sendCmd ( ch chan Cmd2Value , value Cmd2Value , timeout int64 ) error {
var flag = 0
select {
case ch <- value :
flag = 1
case <- time . After ( time . Second * time . Duration ( timeout ) ) :
flag = 2
}
if flag == 1 {
return nil
} else {
return errors . New ( "send cmd timeout" )
}
}
func ( och * OnlineHistoryConsumerHandler ) Run ( channelID int ) {
for {
select {
case cmd := <- och . chArrays [ channelID ] :
switch cmd . Cmd {
case UserMessages :
msgChannelValue := cmd . Value . ( MsgChannelValue )
msgList := msgChannelValue . msgList
triggerID := msgChannelValue . triggerID
storageMsgList := make ( [ ] * pbMsg . MsgDataToMQ , 0 , 80 )
notStoragepushMsgList := make ( [ ] * pbMsg . MsgDataToMQ , 0 , 80 )
log . Debug ( triggerID , "msg arrived channel" , "channel id" , channelID , msgList , msgChannelValue . userID , len ( msgList ) )
for _ , v := range msgList {
log . Debug ( triggerID , "msg come to storage center" , v . String ( ) )
isHistory := utils . GetSwitchFromOptions ( v . MsgData . Options , constant . IsHistory )
isSenderSync := utils . GetSwitchFromOptions ( v . MsgData . Options , constant . IsSenderSync )
if isHistory {
storageMsgList = append ( storageMsgList , v )
//log.NewWarn(triggerID, "storageMsgList to mongodb client msgID: ", v.MsgData.ClientMsgID)
} else {
if ! ( ! isSenderSync && msgChannelValue . userID == v . MsgData . SendID ) {
notStoragepushMsgList = append ( notStoragepushMsgList , v )
}
}
}
//switch msgChannelValue.msg.MsgData.SessionType {
//case constant.SingleChatType:
//case constant.GroupChatType:
//case constant.NotificationChatType:
//default:
// log.NewError(msgFromMQ.OperationID, "SessionType error", msgFromMQ.String())
// return
//}
log . Debug ( triggerID , "msg storage length" , len ( storageMsgList ) , "push length" , len ( notStoragepushMsgList ) )
err , lastSeq := saveUserChatList ( msgChannelValue . userID , storageMsgList , triggerID )
if err != nil {
singleMsgFailedCount += uint64 ( len ( storageMsgList ) )
log . NewError ( triggerID , "single data insert to redis err" , err . Error ( ) , storageMsgList )
} else {
singleMsgSuccessCountMutex . Lock ( )
singleMsgSuccessCount += uint64 ( len ( storageMsgList ) )
singleMsgSuccessCountMutex . Unlock ( )
och . SendMessageToMongoCH ( msgChannelValue . userID , triggerID , storageMsgList , lastSeq )
go func ( push , storage [ ] * pbMsg . MsgDataToMQ ) {
for _ , v := range storage {
sendMessageToPush ( v , msgChannelValue . userID )
}
for _ , x := range push {
sendMessageToPush ( x , msgChannelValue . userID )
}
} ( notStoragepushMsgList , storageMsgList )
}
}
}
}
}
func ( och * OnlineHistoryConsumerHandler ) SendMessageToMongoCH ( userID string , triggerID string , messages [ ] * pbMsg . MsgDataToMQ , lastSeq uint64 ) {
hashCode := getHashCode ( userID )
channelID := hashCode % ChannelNum
log . Debug ( triggerID , "generate channelID" , hashCode , channelID , userID )
//go func(cID uint32, userID string, messages []*pbMsg.MsgDataToMQ) {
och . chMongoArrays [ channelID ] <- Cmd2Value { Cmd : MongoMessages , Value : MsgChannelValue { userID : userID , msgList : messages , triggerID : triggerID , lastSeq : lastSeq } }
}
func ( och * OnlineHistoryConsumerHandler ) MongoMessageRun ( channelID int ) {
for {
select {
case cmd := <- och . chMongoArrays [ channelID ] :
switch cmd . Cmd {
case MongoMessages :
msgChannelValue := cmd . Value . ( MsgChannelValue )
msgList := msgChannelValue . msgList
triggerID := msgChannelValue . triggerID
userID := msgChannelValue . userID
lastSeq := msgChannelValue . lastSeq
err := db . DB . BatchInsertChat2DB ( userID , msgList , triggerID , lastSeq )
if err != nil {
log . NewError ( triggerID , "single data insert to mongo err" , err . Error ( ) , msgList )
}
for _ , v := range msgList {
if v . MsgData . ContentType == constant . DeleteMessageNotification {
tips := server_api_params . TipsComm { }
DeleteMessageTips := server_api_params . DeleteMessageTips { }
err := proto . Unmarshal ( v . MsgData . Content , & tips )
if err != nil {
log . NewError ( triggerID , "tips unmarshal err:" , err . Error ( ) , v . String ( ) )
continue
}
err = proto . Unmarshal ( tips . Detail , & DeleteMessageTips )
if err != nil {
log . NewError ( triggerID , "deleteMessageTips unmarshal err:" , err . Error ( ) , v . String ( ) )
continue
}
if unexistSeqList , err := db . DB . DelMsgBySeqList ( DeleteMessageTips . UserID , DeleteMessageTips . SeqList , v . OperationID ) ; err != nil {
log . NewError ( v . OperationID , utils . GetSelfFuncName ( ) , "DelMsgBySeqList args: " , DeleteMessageTips . UserID , DeleteMessageTips . SeqList , v . OperationID , err . Error ( ) , unexistSeqList )
}
}
}
}
}
}
}
func ( och * OnlineHistoryConsumerHandler ) MessagesDistributionHandle ( ) {
for {
UserAggregationMsgs := make ( map [ string ] [ ] * pbMsg . MsgDataToMQ , ChannelNum )
select {
case cmd := <- och . msgDistributionCh :
switch cmd . Cmd {
case ConsumerMsgs :
triggerChannelValue := cmd . Value . ( TriggerChannelValue )
triggerID := triggerChannelValue . triggerID
consumerMessages := triggerChannelValue . cmsgList
//Aggregation map[userid]message list
log . Debug ( triggerID , "batch messages come to distribution center" , len ( consumerMessages ) )
for i := 0 ; i < len ( consumerMessages ) ; i ++ {
msgFromMQ := pbMsg . MsgDataToMQ { }
err := proto . Unmarshal ( consumerMessages [ i ] . Value , & msgFromMQ )
if err != nil {
log . Error ( triggerID , "msg_transfer Unmarshal msg err" , "msg" , string ( consumerMessages [ i ] . Value ) , "err" , err . Error ( ) )
return
}
log . Debug ( triggerID , "single msg come to distribution center" , msgFromMQ . String ( ) , string ( consumerMessages [ i ] . Key ) )
if oldM , ok := UserAggregationMsgs [ string ( consumerMessages [ i ] . Key ) ] ; ok {
oldM = append ( oldM , & msgFromMQ )
UserAggregationMsgs [ string ( consumerMessages [ i ] . Key ) ] = oldM
} else {
m := make ( [ ] * pbMsg . MsgDataToMQ , 0 , 100 )
m = append ( m , & msgFromMQ )
UserAggregationMsgs [ string ( consumerMessages [ i ] . Key ) ] = m
}
}
log . Debug ( triggerID , "generate map list users len" , len ( UserAggregationMsgs ) )
for userID , v := range UserAggregationMsgs {
if len ( v ) >= 0 {
hashCode := getHashCode ( userID )
channelID := hashCode % ChannelNum
log . Debug ( triggerID , "generate channelID" , hashCode , channelID , userID )
//go func(cID uint32, userID string, messages []*pbMsg.MsgDataToMQ) {
och . chArrays [ channelID ] <- Cmd2Value { Cmd : UserMessages , Value : MsgChannelValue { userID : userID , msgList : v , triggerID : triggerID } }
//}(channelID, userID, v)
}
}
}
}
}
}
func ( mc * OnlineHistoryConsumerHandler ) handleChatWs2Mongo ( cMsg * sarama . ConsumerMessage , msgKey string , sess sarama . ConsumerGroupSession ) {
msg := cMsg . Value
now := time . Now ( )
msgFromMQ := pbMsg . MsgDataToMQ { }
err := proto . Unmarshal ( msg , & msgFromMQ )
if err != nil {
log . Error ( "msg_transfer Unmarshal msg err" , "" , "msg" , string ( msg ) , "err" , err . Error ( ) )
return
}
operationID := msgFromMQ . OperationID
log . NewInfo ( operationID , "msg come mongo!!!" , "" , "msg" , string ( msg ) )
//Control whether to store offline messages (mongo)
isHistory := utils . GetSwitchFromOptions ( msgFromMQ . MsgData . Options , constant . IsHistory )
//Control whether to store history messages (mysql)
isPersist := utils . GetSwitchFromOptions ( msgFromMQ . MsgData . Options , constant . IsPersistent )
isSenderSync := utils . GetSwitchFromOptions ( msgFromMQ . MsgData . Options , constant . IsSenderSync )
switch msgFromMQ . MsgData . SessionType {
case constant . SingleChatType :
log . NewDebug ( msgFromMQ . OperationID , "msg_transfer msg type = SingleChatType" , isHistory , isPersist )
if isHistory {
err := saveUserChat ( msgKey , & msgFromMQ )
if err != nil {
singleMsgFailedCount ++
log . NewError ( operationID , "single data insert to mongo err" , err . Error ( ) , msgFromMQ . String ( ) )
return
}
singleMsgSuccessCountMutex . Lock ( )
singleMsgSuccessCount ++
singleMsgSuccessCountMutex . Unlock ( )
log . NewDebug ( msgFromMQ . OperationID , "sendMessageToPush cost time " , time . Since ( now ) )
}
if ! isSenderSync && msgKey == msgFromMQ . MsgData . SendID {
} else {
go sendMessageToPush ( & msgFromMQ , msgKey )
}
log . NewDebug ( operationID , "saveSingleMsg cost time " , time . Since ( now ) )
case constant . GroupChatType :
log . NewDebug ( msgFromMQ . OperationID , "msg_transfer msg type = GroupChatType" , isHistory , isPersist )
if isHistory {
err := saveUserChat ( msgFromMQ . MsgData . RecvID , & msgFromMQ )
if err != nil {
log . NewError ( operationID , "group data insert to mongo err" , msgFromMQ . String ( ) , msgFromMQ . MsgData . RecvID , err . Error ( ) )
return
}
groupMsgCount ++
}
go sendMessageToPush ( & msgFromMQ , msgFromMQ . MsgData . RecvID )
log . NewDebug ( operationID , "saveGroupMsg cost time " , time . Since ( now ) )
case constant . NotificationChatType :
log . NewDebug ( msgFromMQ . OperationID , "msg_transfer msg type = NotificationChatType" , isHistory , isPersist )
if isHistory {
err := saveUserChat ( msgKey , & msgFromMQ )
if err != nil {
log . NewError ( operationID , "single data insert to mongo err" , err . Error ( ) , msgFromMQ . String ( ) )
return
}
log . NewDebug ( msgFromMQ . OperationID , "sendMessageToPush cost time " , time . Since ( now ) )
}
if ! isSenderSync && msgKey == msgFromMQ . MsgData . SendID {
} else {
go sendMessageToPush ( & msgFromMQ , msgKey )
}
log . NewDebug ( operationID , "saveUserChat cost time " , time . Since ( now ) )
default :
log . NewError ( msgFromMQ . OperationID , "SessionType error" , msgFromMQ . String ( ) )
return
}
sess . MarkMessage ( cMsg , "" )
log . NewDebug ( msgFromMQ . OperationID , "msg_transfer handle topic data to database success..." , msgFromMQ . String ( ) )
}
func ( och * OnlineHistoryConsumerHandler ) handleChatWs2MongoLowReliability ( cMsg * sarama . ConsumerMessage , msgKey string , sess sarama . ConsumerGroupSession ) {
msg := cMsg . Value
msgFromMQ := pbMsg . MsgDataToMQ { }
err := proto . Unmarshal ( msg , & msgFromMQ )
if err != nil {
log . Error ( "msg_transfer Unmarshal msg err" , "" , "msg" , string ( msg ) , "err" , err . Error ( ) )
return
}
operationID := msgFromMQ . OperationID
log . NewInfo ( operationID , "msg come mongo!!!" , "" , "msg" , string ( msg ) )
//Control whether to store offline messages (mongo)
isHistory := utils . GetSwitchFromOptions ( msgFromMQ . MsgData . Options , constant . IsHistory )
isSenderSync := utils . GetSwitchFromOptions ( msgFromMQ . MsgData . Options , constant . IsSenderSync )
if isHistory {
seq , err := db . DB . IncrUserSeq ( msgKey )
if err != nil {
log . NewError ( operationID , "data insert to redis err" , err . Error ( ) , string ( msg ) )
return
}
sess . MarkMessage ( cMsg , "" )
msgFromMQ . MsgData . Seq = uint32 ( seq )
log . Debug ( operationID , "send ch msg is " , msgFromMQ . String ( ) )
//och.msgCh <- Cmd2Value{Cmd: Msg, Value: MsgChannelValue{msgKey, msgFromMQ}}
//err := saveUserChat(msgKey, &msgFromMQ)
//if err != nil {
// singleMsgFailedCount++
// log.NewError(operationID, "single data insert to mongo err", err.Error(), msgFromMQ.String())
// return
//}
//singleMsgSuccessCountMutex.Lock()
//singleMsgSuccessCount++
//singleMsgSuccessCountMutex.Unlock()
//log.NewDebug(msgFromMQ.OperationID, "sendMessageToPush cost time ", time.Since(now))
} else {
if ! ( ! isSenderSync && msgKey == msgFromMQ . MsgData . SendID ) {
go sendMessageToPush ( & msgFromMQ , msgKey )
}
}
}
func ( OnlineHistoryConsumerHandler ) Setup ( _ sarama . ConsumerGroupSession ) error { return nil }
func ( OnlineHistoryConsumerHandler ) Cleanup ( _ sarama . ConsumerGroupSession ) error { return nil }
//func (och *OnlineHistoryConsumerHandler) ConsumeClaim(sess sarama.ConsumerGroupSession,
// claim sarama.ConsumerGroupClaim) error { // a instance in the consumer group
// log.NewDebug("", "online new session msg come", claim.HighWaterMarkOffset(), claim.Topic(), claim.Partition())
// for msg := range claim.Messages() {
// SetOnlineTopicStatus(OnlineTopicBusy)
// //och.TriggerCmd(OnlineTopicBusy)
// log.NewDebug("", "online kafka get info to mongo", "msgTopic", msg.Topic, "msgPartition", msg.Partition, "online", msg.Offset, claim.HighWaterMarkOffset())
// och.msgHandle[msg.Topic](msg, string(msg.Key), sess)
// if claim.HighWaterMarkOffset()-msg.Offset <= 1 {
// log.Debug("", "online msg consume end", claim.HighWaterMarkOffset(), msg.Offset)
// SetOnlineTopicStatus(OnlineTopicVacancy)
// och.TriggerCmd(OnlineTopicVacancy)
// }
// }
// return nil
//}
func ( och * OnlineHistoryConsumerHandler ) ConsumeClaim ( sess sarama . ConsumerGroupSession ,
claim sarama . ConsumerGroupClaim ) error { // a instance in the consumer group
for {
if sess == nil {
log . NewWarn ( "" , " sess == nil, waiting " )
time . Sleep ( 100 * time . Millisecond )
} else {
break
}
}
rwLock := new ( sync . RWMutex )
log . NewDebug ( "" , "online new session msg come" , claim . HighWaterMarkOffset ( ) , claim . Topic ( ) , claim . Partition ( ) )
cMsg := make ( [ ] * sarama . ConsumerMessage , 0 , 1000 )
t := time . NewTicker ( time . Duration ( 100 ) * time . Millisecond )
var triggerID string
go func ( ) {
for {
select {
//case :
// triggerID = utils.OperationIDGenerator()
//
// log.NewDebug(triggerID, "claim.Messages ", msg)
// cMsg = append(cMsg, msg)
// if len(cMsg) >= 1000 {
// ccMsg := make([]*sarama.ConsumerMessage, 0, 1000)
// for _, v := range cMsg {
// ccMsg = append(ccMsg, v)
// }
// log.Debug(triggerID, "length trigger msg consumer start", len(ccMsg))
// och.msgDistributionCh <- Cmd2Value{Cmd: ConsumerMsgs, Value: TriggerChannelValue{
// triggerID: triggerID, cmsgList: ccMsg}}
// sess.MarkMessage(msg, "")
// cMsg = make([]*sarama.ConsumerMessage, 0, 1000)
// log.Debug(triggerID, "length trigger msg consumer end", len(cMsg))
// }
case <- t . C :
if len ( cMsg ) > 0 {
rwLock . Lock ( )
ccMsg := make ( [ ] * sarama . ConsumerMessage , 0 , 1000 )
for _ , v := range cMsg {
ccMsg = append ( ccMsg , v )
}
cMsg = make ( [ ] * sarama . ConsumerMessage , 0 , 1000 )
rwLock . Unlock ( )
split := 1000
triggerID = utils . OperationIDGenerator ( )
log . NewWarn ( triggerID , "timer trigger msg consumer start" , len ( ccMsg ) )
for i := 0 ; i < len ( ccMsg ) / split ; i ++ {
//log.Debug()
och . msgDistributionCh <- Cmd2Value { Cmd : ConsumerMsgs , Value : TriggerChannelValue {
triggerID : triggerID , cmsgList : ccMsg [ i * split : ( i + 1 ) * split ] } }
}
if ( len ( ccMsg ) % split ) > 0 {
och . msgDistributionCh <- Cmd2Value { Cmd : ConsumerMsgs , Value : TriggerChannelValue {
triggerID : triggerID , cmsgList : ccMsg [ split * ( len ( ccMsg ) / split ) : ] } }
}
//sess.MarkMessage(ccMsg[len(cMsg)-1], "")
log . NewWarn ( triggerID , "timer trigger msg consumer end" , len ( cMsg ) )
}
}
}
} ( )
for msg := range claim . Messages ( ) {
//msgFromMQ := pbMsg.MsgDataToMQ{}
//err := proto.Unmarshal(msg.Value, &msgFromMQ)
//if err != nil {
// log.Error(triggerID, "msg_transfer Unmarshal msg err", "msg", string(msg.Value), "err", err.Error())
//}
//userID := string(msg.Key)
//hashCode := getHashCode(userID)
//channelID := hashCode % ChannelNum
//log.Debug(triggerID, "generate channelID", hashCode, channelID, userID)
////go func(cID uint32, userID string, messages []*pbMsg.MsgDataToMQ) {
//och.chArrays[channelID] <- Cmd2Value{Cmd: UserMessages, Value: MsgChannelValue{userID: userID, msgList: []*pbMsg.MsgDataToMQ{&msgFromMQ}, triggerID: msgFromMQ.OperationID}}
//sess.MarkMessage(msg, "")
rwLock . Lock ( )
cMsg = append ( cMsg , msg )
rwLock . Unlock ( )
sess . MarkMessage ( msg , "" )
//och.TriggerCmd(OnlineTopicBusy)
//log.NewDebug("", "online kafka get info to mongo", "msgTopic", msg.Topic, "msgPartition", msg.Partition, "online", msg.Offset, claim.HighWaterMarkOffset())
}
return nil
}
//func (och *OnlineHistoryConsumerHandler) ConsumeClaim(sess sarama.ConsumerGroupSession,
// claim sarama.ConsumerGroupClaim) error { // a instance in the consumer group
//
// for {
// if sess == nil {
// log.NewWarn("", " sess == nil, waiting ")
// time.Sleep(100 * time.Millisecond)
// } else {
// break
// }
// }
//
// log.NewDebug("", "online new session msg come", claim.HighWaterMarkOffset(), claim.Topic(), claim.Partition())
// cMsg := make([]*sarama.ConsumerMessage, 0, 1000)
// t := time.NewTicker(time.Duration(100) * time.Millisecond)
// var triggerID string
// for msg := range claim.Messages() {
// cMsg = append(cMsg, msg)
// //och.TriggerCmd(OnlineTopicBusy)
// select {
// //case :
// // triggerID = utils.OperationIDGenerator()
// //
// // log.NewDebug(triggerID, "claim.Messages ", msg)
// // cMsg = append(cMsg, msg)
// // if len(cMsg) >= 1000 {
// // ccMsg := make([]*sarama.ConsumerMessage, 0, 1000)
// // for _, v := range cMsg {
// // ccMsg = append(ccMsg, v)
// // }
// // log.Debug(triggerID, "length trigger msg consumer start", len(ccMsg))
// // och.msgDistributionCh <- Cmd2Value{Cmd: ConsumerMsgs, Value: TriggerChannelValue{
// // triggerID: triggerID, cmsgList: ccMsg}}
// // sess.MarkMessage(msg, "")
// // cMsg = make([]*sarama.ConsumerMessage, 0, 1000)
// // log.Debug(triggerID, "length trigger msg consumer end", len(cMsg))
// // }
//
// case <-t.C:
// if len(cMsg) > 0 {
// ccMsg := make([]*sarama.ConsumerMessage, 0, 1000)
// for _, v := range cMsg {
// ccMsg = append(ccMsg, v)
// }
// triggerID = utils.OperationIDGenerator()
// log.Debug(triggerID, "timer trigger msg consumer start", len(ccMsg))
// och.msgDistributionCh <- Cmd2Value{Cmd: ConsumerMsgs, Value: TriggerChannelValue{
// triggerID: triggerID, cmsgList: ccMsg}}
// sess.MarkMessage(cMsg[len(cMsg)-1], "")
// cMsg = make([]*sarama.ConsumerMessage, 0, 1000)
// log.Debug(triggerID, "timer trigger msg consumer end", len(cMsg))
// }
// default:
//
// }
// //log.NewDebug("", "online kafka get info to mongo", "msgTopic", msg.Topic, "msgPartition", msg.Partition, "online", msg.Offset, claim.HighWaterMarkOffset())
//
// }
// return nil
//}
func sendMessageToPush ( message * pbMsg . MsgDataToMQ , pushToUserID string ) {
log . Info ( message . OperationID , "msg_transfer send message to push" , "message" , message . String ( ) )
rpcPushMsg := pbPush . PushMsgReq { OperationID : message . OperationID , MsgData : message . MsgData , PushToUserID : pushToUserID }
mqPushMsg := pbMsg . PushMsgDataToMQ { OperationID : message . OperationID , MsgData : message . MsgData , PushToUserID : pushToUserID }
grpcConn := getcdv3 . GetConn ( config . Config . Etcd . EtcdSchema , strings . Join ( config . Config . Etcd . EtcdAddr , "," ) , config . Config . RpcRegisterName . OpenImPushName )
if grpcConn == nil {
log . Error ( rpcPushMsg . OperationID , "rpc dial failed" , "push data" , rpcPushMsg . String ( ) )
pid , offset , err := producer . SendMessage ( & mqPushMsg )
if err != nil {
log . Error ( mqPushMsg . OperationID , "kafka send failed" , "send data" , message . String ( ) , "pid" , pid , "offset" , offset , "err" , err . Error ( ) )
}
return
}
msgClient := pbPush . NewPushMsgServiceClient ( grpcConn )
_ , err := msgClient . PushMsg ( context . Background ( ) , & rpcPushMsg )
if err != nil {
log . Error ( rpcPushMsg . OperationID , "rpc send failed" , rpcPushMsg . OperationID , "push data" , rpcPushMsg . String ( ) , "err" , err . Error ( ) )
pid , offset , err := producer . SendMessage ( & mqPushMsg )
if err != nil {
log . Error ( message . OperationID , "kafka send failed" , mqPushMsg . OperationID , "send data" , mqPushMsg . String ( ) , "pid" , pid , "offset" , offset , "err" , err . Error ( ) )
}
} else {
log . Info ( message . OperationID , "rpc send success" , rpcPushMsg . OperationID , "push data" , rpcPushMsg . String ( ) )
}
}
// String hashes a string to a unique hashcode.
//
// crc32 returns a uint32, but for our use we need
// and non negative integer. Here we cast to an integer
// and invert it if the result is negative.
func getHashCode ( s string ) uint32 {
return crc32 . ChecksumIEEE ( [ ] byte ( s ) )
}