Fix

Author: maropu

sql/core/src/test/resources/sql-tests/inputs/extract.sql

@@ -108,4 +108,22 @@ select date_part(c, c) from t;
 select date_part(null, c) from t;
 
 select date_part(i, i) from t;
-select date_part(null, i) from t;
+select date_part(null, i) from t;
+
+-- In SPARK-31476, we've supported extract('field', source), too
+select extract('millennium', c) from t;
+select extract('century', c) from t;
+select extract('decade', c) from t;
+select extract('year', c) from t;
+select extract('quarter', c) from t;
+select extract('month', c) from t;
+select extract('week', c) from t;
+select extract('day', c) from t;
+select extract('days', c) from t;
+select extract('dayofweek', c) from t;
+select extract('dow', c) from t;
+select extract('isodow', c) from t;
+select extract('doy', c) from t;
+select extract('hour', c) from t;
+select extract('minute', c) from t;
+select extract('second', c) from t;

sql/core/src/test/resources/sql-tests/results/extract.sql.out

@@ -1,5 +1,5 @@
 -- Automatically generated by SQLQueryTestSuite
--- Number of queries: 84
+-- Number of queries: 100
 
 
 -- !query
@@ -677,3 +677,135 @@ select date_part(null, i) from t
 struct<date_part(NULL, t.`i`):double>
 -- !query output
 NULL
+
+
+-- !query
+select extract('millennium', c) from t
+-- !query schema
+struct<>
+-- !query output
+org.apache.spark.sql.AnalysisException
+Literals of type 'millennium' are currently not supported for the string type.;; line 1 pos 7
+
+
+-- !query
+select extract('century', c) from t
+-- !query schema
+struct<>
+-- !query output
+org.apache.spark.sql.AnalysisException
+Literals of type 'century' are currently not supported for the string type.;; line 1 pos 7
+
+
+-- !query
+select extract('decade', c) from t
+-- !query schema
+struct<>
+-- !query output
+org.apache.spark.sql.AnalysisException
+Literals of type 'decade' are currently not supported for the string type.;; line 1 pos 7
+
+
+-- !query
+select extract('year', c) from t
+-- !query schema
+struct<extract('year' FROM t.`c`):int>
+-- !query output
+2011
+
+
+-- !query
+select extract('quarter', c) from t
+-- !query schema
+struct<extract('quarter' FROM t.`c`):int>
+-- !query output
+2
+
+
+-- !query
+select extract('month', c) from t
+-- !query schema
+struct<extract('month' FROM t.`c`):int>
+-- !query output
+5
+
+
+-- !query
+select extract('week', c) from t
+-- !query schema
+struct<extract('week' FROM t.`c`):int>
+-- !query output
+18
+
+
+-- !query
+select extract('day', c) from t
+-- !query schema
+struct<extract('day' FROM t.`c`):int>
+-- !query output
+6
+
+
+-- !query
+select extract('days', c) from t
+-- !query schema
+struct<extract('days' FROM t.`c`):int>
+-- !query output
+6
+
+
+-- !query
+select extract('dayofweek', c) from t
+-- !query schema
+struct<extract('dayofweek' FROM t.`c`):int>
+-- !query output
+6
+
+
+-- !query
+select extract('dow', c) from t
+-- !query schema
+struct<extract('dow' FROM t.`c`):int>
+-- !query output
+6
+
+
+-- !query
+select extract('isodow', c) from t
+-- !query schema
+struct<>
+-- !query output
+org.apache.spark.sql.AnalysisException
+Literals of type 'isodow' are currently not supported for the string type.;; line 1 pos 7
+
+
+-- !query
+select extract('doy', c) from t
+-- !query schema
+struct<extract('doy' FROM t.`c`):int>
+-- !query output
+126
+
+
+-- !query
+select extract('hour', c) from t
+-- !query schema
+struct<extract('hour' FROM t.`c`):int>
+-- !query output
+7
+
+
+-- !query
+select extract('minute', c) from t
+-- !query schema
+struct<extract('minute' FROM t.`c`):int>
+-- !query output
+8
+
+
+-- !query
+select extract('second', c) from t
+-- !query schema
+struct<extract('second' FROM t.`c`):decimal(8,6)>
+-- !query output
+9.123456