Skip to content

Commit 1510f62

Browse files
committed
upmerge and improve benchmark
1 parent b0548a0 commit 1510f62

1 file changed

Lines changed: 14 additions & 5 deletions

File tree

spark/src/test/scala/org/apache/spark/sql/benchmark/CometDatetimeExpressionBenchmark.scala

Lines changed: 14 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -52,11 +52,20 @@ object CometDatetimeExpressionBenchmark extends CometBenchmarkBase {
5252
prepareTable(
5353
dir,
5454
spark.sql(s"select timestamp_micros(cast(value/100000 as integer)) as ts FROM $tbl"))
55-
Seq("YEAR", "MONTH", "DAY", "HOUR", "MINUTE", "SECOND", "WEEK", "QUARTER").foreach {
56-
level =>
57-
val name = s"Timestamp Truncate - $level"
58-
val query = s"select date_trunc('$level', ts) from parquetV1Table"
59-
runExpressionBenchmark(name, values, query)
55+
Seq(
56+
"YEAR",
57+
"QUARTER",
58+
"MONTH",
59+
"WEEK",
60+
"DAY",
61+
"HOUR",
62+
"MINUTE",
63+
"SECOND",
64+
"MILLISECOND",
65+
"MICROSECOND").foreach { level =>
66+
val name = s"Timestamp Truncate - $level"
67+
val query = s"select date_trunc('$level', ts) from parquetV1Table"
68+
runExpressionBenchmark(name, values, query)
6069
}
6170
}
6271
}

0 commit comments

Comments
 (0)