add timeout to decider delegation

addresses review comment ava-labs/icm-services#344 (comment)
avalabsorg · Jul 22, 2024 · e013687 · e013687
1 parent 6402790
commit e013687
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/messages/teleporter/message_handler.go b/messages/teleporter/message_handler.go
@@ -222,9 +222,10 @@ func (m *messageHandler) deciderRejectedMessage() (bool, error) {
 		return false, err
 	}
 
-	// TODO: add a timeout to the context
+	ctx, cancelCtx := context.WithTimeout(context.Background(), 30*time.Second)
+	defer cancelCtx()
 	response, err := m.deciderClient.ShouldSendMessage(
-		context.Background(),
+		ctx,
 		&pbDecider.ShouldSendMessageRequest{
 			NetworkId:           m.unsignedMessage.NetworkID,
 			SourceChainId:       m.unsignedMessage.SourceChainID[:],