validate: 1 minute timeout (#2183)

#2182
PeerDB-io · Oct 24, 2024 · 2c9fb2e · 2c9fb2e
1 parent 5c3df7b
commit 2c9fb2e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/flow/cmd/validate_peer.go b/flow/cmd/validate_peer.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	"log/slog"
+	"time"
 
 	"github.com/PeerDB-io/peer-flow/connectors"
 	connpostgres "github.com/PeerDB-io/peer-flow/connectors/postgres"
@@ -16,6 +17,8 @@ func (h *FlowRequestHandler) ValidatePeer(
 	ctx context.Context,
 	req *protos.ValidatePeerRequest,
 ) (*protos.ValidatePeerResponse, error) {
+	ctx, cancelCtx := context.WithTimeout(ctx, time.Minute)
+	defer cancelCtx()
 	if req.Peer == nil {
 		return &protos.ValidatePeerResponse{
 			Status:  protos.ValidatePeerStatus_INVALID,