diff options
| author | max42 <[email protected]> | 2023-06-30 03:37:03 +0300 |
|---|---|---|
| committer | max42 <[email protected]> | 2023-06-30 03:37:03 +0300 |
| commit | fac2bd72b4b31ec3238292caf8fb2a8aaa6d6c4a (patch) | |
| tree | b8cbc1deb00309c7f1a7ab6df520a76cf0b5c6d7 /yt/cpp/mapreduce/client/operation_helpers.cpp | |
| parent | 7bf166b1a7ed0af927f230022b245af618e998c1 (diff) | |
YT-19324: move YT provider to ydb/library/yql
This commit is formed by the following script: https://paste.yandex-team.ru/6f92e4b8-efc5-4d34-948b-15ee2accd7e7/text.
This commit has zero effect on all projects that depend on YQL.
The summary of changes:
- `yql/providers/yt -> ydb/library/yql/providers/yt `- the whole implementation of YT provider is moved into YDB code base for further export as a part of YT YQL plugin shared library;
- `yql/providers/stat/{expr_nodes,uploader} -> ydb/library/yql/providers/stat/{expr_nodes,uploader}` - a small interface without implementation and the description of stat expr nodes;
- `yql/core/extract_predicate/ut -> ydb/library/yql/core/extract_predicate/ut`;
- `yql/core/{ut,ut_common} -> ydb/library/yql/core/{ut,ut_common}`;
- `yql/core` is gone;
- `yql/library/url_preprocessing -> ydb/library/yql/core/url_preprocessing`.
**NB**: all new targets inside `ydb/` are under `IF (NOT CMAKE_EXPORT)` clause which disables them from open-source cmake generation and ya make build. They will be enabled in the subsequent commits.
Diffstat (limited to 'yt/cpp/mapreduce/client/operation_helpers.cpp')
| -rw-r--r-- | yt/cpp/mapreduce/client/operation_helpers.cpp | 91 |
1 files changed, 91 insertions, 0 deletions
diff --git a/yt/cpp/mapreduce/client/operation_helpers.cpp b/yt/cpp/mapreduce/client/operation_helpers.cpp new file mode 100644 index 00000000000..abb21856622 --- /dev/null +++ b/yt/cpp/mapreduce/client/operation_helpers.cpp @@ -0,0 +1,91 @@ +#include "operation_helpers.h" + +#include <yt/cpp/mapreduce/common/retry_lib.h> + +#include <yt/cpp/mapreduce/interface/config.h> + +#include <yt/cpp/mapreduce/interface/logging/yt_log.h> + +#include <yt/cpp/mapreduce/raw_client/raw_requests.h> + +#include <yt/cpp/mapreduce/http/context.h> +#include <yt/cpp/mapreduce/http/requests.h> + +#include <util/string/builder.h> + +#include <util/system/mutex.h> +#include <util/system/rwlock.h> + +namespace NYT::NDetail { + +//////////////////////////////////////////////////////////////////////////////// + +ui64 RoundUpFileSize(ui64 size) +{ + constexpr ui64 roundUpTo = 4ull << 10; + return (size + roundUpTo - 1) & ~(roundUpTo - 1); +} + +bool UseLocalModeOptimization(const TClientContext& context, const IClientRetryPolicyPtr& clientRetryPolicy) +{ + if (!context.Config->EnableLocalModeOptimization) { + return false; + } + + static THashMap<TString, bool> localModeMap; + static TRWMutex mutex; + + { + TReadGuard guard(mutex); + auto it = localModeMap.find(context.ServerName); + if (it != localModeMap.end()) { + return it->second; + } + } + + bool isLocalMode = false; + TString localModeAttr("//sys/@local_mode_fqdn"); + // We don't want to pollute logs with errors about failed request, + // so we check if path exists before getting it. + if (NRawClient::Exists(clientRetryPolicy->CreatePolicyForGenericRequest(), + context, + TTransactionId(), + localModeAttr, + TExistsOptions().ReadFrom(EMasterReadKind::Cache))) + { + auto fqdnNode = NRawClient::TryGet( + clientRetryPolicy->CreatePolicyForGenericRequest(), + context, + TTransactionId(), + localModeAttr, + TGetOptions().ReadFrom(EMasterReadKind::Cache)); + if (!fqdnNode.IsUndefined()) { + auto fqdn = fqdnNode.AsString(); + isLocalMode = (fqdn == TProcessState::Get()->FqdnHostName); + YT_LOG_DEBUG("Checking local mode; LocalModeFqdn: %v FqdnHostName: %v IsLocalMode: %v", + fqdn, + TProcessState::Get()->FqdnHostName, + isLocalMode ? "true" : "false"); + } + } + + { + TWriteGuard guard(mutex); + localModeMap[context.ServerName] = isLocalMode; + } + + return isLocalMode; +} + +TString GetOperationWebInterfaceUrl(TStringBuf serverName, TOperationId operationId) +{ + serverName.ChopSuffix(":80"); + serverName.ChopSuffix(".yt.yandex-team.ru"); + serverName.ChopSuffix(".yt.yandex.net"); + return ::TStringBuilder() << "https://yt.yandex-team.ru/" << serverName << + "/operations/" << GetGuidAsString(operationId); +} + +//////////////////////////////////////////////////////////////////////////////// + +} // namespace NYT::NDetail |
