Avoids using context.Background() in transaction streamer

OffchainLabs · Jan 15, 2025 · 53701a7 · 53701a7
1 parent da6022a
commit 53701a7
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/arbnode/transaction_streamer.go b/arbnode/transaction_streamer.go
@@ -497,7 +497,7 @@ func (s *TransactionStreamer) GetProcessedMessageCount() (arbutil.MessageIndex,
 	if err != nil {
 		return 0, err
 	}
-	digestedHead, err := s.exec.HeadMessageNumber().Await(context.Background())
+	digestedHead, err := s.exec.HeadMessageNumber().Await(s.GetContext())
 	if err != nil {
 		return 0, err
 	}
@@ -1093,7 +1093,11 @@ func (s *TransactionStreamer) ResultAtCount(count arbutil.MessageIndex) (*execut
 	}
 	log.Info(FailedToGetMsgResultFromDB, "count", count)
 
-	msgResult, err := s.exec.ResultAtPos(pos).Await(context.Background())
+	ctx := context.Background()
+	if s.Started() {
+		ctx = s.GetContext()
+	}
+	msgResult, err := s.exec.ResultAtPos(pos).Await(ctx)
 	if err != nil {
 		return nil, err
 	}