Tableau fix getDate with calendar

Author: mzitnik

client-v2/src/main/java/com/clickhouse/client/api/data_formats/internal/AbstractBinaryFormatReader.java

@@ -33,6 +33,7 @@
 import java.time.LocalDateTime;
 import java.time.ZoneOffset;
 import java.time.ZonedDateTime;
+import java.time.format.DateTimeFormatter;
 import java.time.temporal.ChronoUnit;
 import java.util.*;
 import java.util.concurrent.ConcurrentHashMap;
@@ -57,7 +58,7 @@ public abstract class AbstractBinaryFormatReader implements ClickHouseBinaryForm
 
     private volatile boolean hasNext = true;
 
-
+    private DateTimeFormatter dateTimeFormatter = DateTimeFormatter.ofPattern("yyyy-MM-dd");
     private volatile boolean initialState = true; // reader is in initial state, no records have been read yet
 
     protected AbstractBinaryFormatReader(InputStream inputStream, QuerySettings querySettings, TableSchema schema,
@@ -279,6 +280,13 @@ public String getString(String colName) {
             return null;
         } else if (value instanceof String) {
             return (String) value;
+        } else if (value instanceof ZonedDateTime) {
+            ClickHouseDataType dataType = schema.getColumnByName(colName).getDataType();
+            ZonedDateTime zdt = (ZonedDateTime) value;
+            if (dataType == ClickHouseDataType.Date) {
+                return zdt.format(dateTimeFormatter).toString();
+            }
+            return value.toString();
         } else {
             ClickHouseDataType dataType = schema.getColumnByName(colName).getDataType();
             if (dataType == ClickHouseDataType.Enum8 || dataType == ClickHouseDataType.Enum16) {

jdbc-v2/src/main/java/com/clickhouse/jdbc/ResultSetImpl.java

@@ -13,7 +13,9 @@
 import java.time.LocalDate;
 import java.time.LocalDateTime;
 import java.util.Calendar;
+import java.util.GregorianCalendar;
 import java.util.Map;
+import java.util.TimeZone;
 
 import com.clickhouse.client.api.data_formats.ClickHouseBinaryFormatReader;
 import com.clickhouse.client.api.metadata.TableSchema;
@@ -33,6 +35,7 @@ public class ResultSetImpl implements ResultSet, JdbcV2Wrapper {
     private boolean closed;
     private final StatementImpl parentStatement;
     private boolean wasNull;
+    private final Calendar defaultCalendar;
 
     public ResultSetImpl(StatementImpl parentStatement, QueryResponse response, ClickHouseBinaryFormatReader reader) {
         this.parentStatement = parentStatement;
@@ -41,6 +44,7 @@ public ResultSetImpl(StatementImpl parentStatement, QueryResponse response, Clic
         this.metaData = new com.clickhouse.jdbc.metadata.ResultSetMetaData(this);
         this.closed = false;
         this.wasNull = false;
+        this.defaultCalendar = new GregorianCalendar(TimeZone.getTimeZone("UTC"));
     }
 
     private void checkClosed() throws SQLException {
@@ -1234,7 +1238,15 @@ public Date getDate(int columnIndex, Calendar cal) throws SQLException {
     @Override
     public Date getDate(String columnLabel, Calendar cal) throws SQLException {
         checkClosed();
-        return getDate(columnLabel);
+        Date date = getDate(columnLabel);
+        if (date == null) {
+            return null;
+        }
+        LocalDate d = date.toLocalDate();
+        Calendar c = (Calendar)( cal != null ? cal : defaultCalendar).clone();
+        c.clear();
+        c.set(d.getYear(), d.getMonthValue() - 1, d.getDayOfMonth(), 0, 0, 0);
+        return new Date(c.getTimeInMillis());
     }
 
     @Override