refactor: enhance SQL query formatting for PostgreSQL compatibility in dry run and validation processes

cmd/go_migrate/dryrun.go

@@ -8,6 +8,7 @@ import (
 	"git.ksdemosapps.com/kylesoda/go-migrate/internal/app/azure"
 	"git.ksdemosapps.com/kylesoda/go-migrate/internal/app/config"
 	dbwrapper "git.ksdemosapps.com/kylesoda/go-migrate/internal/app/db-wrapper"
+	db_dialects "git.ksdemosapps.com/kylesoda/go-migrate/internal/app/db-wrapper/db_dialects"
 	log "github.com/sirupsen/logrus"
 )
 
@@ -67,9 +68,15 @@ func dryRunCountSourceRows(
 	for range maxParallelWorkers {
 		wg.Go(func() {
 			for job := range chJobs {
+				var sourceTableDisplay string
+				if sourceDb.GetDialect() == db_dialects.Postgres {
+					sourceTableDisplay = fmt.Sprintf(`"%s"."%s"`, job.SourceTable.Schema, job.SourceTable.Table)
+				} else {
+					sourceTableDisplay = fmt.Sprintf("[%s].[%s]", job.SourceTable.Schema, job.SourceTable.Table)
+				}
 				res := DryRunResult{
 					JobName:     job.Name,
-					SourceTable: fmt.Sprintf("[%s].[%s]", job.SourceTable.Schema, job.SourceTable.Table),
+					SourceTable: sourceTableDisplay,
 				}
 				count, err := countSourceRows(ctx, sourceDb, job)
 				if err != nil {

cmd/go_migrate/validate.go

@@ -8,6 +8,7 @@ import (
 
 	"git.ksdemosapps.com/kylesoda/go-migrate/internal/app/config"
 	dbwrapper "git.ksdemosapps.com/kylesoda/go-migrate/internal/app/db-wrapper"
+	db_dialects "git.ksdemosapps.com/kylesoda/go-migrate/internal/app/db-wrapper/db_dialects"
 	log "github.com/sirupsen/logrus"
 )
 
@@ -32,6 +33,32 @@ func countSourceRows(ctx context.Context, db dbwrapper.DbWrapper, job config.Job
 		args  []any
 	)
 
+	if db.GetDialect() == db_dialects.Postgres {
+		if hasRange && job.SourceTable.PrimaryKey != "" {
+			query = fmt.Sprintf(`SELECT COUNT(*) FROM "%s"."%s" WHERE 1=1`, schema, table)
+			paramIdx := 0
+			if job.Range.Min != nil {
+				paramIdx++
+				op := ">"
+				if job.Range.IsMinInclusive {
+					op = ">="
+				}
+				query += fmt.Sprintf(` AND "%s" %s $%d`, job.SourceTable.PrimaryKey, op, paramIdx)
+				args = append(args, *job.Range.Min)
+			}
+			if job.Range.Max != nil {
+				paramIdx++
+				op := "<"
+				if job.Range.IsMaxInclusive {
+					op = "<="
+				}
+				query += fmt.Sprintf(` AND "%s" %s $%d`, job.SourceTable.PrimaryKey, op, paramIdx)
+				args = append(args, *job.Range.Max)
+			}
+		} else {
+			query = fmt.Sprintf(`SELECT COUNT(*) FROM "%s"."%s"`, schema, table)
+		}
+	} else {
 		if hasRange && job.SourceTable.PrimaryKey != "" {
 			query = fmt.Sprintf("SELECT COUNT_BIG(*) FROM [%s].[%s] WHERE 1=1", schema, table)
 			if job.Range.Min != nil {
@@ -53,6 +80,7 @@ func countSourceRows(ctx context.Context, db dbwrapper.DbWrapper, job config.Job
 		} else {
 			query = fmt.Sprintf("SELECT COUNT_BIG(*) FROM [%s].[%s]", schema, table)
 		}
+	}
 
 	var count int64
 	if err := db.QueryRow(ctx, query, args...).Scan(&count); err != nil {
@@ -64,7 +92,13 @@ func countSourceRows(ctx context.Context, db dbwrapper.DbWrapper, job config.Job
 func countTargetRows(ctx context.Context, db dbwrapper.DbWrapper, job config.Job) (int64, error) {
 	schema := job.TargetTable.Schema
 	table := job.TargetTable.Table
-	query := fmt.Sprintf(`SELECT COUNT(*) FROM "%s"."%s"`, schema, table)
+
+	var query string
+	if db.GetDialect() == db_dialects.Postgres {
+		query = fmt.Sprintf(`SELECT COUNT(*) FROM "%s"."%s"`, schema, table)
+	} else {
+		query = fmt.Sprintf("SELECT COUNT_BIG(*) FROM [%s].[%s]", schema, table)
+	}
 
 	var count int64
 	if err := db.QueryRow(ctx, query).Scan(&count); err != nil {
@@ -74,10 +108,22 @@ func countTargetRows(ctx context.Context, db dbwrapper.DbWrapper, job config.Job
 }
 
 func validateJob(ctx context.Context, sourceDb, targetDb dbwrapper.DbWrapper, job config.Job) ValidationResult {
+	var sourceTable, targetTable string
+	if sourceDb.GetDialect() == db_dialects.Postgres {
+		sourceTable = fmt.Sprintf(`"%s"."%s"`, job.SourceTable.Schema, job.SourceTable.Table)
+	} else {
+		sourceTable = fmt.Sprintf("[%s].[%s]", job.SourceTable.Schema, job.SourceTable.Table)
+	}
+	if targetDb.GetDialect() == db_dialects.Postgres {
+		targetTable = fmt.Sprintf(`"%s"."%s"`, job.TargetTable.Schema, job.TargetTable.Table)
+	} else {
+		targetTable = fmt.Sprintf("[%s].[%s]", job.TargetTable.Schema, job.TargetTable.Table)
+	}
+
 	result := ValidationResult{
 		JobName:     job.Name,
-		SourceTable: fmt.Sprintf("[%s].[%s]", job.SourceTable.Schema, job.SourceTable.Table),
+		SourceTable: sourceTable,
-		TargetTable: fmt.Sprintf(`"%s"."%s"`, job.TargetTable.Schema, job.TargetTable.Table),
+		TargetTable: targetTable,
 	}
 
 	var (