/** * <p>createPutAdapter.</p> * * @param config a {@link org.apache.hadoop.conf.Configuration} object. * @param options a {@link com.google.cloud.bigtable.config.BigtableOptions} object. * @return a {@link com.google.cloud.bigtable.hbase.adapters.PutAdapter} object. */ public static PutAdapter createPutAdapter(Configuration config, BigtableOptions options) { boolean setClientTimestamp = !options.getRetryOptions().allowRetriesWithoutTimestamp(); return new PutAdapter(config.getInt("hbase.client.keyvalue.maxsize", -1), setClientTimestamp); }
/** * <p>createPutAdapter.</p> * * @param config a {@link org.apache.hadoop.conf.Configuration} object. * @param options a {@link com.google.cloud.bigtable.config.BigtableOptions} object. * @return a {@link com.google.cloud.bigtable.hbase.adapters.PutAdapter} object. */ public static PutAdapter createPutAdapter(Configuration config, BigtableOptions options) { boolean setClientTimestamp = !options.getRetryOptions().allowRetriesWithoutTimestamp(); return new PutAdapter(config.getInt("hbase.client.keyvalue.maxsize", -1), setClientTimestamp); }
private WatchdogInterceptor setupWatchdog() { Preconditions.checkState(watchdog == null, "Watchdog already setup"); watchdog = new Watchdog(Clock.SYSTEM, options.getRetryOptions().getReadPartialRowTimeoutMillis()); watchdog.start(BigtableSessionSharedThreadPools.getInstance().getRetryExecutor()); return new WatchdogInterceptor( ImmutableSet.<MethodDescriptor<?, ?>>of(BigtableGrpc.getReadRowsMethod()), watchdog); }
.getCredentialsInterceptor(options.getCredentialOptions(), options.getRetryOptions()); if (credentialsInterceptor != null) { interceptorList.add(credentialsInterceptor);
@Test public void testReadWithBigTableOptionsSetsRetryOptions() { final int initialBackoffMillis = -1; BigtableOptions.Builder optionsBuilder = BIGTABLE_OPTIONS.toBuilder(); RetryOptions.Builder retryOptionsBuilder = new RetryOptions.Builder(); retryOptionsBuilder.setInitialBackoffMillis(initialBackoffMillis); optionsBuilder.setRetryOptions(retryOptionsBuilder.build()); BigtableIO.Read read = BigtableIO.read().withBigtableOptions(optionsBuilder.build()); BigtableOptions options = read.getBigtableOptions(); assertEquals(initialBackoffMillis, options.getRetryOptions().getInitialBackoffMillis()); assertThat(options.getRetryOptions(), Matchers.equalTo(retryOptionsBuilder.build())); }
RetryOptions retryOptions = options.getRetryOptions(); CredentialOptions credentialOptions = options.getCredentialOptions();
this.retryOptions = bigtableOptions.getRetryOptions();
Predicates.<CreateTableFromSnapshotRequest>alwaysFalse()); this.retryOptions = bigtableOptions.getRetryOptions(); this.retryExecutorService = retryExecutorService;
@Test public void testWriteWithBigTableOptionsSetsBulkOptionsAndRetryOptions() { final int maxInflightRpcs = 1; final int initialBackoffMillis = -1; BigtableOptions.Builder optionsBuilder = BIGTABLE_OPTIONS.toBuilder(); BulkOptions.Builder bulkOptionsBuilder = new BulkOptions.Builder(); bulkOptionsBuilder.setMaxInflightRpcs(maxInflightRpcs); RetryOptions.Builder retryOptionsBuilder = new RetryOptions.Builder(); retryOptionsBuilder.setInitialBackoffMillis(initialBackoffMillis); optionsBuilder .setBulkOptions(bulkOptionsBuilder.build()) .setRetryOptions(retryOptionsBuilder.build()); BigtableIO.Write write = BigtableIO.write().withBigtableOptions(optionsBuilder.build()); BigtableOptions options = write.getBigtableOptions(); assertEquals(true, options.getBulkOptions().useBulkApi()); assertEquals(maxInflightRpcs, options.getBulkOptions().getMaxInflightRpcs()); assertEquals(initialBackoffMillis, options.getRetryOptions().getInitialBackoffMillis()); assertThat( options.getBulkOptions(), Matchers.equalTo(bulkOptionsBuilder.setUseBulkApi(true).build())); assertThat(options.getRetryOptions(), Matchers.equalTo(retryOptionsBuilder.build())); }