Skip to content

Commit c91821e

Browse files
scotthartscotthart
authored
impl(generator): add Start and Await overloads to connections (googleapis#14348)
1 parent 309b263 commit c91821e

2,030 files changed

Lines changed: 201049 additions & 1525 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

‎.typos.toml‎

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,5 +61,8 @@ extend-ignore-identifiers-re = [
6161
# This pattern can be found in long hex strings.
6262
"[0-9]caf[0-9]",
6363
# This is a GCS dual-region name.
64-
"NAM4"
64+
"NAM4",
65+
# Metadata is mispelled in this proto message name found in
66+
# google/cloud/aiplatform/v1/vizier_service.proto
67+
"CheckTrialEarlyStoppingStateMetatdata"
6568
]

‎generator/integration_tests/golden/v1/golden_kitchen_sink_connection.h‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
#include "generator/integration_tests/golden/v1/golden_kitchen_sink_connection_idempotency_policy.h"
2323
#include "generator/integration_tests/golden/v1/internal/golden_kitchen_sink_retry_traits.h"
2424
#include "google/cloud/backoff_policy.h"
25+
#include "google/cloud/experimental_tag.h"
2526
#include "google/cloud/future.h"
2627
#include "google/cloud/internal/async_read_write_stream_impl.h"
2728
#include "google/cloud/internal/retry_policy_impl.h"

‎generator/integration_tests/golden/v1/golden_rest_only_connection.h‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
#include "generator/integration_tests/golden/v1/golden_rest_only_connection_idempotency_policy.h"
2323
#include "generator/integration_tests/golden/v1/internal/golden_rest_only_retry_traits.h"
2424
#include "google/cloud/backoff_policy.h"
25+
#include "google/cloud/experimental_tag.h"
2526
#include "google/cloud/internal/retry_policy_impl.h"
2627
#include "google/cloud/options.h"
2728
#include "google/cloud/status_or.h"

‎generator/integration_tests/golden/v1/golden_thing_admin_connection.cc‎

Lines changed: 80 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,22 @@ GoldenThingAdminConnection::CreateDatabase(
5252
Status(StatusCode::kUnimplemented, "not implemented"));
5353
}
5454

55+
StatusOr<google::longrunning::Operation>
56+
GoldenThingAdminConnection::CreateDatabase(
57+
ExperimentalTag, NoAwaitTag,
58+
google::test::admin::database::v1::CreateDatabaseRequest const&) {
59+
return StatusOr<google::longrunning::Operation>(
60+
Status(StatusCode::kUnimplemented, "not implemented"));
61+
}
62+
63+
future<StatusOr<google::test::admin::database::v1::Database>>
64+
GoldenThingAdminConnection::CreateDatabase(
65+
ExperimentalTag, google::longrunning::Operation const&) {
66+
return google::cloud::make_ready_future<
67+
StatusOr<google::test::admin::database::v1::Database>>(
68+
Status(StatusCode::kUnimplemented, "not implemented"));
69+
}
70+
5571
StatusOr<google::test::admin::database::v1::Database>
5672
GoldenThingAdminConnection::GetDatabase(
5773
google::test::admin::database::v1::GetDatabaseRequest const&) {
@@ -66,6 +82,22 @@ GoldenThingAdminConnection::UpdateDatabaseDdl(
6682
Status(StatusCode::kUnimplemented, "not implemented"));
6783
}
6884

85+
StatusOr<google::longrunning::Operation>
86+
GoldenThingAdminConnection::UpdateDatabaseDdl(
87+
ExperimentalTag, NoAwaitTag,
88+
google::test::admin::database::v1::UpdateDatabaseDdlRequest const&) {
89+
return StatusOr<google::longrunning::Operation>(
90+
Status(StatusCode::kUnimplemented, "not implemented"));
91+
}
92+
93+
future<StatusOr<google::test::admin::database::v1::UpdateDatabaseDdlMetadata>>
94+
GoldenThingAdminConnection::UpdateDatabaseDdl(
95+
ExperimentalTag, google::longrunning::Operation const&) {
96+
return google::cloud::make_ready_future<
97+
StatusOr<google::test::admin::database::v1::UpdateDatabaseDdlMetadata>>(
98+
Status(StatusCode::kUnimplemented, "not implemented"));
99+
}
100+
69101
Status
70102
GoldenThingAdminConnection::DropDatabase(
71103
google::test::admin::database::v1::DropDatabaseRequest const&) {
@@ -104,6 +136,22 @@ GoldenThingAdminConnection::CreateBackup(
104136
Status(StatusCode::kUnimplemented, "not implemented"));
105137
}
106138

139+
StatusOr<google::longrunning::Operation>
140+
GoldenThingAdminConnection::CreateBackup(
141+
ExperimentalTag, NoAwaitTag,
142+
google::test::admin::database::v1::CreateBackupRequest const&) {
143+
return StatusOr<google::longrunning::Operation>(
144+
Status(StatusCode::kUnimplemented, "not implemented"));
145+
}
146+
147+
future<StatusOr<google::test::admin::database::v1::Backup>>
148+
GoldenThingAdminConnection::CreateBackup(
149+
ExperimentalTag, google::longrunning::Operation const&) {
150+
return google::cloud::make_ready_future<
151+
StatusOr<google::test::admin::database::v1::Backup>>(
152+
Status(StatusCode::kUnimplemented, "not implemented"));
153+
}
154+
107155
StatusOr<google::test::admin::database::v1::Backup>
108156
GoldenThingAdminConnection::GetBackup(
109157
google::test::admin::database::v1::GetBackupRequest const&) {
@@ -136,6 +184,22 @@ GoldenThingAdminConnection::RestoreDatabase(
136184
Status(StatusCode::kUnimplemented, "not implemented"));
137185
}
138186

187+
StatusOr<google::longrunning::Operation>
188+
GoldenThingAdminConnection::RestoreDatabase(
189+
ExperimentalTag, NoAwaitTag,
190+
google::test::admin::database::v1::RestoreDatabaseRequest const&) {
191+
return StatusOr<google::longrunning::Operation>(
192+
Status(StatusCode::kUnimplemented, "not implemented"));
193+
}
194+
195+
future<StatusOr<google::test::admin::database::v1::Database>>
196+
GoldenThingAdminConnection::RestoreDatabase(
197+
ExperimentalTag, google::longrunning::Operation const&) {
198+
return google::cloud::make_ready_future<
199+
StatusOr<google::test::admin::database::v1::Database>>(
200+
Status(StatusCode::kUnimplemented, "not implemented"));
201+
}
202+
139203
StreamRange<google::longrunning::Operation> GoldenThingAdminConnection::ListDatabaseOperations(
140204
google::test::admin::database::v1::ListDatabaseOperationsRequest) { // NOLINT(performance-unnecessary-value-param)
141205
return google::cloud::internal::MakeUnimplementedPaginationRange<
@@ -156,6 +220,22 @@ GoldenThingAdminConnection::LongRunningWithoutRouting(
156220
Status(StatusCode::kUnimplemented, "not implemented"));
157221
}
158222

223+
StatusOr<google::longrunning::Operation>
224+
GoldenThingAdminConnection::LongRunningWithoutRouting(
225+
ExperimentalTag, NoAwaitTag,
226+
google::test::admin::database::v1::RestoreDatabaseRequest const&) {
227+
return StatusOr<google::longrunning::Operation>(
228+
Status(StatusCode::kUnimplemented, "not implemented"));
229+
}
230+
231+
future<StatusOr<google::test::admin::database::v1::Database>>
232+
GoldenThingAdminConnection::LongRunningWithoutRouting(
233+
ExperimentalTag, google::longrunning::Operation const&) {
234+
return google::cloud::make_ready_future<
235+
StatusOr<google::test::admin::database::v1::Database>>(
236+
Status(StatusCode::kUnimplemented, "not implemented"));
237+
}
238+
159239
future<StatusOr<google::test::admin::database::v1::Database>>
160240
GoldenThingAdminConnection::AsyncGetDatabase(
161241
google::test::admin::database::v1::GetDatabaseRequest const&) {

‎generator/integration_tests/golden/v1/golden_thing_admin_connection.h‎

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,10 @@
2222
#include "generator/integration_tests/golden/v1/golden_thing_admin_connection_idempotency_policy.h"
2323
#include "generator/integration_tests/golden/v1/internal/golden_thing_admin_retry_traits.h"
2424
#include "google/cloud/backoff_policy.h"
25+
#include "google/cloud/experimental_tag.h"
2526
#include "google/cloud/future.h"
2627
#include "google/cloud/internal/retry_policy_impl.h"
28+
#include "google/cloud/no_await_tag.h"
2729
#include "google/cloud/options.h"
2830
#include "google/cloud/polling_policy.h"
2931
#include "google/cloud/status_or.h"
@@ -187,12 +189,24 @@ class GoldenThingAdminConnection {
187189
virtual future<StatusOr<google::test::admin::database::v1::Database>>
188190
CreateDatabase(google::test::admin::database::v1::CreateDatabaseRequest const& request);
189191

192+
virtual StatusOr<google::longrunning::Operation>
193+
CreateDatabase(ExperimentalTag, NoAwaitTag, google::test::admin::database::v1::CreateDatabaseRequest const& request);
194+
195+
virtual future<StatusOr<google::test::admin::database::v1::Database>>
196+
CreateDatabase(ExperimentalTag, google::longrunning::Operation const& operation);
197+
190198
virtual StatusOr<google::test::admin::database::v1::Database>
191199
GetDatabase(google::test::admin::database::v1::GetDatabaseRequest const& request);
192200

193201
virtual future<StatusOr<google::test::admin::database::v1::UpdateDatabaseDdlMetadata>>
194202
UpdateDatabaseDdl(google::test::admin::database::v1::UpdateDatabaseDdlRequest const& request);
195203

204+
virtual StatusOr<google::longrunning::Operation>
205+
UpdateDatabaseDdl(ExperimentalTag, NoAwaitTag, google::test::admin::database::v1::UpdateDatabaseDdlRequest const& request);
206+
207+
virtual future<StatusOr<google::test::admin::database::v1::UpdateDatabaseDdlMetadata>>
208+
UpdateDatabaseDdl(ExperimentalTag, google::longrunning::Operation const& operation);
209+
196210
virtual Status
197211
DropDatabase(google::test::admin::database::v1::DropDatabaseRequest const& request);
198212

@@ -211,6 +225,12 @@ class GoldenThingAdminConnection {
211225
virtual future<StatusOr<google::test::admin::database::v1::Backup>>
212226
CreateBackup(google::test::admin::database::v1::CreateBackupRequest const& request);
213227

228+
virtual StatusOr<google::longrunning::Operation>
229+
CreateBackup(ExperimentalTag, NoAwaitTag, google::test::admin::database::v1::CreateBackupRequest const& request);
230+
231+
virtual future<StatusOr<google::test::admin::database::v1::Backup>>
232+
CreateBackup(ExperimentalTag, google::longrunning::Operation const& operation);
233+
214234
virtual StatusOr<google::test::admin::database::v1::Backup>
215235
GetBackup(google::test::admin::database::v1::GetBackupRequest const& request);
216236

@@ -226,6 +246,12 @@ class GoldenThingAdminConnection {
226246
virtual future<StatusOr<google::test::admin::database::v1::Database>>
227247
RestoreDatabase(google::test::admin::database::v1::RestoreDatabaseRequest const& request);
228248

249+
virtual StatusOr<google::longrunning::Operation>
250+
RestoreDatabase(ExperimentalTag, NoAwaitTag, google::test::admin::database::v1::RestoreDatabaseRequest const& request);
251+
252+
virtual future<StatusOr<google::test::admin::database::v1::Database>>
253+
RestoreDatabase(ExperimentalTag, google::longrunning::Operation const& operation);
254+
229255
virtual StreamRange<google::longrunning::Operation>
230256
ListDatabaseOperations(google::test::admin::database::v1::ListDatabaseOperationsRequest request);
231257

@@ -235,6 +261,12 @@ class GoldenThingAdminConnection {
235261
virtual future<StatusOr<google::test::admin::database::v1::Database>>
236262
LongRunningWithoutRouting(google::test::admin::database::v1::RestoreDatabaseRequest const& request);
237263

264+
virtual StatusOr<google::longrunning::Operation>
265+
LongRunningWithoutRouting(ExperimentalTag, NoAwaitTag, google::test::admin::database::v1::RestoreDatabaseRequest const& request);
266+
267+
virtual future<StatusOr<google::test::admin::database::v1::Database>>
268+
LongRunningWithoutRouting(ExperimentalTag, google::longrunning::Operation const& operation);
269+
238270
virtual future<StatusOr<google::test::admin::database::v1::Database>>
239271
AsyncGetDatabase(google::test::admin::database::v1::GetDatabaseRequest const& request);
240272

0 commit comments

Comments
 (0)