impl(generator): add synchronous LRO method on stubs and decorators (googleapis#14341)

Author: scotthart

generator/integration_tests/golden/v1/internal/golden_thing_admin_auth_decorator.cc

@@ -59,6 +59,16 @@ GoldenThingAdminAuth::AsyncCreateDatabase(
       });
 }
 
+StatusOr<google::longrunning::Operation>
+GoldenThingAdminAuth::CreateDatabase(
+      grpc::ClientContext& context,
+      Options options,
+      google::test::admin::database::v1::CreateDatabaseRequest const& request) {
+  auto status = auth_->ConfigureContext(context);
+  if (!status.ok()) return status;
+  return child_->CreateDatabase(context, options, request);
+}
+
 StatusOr<google::test::admin::database::v1::Database> GoldenThingAdminAuth::GetDatabase(
     grpc::ClientContext& context,
     Options const& options,
@@ -87,6 +97,16 @@ GoldenThingAdminAuth::AsyncUpdateDatabaseDdl(
       });
 }
 
+StatusOr<google::longrunning::Operation>
+GoldenThingAdminAuth::UpdateDatabaseDdl(
+      grpc::ClientContext& context,
+      Options options,
+      google::test::admin::database::v1::UpdateDatabaseDdlRequest const& request) {
+  auto status = auth_->ConfigureContext(context);
+  if (!status.ok()) return status;
+  return child_->UpdateDatabaseDdl(context, options, request);
+}
+
 Status GoldenThingAdminAuth::DropDatabase(
     grpc::ClientContext& context,
     Options const& options,
@@ -151,6 +171,16 @@ GoldenThingAdminAuth::AsyncCreateBackup(
       });
 }
 
+StatusOr<google::longrunning::Operation>
+GoldenThingAdminAuth::CreateBackup(
+      grpc::ClientContext& context,
+      Options options,
+      google::test::admin::database::v1::CreateBackupRequest const& request) {
+  auto status = auth_->ConfigureContext(context);
+  if (!status.ok()) return status;
+  return child_->CreateBackup(context, options, request);
+}
+
 StatusOr<google::test::admin::database::v1::Backup> GoldenThingAdminAuth::GetBackup(
     grpc::ClientContext& context,
     Options const& options,
@@ -206,6 +236,16 @@ GoldenThingAdminAuth::AsyncRestoreDatabase(
       });
 }
 
+StatusOr<google::longrunning::Operation>
+GoldenThingAdminAuth::RestoreDatabase(
+      grpc::ClientContext& context,
+      Options options,
+      google::test::admin::database::v1::RestoreDatabaseRequest const& request) {
+  auto status = auth_->ConfigureContext(context);
+  if (!status.ok()) return status;
+  return child_->RestoreDatabase(context, options, request);
+}
+
 StatusOr<google::test::admin::database::v1::ListDatabaseOperationsResponse> GoldenThingAdminAuth::ListDatabaseOperations(
     grpc::ClientContext& context,
     Options const& options,
@@ -243,6 +283,16 @@ GoldenThingAdminAuth::AsyncLongRunningWithoutRouting(
       });
 }
 
+StatusOr<google::longrunning::Operation>
+GoldenThingAdminAuth::LongRunningWithoutRouting(
+      grpc::ClientContext& context,
+      Options options,
+      google::test::admin::database::v1::RestoreDatabaseRequest const& request) {
+  auto status = auth_->ConfigureContext(context);
+  if (!status.ok()) return status;
+  return child_->LongRunningWithoutRouting(context, options, request);
+}
+
 future<StatusOr<google::test::admin::database::v1::Database>>
 GoldenThingAdminAuth::AsyncGetDatabase(
       google::cloud::CompletionQueue& cq,

generator/integration_tests/golden/v1/internal/golden_thing_admin_auth_decorator.h

@@ -50,6 +50,11 @@ class GoldenThingAdminAuth : public GoldenThingAdminStub {
       google::cloud::internal::ImmutableOptions options,
       google::test::admin::database::v1::CreateDatabaseRequest const& request) override;
 
+  StatusOr<google::longrunning::Operation> CreateDatabase(
+      grpc::ClientContext& context,
+      Options options,
+      google::test::admin::database::v1::CreateDatabaseRequest const& request) override;
+
   StatusOr<google::test::admin::database::v1::Database> GetDatabase(
       grpc::ClientContext& context,
       Options const& options,
@@ -61,6 +66,11 @@ class GoldenThingAdminAuth : public GoldenThingAdminStub {
       google::cloud::internal::ImmutableOptions options,
       google::test::admin::database::v1::UpdateDatabaseDdlRequest const& request) override;
 
+  StatusOr<google::longrunning::Operation> UpdateDatabaseDdl(
+      grpc::ClientContext& context,
+      Options options,
+      google::test::admin::database::v1::UpdateDatabaseDdlRequest const& request) override;
+
   Status DropDatabase(
       grpc::ClientContext& context,
       Options const& options,
@@ -92,6 +102,11 @@ class GoldenThingAdminAuth : public GoldenThingAdminStub {
       google::cloud::internal::ImmutableOptions options,
       google::test::admin::database::v1::CreateBackupRequest const& request) override;
 
+  StatusOr<google::longrunning::Operation> CreateBackup(
+      grpc::ClientContext& context,
+      Options options,
+      google::test::admin::database::v1::CreateBackupRequest const& request) override;
+
   StatusOr<google::test::admin::database::v1::Backup> GetBackup(
       grpc::ClientContext& context,
       Options const& options,
@@ -118,6 +133,11 @@ class GoldenThingAdminAuth : public GoldenThingAdminStub {
       google::cloud::internal::ImmutableOptions options,
       google::test::admin::database::v1::RestoreDatabaseRequest const& request) override;
 
+  StatusOr<google::longrunning::Operation> RestoreDatabase(
+      grpc::ClientContext& context,
+      Options options,
+      google::test::admin::database::v1::RestoreDatabaseRequest const& request) override;
+
   StatusOr<google::test::admin::database::v1::ListDatabaseOperationsResponse> ListDatabaseOperations(
       grpc::ClientContext& context,
       Options const& options,
@@ -134,6 +154,11 @@ class GoldenThingAdminAuth : public GoldenThingAdminStub {
       google::cloud::internal::ImmutableOptions options,
       google::test::admin::database::v1::RestoreDatabaseRequest const& request) override;
 
+  StatusOr<google::longrunning::Operation> LongRunningWithoutRouting(
+      grpc::ClientContext& context,
+      Options options,
+      google::test::admin::database::v1::RestoreDatabaseRequest const& request) override;
+
   future<StatusOr<google::test::admin::database::v1::Database>> AsyncGetDatabase(
       google::cloud::CompletionQueue& cq,
       std::shared_ptr<grpc::ClientContext> context,

generator/integration_tests/golden/v1/internal/golden_thing_admin_logging_decorator.cc

@@ -67,6 +67,20 @@ GoldenThingAdminLogging::AsyncCreateDatabase(
       tracing_options_);
 }
 
+StatusOr<google::longrunning::Operation>
+GoldenThingAdminLogging::CreateDatabase(
+      grpc::ClientContext& context,
+      Options options,
+      google::test::admin::database::v1::CreateDatabaseRequest const& request) {
+  return google::cloud::internal::LogWrapper(
+      [this](grpc::ClientContext& context,
+             Options const& options,
+             google::test::admin::database::v1::CreateDatabaseRequest const& request) {
+        return child_->CreateDatabase(context, options, request);
+      },
+      context, options, request, __func__, tracing_options_);
+}
+
 StatusOr<google::test::admin::database::v1::Database>
 GoldenThingAdminLogging::GetDatabase(
     grpc::ClientContext& context,
@@ -99,6 +113,20 @@ GoldenThingAdminLogging::AsyncUpdateDatabaseDdl(
       tracing_options_);
 }
 
+StatusOr<google::longrunning::Operation>
+GoldenThingAdminLogging::UpdateDatabaseDdl(
+      grpc::ClientContext& context,
+      Options options,
+      google::test::admin::database::v1::UpdateDatabaseDdlRequest const& request) {
+  return google::cloud::internal::LogWrapper(
+      [this](grpc::ClientContext& context,
+             Options const& options,
+             google::test::admin::database::v1::UpdateDatabaseDdlRequest const& request) {
+        return child_->UpdateDatabaseDdl(context, options, request);
+      },
+      context, options, request, __func__, tracing_options_);
+}
+
 Status
 GoldenThingAdminLogging::DropDatabase(
     grpc::ClientContext& context,
@@ -187,6 +215,20 @@ GoldenThingAdminLogging::AsyncCreateBackup(
       tracing_options_);
 }
 
+StatusOr<google::longrunning::Operation>
+GoldenThingAdminLogging::CreateBackup(
+      grpc::ClientContext& context,
+      Options options,
+      google::test::admin::database::v1::CreateBackupRequest const& request) {
+  return google::cloud::internal::LogWrapper(
+      [this](grpc::ClientContext& context,
+             Options const& options,
+             google::test::admin::database::v1::CreateBackupRequest const& request) {
+        return child_->CreateBackup(context, options, request);
+      },
+      context, options, request, __func__, tracing_options_);
+}
+
 StatusOr<google::test::admin::database::v1::Backup>
 GoldenThingAdminLogging::GetBackup(
     grpc::ClientContext& context,
@@ -261,6 +303,20 @@ GoldenThingAdminLogging::AsyncRestoreDatabase(
       tracing_options_);
 }
 
+StatusOr<google::longrunning::Operation>
+GoldenThingAdminLogging::RestoreDatabase(
+      grpc::ClientContext& context,
+      Options options,
+      google::test::admin::database::v1::RestoreDatabaseRequest const& request) {
+  return google::cloud::internal::LogWrapper(
+      [this](grpc::ClientContext& context,
+             Options const& options,
+             google::test::admin::database::v1::RestoreDatabaseRequest const& request) {
+        return child_->RestoreDatabase(context, options, request);
+      },
+      context, options, request, __func__, tracing_options_);
+}
+
 StatusOr<google::test::admin::database::v1::ListDatabaseOperationsResponse>
 GoldenThingAdminLogging::ListDatabaseOperations(
     grpc::ClientContext& context,
@@ -307,6 +363,20 @@ GoldenThingAdminLogging::AsyncLongRunningWithoutRouting(
       tracing_options_);
 }
 
+StatusOr<google::longrunning::Operation>
+GoldenThingAdminLogging::LongRunningWithoutRouting(
+      grpc::ClientContext& context,
+      Options options,
+      google::test::admin::database::v1::RestoreDatabaseRequest const& request) {
+  return google::cloud::internal::LogWrapper(
+      [this](grpc::ClientContext& context,
+             Options const& options,
+             google::test::admin::database::v1::RestoreDatabaseRequest const& request) {
+        return child_->LongRunningWithoutRouting(context, options, request);
+      },
+      context, options, request, __func__, tracing_options_);
+}
+
 future<StatusOr<google::test::admin::database::v1::Database>>
 GoldenThingAdminLogging::AsyncGetDatabase(
       google::cloud::CompletionQueue& cq,

generator/integration_tests/golden/v1/internal/golden_thing_admin_logging_decorator.h

@@ -50,6 +50,11 @@ class GoldenThingAdminLogging : public GoldenThingAdminStub {
       google::cloud::internal::ImmutableOptions options,
       google::test::admin::database::v1::CreateDatabaseRequest const& request) override;
 
+  StatusOr<google::longrunning::Operation> CreateDatabase(
+      grpc::ClientContext& context,
+      Options options,
+      google::test::admin::database::v1::CreateDatabaseRequest const& request) override;
+
   StatusOr<google::test::admin::database::v1::Database> GetDatabase(
       grpc::ClientContext& context,
       Options const& options,
@@ -61,6 +66,11 @@ class GoldenThingAdminLogging : public GoldenThingAdminStub {
       google::cloud::internal::ImmutableOptions options,
       google::test::admin::database::v1::UpdateDatabaseDdlRequest const& request) override;
 
+  StatusOr<google::longrunning::Operation> UpdateDatabaseDdl(
+      grpc::ClientContext& context,
+      Options options,
+      google::test::admin::database::v1::UpdateDatabaseDdlRequest const& request) override;
+
   Status DropDatabase(
       grpc::ClientContext& context,
       Options const& options,
@@ -92,6 +102,11 @@ class GoldenThingAdminLogging : public GoldenThingAdminStub {
       google::cloud::internal::ImmutableOptions options,
       google::test::admin::database::v1::CreateBackupRequest const& request) override;
 
+  StatusOr<google::longrunning::Operation> CreateBackup(
+      grpc::ClientContext& context,
+      Options options,
+      google::test::admin::database::v1::CreateBackupRequest const& request) override;
+
   StatusOr<google::test::admin::database::v1::Backup> GetBackup(
       grpc::ClientContext& context,
       Options const& options,
@@ -118,6 +133,11 @@ class GoldenThingAdminLogging : public GoldenThingAdminStub {
       google::cloud::internal::ImmutableOptions options,
       google::test::admin::database::v1::RestoreDatabaseRequest const& request) override;
 
+  StatusOr<google::longrunning::Operation> RestoreDatabase(
+      grpc::ClientContext& context,
+      Options options,
+      google::test::admin::database::v1::RestoreDatabaseRequest const& request) override;
+
   StatusOr<google::test::admin::database::v1::ListDatabaseOperationsResponse> ListDatabaseOperations(
       grpc::ClientContext& context,
       Options const& options,
@@ -134,6 +154,11 @@ class GoldenThingAdminLogging : public GoldenThingAdminStub {
       google::cloud::internal::ImmutableOptions options,
       google::test::admin::database::v1::RestoreDatabaseRequest const& request) override;
 
+  StatusOr<google::longrunning::Operation> LongRunningWithoutRouting(
+      grpc::ClientContext& context,
+      Options options,
+      google::test::admin::database::v1::RestoreDatabaseRequest const& request) override;
+
   future<StatusOr<google::test::admin::database::v1::Database>> AsyncGetDatabase(
       google::cloud::CompletionQueue& cq,
       std::shared_ptr<grpc::ClientContext> context,

generator/integration_tests/golden/v1/internal/golden_thing_admin_metadata_decorator.cc

@@ -64,6 +64,15 @@ GoldenThingAdminMetadata::AsyncCreateDatabase(
       cq, std::move(context), std::move(options), request);
 }
 
+StatusOr<google::longrunning::Operation>
+GoldenThingAdminMetadata::CreateDatabase(
+    grpc::ClientContext& context,
+    Options options,
+    google::test::admin::database::v1::CreateDatabaseRequest const& request) {
+  SetMetadata(context, options, absl::StrCat("parent=", internal::UrlEncode(request.parent())));
+  return child_->CreateDatabase(context, options, request);
+}
+
 StatusOr<google::test::admin::database::v1::Database>
 GoldenThingAdminMetadata::GetDatabase(
     grpc::ClientContext& context,
@@ -84,6 +93,15 @@ GoldenThingAdminMetadata::AsyncUpdateDatabaseDdl(
       cq, std::move(context), std::move(options), request);
 }
 
+StatusOr<google::longrunning::Operation>
+GoldenThingAdminMetadata::UpdateDatabaseDdl(
+    grpc::ClientContext& context,
+    Options options,
+    google::test::admin::database::v1::UpdateDatabaseDdlRequest const& request) {
+  SetMetadata(context, options, absl::StrCat("database=", internal::UrlEncode(request.database())));
+  return child_->UpdateDatabaseDdl(context, options, request);
+}
+
 Status
 GoldenThingAdminMetadata::DropDatabase(
     grpc::ClientContext& context,
@@ -180,6 +198,15 @@ GoldenThingAdminMetadata::AsyncCreateBackup(
       cq, std::move(context), std::move(options), request);
 }
 
+StatusOr<google::longrunning::Operation>
+GoldenThingAdminMetadata::CreateBackup(
+    grpc::ClientContext& context,
+    Options options,
+    google::test::admin::database::v1::CreateBackupRequest const& request) {
+  SetMetadata(context, options, absl::StrCat("parent=", internal::UrlEncode(request.parent())));
+  return child_->CreateBackup(context, options, request);
+}
+
 StatusOr<google::test::admin::database::v1::Backup>
 GoldenThingAdminMetadata::GetBackup(
     grpc::ClientContext& context,
@@ -227,6 +254,15 @@ GoldenThingAdminMetadata::AsyncRestoreDatabase(
       cq, std::move(context), std::move(options), request);
 }
 
+StatusOr<google::longrunning::Operation>
+GoldenThingAdminMetadata::RestoreDatabase(
+    grpc::ClientContext& context,
+    Options options,
+    google::test::admin::database::v1::RestoreDatabaseRequest const& request) {
+  SetMetadata(context, options, absl::StrCat("parent=", internal::UrlEncode(request.parent())));
+  return child_->RestoreDatabase(context, options, request);
+}
+
 StatusOr<google::test::admin::database::v1::ListDatabaseOperationsResponse>
 GoldenThingAdminMetadata::ListDatabaseOperations(
     grpc::ClientContext& context,
@@ -256,6 +292,15 @@ GoldenThingAdminMetadata::AsyncLongRunningWithoutRouting(
       cq, std::move(context), std::move(options), request);
 }
 
+StatusOr<google::longrunning::Operation>
+GoldenThingAdminMetadata::LongRunningWithoutRouting(
+    grpc::ClientContext& context,
+    Options options,
+    google::test::admin::database::v1::RestoreDatabaseRequest const& request) {
+  SetMetadata(context, options);
+  return child_->LongRunningWithoutRouting(context, options, request);
+}
+
 future<StatusOr<google::test::admin::database::v1::Database>>
 GoldenThingAdminMetadata::AsyncGetDatabase(
       google::cloud::CompletionQueue& cq,