Skip to content

Commit

Permalink
chore: add logging when flushing data
Browse files Browse the repository at this point in the history
  • Loading branch information
gintarasm committed Feb 6, 2024
1 parent 08c3a08 commit 205b80b
Showing 1 changed file with 2 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -99,8 +99,10 @@ public void close() {

@Override
public void flush(boolean endOfInput) {
logger.info("Flushing BigQuery writer {} data. Inflight request count {}", this.sinkInitContext.getSubtaskId(), inflightRequestCount.getRegisteredParties());
checkAsyncException();
inflightRequestCount.arriveAndAwaitAdvance();
logger.info("BigQuery writer {} data flushed. Inflight request count {}", this.sinkInitContext.getSubtaskId(), inflightRequestCount.getRegisteredParties());
}

static class AppendCallBack<A> implements ApiFutureCallback<AppendRowsResponse> {
Expand Down

0 comments on commit 205b80b

Please sign in to comment.