forked from IBM/sarama
-
Notifications
You must be signed in to change notification settings - Fork 0
/
offset_request.go
113 lines (101 loc) · 2.3 KB
/
offset_request.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
package sarama
type offsetRequestBlock struct {
Time int64
MaxOffsets int32
}
func (r *offsetRequestBlock) Encode(pe packetEncoder) error {
pe.putInt64(int64(r.Time))
pe.putInt32(r.MaxOffsets)
return nil
}
func (r *offsetRequestBlock) Decode(pd packetDecoder) (err error) {
if r.Time, err = pd.getInt64(); err != nil {
return err
}
if r.MaxOffsets, err = pd.getInt32(); err != nil {
return err
}
return nil
}
type OffsetRequest struct {
Blocks map[string]map[int32]*offsetRequestBlock
}
func (r *OffsetRequest) Encode(pe packetEncoder) error {
pe.putInt32(-1) // replica ID is always -1 for clients
err := pe.putArrayLength(len(r.Blocks))
if err != nil {
return err
}
for topic, partitions := range r.Blocks {
err = pe.putString(topic)
if err != nil {
return err
}
err = pe.putArrayLength(len(partitions))
if err != nil {
return err
}
for partition, block := range partitions {
pe.putInt32(partition)
if err = block.Encode(pe); err != nil {
return err
}
}
}
return nil
}
func (r *OffsetRequest) Decode(pd packetDecoder) error {
// Ignore replica ID
if _, err := pd.getInt32(); err != nil {
return err
}
blockCount, err := pd.getArrayLength()
if err != nil {
return err
}
if blockCount == 0 {
return nil
}
r.Blocks = make(map[string]map[int32]*offsetRequestBlock)
for i := 0; i < blockCount; i++ {
topic, err := pd.getString()
if err != nil {
return err
}
partitionCount, err := pd.getArrayLength()
if err != nil {
return err
}
r.Blocks[topic] = make(map[int32]*offsetRequestBlock)
for j := 0; j < partitionCount; j++ {
partition, err := pd.getInt32()
if err != nil {
return err
}
block := &offsetRequestBlock{}
if err := block.Decode(pd); err != nil {
return err
}
r.Blocks[topic][partition] = block
}
}
return nil
}
func (r *OffsetRequest) Key() int16 {
return 2
}
func (r *OffsetRequest) Version() int16 {
return 0
}
func (r *OffsetRequest) AddBlock(topic string, partitionID int32, time int64, maxOffsets int32) {
if r.Blocks == nil {
r.Blocks = make(map[string]map[int32]*offsetRequestBlock)
}
if r.Blocks[topic] == nil {
r.Blocks[topic] = make(map[int32]*offsetRequestBlock)
}
tmp := new(offsetRequestBlock)
tmp.Time = time
tmp.MaxOffsets = maxOffsets
r.Blocks[topic][partitionID] = tmp
}