Update RealizedRelationalResultCSVSerializer.java

PrateekGarg-gs · Oct 6, 2023 · af2094f · af2094f
1 parent 90f16d4
commit af2094f
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/...plan/execution/stores/relational/serialization/RealizedRelationalResultCSVSerializer.java b/...plan/execution/stores/relational/serialization/RealizedRelationalResultCSVSerializer.java
@@ -39,6 +39,7 @@ public class RealizedRelationalResultCSVSerializer extends CsvSerializer
     private String databaseTimeZone;
     private boolean withHeader;
     private boolean withTransform;
+    private boolean normalizeWithTimeZone;
     private CSVFormat csvFormat;
 
     public RealizedRelationalResultCSVSerializer(RealizedRelationalResult realizedRelationalResult, String databaseTimeZone)
@@ -52,12 +53,18 @@ public RealizedRelationalResultCSVSerializer(RealizedRelationalResult realizedRe
     }
 
     public RealizedRelationalResultCSVSerializer(RealizedRelationalResult realizedRelationalResult, String databaseTimeZone, boolean withHeader, boolean withTransform, CSVFormat csvFormat)
+    {
+        this(realizedRelationalResult, databaseTimeZone, withHeader, withTransform, csvFormat, true);
+    }
+
+    public RealizedRelationalResultCSVSerializer(RealizedRelationalResult realizedRelationalResult, String databaseTimeZone, boolean withHeader, boolean withTransform, CSVFormat csvFormat, Boolean normalizeWithTimeZone)
     {
         this.realizedRelationalResult = realizedRelationalResult;
         this.withHeader = withHeader;
         this.withTransform = withTransform;
         this.databaseTimeZone = databaseTimeZone;
         this.csvFormat = csvFormat;
+        this.normalizeWithTimeZone = normalizeWithTimeZone;
     }
 
     @Override
@@ -84,7 +91,7 @@ public void stream(OutputStream targetStream)
                 }
                 for (List<Object> row : rows)
                 {
-                    List<Object> normalizedRow = row.stream().map(x -> x != null ? ResultNormalizer.normalizeToSql(x, this.databaseTimeZone) : null).collect(Collectors.toList());
+                    List<Object> normalizedRow = row.stream().map(x -> x != null ? ResultNormalizer.normalizeToSql(x, this.databaseTimeZone, this.normalizeWithTimeZone) : null).collect(Collectors.toList());
                     csvPrinter.printRecord(normalizedRow);
                 }