Correct add list partition sql (selectdb#92)

lsy3993 · May 28, 2024 · c522ed5 · c522ed5
1 parent da6573f
commit c522ed5
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pkg/ccr/base/spec.go b/pkg/ccr/base/spec.go
@@ -846,6 +846,14 @@ func correctAddPartitionSql(addPartitionSql string, addPartition *record.AddPart
 		re := regexp.MustCompile(`VALUES \[\(\), \(\)\) \(.*\)`)
 		addPartitionSql = re.ReplaceAllString(addPartitionSql, "")
 	}
+	if strings.Contains(addPartitionSql, "VALUES IN (((") {
+		re := regexp.MustCompile(`VALUES IN \(\(\((.*)\)\)\)`)
+		matches := re.FindStringSubmatch(addPartitionSql)
+		if len(matches) > 1 {
+			replace := fmt.Sprintf("VALUES IN ((%s))", matches[1])
+			addPartitionSql = re.ReplaceAllString(addPartitionSql, replace)
+		}
+	}
 	if addPartition.IsTemp && !strings.Contains(addPartitionSql, "ADD TEMPORARY PARTITION") {
 		addPartitionSql = strings.ReplaceAll(addPartitionSql, "ADD PARTITION", "ADD TEMPORARY PARTITION")
 	}