feat: add block type to blockfetch Block callback (#684)

blinklabs-io · Aug 16, 2024 · 6f38096 · 6f38096
1 parent 996c877
commit 6f38096
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/cmd/gouroboros/chainsync.go b/cmd/gouroboros/chainsync.go
@@ -324,6 +324,7 @@ func chainSyncRollForwardHandler(
 
 func blockFetchBlockHandler(
 	ctx blockfetch.CallbackContext,
+	blockType uint,
 	blockData ledger.Block,
 ) error {
 	switch block := blockData.(type) {

diff --git a/protocol/blockfetch/blockfetch.go b/protocol/blockfetch/blockfetch.go
@@ -101,7 +101,7 @@ type CallbackContext struct {
 }
 
 // Callback function types
-type BlockFunc func(CallbackContext, ledger.Block) error
+type BlockFunc func(CallbackContext, uint, ledger.Block) error
 type RequestRangeFunc func(CallbackContext, common.Point, common.Point) error
 
 func New(protoOptions protocol.ProtocolOptions, cfg *Config) *BlockFetch {

diff --git a/protocol/blockfetch/client.go b/protocol/blockfetch/client.go
@@ -191,7 +191,7 @@ func (c *Client) handleBlock(msgGeneric protocol.Message) error {
 	}
 	// We use the callback when requesting ranges and the internal channel for a single block
 	if c.blockUseCallback {
-		if err := c.config.BlockFunc(c.callbackContext, blk); err != nil {
+		if err := c.config.BlockFunc(c.callbackContext, wrappedBlock.Type, blk); err != nil {
 			return err
 		}
 	} else {