fix(multirepo): handle out-of-order dependencies during JSONL import (#414)
* bd sync: 2025-11-29 00:08:58 * fix(multirepo): handle out-of-order dependencies during JSONL import Fixes #413. When importing issues from multi-repo JSONL files, if issue A (line 1) has a dependency on issue B (line 5), the import would fail with FK constraint error because B doesn't exist yet. Solution: - Disable FK checks at start of importJSONLFile() - Re-enable FK checks before commit - Run PRAGMA foreign_key_check to validate data integrity - Fail with clear error if orphaned dependencies are detected This allows out-of-order dependencies while still catching corrupted data. --------- Co-authored-by: Shaun Cutts <shauncutts@factfiber.com>
This commit is contained in:
@@ -117,6 +117,7 @@ func (s *SQLiteStorage) hydrateFromRepo(ctx context.Context, repoPath, sourceRep
|
||||
}
|
||||
|
||||
// importJSONLFile imports issues from a JSONL file, setting the source_repo field.
|
||||
// Disables FK checks during import to handle out-of-order dependencies.
|
||||
func (s *SQLiteStorage) importJSONLFile(ctx context.Context, jsonlPath, sourceRepo string) (int, error) {
|
||||
file, err := os.Open(jsonlPath) // #nosec G304 -- jsonlPath is from trusted source
|
||||
if err != nil {
|
||||
@@ -138,8 +139,22 @@ func (s *SQLiteStorage) importJSONLFile(ctx context.Context, jsonlPath, sourceRe
|
||||
count := 0
|
||||
lineNum := 0
|
||||
|
||||
// Get exclusive connection to ensure PRAGMA applies
|
||||
conn, err := s.db.Conn(ctx)
|
||||
if err != nil {
|
||||
return 0, fmt.Errorf("failed to get connection: %w", err)
|
||||
}
|
||||
defer func() { _ = conn.Close() }()
|
||||
|
||||
// Disable foreign keys on this connection to handle out-of-order deps
|
||||
// (issue A may depend on issue B that appears later in the file)
|
||||
_, err = conn.ExecContext(ctx, `PRAGMA foreign_keys = OFF`)
|
||||
if err != nil {
|
||||
return 0, fmt.Errorf("failed to disable foreign keys: %w", err)
|
||||
}
|
||||
|
||||
// Begin transaction for bulk import
|
||||
tx, err := s.db.BeginTx(ctx, nil)
|
||||
tx, err := conn.BeginTx(ctx, nil)
|
||||
if err != nil {
|
||||
return 0, fmt.Errorf("failed to begin transaction: %w", err)
|
||||
}
|
||||
@@ -179,6 +194,28 @@ func (s *SQLiteStorage) importJSONLFile(ctx context.Context, jsonlPath, sourceRe
|
||||
return 0, fmt.Errorf("failed to read JSONL file: %w", err)
|
||||
}
|
||||
|
||||
// Re-enable foreign keys before commit to validate data integrity
|
||||
_, err = conn.ExecContext(ctx, `PRAGMA foreign_keys = ON`)
|
||||
if err != nil {
|
||||
return 0, fmt.Errorf("failed to re-enable foreign keys: %w", err)
|
||||
}
|
||||
|
||||
// Validate FK constraints on imported data
|
||||
rows, err := conn.QueryContext(ctx, `PRAGMA foreign_key_check`)
|
||||
if err != nil {
|
||||
return 0, fmt.Errorf("failed to check foreign keys: %w", err)
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
if rows.Next() {
|
||||
var table, rowid, parent, fkid string
|
||||
_ = rows.Scan(&table, &rowid, &parent, &fkid)
|
||||
return 0, fmt.Errorf(
|
||||
"foreign key violation in imported data: table=%s rowid=%s parent=%s",
|
||||
table, rowid, parent,
|
||||
)
|
||||
}
|
||||
|
||||
if err := tx.Commit(); err != nil {
|
||||
return 0, fmt.Errorf("failed to commit transaction: %w", err)
|
||||
}
|
||||
@@ -197,7 +234,7 @@ func (s *SQLiteStorage) upsertIssueInTx(ctx context.Context, tx *sql.Tx, issue *
|
||||
// Check if issue exists
|
||||
var existingID string
|
||||
err := tx.QueryRowContext(ctx, `SELECT id FROM issues WHERE id = ?`, issue.ID).Scan(&existingID)
|
||||
|
||||
|
||||
if err == sql.ErrNoRows {
|
||||
// Issue doesn't exist - insert it
|
||||
_, err = tx.ExecContext(ctx, `
|
||||
|
||||
Reference in New Issue
Block a user