diff options
author | udovichenko-r <udovichenko-r@yandex-team.com> | 2024-11-19 14:58:38 +0300 |
---|---|---|
committer | udovichenko-r <udovichenko-r@yandex-team.com> | 2024-11-19 15:16:27 +0300 |
commit | 24521403b1c44303e043ba540c09b1fe991c7474 (patch) | |
tree | 341d1e7206bc7c143d04d2d96f05b6dc0655606d /yql/essentials/tests/sql/suites/aggr_factory/udaf.sql | |
parent | 72b3cd51dc3fb9d16975d353ea82fd85701393cc (diff) | |
download | ydb-24521403b1c44303e043ba540c09b1fe991c7474.tar.gz |
YQL-19206 Move contrib/ydb/library/yql/tests/sql/suites -> yql/essentials/tests/sql/suites
commit_hash:d0ef1f92b09c94db7c2408f946d2a4c62b603f00
Diffstat (limited to 'yql/essentials/tests/sql/suites/aggr_factory/udaf.sql')
-rw-r--r-- | yql/essentials/tests/sql/suites/aggr_factory/udaf.sql | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/yql/essentials/tests/sql/suites/aggr_factory/udaf.sql b/yql/essentials/tests/sql/suites/aggr_factory/udaf.sql new file mode 100644 index 0000000000..d419f8767f --- /dev/null +++ b/yql/essentials/tests/sql/suites/aggr_factory/udaf.sql @@ -0,0 +1,26 @@ +/* syntax version 1 */ +/* postgres can not */ +-- count example +$create = ($_item, $_parent) -> { return 1 }; +$add = ($state, $_item, $_parent) -> { return 1 + $state }; +$merge = ($state1, $state2) -> { return $state1 + $state2 }; +$get_result = ($state) -> { return $state }; +$serialize = ($state) -> { return $state }; +$deserialize = ($state) -> { return $state }; +$default = 0; + +$f = AGGREGATION_FACTORY("udaf", $create, $add, $merge, $get_result, $serialize, $deserialize, $default); +$t = AsList(AsStruct(1 as a),AsStruct(2 as a)); + +select Yql::Aggregate($t, AsTuple(), AsTuple(AsTuple(AsAtom("res"), $f( + ListItemType(TypeOf($t)), ($z)->{return $z.a})))); + +$t = AsList(AsStruct(1/0 as a),AsStruct(2/0 as a)); +select Yql::Aggregate($t, AsTuple(), AsTuple(AsTuple(AsAtom("res"), $f( + ListItemType(TypeOf($t)), ($z)->{return $z.a})))); + +use plato; +insert into @a select * from as_table($t); +commit; +select AGGREGATE_BY(a,$f) from @a; +select AGGREGATE_BY(distinct a,$f) from @a; |