refactor: add context parameter to all executor methods (#31)

* refactor: add context parameter to all executor methods

This commit adds context.Context as the first parameter to all executor methods and updates relevant function calls and mocks accordingly. This change enhances context propagation and allows for better control over request lifecycles and timeouts.

Resolves #25.

* refactor: replace `context.TODO()` with `ctx` arg

execution.go

@@ -1,6 +1,7 @@
 package execution
 
 import (
+	"context"
 	"time"
 
 	"github.com/rollkit/go-execution/types"
@@ -9,33 +10,14 @@ import (
 // Executor defines a common interface for interacting with the execution client.
 type Executor interface {
 	// InitChain initializes the blockchain with genesis information.
-	InitChain(
-		genesisTime time.Time,
-		initialHeight uint64,
-		chainID string,
-	) (
-		stateRoot types.Hash,
-		maxBytes uint64,
-		err error,
-	)
+	InitChain(ctx context.Context, genesisTime time.Time, initialHeight uint64, chainID string) (stateRoot types.Hash, maxBytes uint64, err error)
 
 	// GetTxs retrieves all available transactions from the execution client's mempool.
-	GetTxs() ([]types.Tx, error)
+	GetTxs(ctx context.Context) ([]types.Tx, error)
 
 	// ExecuteTxs executes a set of transactions to produce a new block header.
-	ExecuteTxs(
-		txs []types.Tx,
-		blockHeight uint64,
-		timestamp time.Time,
-		prevStateRoot types.Hash,
-	) (
-		updatedStateRoot types.Hash,
-		maxBytes uint64,
-		err error,
-	)
+	ExecuteTxs(ctx context.Context, txs []types.Tx, blockHeight uint64, timestamp time.Time, prevStateRoot types.Hash) (updatedStateRoot types.Hash, maxBytes uint64, err error)
 
 	// SetFinal marks a block at the given height as final.
-	SetFinal(
-		blockHeight uint64,
-	) error
+	SetFinal(ctx context.Context, blockHeight uint64) error
 }

proxy/grpc/client.go

@@ -51,10 +51,7 @@ func (c *Client) Stop() error {
 }
 
 // InitChain initializes the blockchain with genesis information.
-func (c *Client) InitChain(genesisTime time.Time, initialHeight uint64, chainID string) (types.Hash, uint64, error) {
-	ctx, cancel := context.WithTimeout(context.Background(), c.config.DefaultTimeout)
-	defer cancel()
-
+func (c *Client) InitChain(ctx context.Context, genesisTime time.Time, initialHeight uint64, chainID string) (types.Hash, uint64, error) {
 	resp, err := c.client.InitChain(ctx, &pb.InitChainRequest{
 		GenesisTime:   genesisTime.Unix(),
 		InitialHeight: initialHeight,
@@ -71,10 +68,7 @@ func (c *Client) InitChain(genesisTime time.Time, initialHeight uint64, chainID
 }
 
 // GetTxs retrieves all available transactions from the execution client's mempool.
-func (c *Client) GetTxs() ([]types.Tx, error) {
-	ctx, cancel := context.WithTimeout(context.Background(), c.config.DefaultTimeout)
-	defer cancel()
-
+func (c *Client) GetTxs(ctx context.Context) ([]types.Tx, error) {
 	resp, err := c.client.GetTxs(ctx, &pb.GetTxsRequest{})
 	if err != nil {
 		return nil, err
@@ -89,10 +83,7 @@ func (c *Client) GetTxs() ([]types.Tx, error) {
 }
 
 // ExecuteTxs executes a set of transactions to produce a new block header.
-func (c *Client) ExecuteTxs(txs []types.Tx, blockHeight uint64, timestamp time.Time, prevStateRoot types.Hash) (types.Hash, uint64, error) {
-	ctx, cancel := context.WithTimeout(context.Background(), c.config.DefaultTimeout)
-	defer cancel()
-
+func (c *Client) ExecuteTxs(ctx context.Context, txs []types.Tx, blockHeight uint64, timestamp time.Time, prevStateRoot types.Hash) (types.Hash, uint64, error) {
 	req := &pb.ExecuteTxsRequest{
 		Txs:           make([][]byte, len(txs)),
 		BlockHeight:   blockHeight,
@@ -115,10 +106,7 @@ func (c *Client) ExecuteTxs(txs []types.Tx, blockHeight uint64, timestamp time.T
 }
 
 // SetFinal marks a block at the given height as final.
-func (c *Client) SetFinal(blockHeight uint64) error {
-	ctx, cancel := context.WithTimeout(context.Background(), c.config.DefaultTimeout)
-	defer cancel()
-
+func (c *Client) SetFinal(ctx context.Context, blockHeight uint64) error {
 	_, err := c.client.SetFinal(ctx, &pb.SetFinalRequest{
 		BlockHeight: blockHeight,
 	})