From dee5ef359ee7e109cc0112fbfa66ab696ecf489e Mon Sep 17 00:00:00 2001
From: vvvv <vvvv@yandex-team.com>
Date: Mon, 3 Feb 2025 18:31:19 +0300
Subject: Support Udf settings along with ScriptUdf in YT expr traits

init
commit_hash:ebd39f1619c60fc10e21961b0df0c4539ff2b991
---
 yt/yql/providers/yt/lib/expr_traits/yql_expr_traits.cpp | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

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)) {
-- 
cgit v1.3