diff options
| -rw-r--r-- | yt/yql/providers/yt/lib/expr_traits/yql_expr_traits.cpp | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/yt/yql/providers/yt/lib/expr_traits/yql_expr_traits.cpp b/yt/yql/providers/yt/lib/expr_traits/yql_expr_traits.cpp index 0e2e6de57fb..a55d9baf659 100644 --- a/yt/yql/providers/yt/lib/expr_traits/yql_expr_traits.cpp +++ b/yt/yql/providers/yt/lib/expr_traits/yql_expr_traits.cpp @@ -70,6 +70,21 @@ namespace NYql { if (memoryUsage && node.Child(0)->Content().StartsWith("UserSessions.")) { (*memoryUsage)["UserSessions module"] = 512_MB; // Take into account only once } + + if (cpuUsage) { + if (auto cpuSetting = GetSetting(*node.Child(7), "cpu")) { + double usage = FromString<double>(cpuSetting->Child(1)->Content()); + if (auto prev = cpuUsage->FindPtr("Udf from settings arg")) { + usage *= *prev; + } + (*cpuUsage)["Udf from settings arg"] = usage; + } + } + if (memoryUsage) { + if (auto extraMemSetting = GetSetting(*node.Child(7), "extraMem")) { + (*memoryUsage)["Udf from settings arg"] += FromString<ui64>(extraMemSetting->Child(1)->Content()); + } + } } if (NNodes::TYtTableContent::Match(&node)) { |