refactor: update framework methods to use new response format

codecrafters-io · Oct 28, 2024 · baa4c1c · baa4c1c
1 parent 61cbb95
commit baa4c1c
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/protocol/api/api_versions.go b/protocol/api/api_versions.go
@@ -108,7 +108,7 @@ func ApiVersions(b *protocol.Broker, requestBody *ApiVersionsRequestBody) (*ApiV
 		return nil, err
 	}
 
-	_, apiVersionsResponse, err := DecodeApiVersionsHeaderAndResponse(response, requestBody.Version, logger.GetLogger(true, ""))
+	_, apiVersionsResponse, err := DecodeApiVersionsHeaderAndResponse(response.Payload, requestBody.Version, logger.GetLogger(true, ""))
 	if err != nil {
 		return nil, err
 	}

diff --git a/protocol/api/describe_topic_partitions.go b/protocol/api/describe_topic_partitions.go
@@ -109,7 +109,7 @@ func DescribeTopicPartitions(b *protocol.Broker) (*DescribeTopicPartitionsRespon
 		return nil, err
 	}
 
-	_, DescribeTopicPartitionsResponse, err := DecodeDescribeTopicPartitionsHeaderAndResponse(response, logger.GetLogger(true, ""))
+	_, DescribeTopicPartitionsResponse, err := DecodeDescribeTopicPartitionsHeaderAndResponse(response.Payload, logger.GetLogger(true, ""))
 	if err != nil {
 		return nil, err
 	}

diff --git a/protocol/api/fetch.go b/protocol/api/fetch.go
@@ -2,6 +2,7 @@ package kafkaapi
 
 import (
 	"fmt"
+
 	realdecoder "github.com/codecrafters-io/kafka-tester/protocol/decoder"
 	realencoder "github.com/codecrafters-io/kafka-tester/protocol/encoder"
 
@@ -125,9 +126,9 @@ func fetch(b *protocol.Broker, requestBody *FetchRequestBody, logger *logger.Log
 		return nil, err
 	}
 
-	protocol.PrintHexdump(response)
+	protocol.PrintHexdump(response.RawBytes)
 
-	_, fetchResponse, err := DecodeFetchHeaderAndResponse(response, 16, logger)
+	_, fetchResponse, err := DecodeFetchHeaderAndResponse(response.Payload, 16, logger)
 	if err != nil {
 		return nil, err
 	}