postgres-ai
diff --git a/‎internal/retrieval/engine/postgres/logical/dump.go
Copy file name to clipboardExpand all lines: internal/retrieval/engine/postgres/logical/dump.go
+4-9Lines changed: 4 additions & 9 deletions b/‎internal/retrieval/engine/postgres/logical/dump.go
Copy file name to clipboardExpand all lines: internal/retrieval/engine/postgres/logical/dump.go
+4-9Lines changed: 4 additions & 9 deletions
@@ -335,7 +335,7 @@ func (d *DumpJob) Run(ctx context.Context) (err error) {
 		}
 	}
 
-	if err := d.cleanupDumpLocation(ctx, dumpCont.ID); err != nil {
+	if err := d.cleanupDumpLocation(ctx, dumpCont.ID, dbList); err != nil {
 		return err
 	}
 
@@ -406,23 +406,18 @@ func (d *DumpJob) getPassword() string {
 	return pwd
 }
 
-func (d *DumpJob) cleanupDumpLocation(ctx context.Context, dumpContID string) error {
+func (d *DumpJob) cleanupDumpLocation(ctx context.Context, dumpContID string, dbList map[string]DumpDefinition) error {
 	if d.DumpOptions.DumpLocation == "" || d.DumpOptions.Restore.Enabled {
 		return nil
 	}
 
-	ls, err := tools.LsContainerDirectory(ctx, d.dockerClient, dumpContID, d.DumpOptions.DumpLocation)
-	if err != nil {
-		return errors.Wrap(err, "failed to clean up dump location")
-	}
-
-	if len(ls) == 0 {
+	if len(dbList) == 0 {
 		return nil
 	}
 
 	cleanupCmd := []string{"rm", "-rf"}
 
-	for _, dbName := range ls {
+	for dbName := range dbList {
 		cleanupCmd = append(cleanupCmd, path.Join(d.DumpOptions.DumpLocation, dbName))
 	}
Original file line number	Diff line number	Diff line change
`@@ -335,7 +335,7 @@ func (d *DumpJob) Run(ctx context.Context) (err error) {`
`335`	`335`	`}`
`336`	`336`	`}`
`337`	`337`
`338`		`- if err := d.cleanupDumpLocation(ctx, dumpCont.ID); err != nil {`
	`338`	`+ if err := d.cleanupDumpLocation(ctx, dumpCont.ID, dbList); err != nil {`
`339`	`339`	`return err`
`340`	`340`	`}`
`341`	`341`
`@@ -406,23 +406,18 @@ func (d *DumpJob) getPassword() string {`
`406`	`406`	`return pwd`
`407`	`407`	`}`
`408`	`408`
`409`		`-func (d *DumpJob) cleanupDumpLocation(ctx context.Context, dumpContID string) error {`
	`409`	`+func (d *DumpJob) cleanupDumpLocation(ctx context.Context, dumpContID string, dbList map[string]DumpDefinition) error {`
`410`	`410`	`if d.DumpOptions.DumpLocation == "" \|\| d.DumpOptions.Restore.Enabled {`
`411`	`411`	`return nil`
`412`	`412`	`}`
`413`	`413`
`414`		`- ls, err := tools.LsContainerDirectory(ctx, d.dockerClient, dumpContID, d.DumpOptions.DumpLocation)`
`415`		`- if err != nil {`
`416`		`- return errors.Wrap(err, "failed to clean up dump location")`
`417`		`- }`
`418`		`-`
`419`		`- if len(ls) == 0 {`
	`414`	`+ if len(dbList) == 0 {`
`420`	`415`	`return nil`
`421`	`416`	`}`
`422`	`417`
`423`	`418`	`cleanupCmd := []string{"rm", "-rf"}`
`424`	`419`
`425`		`- for _, dbName := range ls {`
	`420`	`+ for dbName := range dbList {`
`426`	`421`	`cleanupCmd = append(cleanupCmd, path.Join(d.DumpOptions.DumpLocation, dbName))`
`427`	`422`	`}`
`428`	`423`