add withDefaultTimeZone and epochDaysToMicros => DateTimeUtils.epochDaysToMicros

Author: TJX2014

sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/collectionOperations.scala

@@ -2625,6 +2625,16 @@ object Sequence {
           stepMicros + stepMonths * microsPerMonth + stepDays * microsPerDay
         val startMicros: Long = num.toLong(start) * scale
         val stopMicros: Long = num.toLong(stop) * scale
+
+        // Date to timestamp is not equal from GMT and Chicago timezones
+        val (startMicros, stopMicros) = if (scale == 1) {
+          (num.toLong(start), num.toLong(stop))
+        }
+        else {
+          (DateTimeUtils.epochDaysToMicros(num.toInt(start), zoneId),
+            DateTimeUtils.epochDaysToMicros(num.toInt(stop), zoneId))
+        }
+
         val maxEstimatedArrayLength =
           getSequenceLength(startMicros, stopMicros, intervalStepInMicros)
 

sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/expressions/CollectionExpressionsSuite.scala

@@ -1854,17 +1854,18 @@ class CollectionExpressionsSuite extends SparkFunSuite with ExpressionEvalHelper
       Literal(Date.valueOf("2018-01-01")),
       Literal(stringToInterval("interval 1 year"))),
       Seq(Date.valueOf("2018-01-01")))
-=======
+
   test("SPARK-31982: sequence doesn't handle date increments that cross DST") {
     Array("America/Chicago", "GMT", "Asia/Shanghai").foreach(tz => {
-      checkEvaluation(Sequence(
-        Cast(Literal("2011-03-01"), DateType),
-        Cast(Literal("2011-04-01"), DateType),
-        Option(Literal(stringToInterval("interval 1 month"))),
-        Option(tz)),
-        Seq(
-          Date.valueOf("2011-03-01"), Date.valueOf("2011-04-01")))
+      DateTimeTestUtils.withDefaultTimeZone(DateTimeUtils.getTimeZone(tz).toZoneId) {
+        checkEvaluation(Sequence(
+          Cast(Literal("2011-03-01"), DateType),
+          Cast(Literal("2011-04-01"), DateType),
+          Option(Literal(stringToInterval("interval 1 month"))),
+          Option(tz)),
+          Seq(
+            Date.valueOf("2011-03-01"), Date.valueOf("2011-04-01")))
+      }
     })
->>>>>>> Asia timezone fix
   }
 }