1

Author: wyxxxcat

pkg/ccr/base/spec.go

@@ -631,6 +631,7 @@ func (s *Spec) CreateTableOrView(createTable *record.CreateTable, srcDatabase st
 	}
 
 	createSql = AddDBPrefixToCreateTableOrViewSql(s.Database, createSql)
+	createSql = ReplaceDBNameForCreateSql(createSql, s.Database, s.Database)
 
 	// Compatible with doris 2.1.x, see apache/doris#44834 for details.
 	for strings.Contains(createSql, "MAXVALUEMAXVALUE") {
@@ -1550,3 +1551,9 @@ func AddDBPrefixToCreateTableOrViewSql(dbName, createSql string) string {
 	}
 	return createSql
 }
+
+func ReplaceDBNameForCreateSql(createSql, srcDbName, destName string) string {
+	re := regexp.MustCompile(fmt.Sprintf(`\b%s\b`, srcDbName))
+	createSql = re.ReplaceAllString(createSql, destName)
+	return createSql
+}

pkg/ccr/base/spec_test.go

@@ -26,3 +26,20 @@ func TestAddDBPrefixToCreateTableOrViewSql(t *testing.T) {
 		}
 	}
 }
+
+func TestReplaceDBNameForCreateSql(t *testing.T) {
+	type TestCase struct {
+		origin, expect string
+	}
+
+	testCases := []TestCase{
+		{"CREATE VIEW `target_db`.`v` AS (select `origin_db`.`func`(`internal`.`target_db`.`create_view_table1`.`id`) as `c1`,abs(`internal`.`target_db`.`create_view_table1`.`id`) from `internal`.`target_db`.`create_view_table1`)", "CREATE VIEW `target_db`.`v` AS (select `target_db`.`func`(`internal`.`target_db`.`create_view_table1`.`id`) as `c1`,abs(`internal`.`target_db`.`create_view_table1`.`id`) from `internal`.`target_db`.`create_view_table1`)"},
+		{"CREATE VIEW `target_db`.`v` AS (select `origin_db_not_replace`.`func`(`internal`.`target_db`.`create_view_table1`.`id`) as `c1`,abs(`internal`.`target_db`.`create_view_table1`.`id`) from `internal`.`target_db`.`create_view_table1`)", "CREATE VIEW `target_db`.`v` AS (select `origin_db_not_replace`.`func`(`internal`.`target_db`.`create_view_table1`.`id`) as `c1`,abs(`internal`.`target_db`.`create_view_table1`.`id`) from `internal`.`target_db`.`create_view_table1`)"},
+	}
+
+	for i, c := range testCases {
+		if actual := base.ReplaceDBNameForCreateSql(c.origin, "origin_db", "target_db"); actual != c.expect {
+			t.Errorf("case %d failed, expect %s, but got %s", i, c.expect, actual)
+		}
+	}
+}