Add a getter for replica ID

CiscoM31 · Oct 26, 2018 · 8869e90 · 8869e90
1 parent ab3ad49
commit 8869e90
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/offset_request.go b/offset_request.go
@@ -130,6 +130,13 @@ func (r *OffsetRequest) SetReplicaID(id int32) {
 	r.replicaID = &r.storeReplicaID
 }
 
+func (r *OffsetRequest) ReplicaID() int32 {
+	if r.replicaID == nil {
+		return -1
+	}
+	return r.storeReplicaID
+}
+
 func (r *OffsetRequest) AddBlock(topic string, partitionID int32, time int64, maxOffsets int32) {
 	if r.blocks == nil {
 		r.blocks = make(map[string]map[int32]*offsetRequestBlock)

diff --git a/offset_request_test.go b/offset_request_test.go
@@ -48,6 +48,12 @@ func TestOffsetRequestV1(t *testing.T) {
 
 func TestOffsetRequestReplicaID(t *testing.T) {
 	request := new(OffsetRequest)
-	request.SetReplicaID(42)
+	replicaID := int32(42)
+	request.SetReplicaID(replicaID)
+
+	if found := request.ReplicaID(); found != replicaID {
+		t.Errorf("replicaID: expected %v, found %v", replicaID, found)
+	}
+
 	testRequest(t, "with replica ID", request, offsetRequestReplicaID)
 }