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 16, 2024
1 parent a08b0fd commit 42e28b7
Showing 1 changed file with 14 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -1124,6 +1124,20 @@ 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 42e28b7

Please sign in to comment.