feat: handle bidtrace 404

ultrasoundmoney · Feb 12, 2024 · 77dde8a · 77dde8a
1 parent 527599e
commit 77dde8a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/datastore/auction_api.go b/datastore/auction_api.go
@@ -27,6 +27,7 @@ func GetPayloadContents(slot uint64, proposerPubkey, blockHash string) (*builder
 	if err != nil {
 		return nil, err
 	}
+	defer resp.Body.Close()
 
 	if resp.StatusCode == 404 {
 		return nil, ErrExecutionPayloadNotFound
@@ -60,6 +61,11 @@ func GetBidTrace(slot uint64, proposerPubkey, blockHash string) (*common.BidTrac
 	if err != nil {
 		return nil, err
 	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode == 404 {
+		return nil, ErrBidTraceNotFound
+	}
 
 	body, err := io.ReadAll(resp.Body)
 	if err != nil {

diff --git a/datastore/datastore.go b/datastore/datastore.go
@@ -21,6 +21,7 @@ import (
 )
 
 var ErrExecutionPayloadNotFound = errors.New("execution payload not found")
+var ErrBidTraceNotFound = errors.New("bidtrace not found")
 
 type GetHeaderResponseKey struct {
 	Slot           uint64