5887ab5ddb
* feat(tracker): update message schema with BatchMetadata; separate message-related responsibilities; add message size * chore(docker): removing edge busybox, as the main repo updated * feat(backend): updated message protocol templates * feat(backend): added support of message size * feat(backend): implemented iterator for new message protocol (with message size) Co-authored-by: Alex Kaminskii <alex@openreplay.com>
56 lines
1.1 KiB
Go
56 lines
1.1 KiB
Go
package messages
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"fmt"
|
|
"io"
|
|
)
|
|
|
|
type SessionSearch struct {
|
|
message
|
|
Timestamp uint64
|
|
Partition uint64
|
|
}
|
|
|
|
func (msg *SessionSearch) Encode() []byte {
|
|
buf := make([]byte, 11)
|
|
buf[0] = 127
|
|
p := 1
|
|
p = WriteUint(msg.Timestamp, buf, p)
|
|
p = WriteUint(msg.Partition, buf, p)
|
|
return buf[:p]
|
|
}
|
|
|
|
func (msg *SessionSearch) EncodeWithIndex() []byte {
|
|
encoded := msg.Encode()
|
|
if IsIOSType(msg.TypeID()) {
|
|
return encoded
|
|
}
|
|
data := make([]byte, len(encoded)+8)
|
|
copy(data[8:], encoded[:])
|
|
binary.LittleEndian.PutUint64(data[0:], msg.Meta().Index)
|
|
return data
|
|
}
|
|
|
|
func (msg *SessionSearch) Decode() Message {
|
|
return msg
|
|
}
|
|
|
|
func (msg *SessionSearch) TypeID() int {
|
|
return 127
|
|
}
|
|
|
|
func DecodeExtraMessage(code string, reader io.Reader) (Message, error) {
|
|
var err error
|
|
if code != "127" {
|
|
return nil, fmt.Errorf("unknown message code: %s", code)
|
|
}
|
|
msg := &SessionSearch{}
|
|
if msg.Timestamp, err = ReadUint(reader); err != nil {
|
|
return nil, fmt.Errorf("can't read message timestamp: %s", err)
|
|
}
|
|
if msg.Partition, err = ReadUint(reader); err != nil {
|
|
return nil, fmt.Errorf("can't read last partition: %s", err)
|
|
}
|
|
return msg, nil
|
|
}
|