aboutsummaryrefslogtreecommitdiffstats
path: root/yql/essentials/tests/s-expressions/suites/Udf/Percentiles.yqls
diff options
context:
space:
mode:
authorrobot-piglet <robot-piglet@yandex-team.com>2024-12-13 15:14:50 +0300
committerrobot-piglet <robot-piglet@yandex-team.com>2024-12-13 16:07:36 +0300
commit5cbdc6f2bc811da3e6104da774a30b25a54a8708 (patch)
tree68f3ddb0d928332e5ea73853f40f4670f4babb65 /yql/essentials/tests/s-expressions/suites/Udf/Percentiles.yqls
parent1402a032649d20ffb80a541f706afcd31cd897fa (diff)
downloadydb-5cbdc6f2bc811da3e6104da774a30b25a54a8708.tar.gz
Intermediate changes
commit_hash:27af5f8a0a7836d78ffdc205213f07f79dd9f7ad
Diffstat (limited to 'yql/essentials/tests/s-expressions/suites/Udf/Percentiles.yqls')
-rw-r--r--yql/essentials/tests/s-expressions/suites/Udf/Percentiles.yqls45
1 files changed, 45 insertions, 0 deletions
diff --git a/yql/essentials/tests/s-expressions/suites/Udf/Percentiles.yqls b/yql/essentials/tests/s-expressions/suites/Udf/Percentiles.yqls
new file mode 100644
index 0000000000..0a15c6095a
--- /dev/null
+++ b/yql/essentials/tests/s-expressions/suites/Udf/Percentiles.yqls
@@ -0,0 +1,45 @@
+(
+#comment
+(let udfTDigest_Create (Udf 'Stat.TDigest_Create))
+(let udfTDigest_AddValue (Udf 'Stat.TDigest_AddValue))
+(let udfTDigest_GetPercentile (Udf 'Stat.TDigest_GetPercentile))
+(let udfTDigest_Serialize (Udf 'Stat.TDigest_Serialize))
+(let udfTDigest_Deserialize (Udf 'Stat.TDigest_Deserialize))
+(let udfTDigest_Merge (Udf 'Stat.TDigest_Merge))
+(let mr_source (DataSource 'yt 'plato))
+(let x (Read! world mr_source (Key '('table (String 'Input))) '('key 'subkey 'value) '()))
+(let world (Left! x))
+(let table1 (Right! x))
+(let preMap (lambda '(item) (Just item)))
+(let keyExtractor (lambda '(item) (Member item 'key)))
+(let init (lambda '(key item) (Apply udfTDigest_Create (Unwrap (FromString (Member item 'value) 'Double)))))
+(let update (lambda '(key item state) (Apply udfTDigest_AddValue state (Unwrap (FromString (Member item 'value) 'Double)))))
+(let finish (lambda '(key state) (block '(
+ (let s (Struct))
+ (let s (AddMember s 'key key))
+ (let s (AddMember s 'subkey (String '.)))
+ (let s (AddMember s 'value (Apply udfTDigest_Serialize state)))
+ (return (Just s))
+))))
+(let table2 (CombineByKey table1 preMap keyExtractor init update finish))
+(let listHandler (lambda '(stream) (FlatMap stream (lambda '(pair) (block '(
+ (let key (Nth pair '0))
+ (let list (ForwardList (Nth pair '1)))
+ (let init (lambda '(item) (Apply udfTDigest_Deserialize (Member item 'value))))
+ (let update (lambda '(item state) (Apply udfTDigest_Merge (Apply udfTDigest_Deserialize (Member item 'value)) state)))
+ (let state (Unwrap (Fold1 list init update)))
+ (let s (Struct))
+ (let s (AddMember s 'key key))
+ (let s (AddMember s 'subkey (String '.)))
+ (let g1 (ToString (Apply udfTDigest_GetPercentile state (Double '0.75))))
+ (let g2 (ToString (Apply udfTDigest_GetPercentile state (Double '0.999))))
+ (let s (AddMember s 'value (Concat (Concat g1 (String '" ")) g2)))
+ (let ret (Just s))
+ (return ret)
+))))))
+(let table3 (PartitionByKey table2 keyExtractor (Void) (Void) listHandler))
+(let mr_sink (DataSink 'yt 'plato))
+(let world (Write! world mr_sink (Key '('table (String 'Output))) table3 '('('mode 'append))))
+(let world (Commit! world mr_sink))
+(return world)
+)