From 39c0d995020e3a174d4fdb58d7c62850c651b725 Mon Sep 17 00:00:00 2001 From: Kylesoda <249518290+kylesoda@users.noreply.github.com> Date: Thu, 16 Apr 2026 13:19:46 -0500 Subject: [PATCH] feat: extend context timeout to 1 minute for database queries in mssql.go and postgres.go --- internal/app/etl/table_analyzers/mssql.go | 4 ++-- internal/app/etl/table_analyzers/postgres.go | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/internal/app/etl/table_analyzers/mssql.go b/internal/app/etl/table_analyzers/mssql.go index f21a3ab..4faaf21 100644 --- a/internal/app/etl/table_analyzers/mssql.go +++ b/internal/app/etl/table_analyzers/mssql.go @@ -184,7 +184,7 @@ JOIN sys.partitions p ON t.object_id = p.object_id WHERE s.name = @schema AND t.name = @table AND p.index_id IN (0, 1) GROUP BY t.name` - ctxTimeout, cancel := context.WithTimeout(ctx, time.Second*20) + ctxTimeout, cancel := context.WithTimeout(ctx, 1*time.Minute) defer cancel() var rowsCount int64 @@ -216,7 +216,7 @@ ORDER BY batch_id`, tableInfo.Schema, tableInfo.Table) - ctxTimeout, cancel := context.WithTimeout(ctx, time.Second*20) + ctxTimeout, cancel := context.WithTimeout(ctx, 1*time.Minute) defer cancel() rows, err := ta.db.Query(ctxTimeout, query, sql.Named("maxPartitions", maxPartitions)) diff --git a/internal/app/etl/table_analyzers/postgres.go b/internal/app/etl/table_analyzers/postgres.go index 9489c78..8aac15d 100644 --- a/internal/app/etl/table_analyzers/postgres.go +++ b/internal/app/etl/table_analyzers/postgres.go @@ -125,7 +125,7 @@ func (ta *PostgresTableAnalyzer) QueryColumnTypes( ctx context.Context, tableInfo config.TableInfo, ) ([]models.ColumnType, error) { - localCtx, cancel := context.WithTimeout(ctx, 20*time.Second) + localCtx, cancel := context.WithTimeout(ctx, 1*time.Minute) defer cancel() rows, err := ta.db.Query(localCtx, postgresColumnMetadataQuery, tableInfo.Schema, tableInfo.Table)