yql/essentials/tests/sql/suites/optimizers/yt_shuffle_by_keys.sql


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

--Test, that YT optimizer can rewrite ShuffleByKeys with PartitionsByKes
USE plato;

$input = PROCESS Input;

SELECT YQL::ShuffleByKeys(
            $input, 
            ($_)->("dsdsa"), 
            ($_)->([1]) -- list
        );

SELECT YQL::ShuffleByKeys(
            $input, 
            ($_)->(12), 
            ($_)->(Just(2)) -- optional
        );
        
SELECT YQL::ShuffleByKeys(
            $input, 
            ($_)->(true), 
            ($_)->(YQL::ToStream([3])) -- stream
        );