24fcc2011f
Nereids function case do not have a order by clause, so the result will be unstable, so order by is added to ensure stability.
247 lines
7.5 KiB
Groovy
247 lines
7.5 KiB
Groovy
// Licensed to the Apache Software Foundation (ASF) under one
|
|
// or more contributor license agreements. See the NOTICE file
|
|
// distributed with this work for additional information
|
|
// regarding copyright ownership. The ASF licenses this file
|
|
// to you under the Apache License, Version 2.0 (the
|
|
// "License"); you may not use this file except in compliance
|
|
// with the License. You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing,
|
|
// software distributed under the License is distributed on an
|
|
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
// KIND, either express or implied. See the License for the
|
|
// specific language governing permissions and limitations
|
|
// under the License.
|
|
|
|
suite("nereids_function") {
|
|
sql "SET enable_nereids_planner=true"
|
|
sql "SET enable_fallback_to_original_planner=false"
|
|
|
|
order_qt_max """
|
|
SELECT max(lo_discount), max(lo_extendedprice) AS max_extendedprice FROM lineorder;
|
|
"""
|
|
|
|
order_qt_min """
|
|
SELECT min(lo_discount), min(lo_extendedprice) AS min_extendedprice FROM lineorder;
|
|
"""
|
|
|
|
order_qt_max_and_min """
|
|
SELECT max(lo_extendedprice), min(lo_discount) FROM lineorder;
|
|
"""
|
|
|
|
order_qt_count """
|
|
SELECT count(c_city), count(*) AS custdist FROM customer;
|
|
"""
|
|
|
|
order_qt_distinct_count """
|
|
SELECT count(distinct c_custkey + 1) AS custdist FROM customer group by c_city;
|
|
"""
|
|
|
|
order_qt_distinct_count_group_by_distributed_key """
|
|
SELECT c_custkey, count(distinct c_custkey + 1) AS custdist FROM customer group by c_custkey;
|
|
"""
|
|
|
|
order_qt_avg """
|
|
SELECT avg(lo_tax), avg(lo_extendedprice) AS avg_extendedprice FROM lineorder;
|
|
"""
|
|
|
|
qt_string_arithmetic """
|
|
SELECT "1" + "2", "1" - "2";
|
|
"""
|
|
|
|
// variable length function
|
|
test {
|
|
sql "select coalesce(1), coalesce(null, 2), coalesce(null, 3), coalesce(4, null, 2)"
|
|
result([[1, 2, 3, 4]])
|
|
}
|
|
|
|
// nested function
|
|
test {
|
|
sql "select cast(date('1994-01-01') + interval '1' YEAR as varchar)"
|
|
result([["1995-01-01"]])
|
|
}
|
|
|
|
test {
|
|
sql "select substring(substring('1994-01-01', 5), 3)"
|
|
result([["1-01"]])
|
|
}
|
|
|
|
// numbers: table valued function
|
|
test {
|
|
sql "select `number` from numbers(number = 10, backend_num = 1)"
|
|
result([[0L], [1L], [2L], [3L], [4L], [5L], [6L], [7L], [8L], [9L]])
|
|
}
|
|
|
|
// numbers: table valued function
|
|
test {
|
|
sql """
|
|
select
|
|
a.number as num1, b.number as num2
|
|
from
|
|
numbers("number" = "10") a
|
|
inner join numbers("number" = "10") b on a.number=b.number order by 1,2;
|
|
"""
|
|
result([[0L, 0L], [1L, 1L], [2L, 2L], [3L, 3L], [4L, 4L], [5L, 5L], [6L, 6L], [7L, 7L], [8L, 8L], [9L, 9L]])
|
|
}
|
|
|
|
qt_subquery1 """ select * from numbers("number" = "10") where number = (select number from numbers("number" = "10") where number=1); """
|
|
qt_subquery2 """ select * from numbers("number" = "10") where number in (select number from numbers("number" = "10") where number>5); """
|
|
qt_subquery3 """ select a.number from numbers("number" = "10") a where number in (select number from numbers("number" = "10") b where a.number=b.number); """
|
|
|
|
test {
|
|
sql """select `number` from numbers("number" = -1, 'backend_num' = `1`)"""
|
|
result([])
|
|
}
|
|
|
|
test {
|
|
sql "select `number` from numbers(a=2)"
|
|
exception "Can not build NumbersTableValuedFunction by numbers('a' = '2')"
|
|
}
|
|
|
|
test {
|
|
sql """select a.`number` from numbers(number = 3) a"""
|
|
result([[0L], [1L], [2L]])
|
|
}
|
|
|
|
test {
|
|
sql """select b.number from (select * from numbers(number = 3) a)b"""
|
|
result([[0L], [1L], [2L]])
|
|
}
|
|
|
|
test {
|
|
sql "select from_unixtime(1249488000, 'yyyyMMdd')"
|
|
result([["20090806"]])
|
|
}
|
|
|
|
test {
|
|
sql "select convert_to('abc', cast(number as varchar)) from numbers('number'='1')"
|
|
exception "must be a constant"
|
|
}
|
|
|
|
test {
|
|
sql """select "1" == "123", "%%" == "%%" """
|
|
result([[false, true]])
|
|
}
|
|
|
|
qt_floor """
|
|
SELECT floor(2.1);
|
|
"""
|
|
|
|
qt_ceil """
|
|
SELECT ceil(2.1);
|
|
"""
|
|
|
|
test {
|
|
sql "select left('abcd', 3), right('abcd', 3)"
|
|
result([['abc', 'bcd']])
|
|
}
|
|
|
|
// test window_funnel function
|
|
sql """ DROP TABLE IF EXISTS window_funnel_test """
|
|
sql """
|
|
CREATE TABLE IF NOT EXISTS window_funnel_test (
|
|
xwho varchar(50) NULL COMMENT 'xwho',
|
|
xwhen datetimev2(3) COMMENT 'xwhen',
|
|
xwhat int NULL COMMENT 'xwhat'
|
|
)
|
|
DUPLICATE KEY(xwho)
|
|
DISTRIBUTED BY HASH(xwho) BUCKETS 3
|
|
PROPERTIES (
|
|
"replication_num" = "1"
|
|
);
|
|
"""
|
|
|
|
sql "INSERT into window_funnel_test (xwho, xwhen, xwhat) VALUES('1', '2022-03-12 10:41:00.111111', 1)"
|
|
sql "INSERT INTO window_funnel_test (xwho, xwhen, xwhat) VALUES('1', '2022-03-12 13:28:02.111111', 2)"
|
|
sql "INSERT INTO window_funnel_test (xwho, xwhen, xwhat) VALUES('1', '2022-03-12 16:15:01.111111', 3)"
|
|
sql "INSERT INTO window_funnel_test (xwho, xwhen, xwhat) VALUES('1', '2022-03-12 19:05:04.111111', 4)"
|
|
|
|
qt_window_funnel """
|
|
select
|
|
window_funnel(
|
|
1,
|
|
'default',
|
|
t.xwhen,
|
|
t.xwhat = 1,
|
|
t.xwhat = 2
|
|
) AS level
|
|
from window_funnel_test t;
|
|
"""
|
|
qt_window_funnel """
|
|
select
|
|
window_funnel(
|
|
20000,
|
|
'default',
|
|
t.xwhen,
|
|
t.xwhat = 1,
|
|
t.xwhat = 2
|
|
) AS level
|
|
from window_funnel_test t;
|
|
"""
|
|
|
|
test {
|
|
sql "select cast(1.2 as integer);"
|
|
result([[1]])
|
|
}
|
|
|
|
// scalar function
|
|
sql """
|
|
DROP TABLE IF EXISTS running_difference_test
|
|
"""
|
|
|
|
sql """
|
|
CREATE TABLE running_difference_test (
|
|
`id` int NOT NULL COMMENT 'id' ,
|
|
`day` date COMMENT 'day',
|
|
`time_val` datetime COMMENT 'time_val',
|
|
`double_num` double NULL COMMENT 'double_num',
|
|
`decimal_num` decimal(9, 6) NULL COMMENT 'decimal_num'
|
|
)
|
|
DUPLICATE KEY(id)
|
|
DISTRIBUTED BY HASH(id) BUCKETS 3
|
|
PROPERTIES (
|
|
"replication_num" = "1"
|
|
);
|
|
"""
|
|
|
|
sql """
|
|
INSERT into running_difference_test values
|
|
('1', '2022-10-28', '2022-03-12 10:41:00', null, 2.66),
|
|
('2','2022-10-27', '2022-03-12 10:41:02', 2.6, 2.67),
|
|
('3','2022-10-28', '2022-03-12 10:41:03', 2.5, 2.76),
|
|
('4','2022-9-29', '2022-03-12 10:41:03', null, 6.756),
|
|
('5','2022-10-31', '2022-03-12 10:42:01', 3.3, 7.545),
|
|
('6', '2022-11-08', '2022-03-12 11:05:04', 4.7, 3.5464);
|
|
"""
|
|
|
|
sql """
|
|
SELECT
|
|
id,
|
|
day,
|
|
time_val,
|
|
double_num,
|
|
running_difference(id) AS delta1,
|
|
running_difference(day) AS delta2,
|
|
running_difference(time_val) AS delta3,
|
|
running_difference(double_num) AS delta4,
|
|
running_difference(decimal_num) AS delta5
|
|
FROM (
|
|
SELECT
|
|
id,
|
|
day,
|
|
time_val,
|
|
double_num,
|
|
decimal_num
|
|
FROM running_difference_test
|
|
ORDER BY id ASC
|
|
) as runningDifference
|
|
"""
|
|
|
|
qt_regexp_extract_all """
|
|
SELECT regexp_extract_all('AbCdE', '([[:lower:]]+)C([[:lower:]]+)')
|
|
"""
|
|
}
|
|
|