feat: Revised implementation

* Checked table object initialization order * Added support end-to-end FK/PK references
GreenmaskIO · Nov 1, 2024 · 29f466f · 29f466f
1 parent 40274fe
commit 29f466f
Show file tree

Hide file tree

Showing 10 changed files with 323 additions and 103 deletions.
diff --git a/internal/db/postgres/cmd/dump.go b/internal/db/postgres/cmd/dump.go
@@ -288,6 +288,9 @@ func (d *Dump) taskProducer(ctx context.Context, tasks chan<- dumpers.DumpTask)
 			var task dumpers.DumpTask
 			switch v := dumpObj.(type) {
 			case *entries.Table:
+				if v.RelKind == 'p' {
+					continue
+				}
 				task = dumpers.NewTableDumper(v, d.validate, d.validateRowsLimit, d.pgDumpOptions.Pgzip)
 			case *entries.Sequence:
 				task = dumpers.NewSequenceDumper(v)
@@ -355,7 +358,7 @@ func (d *Dump) setDumpDependenciesGraph(tables []*entries.Table) {
 	d.dumpDependenciesGraph = make(map[int32][]int32)
 	for _, oid := range sortedOids {
 		idx := slices.IndexFunc(tables, func(entry *entries.Table) bool {
-			return entry.Oid == oid
+			return entry.Oid == oid || entry.RootPtOid == oid
 		})
 		if idx == -1 {
 			panic(fmt.Sprintf("table not found: oid=%d", oid))