Skip to content

Commit 6bc1b72

Browse files
authored
Add RemoteData to YQL operation progress. Add cluster_name for YT YQL operations (#4648)
1 parent de53293 commit 6bc1b72

File tree

4 files changed

+17
-1
lines changed

4 files changed

+17
-1
lines changed

ydb/library/yql/core/progress_merger/progress_merger.cpp

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,12 @@ bool TNodeProgressBase::MergeWith(const TOperationProgress& p) {
6464
Stages_.push_back(p.Stage);
6565
dirty = true;
6666
}
67+
68+
// (6) remote data
69+
if (!p.RemoteData.empty() && p.RemoteData != Progress_.RemoteData) {
70+
Progress_.RemoteData = p.RemoteData;
71+
dirty = true;
72+
}
6773
return Dirty_ = dirty;
6874
}
6975

ydb/library/yql/core/yql_execution.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@ namespace NYql {
4040
TStage Stage;
4141

4242
TString RemoteId;
43+
THashMap<TString, TString> RemoteData;
4344

4445
struct TCounters {
4546
ui64 Completed = 0ULL;

ydb/library/yql/providers/yt/gateway/native/yql_yt_op_tracker.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ TFuture<void> TOperationTracker::MakeOperationWaiter(const NYT::IOperationPtr& o
9898
auto filter = NYT::TOperationAttributeFilter();
9999
filter.Add(NYT::EOperationAttribute::State);
100100

101-
auto checker = [future, operation, ytServer, progress, progressWriter, filter] () mutable {
101+
auto checker = [future, operation, ytServer, progress, progressWriter, filter, ytClusterName] () mutable {
102102
bool done = future.Wait(TDuration::Zero());
103103

104104
if (!done) {
@@ -108,6 +108,7 @@ TFuture<void> TOperationTracker::MakeOperationWaiter(const NYT::IOperationPtr& o
108108
if (!progress.RemoteId) {
109109
progress.RemoteId = ytServer + "/" + GetGuidAsString(operation->GetId());
110110
}
111+
progress.RemoteData["cluster_name"] = ytClusterName;
111112
if (auto briefProgress = operation->GetBriefProgress()) {
112113
progress.Counters.ConstructInPlace();
113114
progress.Counters->Completed = briefProgress->Completed;

ydb/library/yql/yt/native/progress_merger.cpp

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,14 @@ void TNodeProgress::Serialize(::NYson::TYsonWriter& writer) const
2121
writer.OnKeyedItem("remoteId");
2222
writer.OnStringScalar(Progress_.RemoteId);
2323

24+
writer.OnKeyedItem("remoteData");
25+
writer.OnBeginMap();
26+
for (const auto& it : Progress_.RemoteData) {
27+
writer.OnKeyedItem(it.first);
28+
writer.OnStringScalar(it.second);
29+
}
30+
writer.OnEndMap();
31+
2432
writer.OnKeyedItem("stages");
2533
writer.OnBeginMap();
2634
for (size_t index = 0; index < Stages_.size(); index++) {

0 commit comments

Comments
 (0)