Folding improvement (#19832)

Author: dorooleg

ydb/core/kqp/opt/kqp_constant_folding_transformer.cpp

@@ -39,6 +39,13 @@ namespace {
             return;
         }
 
+        if (TCoAsStruct::Match(input.Get())) {
+            for (auto child : TExprBase(input).Cast<TCoAsStruct>()) {
+                ExtractConstantExprs(child.Item(1).Ptr(), replaces, ctx);
+            }
+            return;
+        }
+
         if (input->IsCallable() && input->Content() != "EvaluateExpr") {
             if (input->ChildrenSize() >= 1) {
                 for (size_t i = 0; i < input->ChildrenSize(); i++) {

ydb/core/kqp/ut/olap/kqp_olap_ut.cpp

@@ -3887,5 +3887,56 @@ Y_UNIT_TEST_SUITE(KqpOlap) {
             CompareYson(result, R"([[20u]])");
         }
     }
+
+    Y_UNIT_TEST(GroupByWithMakeDatetime) {
+        auto settings = TKikimrSettings()
+            .SetWithSampleTables(false);
+        TKikimrRunner kikimr(settings);
+
+        auto tableClient = kikimr.GetTableClient();
+        auto tableSession = tableClient.CreateSession().GetValueSync().GetSession();
+
+        auto queryClient = kikimr.GetQueryClient();
+        auto result = queryClient.GetSession().GetValueSync();
+        NStatusHelpers::ThrowOnError(result);
+        auto querySession = result.GetSession();
+
+        {
+            auto result = tableSession.ExecuteSchemeQuery(R"(
+                CREATE TABLE `/Root/query_stat` (
+                    ts      Timestamp NOT NULL,
+                    folder_id String,
+                    primary key(ts)	
+                )
+                PARTITION BY HASH(ts)
+                WITH (STORE = COLUMN);
+            )").GetValueSync();
+            UNIT_ASSERT_C(result.IsSuccess(), result.GetIssues().ToString());
+        }
+
+        {
+            auto result = querySession.ExecuteQuery(R"(
+                    INSERT INTO `/Root/query_stat` (ts, folder_id) 
+                    VALUES (
+                        CurrentUtcTimestamp(),
+                        "abc"
+                    )
+                )", NYdb::NQuery::TTxControl::NoTx()).GetValueSync();
+            UNIT_ASSERT_C(result.IsSuccess(), result.GetIssues().ToString());
+        }
+
+        {
+            auto result = querySession.ExecuteQuery(R"(
+                    SELECT 
+                        ts1, count(*)
+                    FROM 
+                        query_stat 
+                    where 
+                        folder_id not in [ "b1g0gammoel2iuh0hir6" ] 
+                    GROUP BY DateTime::MakeDatetime(DateTime::StartOf(ts, DateTime::IntervalFromDays(1))) as ts1
+                )", NYdb::NQuery::TTxControl::NoTx()).GetValueSync();
+            UNIT_ASSERT_C(result.IsSuccess(), result.GetIssues().ToString());
+        }
+    }
 }
 }