Skip to content
This repository was archived by the owner on Sep 16, 2023. It is now read-only.

feat: new parameters in FlexTemplateRuntimeEnvironment fix: added google.api.http annotations to RPCs #122

Merged
gcf-merge-on-green merged 2 commits into from
Feb 3, 2022
Merged

feat: new parameters in FlexTemplateRuntimeEnvironment fix: added google.api.http annotations to RPCs #122

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
b111e23
feat: new parameters in FlexTemplateRuntimeEnvironment fix: added goo…
gcf-owl-bot[bot] Feb 1, 2022
ef54d36
🦉 Updates from OwlBot
gcf-owl-bot[bot] Feb 1, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 3 additions & 1 deletion google-cloud-dataflow/src/main/java/com/google/dataflow/v1beta3/package-info.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,9 @@
*/

/**
* The interfaces provided are listed below, along with usage samples.
* A client to Dataflow API
*
* <p>The interfaces provided are listed below, along with usage samples.
*
* <p>======================= JobsV1Beta3Client =======================
*
Expand Down
8 changes: 8 additions & 0 deletions ...dataflow/src/main/java/com/google/dataflow/v1beta3/stub/GrpcFlexTemplatesServiceStub.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@
import com.google.api.gax.grpc.GrpcStubCallableFactory;
import com.google.api.gax.rpc.ClientContext;
import com.google.api.gax.rpc.UnaryCallable;
import com.google.common.collect.ImmutableMap;
import com.google.dataflow.v1beta3.LaunchFlexTemplateRequest;
import com.google.dataflow.v1beta3.LaunchFlexTemplateResponse;
import com.google.longrunning.stub.GrpcOperationsStub;
Expand Down Expand Up @@ -103,6 +104,13 @@ protected GrpcFlexTemplatesServiceStub(
launchFlexTemplateTransportSettings =
GrpcCallSettings.<LaunchFlexTemplateRequest, LaunchFlexTemplateResponse>newBuilder()
.setMethodDescriptor(launchFlexTemplateMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
params.put("location", String.valueOf(request.getLocation()));
params.put("project_id", String.valueOf(request.getProjectId()));
return params.build();
})
.build();

this.launchFlexTemplateCallable =
Expand Down
45 changes: 45 additions & 0 deletions ...le-cloud-dataflow/src/main/java/com/google/dataflow/v1beta3/stub/GrpcJobsV1Beta3Stub.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@
import com.google.api.gax.grpc.GrpcStubCallableFactory;
import com.google.api.gax.rpc.ClientContext;
import com.google.api.gax.rpc.UnaryCallable;
import com.google.common.collect.ImmutableMap;
import com.google.dataflow.v1beta3.CheckActiveJobsRequest;
import com.google.dataflow.v1beta3.CheckActiveJobsResponse;
import com.google.dataflow.v1beta3.CreateJobRequest;
Expand Down Expand Up @@ -170,22 +171,58 @@ protected GrpcJobsV1Beta3Stub(
GrpcCallSettings<CreateJobRequest, Job> createJobTransportSettings =
GrpcCallSettings.<CreateJobRequest, Job>newBuilder()
.setMethodDescriptor(createJobMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
params.put("location", String.valueOf(request.getLocation()));
params.put("project_id", String.valueOf(request.getProjectId()));
return params.build();
})
.build();
GrpcCallSettings<GetJobRequest, Job> getJobTransportSettings =
GrpcCallSettings.<GetJobRequest, Job>newBuilder()
.setMethodDescriptor(getJobMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
params.put("job_id", String.valueOf(request.getJobId()));
params.put("location", String.valueOf(request.getLocation()));
params.put("project_id", String.valueOf(request.getProjectId()));
return params.build();
})
.build();
GrpcCallSettings<UpdateJobRequest, Job> updateJobTransportSettings =
GrpcCallSettings.<UpdateJobRequest, Job>newBuilder()
.setMethodDescriptor(updateJobMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
params.put("job_id", String.valueOf(request.getJobId()));
params.put("location", String.valueOf(request.getLocation()));
params.put("project_id", String.valueOf(request.getProjectId()));
return params.build();
})
.build();
GrpcCallSettings<ListJobsRequest, ListJobsResponse> listJobsTransportSettings =
GrpcCallSettings.<ListJobsRequest, ListJobsResponse>newBuilder()
.setMethodDescriptor(listJobsMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
params.put("location", String.valueOf(request.getLocation()));
params.put("project_id", String.valueOf(request.getProjectId()));
return params.build();
})
.build();
GrpcCallSettings<ListJobsRequest, ListJobsResponse> aggregatedListJobsTransportSettings =
GrpcCallSettings.<ListJobsRequest, ListJobsResponse>newBuilder()
.setMethodDescriptor(aggregatedListJobsMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
params.put("project_id", String.valueOf(request.getProjectId()));
return params.build();
})
.build();
GrpcCallSettings<CheckActiveJobsRequest, CheckActiveJobsResponse>
checkActiveJobsTransportSettings =
Expand All @@ -195,6 +232,14 @@ protected GrpcJobsV1Beta3Stub(
GrpcCallSettings<SnapshotJobRequest, Snapshot> snapshotJobTransportSettings =
GrpcCallSettings.<SnapshotJobRequest, Snapshot>newBuilder()
.setMethodDescriptor(snapshotJobMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
params.put("job_id", String.valueOf(request.getJobId()));
params.put("location", String.valueOf(request.getLocation()));
params.put("project_id", String.valueOf(request.getProjectId()));
return params.build();
})
.build();

this.createJobCallable =
Expand Down
9 changes: 9 additions & 0 deletions ...loud-dataflow/src/main/java/com/google/dataflow/v1beta3/stub/GrpcMessagesV1Beta3Stub.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@
import com.google.api.gax.grpc.GrpcStubCallableFactory;
import com.google.api.gax.rpc.ClientContext;
import com.google.api.gax.rpc.UnaryCallable;
import com.google.common.collect.ImmutableMap;
import com.google.dataflow.v1beta3.ListJobMessagesRequest;
import com.google.dataflow.v1beta3.ListJobMessagesResponse;
import com.google.longrunning.stub.GrpcOperationsStub;
Expand Down Expand Up @@ -107,6 +108,14 @@ protected GrpcMessagesV1Beta3Stub(
listJobMessagesTransportSettings =
GrpcCallSettings.<ListJobMessagesRequest, ListJobMessagesResponse>newBuilder()
.setMethodDescriptor(listJobMessagesMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
params.put("job_id", String.valueOf(request.getJobId()));
params.put("location", String.valueOf(request.getLocation()));
params.put("project_id", String.valueOf(request.getProjectId()));
return params.build();
})
.build();

this.listJobMessagesCallable =
Expand Down
26 changes: 26 additions & 0 deletions ...cloud-dataflow/src/main/java/com/google/dataflow/v1beta3/stub/GrpcMetricsV1Beta3Stub.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@
import com.google.api.gax.grpc.GrpcStubCallableFactory;
import com.google.api.gax.rpc.ClientContext;
import com.google.api.gax.rpc.UnaryCallable;
import com.google.common.collect.ImmutableMap;
import com.google.dataflow.v1beta3.GetJobExecutionDetailsRequest;
import com.google.dataflow.v1beta3.GetJobMetricsRequest;
import com.google.dataflow.v1beta3.GetStageExecutionDetailsRequest;
Expand Down Expand Up @@ -138,16 +139,41 @@ protected GrpcMetricsV1Beta3Stub(
GrpcCallSettings<GetJobMetricsRequest, JobMetrics> getJobMetricsTransportSettings =
GrpcCallSettings.<GetJobMetricsRequest, JobMetrics>newBuilder()
.setMethodDescriptor(getJobMetricsMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
params.put("job_id", String.valueOf(request.getJobId()));
params.put("location", String.valueOf(request.getLocation()));
params.put("project_id", String.valueOf(request.getProjectId()));
return params.build();
})
.build();
GrpcCallSettings<GetJobExecutionDetailsRequest, JobExecutionDetails>
getJobExecutionDetailsTransportSettings =
GrpcCallSettings.<GetJobExecutionDetailsRequest, JobExecutionDetails>newBuilder()
.setMethodDescriptor(getJobExecutionDetailsMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
params.put("job_id", String.valueOf(request.getJobId()));
params.put("location", String.valueOf(request.getLocation()));
params.put("project_id", String.valueOf(request.getProjectId()));
return params.build();
})
.build();
GrpcCallSettings<GetStageExecutionDetailsRequest, StageExecutionDetails>
getStageExecutionDetailsTransportSettings =
GrpcCallSettings.<GetStageExecutionDetailsRequest, StageExecutionDetails>newBuilder()
.setMethodDescriptor(getStageExecutionDetailsMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
params.put("job_id", String.valueOf(request.getJobId()));
params.put("location", String.valueOf(request.getLocation()));
params.put("project_id", String.valueOf(request.getProjectId()));
params.put("stage_id", String.valueOf(request.getStageId()));
return params.build();
})
.build();

this.getJobMetricsCallable =
Expand Down
25 changes: 25 additions & 0 deletions ...oud-dataflow/src/main/java/com/google/dataflow/v1beta3/stub/GrpcSnapshotsV1Beta3Stub.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@
import com.google.api.gax.grpc.GrpcStubCallableFactory;
import com.google.api.gax.rpc.ClientContext;
import com.google.api.gax.rpc.UnaryCallable;
import com.google.common.collect.ImmutableMap;
import com.google.dataflow.v1beta3.DeleteSnapshotRequest;
import com.google.dataflow.v1beta3.DeleteSnapshotResponse;
import com.google.dataflow.v1beta3.GetSnapshotRequest;
Expand Down Expand Up @@ -126,15 +127,39 @@ protected GrpcSnapshotsV1Beta3Stub(
GrpcCallSettings<GetSnapshotRequest, Snapshot> getSnapshotTransportSettings =
GrpcCallSettings.<GetSnapshotRequest, Snapshot>newBuilder()
.setMethodDescriptor(getSnapshotMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
params.put("location", String.valueOf(request.getLocation()));
params.put("project_id", String.valueOf(request.getProjectId()));
params.put("snapshot_id", String.valueOf(request.getSnapshotId()));
return params.build();
})
.build();
GrpcCallSettings<DeleteSnapshotRequest, DeleteSnapshotResponse>
deleteSnapshotTransportSettings =
GrpcCallSettings.<DeleteSnapshotRequest, DeleteSnapshotResponse>newBuilder()
.setMethodDescriptor(deleteSnapshotMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
params.put("location", String.valueOf(request.getLocation()));
params.put("project_id", String.valueOf(request.getProjectId()));
params.put("snapshot_id", String.valueOf(request.getSnapshotId()));
return params.build();
})
.build();
GrpcCallSettings<ListSnapshotsRequest, ListSnapshotsResponse> listSnapshotsTransportSettings =
GrpcCallSettings.<ListSnapshotsRequest, ListSnapshotsResponse>newBuilder()
.setMethodDescriptor(listSnapshotsMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
params.put("job_id", String.valueOf(request.getJobId()));
params.put("location", String.valueOf(request.getLocation()));
params.put("project_id", String.valueOf(request.getProjectId()));
return params.build();
})
.build();

this.getSnapshotCallable =
Expand Down
22 changes: 22 additions & 0 deletions ...oud-dataflow/src/main/java/com/google/dataflow/v1beta3/stub/GrpcTemplatesServiceStub.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@
import com.google.api.gax.grpc.GrpcStubCallableFactory;
import com.google.api.gax.rpc.ClientContext;
import com.google.api.gax.rpc.UnaryCallable;
import com.google.common.collect.ImmutableMap;
import com.google.dataflow.v1beta3.CreateJobFromTemplateRequest;
import com.google.dataflow.v1beta3.GetTemplateRequest;
import com.google.dataflow.v1beta3.GetTemplateResponse;
Expand Down Expand Up @@ -127,15 +128,36 @@ protected GrpcTemplatesServiceStub(
GrpcCallSettings<CreateJobFromTemplateRequest, Job> createJobFromTemplateTransportSettings =
GrpcCallSettings.<CreateJobFromTemplateRequest, Job>newBuilder()
.setMethodDescriptor(createJobFromTemplateMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
params.put("location", String.valueOf(request.getLocation()));
params.put("project_id", String.valueOf(request.getProjectId()));
return params.build();
})
.build();
GrpcCallSettings<LaunchTemplateRequest, LaunchTemplateResponse>
launchTemplateTransportSettings =
GrpcCallSettings.<LaunchTemplateRequest, LaunchTemplateResponse>newBuilder()
.setMethodDescriptor(launchTemplateMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
params.put("location", String.valueOf(request.getLocation()));
params.put("project_id", String.valueOf(request.getProjectId()));
return params.build();
})
.build();
GrpcCallSettings<GetTemplateRequest, GetTemplateResponse> getTemplateTransportSettings =
GrpcCallSettings.<GetTemplateRequest, GetTemplateResponse>newBuilder()
.setMethodDescriptor(getTemplateMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
params.put("location", String.valueOf(request.getLocation()));
params.put("project_id", String.valueOf(request.getProjectId()));
return params.build();
})
.build();

this.createJobFromTemplateCallable =
Expand Down
21 changes: 7 additions & 14 deletions ...le-cloud-dataflow-v1beta3/src/main/java/com/google/dataflow/v1beta3/AutoscalingEvent.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -567,8 +567,7 @@ public com.google.protobuf.TimestampOrBuilder getTimeOrBuilder() {
*
*
* <pre>
* A short and friendly name for the worker pool this event refers to,
* populated from the value of PoolStageRelation::user_pool_name.
* A short and friendly name for the worker pool this event refers to.
* </pre>
*
* <code>string worker_pool = 7;</code>
Expand All @@ -591,8 +590,7 @@ public java.lang.String getWorkerPool() {
*
*
* <pre>
* A short and friendly name for the worker pool this event refers to,
* populated from the value of PoolStageRelation::user_pool_name.
* A short and friendly name for the worker pool this event refers to.
* </pre>
*
* <code>string worker_pool = 7;</code>
Expand Down Expand Up @@ -1628,8 +1626,7 @@ public com.google.protobuf.TimestampOrBuilder getTimeOrBuilder() {
*
*
* <pre>
* A short and friendly name for the worker pool this event refers to,
* populated from the value of PoolStageRelation::user_pool_name.
* A short and friendly name for the worker pool this event refers to.
* </pre>
*
* <code>string worker_pool = 7;</code>
Expand All @@ -1651,8 +1648,7 @@ public java.lang.String getWorkerPool() {
*
*
* <pre>
* A short and friendly name for the worker pool this event refers to,
* populated from the value of PoolStageRelation::user_pool_name.
* A short and friendly name for the worker pool this event refers to.
* </pre>
*
* <code>string worker_pool = 7;</code>
Expand All @@ -1674,8 +1670,7 @@ public com.google.protobuf.ByteString getWorkerPoolBytes() {
*
*
* <pre>
* A short and friendly name for the worker pool this event refers to,
* populated from the value of PoolStageRelation::user_pool_name.
* A short and friendly name for the worker pool this event refers to.
* </pre>
*
* <code>string worker_pool = 7;</code>
Expand All @@ -1696,8 +1691,7 @@ public Builder setWorkerPool(java.lang.String value) {
*
*
* <pre>
* A short and friendly name for the worker pool this event refers to,
* populated from the value of PoolStageRelation::user_pool_name.
* A short and friendly name for the worker pool this event refers to.
* </pre>
*
* <code>string worker_pool = 7;</code>
Expand All @@ -1714,8 +1708,7 @@ public Builder clearWorkerPool() {
*
*
* <pre>
* A short and friendly name for the worker pool this event refers to,
* populated from the value of PoolStageRelation::user_pool_name.
* A short and friendly name for the worker pool this event refers to.
* </pre>
*
* <code>string worker_pool = 7;</code>
Expand Down
6 changes: 2 additions & 4 deletions ...dataflow-v1beta3/src/main/java/com/google/dataflow/v1beta3/AutoscalingEventOrBuilder.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -157,8 +157,7 @@ public interface AutoscalingEventOrBuilder
*
*
* <pre>
* A short and friendly name for the worker pool this event refers to,
* populated from the value of PoolStageRelation::user_pool_name.
* A short and friendly name for the worker pool this event refers to.
* </pre>
*
* <code>string worker_pool = 7;</code>
Expand All @@ -170,8 +169,7 @@ public interface AutoscalingEventOrBuilder
*
*
* <pre>
* A short and friendly name for the worker pool this event refers to,
* populated from the value of PoolStageRelation::user_pool_name.
* A short and friendly name for the worker pool this event refers to.
* </pre>
*
* <code>string worker_pool = 7;</code>
Expand Down
4 changes: 2 additions & 2 deletions ...e-cloud-dataflow-v1beta3/src/main/java/com/google/dataflow/v1beta3/BigTableIODetails.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@
*
*
* <pre>
* Metadata for a Cloud BigTable connector used by the job.
* Metadata for a Cloud Bigtable connector used by the job.
* </pre>
*
* Protobuf type {@code google.dataflow.v1beta3.BigTableIODetails}
Expand Down Expand Up @@ -455,7 +455,7 @@ protected Builder newBuilderForType(com.google.protobuf.GeneratedMessageV3.Build
*
*
* <pre>
* Metadata for a Cloud BigTable connector used by the job.
* Metadata for a Cloud Bigtable connector used by the job.
* </pre>
*
* Protobuf type {@code google.dataflow.v1beta3.BigTableIODetails}
Expand Down
Loading