🐛 fix(sql-file): 修复整库备份导入失败

Fixes #662
2026-08-09 16:23:27 +08:00 · 2026-07-17 23:37:53 +08:00
parent e24942083a
commit d0a6a97f71
3 changed files with 233 additions and 9 deletions
--- a/internal/app/methods_file_sql_execution_test.go
+++ b/internal/app/methods_file_sql_execution_test.go
@@ -584,3 +584,109 @@ func TestStreamSQLFileKeepsOraclePackageSpecAndBodyTogether(t *testing.T) {
 		t.Fatalf("unexpected third statement: %q", statements[2])
 	}
 }
+
+func TestResolveSQLFileExecutionRunConfigUsesServerConnectionForGoNaviMySQLDatabaseBackup(t *testing.T) {
+	preamble := strings.Join([]string{
+		"-- GoNavi SQL Export",
+		"-- Time: 2026-07-17 00:00:00",
+		"-- Database: restore_target",
+		"",
+		"CREATE DATABASE IF NOT EXISTS `restore_target`;",
+		"",
+		"USE `restore_target`;",
+	}, "\n")
+
+	got := resolveSQLFileExecutionRunConfig(
+		connection.ConnectionConfig{Type: "mysql", Database: "selected_target"},
+		"selected_target",
+		[]byte(preamble),
+	)
+	if got.Database != "" {
+		t.Fatalf("GoNavi MySQL database backup must connect at server level before CREATE/USE, got database=%q", got.Database)
+	}
+}
+
+func TestResolveSQLFileExecutionRunConfigKeepsSelectedDatabaseForRegularSQL(t *testing.T) {
+	got := resolveSQLFileExecutionRunConfig(
+		connection.ConnectionConfig{Type: "mysql", Database: "configured_default"},
+		"selected_target",
+		[]byte("CREATE TABLE demo(id INT);"),
+	)
+	if got.Database != "selected_target" {
+		t.Fatalf("regular SQL must retain the selected database, got database=%q", got.Database)
+	}
+}
+
+func TestResolveSQLFileExecutionRunConfigUsesServerConnectionForLegacyGoNaviMySQLDatabaseBackup(t *testing.T) {
+	preamble := strings.Join([]string{
+		"-- GoNavi SQL Export",
+		"-- Time: 2026-07-11 00:00:00",
+		"-- Database: legacy_restore_target",
+		"",
+		"USE `legacy_restore_target`;",
+	}, "\n")
+
+	got := resolveSQLFileExecutionRunConfig(
+		connection.ConnectionConfig{Type: "mysql", Database: "selected_target"},
+		"selected_target",
+		[]byte(preamble),
+	)
+	if got.Database != "" {
+		t.Fatalf("legacy GoNavi MySQL database backup must connect at server level before USE, got database=%q", got.Database)
+	}
+}
+
+func TestBuildGoNaviMySQLDatabaseBackupBootstrapSQLOnlyForLegacyBackup(t *testing.T) {
+	legacy := goNaviMySQLDatabaseBackupPreamble{databaseName: "legacy_restore_target"}
+	if got := buildGoNaviMySQLDatabaseBackupBootstrapSQL(legacy); got != "CREATE DATABASE IF NOT EXISTS `legacy_restore_target`" {
+		t.Fatalf("unexpected legacy bootstrap SQL: %q", got)
+	}
+
+	current := goNaviMySQLDatabaseBackupPreamble{
+		databaseName:           "current_restore_target",
+		includesCreateDatabase: true,
+	}
+	if got := buildGoNaviMySQLDatabaseBackupBootstrapSQL(current); got != "" {
+		t.Fatalf("backup that already creates its database must not be bootstrapped again, got %q", got)
+	}
+}
+
+func TestExecuteSQLFileStreamRunsGoNaviMySQLDatabaseBackupHeader(t *testing.T) {
+	fakeDB := &fakeSQLFileBatchDB{}
+	input := strings.Join([]string{
+		"-- GoNavi SQL Export",
+		"-- Database: restore_target",
+		"CREATE DATABASE IF NOT EXISTS `restore_target`;",
+		"USE `restore_target`;",
+		"SET FOREIGN_KEY_CHECKS=0;",
+		"CREATE TABLE users(id INT PRIMARY KEY);",
+		"INSERT INTO users(id) VALUES (1);",
+		"SET FOREIGN_KEY_CHECKS=1;",
+	}, "\n")
+
+	result, err := executeSQLFileStream(context.Background(), fakeDB, strings.NewReader(input), sqlFileExecutionOptions{
+		DBType:             "mysql",
+		BatchMaxStatements: 100,
+		BatchMaxBytes:      1024,
+	}, nil)
+	if err != nil {
+		t.Fatalf("executeSQLFileStream returned error: %v", err)
+	}
+	if result.Executed != 6 || result.Failed != 0 {
+		t.Fatalf("expected complete database backup header and statements to execute, got %#v", result)
+	}
+	joinedExec := strings.Join(fakeDB.execQueries, "\n")
+	for _, expected := range []string{
+		"CREATE DATABASE IF NOT EXISTS `restore_target`",
+		"USE `restore_target`",
+		"CREATE TABLE users(id INT PRIMARY KEY)",
+		"SET FOREIGN_KEY_CHECKS=1",
+	} {
+		if !strings.Contains(joinedExec, expected) {
+			t.Fatalf("expected backup statement %q to execute, queries=%#v", expected, fakeDB.execQueries)
+		}
+	}
+	if len(fakeDB.batchQueries) != 1 || !strings.Contains(fakeDB.batchQueries[0], "INSERT INTO users(id) VALUES (1)") {
+		t.Fatalf("expected INSERT data to be batched after schema restore, batches=%#v", fakeDB.batchQueries)
+	}
+}