Update generated libraries

googleapis · Feb 15, 2024 · fedfab9 · fedfab9
1 parent 0f651f1
commit fedfab9
Show file tree

Hide file tree

Showing 3 changed files with 41 additions and 13 deletions.
diff --git a/google/cloud/storagecontrol/v2/internal/storage_control_connection_impl.cc b/google/cloud/storagecontrol/v2/internal/storage_control_connection_impl.cc
@@ -70,41 +70,53 @@ StatusOr<google::storage::control::v2::Folder>
 StorageControlConnectionImpl::CreateFolder(
     google::storage::control::v2::CreateFolderRequest const& request) {
   auto current = google::cloud::internal::SaveCurrentOptions();
+  auto request_copy = request;
+  if (request_copy.request_id().empty()) {
+    request_copy.set_request_id(invocation_id_generator_->MakeInvocationId());
+  }
   return google::cloud::internal::RetryLoop(
       retry_policy(*current), backoff_policy(*current),
-      idempotency_policy(*current)->CreateFolder(request),
+      idempotency_policy(*current)->CreateFolder(request_copy),
       [this](grpc::ClientContext& context,
              google::storage::control::v2::CreateFolderRequest const& request) {
         return stub_->CreateFolder(context, request);
       },
-      request, __func__);
+      request_copy, __func__);
 }
 
 Status StorageControlConnectionImpl::DeleteFolder(
     google::storage::control::v2::DeleteFolderRequest const& request) {
   auto current = google::cloud::internal::SaveCurrentOptions();
+  auto request_copy = request;
+  if (request_copy.request_id().empty()) {
+    request_copy.set_request_id(invocation_id_generator_->MakeInvocationId());
+  }
   return google::cloud::internal::RetryLoop(
       retry_policy(*current), backoff_policy(*current),
-      idempotency_policy(*current)->DeleteFolder(request),
+      idempotency_policy(*current)->DeleteFolder(request_copy),
       [this](grpc::ClientContext& context,
              google::storage::control::v2::DeleteFolderRequest const& request) {
         return stub_->DeleteFolder(context, request);
       },
-      request, __func__);
+      request_copy, __func__);
 }
 
 StatusOr<google::storage::control::v2::Folder>
 StorageControlConnectionImpl::GetFolder(
     google::storage::control::v2::GetFolderRequest const& request) {
   auto current = google::cloud::internal::SaveCurrentOptions();
+  auto request_copy = request;
+  if (request_copy.request_id().empty()) {
+    request_copy.set_request_id(invocation_id_generator_->MakeInvocationId());
+  }
   return google::cloud::internal::RetryLoop(
       retry_policy(*current), backoff_policy(*current),
-      idempotency_policy(*current)->GetFolder(request),
+      idempotency_policy(*current)->GetFolder(request_copy),
       [this](grpc::ClientContext& context,
              google::storage::control::v2::GetFolderRequest const& request) {
         return stub_->GetFolder(context, request);
       },
-      request, __func__);
+      request_copy, __func__);
 }
 
 StreamRange<google::storage::control::v2::Folder>
@@ -143,6 +155,9 @@ StorageControlConnectionImpl::RenameFolder(
     google::storage::control::v2::RenameFolderRequest const& request) {
   auto current = google::cloud::internal::SaveCurrentOptions();
   auto request_copy = request;
+  if (request_copy.request_id().empty()) {
+    request_copy.set_request_id(invocation_id_generator_->MakeInvocationId());
+  }
   auto const idempotent =
       idempotency_policy(*current)->RenameFolder(request_copy);
   return google::cloud::internal::AsyncLongRunningOperation<
@@ -179,13 +194,17 @@ StatusOr<google::storage::control::v2::StorageLayout>
 StorageControlConnectionImpl::GetStorageLayout(
     google::storage::control::v2::GetStorageLayoutRequest const& request) {
   auto current = google::cloud::internal::SaveCurrentOptions();
+  auto request_copy = request;
+  if (request_copy.request_id().empty()) {
+    request_copy.set_request_id(invocation_id_generator_->MakeInvocationId());
+  }
   return google::cloud::internal::RetryLoop(
       retry_policy(*current), backoff_policy(*current),
-      idempotency_policy(*current)->GetStorageLayout(request),
+      idempotency_policy(*current)->GetStorageLayout(request_copy),
       [this](grpc::ClientContext& context,
              google::storage::control::v2::GetStorageLayoutRequest const&
                  request) { return stub_->GetStorageLayout(context, request); },
-      request, __func__);
+      request_copy, __func__);
 }
 
 GOOGLE_CLOUD_CPP_INLINE_NAMESPACE_END

diff --git a/google/cloud/storagecontrol/v2/internal/storage_control_connection_impl.h b/google/cloud/storagecontrol/v2/internal/storage_control_connection_impl.h
@@ -27,6 +27,7 @@
 #include "google/cloud/background_threads.h"
 #include "google/cloud/backoff_policy.h"
 #include "google/cloud/future.h"
+#include "google/cloud/internal/invocation_id_generator.h"
 #include "google/cloud/options.h"
 #include "google/cloud/polling_policy.h"
 #include "google/cloud/status_or.h"
@@ -77,6 +78,9 @@ class StorageControlConnectionImpl
   std::unique_ptr<google::cloud::BackgroundThreads> background_;
   std::shared_ptr<storagecontrol_v2_internal::StorageControlStub> stub_;
   Options options_;
+  std::shared_ptr<google::cloud::internal::InvocationIdGenerator>
+      invocation_id_generator_ =
+          std::make_shared<google::cloud::internal::InvocationIdGenerator>();
 };
 
 GOOGLE_CLOUD_CPP_INLINE_NAMESPACE_END

diff --git a/google/cloud/storagecontrol/v2/storage_control_connection_idempotency_policy.cc b/google/cloud/storagecontrol/v2/storage_control_connection_idempotency_policy.cc
@@ -35,17 +35,20 @@ StorageControlConnectionIdempotencyPolicy::clone() const {
 }
 
 Idempotency StorageControlConnectionIdempotencyPolicy::CreateFolder(
-    google::storage::control::v2::CreateFolderRequest const&) {
+    google::storage::control::v2::CreateFolderRequest const& request) {
+  if (!request.request_id().empty()) return Idempotency::kIdempotent;
   return Idempotency::kNonIdempotent;
 }
 
 Idempotency StorageControlConnectionIdempotencyPolicy::DeleteFolder(
-    google::storage::control::v2::DeleteFolderRequest const&) {
+    google::storage::control::v2::DeleteFolderRequest const& request) {
+  if (!request.request_id().empty()) return Idempotency::kIdempotent;
   return Idempotency::kNonIdempotent;
 }
 
 Idempotency StorageControlConnectionIdempotencyPolicy::GetFolder(
-    google::storage::control::v2::GetFolderRequest const&) {
+    google::storage::control::v2::GetFolderRequest const& request) {
+  if (!request.request_id().empty()) return Idempotency::kIdempotent;
   return Idempotency::kIdempotent;
 }
 
@@ -55,12 +58,14 @@ Idempotency StorageControlConnectionIdempotencyPolicy::ListFolders(
 }
 
 Idempotency StorageControlConnectionIdempotencyPolicy::RenameFolder(
-    google::storage::control::v2::RenameFolderRequest const&) {
+    google::storage::control::v2::RenameFolderRequest const& request) {
+  if (!request.request_id().empty()) return Idempotency::kIdempotent;
   return Idempotency::kNonIdempotent;
 }
 
 Idempotency StorageControlConnectionIdempotencyPolicy::GetStorageLayout(
-    google::storage::control::v2::GetStorageLayoutRequest const&) {
+    google::storage::control::v2::GetStorageLayoutRequest const& request) {
+  if (!request.request_id().empty()) return Idempotency::kIdempotent;
   return Idempotency::kIdempotent;
 }