zhangdeman/event
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

redis事件多分区消费

Browse Source
This commit is contained in:
白茶清欢 2024-06-27 12:00:15 +08:00
parent 414ad47e72
commit 7fe6003db8
1 changed files with 20 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
redis_pub_sub.go
View File

@ -152,27 +152,28 @@ func (r *RedisEventPubSub) ConsumeEvent(handler abstract.EventHandler, successCa
if nil == successCallback { if nil == successCallback {
successCallback = define.DefaultSuccessCallbackHandler successCallback = define.DefaultSuccessCallbackHandler
} }
go func() { for partition := 0; partition < r.pubSubConfig.PartitionNum; partition++ {
defer func() { go func(realPartition int) {
if panicErr := recover(); nil != panicErr { defer func() {
fmt.Println(r) if panicErr := recover(); nil != panicErr {
} r.panicCallback(panicErr, nil, nil)
}() }
for !r.isStop || (r.isStop && len(messageChan) == 0) { }()
select { for !r.isStop || (r.isStop && len(messageChan) == 0) {
case eventData := <-messageChan: select {
handlerResult, handlerErr := handler(eventData) case eventData := <-messageChan:
if nil != handlerErr { handlerResult, handlerErr := handler(eventData)
// 失败回调 if nil != handlerErr {
failureCallback(eventData, handlerResult, handlerErr) // 失败回调
break failureCallback(eventData, handlerResult, handlerErr)
} else { } else {
// 成功回调 // 成功回调
successCallback(eventData, handlerResult) successCallback(eventData, handlerResult)
}
} }
} }
} }(partition)
}() }
return nil return nil
} }