From e01368776d57a5cfd8d44d66dd62bce7dce40f6a Mon Sep 17 00:00:00 2001
From: "F. Eugene Aumson" <feuGeneA@users.noreply.github.com>
Date: Mon, 22 Jul 2024 15:34:02 +0000
Subject: [PATCH] add timeout to decider delegation

addresses review comment https://github.com/ava-labs/awm-relayer/pull/344#discussion_r1680066442
---
 messages/teleporter/message_handler.go | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/messages/teleporter/message_handler.go b/messages/teleporter/message_handler.go
index 91d3c0e8..2299f040 100644
--- 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[:],