Merge pull request #11788 from ydb-platform/mergelibs-241120-1113

Library import 241120-1113

1 files changed, 21 insertions, 0 deletions

diff --git a/yql/essentials/tests/sql/suites/aggregate/group_by_session_extended.sql b/yql/essentials/tests/sql/suites/aggregate/group_by_session_extended.sql
new file mode 100644
index 0000000000..1b9af1d0d8
--- /dev/null
+++ b/yql/essentials/tests/sql/suites/aggregate/group_by_session_extended.sql
@@ -0,0 +1,21 @@
+/* syntax version 1 */
+/* postgres can not */
+
+$init = ($row) -> (AsStruct($row.ts ?? 0 as value, 1 as count));
+
+$calculate = ($_row, $state) -> ($state.value);
+-- split partition into two-element grooups, make session key to be cumulative sum of ts from partition start
+$update = ($row, $state) -> {
+  $state = AsStruct($state.count + 1 as count, $state.value as value);
+  $state = AsStruct($state.count as count, $state.value + ($row.ts ?? 0) as value);
+  return  AsTuple(Unwrap($state.count % 2) == 1, $state);
+};
+
+SELECT
+    user,
+    SessionStart() as session_start,
+    ListSort(AGGREGATE_LIST(ts ?? 100500)) as session,
+    COUNT(1) as session_len
+FROM plato.Input
+GROUP BY SessionWindow(ts, $init, $update, $calculate), user
+ORDER BY user, session_start;