Browse Source

Decode also replica ID

Jaime Soriano Pastor 6 years ago
parent
commit
ab3ad49b5b
2 changed files with 7 additions and 3 deletions
  1. 6 2
      offset_request.go
  2. 1 1
      offset_request_test.go

+ 6 - 2
offset_request.go

@@ -67,10 +67,14 @@ func (r *OffsetRequest) encode(pe packetEncoder) error {
 func (r *OffsetRequest) decode(pd packetDecoder, version int16) error {
 	r.Version = version
 
-	// Ignore replica ID
-	if _, err := pd.getInt32(); err != nil {
+	replicaID, err := pd.getInt32()
+	if err != nil {
 		return err
 	}
+	if replicaID >= 0 {
+		r.SetReplicaID(replicaID)
+	}
+
 	blockCount, err := pd.getArrayLength()
 	if err != nil {
 		return err

+ 1 - 1
offset_request_test.go

@@ -49,5 +49,5 @@ func TestOffsetRequestV1(t *testing.T) {
 func TestOffsetRequestReplicaID(t *testing.T) {
 	request := new(OffsetRequest)
 	request.SetReplicaID(42)
-	testRequestEncode(t, "with replica ID", request, offsetRequestReplicaID)
+	testRequest(t, "with replica ID", request, offsetRequestReplicaID)
 }