feat: add range configuration to job settings for enhanced data processing control

config.yaml

@@ -30,6 +30,11 @@ jobs:
       table: MANZANA
     pre_sql: 
       - 'SELECT 1'
+    range:
+      min: 1000000
+      max: 2000000
+      is_min_inclusive: false
+      is_max_inclusive: true
 
   - name: red_puerto
     enabled: true

internal/app/config/migration.go

@@ -50,6 +50,12 @@ type Job struct {
 	PreSQL      []string        `yaml:"pre_sql"`
 	PostSQL     []string        `yaml:"post_sql"`
 	JobConfig   `yaml:",inline"`
+	Range       struct {
+		Min            int64 `yaml:"min"`
+		Max            int64 `yaml:"max"`
+		IsMinInclusive bool  `yaml:"is_min_inclusive"`
+		IsMaxInclusive bool  `yaml:"is_max_inclusive"`
+	}
 }
 
 type MigrationConfig struct {

internal/app/custom_errors/extractor.error.go

@@ -103,8 +103,8 @@ func ExtractorErrorHandler(
 			if err.HasLastId {
 				newPartition.ParentId = err.Partition.Id
 				newPartition.Id = uuid.New()
-				newPartition.LowerLimit = err.LastId
+				newPartition.Range.Min = err.LastId
-				newPartition.IsLowerLimitInclusive = false
+				newPartition.Range.IsMinInclusive = false
 			}
 
 			requeueWithBackoff(ctx, delay, func() {

internal/app/etl/extractors/mssql.go

@@ -107,13 +107,13 @@ func (mssqlEx *MssqlExtractor) ProcessPartition(
 	indexPrimaryKey int,
 	chBatchesOut chan<- models.Batch,
 ) (int, error) {
-	query := buildExtractQueryMssql(tableInfo, columns, partition.ShouldUseRange, partition.IsLowerLimitInclusive)
+	query := buildExtractQueryMssql(tableInfo, columns, partition.HasRange, partition.Range.IsMinInclusive)
 
 	var queryArgs []any
-	if partition.ShouldUseRange {
+	if partition.HasRange {
 		queryArgs = append(queryArgs,
-			sql.Named("min", partition.LowerLimit),
+			sql.Named("min", partition.Range.Min),
-			sql.Named("max", partition.UpperLimit),
+			sql.Named("max", partition.Range.Max),
 		)
 	}
 

internal/app/etl/extractors/postgres.go

@@ -62,7 +62,7 @@ func (postgresEx *PostgresExtractor) ProcessPartition(
 ) (int, error) {
 	query := buildExtractQueryPostgres(tableInfo, columns)
 
-	if partition.ShouldUseRange {
+	if partition.HasRange {
 		return 0, errors.New("Batch config not yet supported")
 	}
 

internal/app/etl/table_analyzers/main.go

@@ -24,7 +24,7 @@ func PartitionRangeGenerator(
 	if rowsCount <= rowsPerPartition {
 		return []models.Partition{{
 			Id:           uuid.New(),
-			ShouldUseRange: false,
+			HasRange:     false,
 			RetryCounter: 0,
 		}}, nil
 

internal/app/etl/table_analyzers/mssql.go

@@ -229,12 +229,14 @@ ORDER BY batch_id`,
 	for rows.Next() {
 		partition := models.Partition{
 			Id:           uuid.New(),
-			ShouldUseRange:        true,
+			HasRange:     true,
 			RetryCounter: 0,
-			IsLowerLimitInclusive: true,
+			Range: models.PartitionRange{
+				IsMinInclusive: true,
+			},
 		}
 
-		if err := rows.Scan(&partition.LowerLimit, &partition.UpperLimit); err != nil {
+		if err := rows.Scan(&partition.Range.Min, &partition.Range.Max); err != nil {
 			return nil, err
 		}
 

internal/app/models/main.go

@@ -11,12 +11,17 @@ type Batch struct {
 	RetryCounter int
 }
 
+type PartitionRange struct {
+	Min            int64
+	Max            int64
+	IsMinInclusive bool
+	IsMaxInclusive bool
+}
+
 type Partition struct {
 	Id           uuid.UUID
 	ParentId     uuid.UUID
-	LowerLimit            int64
+	Range        PartitionRange
-	UpperLimit            int64
+	HasRange     bool
-	IsLowerLimitInclusive bool
-	ShouldUseRange        bool
 	RetryCounter int
 }