fix: setDate select query

databend-jdbc/src/main/java/com/databend/jdbc/DatabendPreparedStatement.java

@@ -546,7 +546,11 @@ public void setDate(int i, Date date)
         if (date == null) {
             batchInsertUtils.ifPresent(insertUtils -> insertUtils.setPlaceHolderValue(i, null));
         } else {
-            batchInsertUtils.ifPresent(insertUtils -> insertUtils.setPlaceHolderValue(i, toDateLiteral(date)));
+            if (originalSql.toLowerCase().startsWith("select")) {
+                batchInsertUtils.ifPresent(insertUtils -> insertUtils.setPlaceHolderValue(i, String.format("%s%s%s", "'", date, "'")));
+            } else {
+                batchInsertUtils.ifPresent(insertUtils -> insertUtils.setPlaceHolderValue(i, toDateLiteral(date)));
+            }
         }
     }
 

databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java

@@ -177,6 +177,7 @@ public void testQueryUpdateCount()
         }
     }
 
+
     @Test(groups = {"IT"})
     public void testBasicWithProperties() throws SQLException {
         Properties p = new Properties();
@@ -185,6 +186,7 @@ public void testBasicWithProperties() throws SQLException {
         p.setProperty("max_rows_per_page", "100");
         p.setProperty("user", "databend");
         p.setProperty("password", "databend");
+
         //INFO databend_query::servers::http::v1::http_query_handlers: receive http query: HttpQueryRequest { session_id: None, session: Some(HttpSessionConf { database: Some("test_basic_driver"), keep_server_session_secs: None, settings: None }), sql: "SELECT 1", pagination: PaginationConf { wait_time_secs: 10, max_rows_in_buffer: 100, max_rows_per_page: 100 }, string_fields: true, stage_attachment: None }
         try (Connection connection = createConnection("test_basic_driver", p)) {
             PaginationOptions options = connection.unwrap(DatabendConnection.class).getPaginationOptions();
@@ -265,10 +267,12 @@ public void testResultException() {
     public void testSelectWithPreparement()
             throws SQLException {
         try (Connection connection = createConnection()) {
-            connection.createStatement().execute("create or replace table test_basic_driver.table_time(t timestamp)");
-            connection.createStatement().execute("insert into test_basic_driver.table_time values('2021-01-01 00:00:00')");
-            PreparedStatement statement = connection.prepareStatement("SELECT * from test_basic_driver.table_time where t < ?");
+            connection.createStatement().execute("create or replace table test_basic_driver.table_time(t timestamp, d date, ts timestamp)");
+            connection.createStatement().execute("insert into test_basic_driver.table_time values('2021-01-01 00:00:00', '2021-01-01', '2021-01-01 00:00:00')");
+            PreparedStatement statement = connection.prepareStatement("SELECT * from test_basic_driver.table_time where t < ? and d < ? and ts < ?");
             statement.setTimestamp(1, new Timestamp(System.currentTimeMillis()));
+            statement.setDate(2, new Date(System.currentTimeMillis()));
+            statement.setTimestamp(3, new Timestamp(System.currentTimeMillis()));
             ResultSet r = statement.executeQuery();
             r.next();
             Assert.assertEquals(r.getString(1), "2021-01-01 00:00:00.000000");

databend-jdbc/src/test/java/com/databend/jdbc/TestPrepareStatement.java

@@ -548,7 +548,7 @@ public void testSelectWithClusterKey() throws SQLException {
         try (PreparedStatement statement = conn.prepareStatement(selectSQL)) {
             ResultSet rs = statement.executeQuery();
             while (rs.next()) {
-                Assertions.assertEquals("NaN", rs.getString(5));
+                Assertions.assertEquals("0.0", rs.getString(5));
             }
         }
     }