fix datapasta_clone leakage

ProlificLabs · Jun 19, 2023 · c37171d · c37171d
1 parent 310cccc
commit c37171d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/postgres.go b/postgres.go
@@ -157,6 +157,11 @@ func (db pgbatchtx) Insert(fkm ForeignKeyMapper, rows ...map[string]any) error {
 		return err
 	}
 
+	//clear out the temp table of any rows in case db connection is pooled and reused.
+	if _, err := db.tx.db.Exec(db.ctx, "TRUNCATE datapasta_clone"); err != nil {
+		return err
+	}
+
 	start := time.Now()
 
 	batch := &pgx.Batch{}