4608911f5b
* feat(backend): removed duplicate code in ender message iterator * feat(backend): removed duplicate code in sink message iterator
30 lines
801 B
Go
30 lines
801 B
Go
package messages
|
|
|
|
type enderIteratorImpl struct {
|
|
coreIterator MessageIterator
|
|
handler MessageHandler
|
|
lastMessage Message
|
|
}
|
|
|
|
func NewEnderMessageIterator(messageHandler MessageHandler, messageFilter []int, autoDecode bool) MessageIterator {
|
|
enderIter := &enderIteratorImpl{
|
|
handler: messageHandler,
|
|
}
|
|
enderIter.coreIterator = NewMessageIterator(enderIter.handle, messageFilter, autoDecode)
|
|
return enderIter
|
|
}
|
|
|
|
func (e *enderIteratorImpl) handle(message Message) {
|
|
e.lastMessage = message
|
|
}
|
|
|
|
func (e *enderIteratorImpl) Iterate(batchData []byte, batchInfo *BatchInfo) {
|
|
// Reset last message
|
|
e.lastMessage = nil
|
|
// Call core iterator
|
|
e.coreIterator.Iterate(batchData, batchInfo)
|
|
// Call handler if last message is not nil
|
|
if e.lastMessage != nil {
|
|
e.handler(e.lastMessage)
|
|
}
|
|
}
|