Skip to content

Commit 299d6c8

Browse files
iskakaushikAmogh-Bharadwajserprex
authored
Add time limit for get variant to be 10s (#3622)
During DropFlow we have get flow metadata, there is a 1 min timeout. But the failure can take longer. --------- Co-authored-by: Amogh Bharadwaj <[email protected]> Co-authored-by: Philip Dubé <[email protected]>
1 parent 02c9c20 commit 299d6c8

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

flow/activities/flowable.go

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1744,8 +1744,13 @@ func (a *FlowableActivity) GetFlowMetadata(
17441744

17451745
// Detect source database variant
17461746
if input.SourceName != "" {
1747-
if srcConn, err := connectors.GetByNameAs[connectors.DatabaseVariantConnector](ctx, nil, a.CatalogPool, input.SourceName); err == nil {
1748-
if variant, variantErr := srcConn.GetDatabaseVariant(ctx); variantErr == nil {
1747+
// Use a short timeout for optional variant detection to avoid consuming entire activity timeout
1748+
variantCtx, cancel := context.WithTimeout(ctx, 10*time.Second)
1749+
defer cancel()
1750+
if srcConn, err := connectors.GetByNameAs[connectors.DatabaseVariantConnector](
1751+
variantCtx, nil, a.CatalogPool, input.SourceName,
1752+
); err == nil {
1753+
if variant, variantErr := srcConn.GetDatabaseVariant(variantCtx); variantErr == nil {
17491754
sourcePeer.Variant = variant
17501755
} else {
17511756
logger.Warn("failed to get source database variant", slog.Any("error", variantErr))

0 commit comments

Comments
 (0)