diff --git a/writers/batchwriter/batchwriter.go b/writers/batchwriter/batchwriter.go index af1b0bcb23..ad85b9fa9f 100644 --- a/writers/batchwriter/batchwriter.go +++ b/writers/batchwriter/batchwriter.go @@ -107,7 +107,10 @@ func (w *BatchWriter) Flush(ctx context.Context) error { if err := w.flushMigrateTables(ctx); err != nil { return err } - return w.flushDeleteStaleTables(ctx) + if err := w.flushDeleteStaleTables(ctx); err != nil { + return err + } + return w.flushDeleteRecordTables(ctx) } func (w *BatchWriter) Close(context.Context) error { @@ -310,6 +313,9 @@ func (w *BatchWriter) Write(ctx context.Context, msgs <-chan message.WriteMessag if err := w.flushDeleteStaleTables(ctx); err != nil { return err } + if err := w.flushDeleteRecordTables(ctx); err != nil { + return err + } if err := w.startWorker(ctx, m); err != nil { return err }