Override aiplatform_v1::FeatureOnlineStoreServiceClient Retry Policies
This shows how to override the retry policies for aiplatform_v1::FeatureOnlineStoreServiceClient:
auto options =
google::cloud::Options{}
.set<google::cloud::aiplatform_v1::
FeatureOnlineStoreServiceConnectionIdempotencyPolicyOption>(
CustomIdempotencyPolicy().clone())
.set<google::cloud::aiplatform_v1::
FeatureOnlineStoreServiceRetryPolicyOption>(
google::cloud::aiplatform_v1::
FeatureOnlineStoreServiceLimitedErrorCountRetryPolicy(3)
.clone())
.set<google::cloud::aiplatform_v1::
FeatureOnlineStoreServiceBackoffPolicyOption>(
google::cloud::ExponentialBackoffPolicy(
/*initial_delay=*/std::chrono::milliseconds(200),
/*maximum_delay=*/std::chrono::seconds(45),
/*scaling=*/2.0)
.clone());
auto connection =
google::cloud::aiplatform_v1::MakeFeatureOnlineStoreServiceConnection(
"location-unused-in-this-example", options);
// c1 and c2 share the same retry policies
auto c1 =
google::cloud::aiplatform_v1::FeatureOnlineStoreServiceClient(connection);
auto c2 =
google::cloud::aiplatform_v1::FeatureOnlineStoreServiceClient(connection);
// You can override any of the policies in a new client. This new client
// will share the policies from c1 (or c2) *except* for the retry policy.
auto c3 = google::cloud::aiplatform_v1::FeatureOnlineStoreServiceClient(
connection, google::cloud::Options{}
.set<google::cloud::aiplatform_v1::
FeatureOnlineStoreServiceRetryPolicyOption>(
google::cloud::aiplatform_v1::
FeatureOnlineStoreServiceLimitedTimeRetryPolicy(
std::chrono::minutes(5))
.clone()));
// You can also override the policies in a single call:
// c3.SomeRpc(..., google::cloud::Options{}
// .set<google::cloud::aiplatform_v1::FeatureOnlineStoreServiceRetryPolicyOption>(
// google::cloud::aiplatform_v1::FeatureOnlineStoreServiceLimitedErrorCountRetryPolicy(10).clone()));
Assuming you have created a custom idempotency policy. Such as:
class CustomIdempotencyPolicy
: public google::cloud::aiplatform_v1::
FeatureOnlineStoreServiceConnectionIdempotencyPolicy {
public:
~CustomIdempotencyPolicy() override = default;
std::unique_ptr<google::cloud::aiplatform_v1::
FeatureOnlineStoreServiceConnectionIdempotencyPolicy>
clone() const override {
return std::make_unique<CustomIdempotencyPolicy>(*this);
}
// Override inherited functions to define as needed.
};