builder .createInstanceOperationSettings() .setInitialCallSettings( UnaryCallSettings .<CreateInstanceRequest, OperationSnapshot>newUnaryCallSettingsBuilder() .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("default")) .build()) .setResponseTransformer( ProtoOperationTransformers.ResponseTransformer.create(Instance.class)) .setMetadataTransformer( ProtoOperationTransformers.MetadataTransformer.create(CreateInstanceMetadata.class)) .setPollingAlgorithm( OperationTimedPollAlgorithm.create( RetrySettings.newBuilder() builder .partialUpdateInstanceOperationSettings() .setInitialCallSettings( UnaryCallSettings .<PartialUpdateInstanceRequest, OperationSnapshot>newUnaryCallSettingsBuilder() .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("default")) .build()) .setResponseTransformer( ProtoOperationTransformers.ResponseTransformer.create(Instance.class)) .setMetadataTransformer( ProtoOperationTransformers.MetadataTransformer.create(UpdateInstanceMetadata.class)) .setPollingAlgorithm(
builder .importProductSetsOperationSettings() .setInitialCallSettings( UnaryCallSettings .<ImportProductSetsRequest, OperationSnapshot>newUnaryCallSettingsBuilder() .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("default")) .build()) .setResponseTransformer( ProtoOperationTransformers.ResponseTransformer.create( ImportProductSetsResponse.class)) .setMetadataTransformer( ProtoOperationTransformers.MetadataTransformer.create(BatchOperationMetadata.class)) .setPollingAlgorithm( OperationTimedPollAlgorithm.create( RetrySettings.newBuilder()
builder .createDatabaseOperationSettings() .setInitialCallSettings( UnaryCallSettings .<CreateDatabaseRequest, OperationSnapshot>newUnaryCallSettingsBuilder() .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("default")) .build()) .setResponseTransformer( ProtoOperationTransformers.ResponseTransformer.create(Database.class)) .setMetadataTransformer( ProtoOperationTransformers.MetadataTransformer.create(CreateDatabaseMetadata.class)) .setPollingAlgorithm( OperationTimedPollAlgorithm.create( RetrySettings.newBuilder() builder .updateDatabaseDdlOperationSettings() .setInitialCallSettings( UnaryCallSettings .<UpdateDatabaseDdlRequest, OperationSnapshot>newUnaryCallSettingsBuilder() .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("default")) .build()) .setResponseTransformer( ProtoOperationTransformers.ResponseTransformer.create(Empty.class)) .setMetadataTransformer( ProtoOperationTransformers.MetadataTransformer.create( UpdateDatabaseDdlMetadata.class))
builder .batchUpdateIntentsOperationSettings() .setInitialCallSettings( UnaryCallSettings .<BatchUpdateIntentsRequest, OperationSnapshot>newUnaryCallSettingsBuilder() .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("default")) .build()) .setResponseTransformer( ProtoOperationTransformers.ResponseTransformer.create( BatchUpdateIntentsResponse.class)) .setMetadataTransformer( ProtoOperationTransformers.MetadataTransformer.create(Struct.class)) .setPollingAlgorithm( OperationTimedPollAlgorithm.create( RetrySettings.newBuilder() builder .batchDeleteIntentsOperationSettings() .setInitialCallSettings( UnaryCallSettings .<BatchDeleteIntentsRequest, OperationSnapshot>newUnaryCallSettingsBuilder() .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("default")) .build()) .setResponseTransformer( ProtoOperationTransformers.ResponseTransformer.create(Empty.class)) .setMetadataTransformer( ProtoOperationTransformers.MetadataTransformer.create(Struct.class))
builder .instantiateWorkflowTemplateOperationSettings() .setInitialCallSettings( UnaryCallSettings .<InstantiateWorkflowTemplateRequest, OperationSnapshot> .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("default")) .build()) .setResponseTransformer( ProtoOperationTransformers.ResponseTransformer.create(Empty.class)) .setMetadataTransformer( ProtoOperationTransformers.MetadataTransformer.create(WorkflowMetadata.class)) .setPollingAlgorithm( OperationTimedPollAlgorithm.create( RetrySettings.newBuilder() builder .instantiateInlineWorkflowTemplateOperationSettings() .setInitialCallSettings( UnaryCallSettings .<InstantiateInlineWorkflowTemplateRequest, OperationSnapshot> .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("default")) .build()) .setResponseTransformer( ProtoOperationTransformers.ResponseTransformer.create(Empty.class)) .setMetadataTransformer( ProtoOperationTransformers.MetadataTransformer.create(WorkflowMetadata.class)) .setPollingAlgorithm(
builder .instantiateWorkflowTemplateOperationSettings() .setInitialCallSettings( UnaryCallSettings .<InstantiateWorkflowTemplateRequest, OperationSnapshot> .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("default")) .build()) .setResponseTransformer( ProtoOperationTransformers.ResponseTransformer.create(Empty.class)) .setMetadataTransformer( ProtoOperationTransformers.MetadataTransformer.create(WorkflowMetadata.class)) .setPollingAlgorithm( OperationTimedPollAlgorithm.create( RetrySettings.newBuilder() builder .instantiateInlineWorkflowTemplateOperationSettings() .setInitialCallSettings( UnaryCallSettings .<InstantiateInlineWorkflowTemplateRequest, OperationSnapshot> .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("default")) .build()) .setResponseTransformer( ProtoOperationTransformers.ResponseTransformer.create(Empty.class)) .setMetadataTransformer( ProtoOperationTransformers.MetadataTransformer.create(WorkflowMetadata.class)) .setPollingAlgorithm(
builder .batchUpdateIntentsOperationSettings() .setInitialCallSettings( UnaryCallSettings .<BatchUpdateIntentsRequest, OperationSnapshot>newUnaryCallSettingsBuilder() .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("default")) .build()) .setResponseTransformer( ProtoOperationTransformers.ResponseTransformer.create( BatchUpdateIntentsResponse.class)) .setMetadataTransformer( ProtoOperationTransformers.MetadataTransformer.create(Struct.class)) .setPollingAlgorithm( OperationTimedPollAlgorithm.create( RetrySettings.newBuilder() builder .batchDeleteIntentsOperationSettings() .setInitialCallSettings( UnaryCallSettings .<BatchDeleteIntentsRequest, OperationSnapshot>newUnaryCallSettingsBuilder() .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("default")) .build()) .setResponseTransformer( ProtoOperationTransformers.ResponseTransformer.create(Empty.class)) .setMetadataTransformer( ProtoOperationTransformers.MetadataTransformer.create(Struct.class))
builder .createDocumentOperationSettings() .setInitialCallSettings( UnaryCallSettings .<CreateDocumentRequest, OperationSnapshot>newUnaryCallSettingsBuilder() .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("default")) .build()) .setResponseTransformer( ProtoOperationTransformers.ResponseTransformer.create(Document.class)) .setMetadataTransformer( ProtoOperationTransformers.MetadataTransformer.create( KnowledgeOperationMetadata.class)) .setPollingAlgorithm( OperationTimedPollAlgorithm.create( RetrySettings.newBuilder() builder .deleteDocumentOperationSettings() .setInitialCallSettings( UnaryCallSettings .<DeleteDocumentRequest, OperationSnapshot>newUnaryCallSettingsBuilder() .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("default")) .build()) .setResponseTransformer( ProtoOperationTransformers.ResponseTransformer.create(Empty.class)) .setMetadataTransformer( ProtoOperationTransformers.MetadataTransformer.create(
.build(), clock); callSettings = callSettings.toBuilder().setPollingAlgorithm(pollingAlgorithm).build();
.setInitialCallSettings(initialCallSettings) .setResponseTransformer( ProtoOperationTransformers.ResponseTransformer.create(Color.class)) .setMetadataTransformer( ProtoOperationTransformers.MetadataTransformer.create(Money.class)) .setPollingAlgorithm(pollingAlgorithm) .build();
.build(), clock); callSettings = callSettings.toBuilder().setPollingAlgorithm(pollingAlgorithm).build();
builder .asyncBatchAnnotateFilesOperationSettings() .setInitialCallSettings( UnaryCallSettings .<AsyncBatchAnnotateFilesRequest, OperationSnapshot>newUnaryCallSettingsBuilder() .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("default")) .build()) .setResponseTransformer( ProtoOperationTransformers.ResponseTransformer.create( AsyncBatchAnnotateFilesResponse.class)) .setMetadataTransformer( ProtoOperationTransformers.MetadataTransformer.create(OperationMetadata.class)) .setPollingAlgorithm( OperationTimedPollAlgorithm.create( RetrySettings.newBuilder()
builder .asyncBatchAnnotateFilesOperationSettings() .setInitialCallSettings( UnaryCallSettings .<AsyncBatchAnnotateFilesRequest, OperationSnapshot>newUnaryCallSettingsBuilder() .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("default")) .build()) .setResponseTransformer( ProtoOperationTransformers.ResponseTransformer.create( AsyncBatchAnnotateFilesResponse.class)) .setMetadataTransformer( ProtoOperationTransformers.MetadataTransformer.create(OperationMetadata.class)) .setPollingAlgorithm( OperationTimedPollAlgorithm.create( RetrySettings.newBuilder()
builder .asyncBatchAnnotateFilesOperationSettings() .setInitialCallSettings( UnaryCallSettings .<AsyncBatchAnnotateFilesRequest, OperationSnapshot>newUnaryCallSettingsBuilder() .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("default")) .build()) .setResponseTransformer( ProtoOperationTransformers.ResponseTransformer.create( AsyncBatchAnnotateFilesResponse.class)) .setMetadataTransformer( ProtoOperationTransformers.MetadataTransformer.create(OperationMetadata.class)) .setPollingAlgorithm( OperationTimedPollAlgorithm.create( RetrySettings.newBuilder()
@Test public void testFutureCallPollCancelOnLongTimeoutExceeded() throws Exception { final int iterationsCount = 1000; String opName = "testFutureCallPollCancelOnLongTimeoutExceeded"; OperationSnapshot initialOperation = getOperation(opName, null, null, null, false); OperationSnapshot[] pollOperations = new OperationSnapshot[iterationsCount]; for (int i = 0; i < iterationsCount; i++) { pollOperations[i] = getOperation(opName, null, null, null, false); } UnaryCallable<Integer, OperationSnapshot> initialCallable = mockGetOpSnapshotCallable(StatusCode.Code.OK, initialOperation); LongRunningClient longRunningClient = mockGetOperation(StatusCode.Code.OK, pollOperations); pollingAlgorithm = OperationTimedPollAlgorithm.create( FAST_RECHECKING_SETTINGS.toBuilder().setTotalTimeout(Duration.ofMillis(1000L)).build(), clock); callSettings = callSettings.toBuilder().setPollingAlgorithm(pollingAlgorithm).build(); OperationCallable<Integer, Color, Currency> callable = FakeCallableFactory.createOperationCallable( initialCallable, callSettings, initialContext, longRunningClient); OperationFuture<Color, Currency> future = callable.futureCall(2, FakeCallContext.createDefault()); assertFutureCancelMetaCancel(future); assertThat(executor.getIterationsCount()).isEqualTo(iterationsCount); }
@Test public void testBuilderFromSettings() throws Exception { OperationCallSettings.Builder<Integer, String, Long> builder = OperationCallSettings.newBuilder(); UnaryCallSettings<Integer, OperationSnapshot> initialCallSettings = UnaryCallSettings.<Integer, OperationSnapshot>newUnaryCallSettingsBuilder() .setRetryableCodes(Code.UNAVAILABLE) .build(); TimedRetryAlgorithm pollingAlgorithm = Mockito.mock(TimedRetryAlgorithm.class); ResponseTransformer responseTransformer = new ResponseTransformer(); MetadataTransformer metadataTransformer = new MetadataTransformer(); builder.setPollingAlgorithm(pollingAlgorithm); builder.setResponseTransformer(responseTransformer); builder.setMetadataTransformer(metadataTransformer); builder.setInitialCallSettings(initialCallSettings); Truth.assertThat(builder.getInitialCallSettings()).isSameAs(initialCallSettings); OperationCallSettings settings = builder.build(); OperationCallSettings.Builder newBuilder = settings.toBuilder(); Truth.assertThat(newBuilder.getPollingAlgorithm()).isSameAs(pollingAlgorithm); Truth.assertThat(newBuilder.getResponseTransformer()).isSameAs(responseTransformer); Truth.assertThat(newBuilder.getMetadataTransformer()).isSameAs(metadataTransformer); Truth.assertThat(newBuilder.getInitialCallSettings()).isNotNull(); Truth.assertThat(newBuilder.getInitialCallSettings().getRetryableCodes().size()).isEqualTo(1); }
@Test public void testBuilder() { OperationCallSettings.Builder<Integer, String, Long> builder = OperationCallSettings.newBuilder(); UnaryCallSettings<Integer, OperationSnapshot> initialCallSettings = UnaryCallSettings.<Integer, OperationSnapshot>newUnaryCallSettingsBuilder() .setRetryableCodes(Code.UNAVAILABLE) .build(); TimedRetryAlgorithm pollingAlgorithm = Mockito.mock(TimedRetryAlgorithm.class); ResponseTransformer responseTransformer = new ResponseTransformer(); MetadataTransformer metadataTransformer = new MetadataTransformer(); builder.setPollingAlgorithm(pollingAlgorithm); builder.setResponseTransformer(responseTransformer); builder.setMetadataTransformer(metadataTransformer); builder.setInitialCallSettings(initialCallSettings); Truth.assertThat(builder.getInitialCallSettings()).isSameAs(initialCallSettings); OperationCallSettings settings = builder.build(); Truth.assertThat(settings.getPollingAlgorithm()).isSameAs(pollingAlgorithm); Truth.assertThat(settings.getResponseTransformer()).isSameAs(responseTransformer); Truth.assertThat(settings.getMetadataTransformer()).isSameAs(metadataTransformer); Truth.assertThat(settings.getInitialCallSettings()).isNotNull(); Truth.assertThat(settings.getInitialCallSettings().getRetryableCodes().size()).isEqualTo(1); }
@Before public void setUp() throws IOException { initialChannel = mock(FakeChannel.class); pollTransportChannel = mock(TransportChannel.class); TransportChannelProvider operationsChannelProvider = mock(TransportChannelProvider.class); when(operationsChannelProvider.getTransportChannel()).thenReturn(pollTransportChannel); clock = new FakeApiClock(0L); executor = RecordingScheduler.create(clock); pollingAlgorithm = OperationTimedPollAlgorithm.create(FAST_RECHECKING_SETTINGS, clock); UnaryCallSettings<Integer, OperationSnapshot> initialCallSettings = UnaryCallSettings.<Integer, OperationSnapshot>newUnaryCallSettingsBuilder() .setRetrySettings(FAST_RETRY_SETTINGS.toBuilder().setMaxAttempts(1).build()) .build(); callSettings = OperationCallSettings.<Integer, Color, Currency>newBuilder() .setInitialCallSettings(initialCallSettings) .setResponseTransformer(new ResponseTransformer()) .setMetadataTransformer(new MetadataTransformer()) .setPollingAlgorithm(pollingAlgorithm) .build(); initialContext = getClientContext(initialChannel, executor); }
protected BigtableTableAdminStubSettings(Builder settingsBuilder) throws IOException { super(settingsBuilder); createTableSettings = settingsBuilder.createTableSettings().build(); createTableFromSnapshotSettings = settingsBuilder.createTableFromSnapshotSettings().build(); createTableFromSnapshotOperationSettings = settingsBuilder.createTableFromSnapshotOperationSettings().build(); listTablesSettings = settingsBuilder.listTablesSettings().build(); getTableSettings = settingsBuilder.getTableSettings().build(); deleteTableSettings = settingsBuilder.deleteTableSettings().build(); modifyColumnFamiliesSettings = settingsBuilder.modifyColumnFamiliesSettings().build(); dropRowRangeSettings = settingsBuilder.dropRowRangeSettings().build(); generateConsistencyTokenSettings = settingsBuilder.generateConsistencyTokenSettings().build(); checkConsistencySettings = settingsBuilder.checkConsistencySettings().build(); snapshotTableSettings = settingsBuilder.snapshotTableSettings().build(); snapshotTableOperationSettings = settingsBuilder.snapshotTableOperationSettings().build(); getSnapshotSettings = settingsBuilder.getSnapshotSettings().build(); listSnapshotsSettings = settingsBuilder.listSnapshotsSettings().build(); deleteSnapshotSettings = settingsBuilder.deleteSnapshotSettings().build(); }
protected WorkflowTemplateServiceStubSettings(Builder settingsBuilder) throws IOException { super(settingsBuilder); createWorkflowTemplateSettings = settingsBuilder.createWorkflowTemplateSettings().build(); getWorkflowTemplateSettings = settingsBuilder.getWorkflowTemplateSettings().build(); instantiateWorkflowTemplateSettings = settingsBuilder.instantiateWorkflowTemplateSettings().build(); instantiateWorkflowTemplateOperationSettings = settingsBuilder.instantiateWorkflowTemplateOperationSettings().build(); instantiateInlineWorkflowTemplateSettings = settingsBuilder.instantiateInlineWorkflowTemplateSettings().build(); instantiateInlineWorkflowTemplateOperationSettings = settingsBuilder.instantiateInlineWorkflowTemplateOperationSettings().build(); updateWorkflowTemplateSettings = settingsBuilder.updateWorkflowTemplateSettings().build(); listWorkflowTemplatesSettings = settingsBuilder.listWorkflowTemplatesSettings().build(); deleteWorkflowTemplateSettings = settingsBuilder.deleteWorkflowTemplateSettings().build(); }