-
Notifications
You must be signed in to change notification settings - Fork 2
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Improve storage classes and add test using bigquery emulator docker c…
…ontainer
- Loading branch information
1 parent
c4b9c45
commit 1ed4e17
Showing
15 changed files
with
414 additions
and
297 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
100 changes: 100 additions & 0 deletions
100
...ium-server-bigquery-sinks/src/main/java/io/debezium/server/bigquery/StreamDataWriter.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,100 @@ | ||
| package io.debezium.server.bigquery; | ||
|
|
||
| import com.google.api.core.ApiFuture; | ||
| import com.google.api.gax.core.FixedExecutorProvider; | ||
| import com.google.api.gax.grpc.InstantiatingGrpcChannelProvider; | ||
| import com.google.api.gax.retrying.RetrySettings; | ||
| import com.google.cloud.bigquery.storage.v1.*; | ||
| import com.google.protobuf.Descriptors.DescriptorValidationException; | ||
| import io.debezium.DebeziumException; | ||
| import org.json.JSONArray; | ||
| import org.threeten.bp.Duration; | ||
|
|
||
| import java.io.IOException; | ||
| import java.util.concurrent.Executors; | ||
| import java.util.concurrent.atomic.AtomicInteger; | ||
|
|
||
|
|
||
| public class StreamDataWriter { | ||
| private static final int MAX_RECREATE_COUNT = 3; | ||
| private final BigQueryWriteClient client; | ||
| private final Boolean ignoreUnknownFields; | ||
| private final InstantiatingGrpcChannelProvider instantiatingGrpcChannelProvider; | ||
| private final TableName parentTable; | ||
| private final Object lock = new Object(); | ||
| JsonStreamWriter streamWriter; | ||
| private AtomicInteger recreateCount = new AtomicInteger(0); | ||
|
|
||
|
|
||
| public StreamDataWriter(TableName parentTable, BigQueryWriteClient client, | ||
| Boolean ignoreUnknownFields, InstantiatingGrpcChannelProvider instantiatingGrpcChannelProvider) | ||
| throws DescriptorValidationException, IOException, InterruptedException { | ||
| this.client = client; | ||
| this.ignoreUnknownFields = ignoreUnknownFields; | ||
| this.instantiatingGrpcChannelProvider = instantiatingGrpcChannelProvider; | ||
| this.parentTable = parentTable; | ||
| } | ||
|
|
||
| public void initialize() | ||
| throws DescriptorValidationException, IOException, InterruptedException { | ||
| streamWriter = createStreamWriter(this.parentTable.toString()); | ||
| } | ||
|
|
||
| private JsonStreamWriter createStreamWriter(String parentTableName) | ||
| throws DescriptorValidationException, IOException, InterruptedException { | ||
| // https://cloud.google.com/bigquery/docs/write-api-streaming | ||
| // Configure in-stream automatic retry settings. | ||
| // Error codes that are immediately retried: | ||
| // * ABORTED, UNAVAILABLE, CANCELLED, INTERNAL, DEADLINE_EXCEEDED | ||
| // Error codes that are retried with exponential backoff: | ||
| // * RESOURCE_EXHAUSTED | ||
| RetrySettings retrySettings = | ||
| RetrySettings.newBuilder() | ||
| .setInitialRetryDelay(Duration.ofMillis(500)) | ||
| .setRetryDelayMultiplier(1.1) | ||
| .setMaxAttempts(5) | ||
| .setMaxRetryDelay(Duration.ofMinutes(1)) | ||
| .build(); | ||
| // Use the JSON stream writer to send records in JSON format. Specify the table name to write | ||
| // to the default stream. | ||
| // For more information about JsonStreamWriter, see: | ||
| // https://googleapis.dev/java/google-cloud-bigquerystorage/latest/com/google/cloud/bigquery/storage/v1/JsonStreamWriter.html | ||
| return JsonStreamWriter.newBuilder(parentTableName, client) | ||
| .setIgnoreUnknownFields(ignoreUnknownFields) | ||
| .setExecutorProvider(FixedExecutorProvider.create(Executors.newScheduledThreadPool(100))) | ||
| .setChannelProvider(instantiatingGrpcChannelProvider) | ||
| .setEnableConnectionPool(true) | ||
| // If value is missing in json and there is a default value configured on bigquery | ||
| // column, apply the default value to the missing value field. | ||
| .setDefaultMissingValueInterpretation(AppendRowsRequest.MissingValueInterpretation.DEFAULT_VALUE) | ||
| .setRetrySettings(retrySettings) | ||
| .build(); | ||
| } | ||
|
|
||
|
|
||
| public void appendSync(JSONArray data) throws DescriptorValidationException, IOException { | ||
| try { | ||
| synchronized (this.lock) { | ||
| if (!streamWriter.isUserClosed() && streamWriter.isClosed() && recreateCount.getAndIncrement() < MAX_RECREATE_COUNT) { | ||
| streamWriter = createStreamWriter(streamWriter.getStreamName()); | ||
| } | ||
| } | ||
|
|
||
| ApiFuture<AppendRowsResponse> future = streamWriter.append(data); | ||
| AppendRowsResponse response = future.get(); | ||
| if (response.hasError()) { | ||
| throw new DebeziumException("Failed to append data to stream. Error Code:" + response.getError().getCode() + " Error Message:" + response.getError().getMessage()); | ||
| } | ||
| } catch (Exception throwable) { | ||
| throw new DebeziumException("Failed to append data to stream " + streamWriter.getStreamName() + "\n" + throwable.getMessage(), | ||
| throwable); | ||
| } | ||
| } | ||
|
|
||
| public void close(BigQueryWriteClient client) { | ||
| if (streamWriter != null) { | ||
| streamWriter.close(); | ||
| client.finalizeWriteStream(streamWriter.getStreamName()); | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.