File tree Expand file tree Collapse file tree 3 files changed +21
-2
lines changed
google-cloud-bigquerystorage/src
main/java/com/google/cloud/bigquery/storage/v1
test/java/com/google/cloud/bigquery/storage/v1
samples/snippets/src/main/java/com/example/bigquerystorage Expand file tree Collapse file tree 3 files changed +21
-2
lines changed Original file line number Diff line number Diff line change @@ -318,7 +318,10 @@ static BigQueryWriteSettings getBigQueryWriteSettings(Builder builder) throws IO
318
318
new BigQueryWriteSettings .Builder ()
319
319
.setTransportChannelProvider (
320
320
BigQueryWriteSettings .defaultGrpcTransportProviderBuilder ()
321
- .setChannelsPerCpu (1 )
321
+ .setKeepAliveTime (org .threeten .bp .Duration .ofMinutes (1 ))
322
+ .setKeepAliveTimeout (org .threeten .bp .Duration .ofMinutes (1 ))
323
+ .setKeepAliveWithoutCalls (true )
324
+ .setChannelsPerCpu (2 )
322
325
.build ())
323
326
.setCredentialsProvider (
324
327
BigQueryWriteSettings .defaultCredentialsProviderBuilder ().build ())
Original file line number Diff line number Diff line change @@ -1413,6 +1413,17 @@ public void testBuilderDefaultSetting() throws Exception {
1413
1413
writeSettings .getCredentialsProvider ().toString ());
1414
1414
assertTrue (
1415
1415
writeSettings .getTransportChannelProvider () instanceof InstantiatingGrpcChannelProvider );
1416
+ assertTrue (
1417
+ ((InstantiatingGrpcChannelProvider ) writeSettings .getTransportChannelProvider ())
1418
+ .getKeepAliveWithoutCalls ());
1419
+ assertEquals (
1420
+ ((InstantiatingGrpcChannelProvider ) writeSettings .getTransportChannelProvider ())
1421
+ .getKeepAliveTimeout (),
1422
+ org .threeten .bp .Duration .ofMinutes (1 ));
1423
+ assertEquals (
1424
+ ((InstantiatingGrpcChannelProvider ) writeSettings .getTransportChannelProvider ())
1425
+ .getKeepAliveTime (),
1426
+ org .threeten .bp .Duration .ofMinutes (1 ));
1416
1427
assertEquals (
1417
1428
BigQueryWriteSettings .getDefaultEndpoint (), writeSettings .getEndpoint ().toString ());
1418
1429
}
Original file line number Diff line number Diff line change 21
21
import com .google .api .core .ApiFuture ;
22
22
import com .google .api .core .ApiFutureCallback ;
23
23
import com .google .api .core .ApiFutures ;
24
+ import com .google .api .gax .core .FixedExecutorProvider ;
24
25
import com .google .cloud .bigquery .BigQuery ;
25
26
import com .google .cloud .bigquery .BigQueryOptions ;
26
27
import com .google .cloud .bigquery .QueryJobConfiguration ;
39
40
import io .grpc .Status .Code ;
40
41
import java .io .IOException ;
41
42
import java .util .Map ;
43
+ import java .util .concurrent .Executors ;
42
44
import java .util .concurrent .Phaser ;
43
45
import java .util .concurrent .atomic .AtomicInteger ;
44
46
import javax .annotation .concurrent .GuardedBy ;
@@ -151,7 +153,10 @@ public void initialize(TableName parentTable)
151
153
// For more information about JsonStreamWriter, see:
152
154
// https://googleapis.dev/java/google-cloud-bigquerystorage/latest/com/google/cloud/bigquery/storage/v1/JsonStreamWriter.html
153
155
streamWriter =
154
- JsonStreamWriter .newBuilder (parentTable .toString (), BigQueryWriteClient .create ()).build ();
156
+ JsonStreamWriter .newBuilder (parentTable .toString (), BigQueryWriteClient .create ())
157
+ .setExecutorProvider (
158
+ FixedExecutorProvider .create (Executors .newScheduledThreadPool (100 )))
159
+ .build ();
155
160
}
156
161
157
162
public void append (AppendContext appendContext )
You can’t perform that action at this time.
0 commit comments