Skip to content

Commit

Permalink
add back public methods
Browse files Browse the repository at this point in the history
  • Loading branch information
Abacn committed Sep 13, 2024
1 parent d751704 commit 6c2ce5b
Showing 1 changed file with 15 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -1119,6 +1119,21 @@ public Write withFlowControl(boolean enableFlowControl) {
.build();
}

/** Returns a new {@link BigtableIO.Write} with client side latency based throttling enabled. */
public Write withThrottlingTargetMs(int throttlingTargetMs) {

BigtableWriteOptions options = getBigtableWriteOptions();
return toBuilder()
.setBigtableWriteOptions(
options.toBuilder().setThrottlingTargetMs(throttlingTargetMs).build())
.build();
}

/** This configuration is removed in Beam 2.60.0, Do not use. */
public Write withThrottlingReportTargetMs(int throttlingReportTargetMs) {
throw new UnsupportedOperationException("withThrottlingReportTargetMs is removed");
}

public Write withErrorHandler(ErrorHandler<BadRecord, ?> badRecordErrorHandler) {
return toBuilder()
.setBadRecordErrorHandler(badRecordErrorHandler)
Expand Down

0 comments on commit 6c2ce5b

Please sign in to comment.