implemented writer statement with limited insert columns

Author: chernser

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

@@ -5,6 +5,7 @@
 import com.clickhouse.client.api.insert.InsertResponse;
 import com.clickhouse.client.api.insert.InsertSettings;
 import com.clickhouse.client.api.metadata.TableSchema;
+import com.clickhouse.data.ClickHouseColumn;
 import com.clickhouse.data.ClickHouseFormat;
 import com.clickhouse.jdbc.internal.ExceptionUtils;
 import com.clickhouse.jdbc.internal.ParsedPreparedStatement;
@@ -29,8 +30,10 @@
 import java.sql.SQLXML;
 import java.sql.Time;
 import java.sql.Timestamp;
+import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Calendar;
+import java.util.List;
 import java.util.concurrent.TimeUnit;
 
 /**
@@ -50,7 +53,16 @@ public WriterStatementImpl(ConnectionImpl connection, String originalSql, TableS
             throws SQLException {
         super(connection, originalSql, parsedStatement);
 
-        this.tableSchema = tableSchema;
+        if (parsedStatement.getInsertColumns() != null) {
+            List<ClickHouseColumn> insertColumns = new ArrayList<>();
+            for (String column : parsedStatement.getInsertColumns()) {
+                insertColumns.add(tableSchema.getColumnByName(column));
+            }
+            this.tableSchema = new TableSchema(tableSchema.getTableName(), tableSchema.getQuery(),
+                    tableSchema.getDatabaseName(), insertColumns);
+        } else {
+            this.tableSchema = tableSchema;
+        }
         try {
             resetWriter();
         } catch (IOException e) {

jdbc-v2/src/test/java/com/clickhouse/jdbc/PreparedStatementTest.java

@@ -611,7 +611,7 @@ Object[][] testBatchInsertWithRowBinary_dp() {
         return new Object[][]{
                 {"INSERT  INTO \n `%s` \nVALUES (?, ?, abs(?), ?)", PreparedStatementImpl.class}, // only string possible (because of abs(?))
                 {"INSERT  INTO\n `%s` \nVALUES (?, ?, ?, ?)", WriterStatementImpl.class}, // row binary writer
-                {" INSERT INTO %s (ts, v1, v2, v3) VALUES (?, ?, ?, ?)", PreparedStatementImpl.class}, // only string supported now
+                {" INSERT INTO %s (ts, v1, v2, v3) VALUES (?, ?, ?, ?)", WriterStatementImpl.class}, // only string supported now
                 {"INSERT INTO %s SELECT ?, ?, ?, ?", PreparedStatementImpl.class}, // only string possible (because of SELECT)
         };
     }