generated code

ci/etc/expected_install_directories

@@ -135,6 +135,10 @@
 ./include/google/cloud/bigquery/datatransfer/v1
 ./include/google/cloud/bigquery/datatransfer/v1/internal
 ./include/google/cloud/bigquery/datatransfer/v1/mocks
+./include/google/cloud/bigquery/job
+./include/google/cloud/bigquery/job/v2
+./include/google/cloud/bigquery/job/v2/internal
+./include/google/cloud/bigquery/job/v2/mocks
 ./include/google/cloud/bigquery/migration
 ./include/google/cloud/bigquery/migration/v2
 ./include/google/cloud/bigquery/migration/v2/internal
@@ -148,6 +152,7 @@
 ./include/google/cloud/bigquery/storage/v1
 ./include/google/cloud/bigquery/storage/v1/internal
 ./include/google/cloud/bigquery/storage/v1/mocks
+./include/google/cloud/bigquery/v2
 ./include/google/cloud/bigtable
 ./include/google/cloud/bigtable/admin
 ./include/google/cloud/bigtable/admin/internal
@@ -1271,6 +1276,8 @@
 ./lib64/cmake/google_cloud_cpp_beyondcorp
 ./lib64/cmake/google_cloud_cpp_beyondcorp_mocks
 ./lib64/cmake/google_cloud_cpp_bigquery
+./lib64/cmake/google_cloud_cpp_bigquery_job
+./lib64/cmake/google_cloud_cpp_bigquery_job_mocks
 ./lib64/cmake/google_cloud_cpp_bigquery_mocks
 ./lib64/cmake/google_cloud_cpp_bigtable
 ./lib64/cmake/google_cloud_cpp_bigtable_mocks

external/googleapis/protodeps/bigquery_v2.deps

@@ -0,0 +1,6 @@
+@com_google_googleapis//google/api:annotations_proto
+@com_google_googleapis//google/api:client_proto
+@com_google_googleapis//google/api:field_behavior_proto
+@com_google_googleapis//google/api:http_proto
+@com_google_googleapis//google/api:launch_stage_proto
+@com_google_googleapis//google/api:resource_proto

external/googleapis/protolists/bigquery_v2.list

@@ -0,0 +1,44 @@
+@com_google_googleapis//google/cloud/bigquery/v2:biglake_config.proto
+@com_google_googleapis//google/cloud/bigquery/v2:clustering.proto
+@com_google_googleapis//google/cloud/bigquery/v2:data_format_options.proto
+@com_google_googleapis//google/cloud/bigquery/v2:dataset.proto
+@com_google_googleapis//google/cloud/bigquery/v2:dataset_reference.proto
+@com_google_googleapis//google/cloud/bigquery/v2:decimal_target_types.proto
+@com_google_googleapis//google/cloud/bigquery/v2:encryption_config.proto
+@com_google_googleapis//google/cloud/bigquery/v2:error.proto
+@com_google_googleapis//google/cloud/bigquery/v2:external_catalog_dataset_options.proto
+@com_google_googleapis//google/cloud/bigquery/v2:external_catalog_table_options.proto
+@com_google_googleapis//google/cloud/bigquery/v2:external_data_config.proto
+@com_google_googleapis//google/cloud/bigquery/v2:external_dataset_reference.proto
+@com_google_googleapis//google/cloud/bigquery/v2:file_set_specification_type.proto
+@com_google_googleapis//google/cloud/bigquery/v2:hive_partitioning.proto
+@com_google_googleapis//google/cloud/bigquery/v2:job.proto
+@com_google_googleapis//google/cloud/bigquery/v2:job_config.proto
+@com_google_googleapis//google/cloud/bigquery/v2:job_creation_reason.proto
+@com_google_googleapis//google/cloud/bigquery/v2:job_reference.proto
+@com_google_googleapis//google/cloud/bigquery/v2:job_stats.proto
+@com_google_googleapis//google/cloud/bigquery/v2:job_status.proto
+@com_google_googleapis//google/cloud/bigquery/v2:json_extension.proto
+@com_google_googleapis//google/cloud/bigquery/v2:location_metadata.proto
+@com_google_googleapis//google/cloud/bigquery/v2:map_target_type.proto
+@com_google_googleapis//google/cloud/bigquery/v2:model.proto
+@com_google_googleapis//google/cloud/bigquery/v2:model_reference.proto
+@com_google_googleapis//google/cloud/bigquery/v2:partitioning_definition.proto
+@com_google_googleapis//google/cloud/bigquery/v2:privacy_policy.proto
+@com_google_googleapis//google/cloud/bigquery/v2:project.proto
+@com_google_googleapis//google/cloud/bigquery/v2:query_parameter.proto
+@com_google_googleapis//google/cloud/bigquery/v2:range_partitioning.proto
+@com_google_googleapis//google/cloud/bigquery/v2:restriction_config.proto
+@com_google_googleapis//google/cloud/bigquery/v2:routine.proto
+@com_google_googleapis//google/cloud/bigquery/v2:routine_reference.proto
+@com_google_googleapis//google/cloud/bigquery/v2:row_access_policy.proto
+@com_google_googleapis//google/cloud/bigquery/v2:row_access_policy_reference.proto
+@com_google_googleapis//google/cloud/bigquery/v2:session_info.proto
+@com_google_googleapis//google/cloud/bigquery/v2:standard_sql.proto
+@com_google_googleapis//google/cloud/bigquery/v2:system_variable.proto
+@com_google_googleapis//google/cloud/bigquery/v2:table.proto
+@com_google_googleapis//google/cloud/bigquery/v2:table_constraints.proto
+@com_google_googleapis//google/cloud/bigquery/v2:table_reference.proto
+@com_google_googleapis//google/cloud/bigquery/v2:table_schema.proto
+@com_google_googleapis//google/cloud/bigquery/v2:time_partitioning.proto
+@com_google_googleapis//google/cloud/bigquery/v2:udf_resource.proto

google/cloud/bigquery/job/v2/internal/job_option_defaults.cc

@@ -0,0 +1,66 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+// Generated by the Codegen C++ plugin.
+// If you make any local changes, they will be lost.
+// source: google/cloud/bigquery/v2/job.proto
+
+#include "google/cloud/bigquery/job/v2/internal/job_option_defaults.h"
+#include "google/cloud/bigquery/job/v2/job_connection.h"
+#include "google/cloud/bigquery/job/v2/job_options.h"
+#include "google/cloud/internal/populate_common_options.h"
+#include "google/cloud/internal/populate_grpc_options.h"
+#include <memory>
+#include <utility>
+
+namespace google {
+namespace cloud {
+namespace bigquery_job_v2_internal {
+GOOGLE_CLOUD_CPP_INLINE_NAMESPACE_BEGIN
+
+namespace {
+auto constexpr kBackoffScaling = 2.0;
+}  // namespace
+
+Options JobServiceDefaultOptions(Options options) {
+  options = internal::PopulateCommonOptions(
+      std::move(options), "GOOGLE_CLOUD_CPP_JOB_SERVICE_ENDPOINT", "",
+      "GOOGLE_CLOUD_CPP_JOB_SERVICE_AUTHORITY", "bigquery.googleapis.com");
+  options = internal::PopulateGrpcOptions(std::move(options));
+  if (!options.has<bigquery_job_v2::JobServiceRetryPolicyOption>()) {
+    options.set<bigquery_job_v2::JobServiceRetryPolicyOption>(
+        bigquery_job_v2::JobServiceLimitedTimeRetryPolicy(
+            std::chrono::minutes(30))
+            .clone());
+  }
+  if (!options.has<bigquery_job_v2::JobServiceBackoffPolicyOption>()) {
+    options.set<bigquery_job_v2::JobServiceBackoffPolicyOption>(
+        ExponentialBackoffPolicy(
+            std::chrono::seconds(0), std::chrono::seconds(1),
+            std::chrono::minutes(5), kBackoffScaling, kBackoffScaling)
+            .clone());
+  }
+  if (!options.has<
+          bigquery_job_v2::JobServiceConnectionIdempotencyPolicyOption>()) {
+    options.set<bigquery_job_v2::JobServiceConnectionIdempotencyPolicyOption>(
+        bigquery_job_v2::MakeDefaultJobServiceConnectionIdempotencyPolicy());
+  }
+
+  return options;
+}
+
+GOOGLE_CLOUD_CPP_INLINE_NAMESPACE_END
+}  // namespace bigquery_job_v2_internal
+}  // namespace cloud
+}  // namespace google

google/cloud/bigquery/job/v2/internal/job_option_defaults.h

@@ -0,0 +1,37 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+// Generated by the Codegen C++ plugin.
+// If you make any local changes, they will be lost.
+// source: google/cloud/bigquery/v2/job.proto
+
+#ifndef GOOGLE_CLOUD_CPP_GOOGLE_CLOUD_BIGQUERY_JOB_V2_INTERNAL_JOB_OPTION_DEFAULTS_H
+#define GOOGLE_CLOUD_CPP_GOOGLE_CLOUD_BIGQUERY_JOB_V2_INTERNAL_JOB_OPTION_DEFAULTS_H
+
+#include "google/cloud/options.h"
+#include "google/cloud/version.h"
+
+namespace google {
+namespace cloud {
+namespace bigquery_job_v2_internal {
+GOOGLE_CLOUD_CPP_INLINE_NAMESPACE_BEGIN
+
+Options JobServiceDefaultOptions(Options options);
+
+GOOGLE_CLOUD_CPP_INLINE_NAMESPACE_END
+}  // namespace bigquery_job_v2_internal
+}  // namespace cloud
+}  // namespace google
+
+#endif  // GOOGLE_CLOUD_CPP_GOOGLE_CLOUD_BIGQUERY_JOB_V2_INTERNAL_JOB_OPTION_DEFAULTS_H

google/cloud/bigquery/job/v2/internal/job_rest_connection_impl.cc

@@ -0,0 +1,164 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+// Generated by the Codegen C++ plugin.
+// If you make any local changes, they will be lost.
+// source: google/cloud/bigquery/v2/job.proto
+
+#include "google/cloud/bigquery/job/v2/internal/job_rest_connection_impl.h"
+#include "google/cloud/bigquery/job/v2/internal/job_rest_stub_factory.h"
+#include "google/cloud/common_options.h"
+#include "google/cloud/credentials.h"
+#include "google/cloud/internal/pagination_range.h"
+#include "google/cloud/internal/rest_retry_loop.h"
+#include "google/cloud/rest_options.h"
+#include <memory>
+#include <utility>
+
+namespace google {
+namespace cloud {
+namespace bigquery_job_v2_internal {
+GOOGLE_CLOUD_CPP_INLINE_NAMESPACE_BEGIN
+
+JobServiceRestConnectionImpl::JobServiceRestConnectionImpl(
+    std::unique_ptr<google::cloud::BackgroundThreads> background,
+    std::shared_ptr<bigquery_job_v2_internal::JobServiceRestStub> stub,
+    Options options)
+    : background_(std::move(background)),
+      stub_(std::move(stub)),
+      options_(internal::MergeOptions(std::move(options),
+                                      JobServiceConnection::options())) {}
+
+StatusOr<google::cloud::bigquery::v2::JobCancelResponse>
+JobServiceRestConnectionImpl::CancelJob(
+    google::cloud::bigquery::v2::CancelJobRequest const& request) {
+  auto current = google::cloud::internal::SaveCurrentOptions();
+  return google::cloud::rest_internal::RestRetryLoop(
+      retry_policy(*current), backoff_policy(*current),
+      idempotency_policy(*current)->CancelJob(request),
+      [this](rest_internal::RestContext& rest_context, Options const& options,
+             google::cloud::bigquery::v2::CancelJobRequest const& request) {
+        return stub_->CancelJob(rest_context, options, request);
+      },
+      *current, request, __func__);
+}
+
+StatusOr<google::cloud::bigquery::v2::Job> JobServiceRestConnectionImpl::GetJob(
+    google::cloud::bigquery::v2::GetJobRequest const& request) {
+  auto current = google::cloud::internal::SaveCurrentOptions();
+  return google::cloud::rest_internal::RestRetryLoop(
+      retry_policy(*current), backoff_policy(*current),
+      idempotency_policy(*current)->GetJob(request),
+      [this](rest_internal::RestContext& rest_context, Options const& options,
+             google::cloud::bigquery::v2::GetJobRequest const& request) {
+        return stub_->GetJob(rest_context, options, request);
+      },
+      *current, request, __func__);
+}
+
+StatusOr<google::cloud::bigquery::v2::Job>
+JobServiceRestConnectionImpl::InsertJob(
+    google::cloud::bigquery::v2::InsertJobRequest const& request) {
+  auto current = google::cloud::internal::SaveCurrentOptions();
+  return google::cloud::rest_internal::RestRetryLoop(
+      retry_policy(*current), backoff_policy(*current),
+      idempotency_policy(*current)->InsertJob(request),
+      [this](rest_internal::RestContext& rest_context, Options const& options,
+             google::cloud::bigquery::v2::InsertJobRequest const& request) {
+        return stub_->InsertJob(rest_context, options, request);
+      },
+      *current, request, __func__);
+}
+
+Status JobServiceRestConnectionImpl::DeleteJob(
+    google::cloud::bigquery::v2::DeleteJobRequest const& request) {
+  auto current = google::cloud::internal::SaveCurrentOptions();
+  return google::cloud::rest_internal::RestRetryLoop(
+      retry_policy(*current), backoff_policy(*current),
+      idempotency_policy(*current)->DeleteJob(request),
+      [this](rest_internal::RestContext& rest_context, Options const& options,
+             google::cloud::bigquery::v2::DeleteJobRequest const& request) {
+        return stub_->DeleteJob(rest_context, options, request);
+      },
+      *current, request, __func__);
+}
+
+StreamRange<google::cloud::bigquery::v2::ListFormatJob>
+JobServiceRestConnectionImpl::ListJobs(
+    google::cloud::bigquery::v2::ListJobsRequest request) {
+  request.clear_page_token();
+  auto current = google::cloud::internal::SaveCurrentOptions();
+  auto idempotency = idempotency_policy(*current)->ListJobs(request);
+  char const* function_name = __func__;
+  return google::cloud::internal::MakePaginationRange<
+      StreamRange<google::cloud::bigquery::v2::ListFormatJob>>(
+      current, std::move(request),
+      [idempotency, function_name, stub = stub_,
+       retry = std::shared_ptr<bigquery_job_v2::JobServiceRetryPolicy>(
+           retry_policy(*current)),
+       backoff = std::shared_ptr<BackoffPolicy>(backoff_policy(*current))](
+          Options const& options,
+          google::cloud::bigquery::v2::ListJobsRequest const& r) {
+        return google::cloud::rest_internal::RestRetryLoop(
+            retry->clone(), backoff->clone(), idempotency,
+            [stub](
+                rest_internal::RestContext& rest_context,
+                Options const& options,
+                google::cloud::bigquery::v2::ListJobsRequest const& request) {
+              return stub->ListJobs(rest_context, options, request);
+            },
+            options, r, function_name);
+      },
+      [](google::cloud::bigquery::v2::JobList r) {
+        std::vector<google::cloud::bigquery::v2::ListFormatJob> result(
+            r.jobs().size());
+        auto& messages = *r.mutable_jobs();
+        std::move(messages.begin(), messages.end(), result.begin());
+        return result;
+      });
+}
+
+StatusOr<google::cloud::bigquery::v2::GetQueryResultsResponse>
+JobServiceRestConnectionImpl::GetQueryResults(
+    google::cloud::bigquery::v2::GetQueryResultsRequest const& request) {
+  auto current = google::cloud::internal::SaveCurrentOptions();
+  return google::cloud::rest_internal::RestRetryLoop(
+      retry_policy(*current), backoff_policy(*current),
+      idempotency_policy(*current)->GetQueryResults(request),
+      [this](
+          rest_internal::RestContext& rest_context, Options const& options,
+          google::cloud::bigquery::v2::GetQueryResultsRequest const& request) {
+        return stub_->GetQueryResults(rest_context, options, request);
+      },
+      *current, request, __func__);
+}
+
+StatusOr<google::cloud::bigquery::v2::QueryResponse>
+JobServiceRestConnectionImpl::Query(
+    google::cloud::bigquery::v2::PostQueryRequest const& request) {
+  auto current = google::cloud::internal::SaveCurrentOptions();
+  return google::cloud::rest_internal::RestRetryLoop(
+      retry_policy(*current), backoff_policy(*current),
+      idempotency_policy(*current)->Query(request),
+      [this](rest_internal::RestContext& rest_context, Options const& options,
+             google::cloud::bigquery::v2::PostQueryRequest const& request) {
+        return stub_->Query(rest_context, options, request);
+      },
+      *current, request, __func__);
+}
+
+GOOGLE_CLOUD_CPP_INLINE_NAMESPACE_END
+}  // namespace bigquery_job_v2_internal
+}  // namespace cloud
+}  // namespace google