forked from IBM/sarama
-
Notifications
You must be signed in to change notification settings - Fork 0
/
offset_fetch_request.go
71 lines (63 loc) · 1.51 KB
/
offset_fetch_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
package sarama
type OffsetFetchRequest struct {
ConsumerGroup string
IVersion int16
Partitions map[string][]int32
}
func (r *OffsetFetchRequest) Encode(pe packetEncoder) (err error) {
if r.IVersion < 0 || r.IVersion > 1 {
return PacketEncodingError{"invalid or unsupported OffsetFetchRequest version field"}
}
if err = pe.putString(r.ConsumerGroup); err != nil {
return err
}
if err = pe.putArrayLength(len(r.Partitions)); err != nil {
return err
}
for topic, partitions := range r.Partitions {
if err = pe.putString(topic); err != nil {
return err
}
if err = pe.putInt32Array(partitions); err != nil {
return err
}
}
return nil
}
func (r *OffsetFetchRequest) Decode(pd packetDecoder) (err error) {
if r.ConsumerGroup, err = pd.getString(); err != nil {
return err
}
partitionCount, err := pd.getArrayLength()
if err != nil {
return err
}
if partitionCount == 0 {
return nil
}
r.Partitions = make(map[string][]int32)
for i := 0; i < partitionCount; i++ {
topic, err := pd.getString()
if err != nil {
return err
}
partitions, err := pd.getInt32Array()
if err != nil {
return err
}
r.Partitions[topic] = partitions
}
return nil
}
func (r *OffsetFetchRequest) Key() int16 {
return 9
}
func (r *OffsetFetchRequest) Version() int16 {
return r.IVersion
}
func (r *OffsetFetchRequest) AddPartition(topic string, partitionID int32) {
if r.Partitions == nil {
r.Partitions = make(map[string][]int32)
}
r.Partitions[topic] = append(r.Partitions[topic], partitionID)
}