fix: snowflake query string builder should follow timezone information from Arrow.

Author: Leo Yan

athena-snowflake/src/main/java/com/amazonaws/athena/connectors/snowflake/SnowflakeQueryStringBuilder.java

@@ -288,6 +288,11 @@ else if (value instanceof Number) {
                 long millis = ((Number) value).longValue();
                 Timestamp timestamp = new Timestamp(millis);
                 SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
+                String timezone = ((ArrowType.Timestamp) arrowType).getTimezone();
+                if (timezone == null || timezone.isEmpty()) {
+                    timezone = "UTC";
+                }
+                sdf.setTimeZone(java.util.TimeZone.getTimeZone(timezone));
                 return sdf.format(timestamp);
             case Time:
             case Interval:

athena-snowflake/src/test/java/com/amazonaws/athena/connectors/snowflake/SnowflakeQueryStringBuilderTest.java

@@ -129,11 +129,29 @@ void testGetObjectForWhereClause_Date() {
     }
 
     @Test
-    void testGetObjectForWhereClause_Timestamp() {
+    void testGetObjectForWhereClause_TimestampWithNullZone() {
+        Object result = queryBuilder.getObjectForWhereClause("created_at", 1711929600000L,  new ArrowType.Timestamp(TimeUnit.MILLISECOND, null));
+        assertEquals("2024-04-01 00:00:00", result);
+    }
+
+    @Test
+    void testGetObjectForWhereClause_TimestampWithUtcZone() {
         Object result = queryBuilder.getObjectForWhereClause("created_at", 1711929600000L,  new ArrowType.Timestamp(TimeUnit.MILLISECOND, "UTC"));
         assertEquals("2024-04-01 00:00:00", result);
     }
 
+    @Test
+    void testGetObjectForWhereClause_TimestampWithLocalZone() {
+        Object result = queryBuilder.getObjectForWhereClause("created_at", 1711929600000L,  new ArrowType.Timestamp(TimeUnit.MILLISECOND, "PST"));
+        assertEquals("2024-03-31 17:00:00", result);
+
+        Object result1 = queryBuilder.getObjectForWhereClause("created_at", 1711929600000L,  new ArrowType.Timestamp(TimeUnit.MILLISECOND, "America/Los_Angeles"));
+        assertEquals("2024-03-31 17:00:00", result1);
+
+        Object result2 = queryBuilder.getObjectForWhereClause("created_at", 1711929600000L,  new ArrowType.Timestamp(TimeUnit.MILLISECOND, "GMT-8:00"));
+        assertEquals("2024-03-31 16:00:00", result2);
+    }
+
     @Test
     void testToPredicateWithUnsupportedType() {
         assertThrows(UnsupportedOperationException.class, () ->