90 lines
1.8 KiB
Go
90 lines
1.8 KiB
Go
|
package sarama
|
||
|
|
|
||
|
|
type MessageBlock struct {
|
||
|
|
Offset int64
|
||
|
|
Msg *Message
|
||
|
|
}
|
||
|
|
|
||
|
|
// Messages convenience helper which returns either all the
|
||
|
|
// messages that are wrapped in this block
|
||
|
|
func (msb *MessageBlock) Messages() []*MessageBlock {
|
||
|
|
if msb.Msg.Set != nil {
|
||
|
|
return msb.Msg.Set.Messages
|
||
|
|
}
|
||
|
|
return []*MessageBlock{msb}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (msb *MessageBlock) encode(pe packetEncoder) error {
|
||
|
|
pe.putInt64(msb.Offset)
|
||
|
|
pe.push(&lengthField{})
|
||
|
|
err := msb.Msg.encode(pe)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return pe.pop()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (msb *MessageBlock) decode(pd packetDecoder) (err error) {
|
||
|
|
if msb.Offset, err = pd.getInt64(); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
if err = pd.push(&lengthField{}); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
msb.Msg = new(Message)
|
||
|
|
if err = msb.Msg.decode(pd); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
if err = pd.pop(); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
type MessageSet struct {
|
||
|
|
PartialTrailingMessage bool // whether the set on the wire contained an incomplete trailing MessageBlock
|
||
|
|
Messages []*MessageBlock
|
||
|
|
}
|
||
|
|
|
||
|
|
func (ms *MessageSet) encode(pe packetEncoder) error {
|
||
|
|
for i := range ms.Messages {
|
||
|
|
err := ms.Messages[i].encode(pe)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (ms *MessageSet) decode(pd packetDecoder) (err error) {
|
||
|
|
ms.Messages = nil
|
||
|
|
|
||
|
|
for pd.remaining() > 0 {
|
||
|
|
msb := new(MessageBlock)
|
||
|
|
err = msb.decode(pd)
|
||
|
|
switch err {
|
||
|
|
case nil:
|
||
|
|
ms.Messages = append(ms.Messages, msb)
|
||
|
|
case ErrInsufficientData:
|
||
|
|
// As an optimization the server is allowed to return a partial message at the
|
||
|
|
// end of the message set. Clients should handle this case. So we just ignore such things.
|
||
|
|
ms.PartialTrailingMessage = true
|
||
|
|
return nil
|
||
|
|
default:
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (ms *MessageSet) addMessage(msg *Message) {
|
||
|
|
block := new(MessageBlock)
|
||
|
|
block.Msg = msg
|
||
|
|
ms.Messages = append(ms.Messages, block)
|
||
|
|
}
|