Align the timezone of decimal partition value with Velox

apache · Nov 21, 2024 · 686446e · 686446e
1 parent d5e5544
commit 686446e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/backends-velox/src/main/scala/org/apache/gluten/backendsapi/velox/VeloxIteratorApi.scala b/backends-velox/src/main/scala/org/apache/gluten/backendsapi/velox/VeloxIteratorApi.scala
@@ -43,7 +43,7 @@ import org.apache.spark.util.{ExecutorManager, SparkDirectoryUtil}
 
 import java.lang.{Long => JLong}
 import java.nio.charset.StandardCharsets
-import java.time.ZoneOffset
+import java.time.ZoneId
 import java.util.{ArrayList => JArrayList, HashMap => JHashMap, Map => JMap, UUID}
 
 import scala.collection.JavaConverters._
@@ -151,8 +151,9 @@ class VeloxIteratorApi extends IteratorApi with Logging {
               case _: DecimalType =>
                 pn.asInstanceOf[Decimal].toJavaBigInteger.toString
               case _: TimestampType =>
+                // Velox uses 'America/Los_Angeles' timezone by default.
                 TimestampFormatter
-                  .getFractionFormatter(ZoneOffset.UTC)
+                  .getFractionFormatter(ZoneId.of("America/Los_Angeles"))
                   .format(pn.asInstanceOf[java.lang.Long])
               case _ => pn.toString
             }

diff --git a/ep/build-velox/src/get_velox.sh b/ep/build-velox/src/get_velox.sh
@@ -16,8 +16,8 @@
 
 set -exu
 
-VELOX_REPO=https://github.com/oap-project/velox.git
-VELOX_BRANCH=2024_11_20
+VELOX_REPO=https://github.com/rui-mo/velox.git
+VELOX_BRANCH=test_tz
 VELOX_HOME=""
 
 OS=`uname -s`