-
Notifications
You must be signed in to change notification settings - Fork 607
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
10 changed files
with
209 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,112 @@ | ||
| #include "fetch_database.h" | ||
|
|
||
| #include <library/cpp/retry/retry_policy.h> | ||
|
|
||
| #include <ydb/core/base/appdata_fwd.h> | ||
|
|
||
| #include <ydb/library/table_creator/table_creator.h> | ||
|
|
||
|
|
||
| namespace NKikimr::NMetadata::NModifications { | ||
|
|
||
| namespace { | ||
|
|
||
| class TDatabaseFetcherActor : public TActorBootstrapped<TDatabaseFetcherActor> { | ||
| using TBase = TActorBootstrapped<TDatabaseFetcherActor>; | ||
| using TRetryPolicy = IRetryPolicy<>; | ||
|
|
||
| public: | ||
| explicit TDatabaseFetcherActor(const TString& database) | ||
| : Database(database) | ||
| {} | ||
|
|
||
| void Registered(TActorSystem* sys, const TActorId& owner) override { | ||
| TBase::Registered(sys, owner); | ||
| Owner = owner; | ||
| } | ||
|
|
||
| void Bootstrap() { | ||
| StartRequest(); | ||
| Become(&TDatabaseFetcherActor::StateFunc); | ||
| } | ||
|
|
||
| void Handle(TEvents::TEvUndelivered::TPtr& ev) { | ||
| if (ev->Get()->Reason == NActors::TEvents::TEvUndelivered::ReasonActorUnknown && ScheduleRetry()) { | ||
| return; | ||
| } | ||
|
|
||
| Reply("Scheme cache is unavailable"); | ||
| } | ||
|
|
||
| void Handle(TEvTxProxySchemeCache::TEvNavigateKeySetResult::TPtr& ev) { | ||
| const auto& results = ev->Get()->Request->ResultSet; | ||
| Y_ABORT_UNLESS(results.size() == 1); | ||
|
|
||
| const auto& result = results[0]; | ||
| if (result.DomainInfo) { | ||
| Serverless = result.DomainInfo->IsServerless(); | ||
| Reply(); | ||
| return; | ||
| } | ||
|
|
||
| if (result.Status == NSchemeCache::TSchemeCacheNavigate::EStatus::LookupError && ScheduleRetry()) { | ||
| return; | ||
| } | ||
|
|
||
| Reply(TStringBuilder() << "Failed to fetch database info: " << result.Status); | ||
| } | ||
|
|
||
| STRICT_STFUNC(StateFunc, | ||
| sFunc(TEvents::TEvWakeup, StartRequest); | ||
| hFunc(TEvents::TEvUndelivered, Handle); | ||
| hFunc(TEvTxProxySchemeCache::TEvNavigateKeySetResult, Handle); | ||
| ) | ||
|
|
||
| private: | ||
| void StartRequest() { | ||
| auto event = NTableCreator::BuildSchemeCacheNavigateRequest( | ||
| {{}}, | ||
| Database ? Database : AppData()->TenantName, | ||
| MakeIntrusive<NACLib::TUserToken>(BUILTIN_ACL_METADATA, TVector<NACLib::TSID>{}) | ||
| ); | ||
| event->ResultSet[0].Operation = NSchemeCache::TSchemeCacheNavigate::OpPath; | ||
| Send(MakeSchemeCacheID(), new TEvTxProxySchemeCache::TEvNavigateKeySet(event.Release()), IEventHandle::FlagTrackDelivery); | ||
| } | ||
|
|
||
| bool ScheduleRetry() { | ||
| if (!RetryState) { | ||
| RetryState = TRetryPolicy::GetFixedIntervalPolicy( | ||
| [](){return ERetryErrorClass::ShortRetry;} | ||
| , TDuration::MilliSeconds(100) | ||
| , TDuration::MilliSeconds(500) | ||
| , 100 | ||
| )->CreateRetryState();; | ||
| } | ||
|
|
||
| if (const auto delay = RetryState->GetNextRetryDelay()) { | ||
| this->Schedule(*delay, new TEvents::TEvWakeup()); | ||
| return true; | ||
| } | ||
|
|
||
| return false; | ||
| } | ||
|
|
||
| void Reply(const std::optional<TString>& errorMessage = std::nullopt) { | ||
| Send(Owner, new TEvFetchDatabaseResponse(Serverless, errorMessage)); | ||
| PassAway(); | ||
| } | ||
|
|
||
| private: | ||
| const TString Database; | ||
| TActorId Owner; | ||
| TRetryPolicy::IRetryState::TPtr RetryState; | ||
| bool Serverless = false; | ||
| }; | ||
|
|
||
| } // anonymous namespace | ||
|
|
||
| IActor* CreateDatabaseFetcherActor(const TString& database) { | ||
| return new TDatabaseFetcherActor(database); | ||
| } | ||
|
|
||
| } // NKikimr::NMetadata::NModifications |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,24 @@ | ||
| #pragma once | ||
|
|
||
| #include "common.h" | ||
|
|
||
| #include <ydb/library/accessor/accessor.h> | ||
|
|
||
|
|
||
| namespace NKikimr::NMetadata::NModifications { | ||
|
|
||
| class TEvFetchDatabaseResponse : public TEventLocal<TEvFetchDatabaseResponse, EvFetchDatabaseResponse> { | ||
| private: | ||
| YDB_READONLY_DEF(bool, Serverless); | ||
| YDB_READONLY_DEF(std::optional<TString>, ErrorString); | ||
|
|
||
| public: | ||
| TEvFetchDatabaseResponse(bool serverless, const std::optional<TString>& errorString) | ||
| : Serverless(serverless) | ||
| , ErrorString(errorString) | ||
| {} | ||
| }; | ||
|
|
||
| IActor* CreateDatabaseFetcherActor(const TString& database); | ||
|
|
||
| } // NKikimr::NMetadata::NModifications |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters