Start of Map To Row

Author: jeffgbutler

src/main/java/org/mybatis/dynamic/sql/insert/MultiRowInsertDSL.java

@@ -28,6 +28,7 @@
 import org.mybatis.dynamic.sql.util.ConstantMapping;
 import org.mybatis.dynamic.sql.util.NullMapping;
 import org.mybatis.dynamic.sql.util.PropertyMapping;
+import org.mybatis.dynamic.sql.util.RowMapping;
 import org.mybatis.dynamic.sql.util.StringConstantMapping;
 
 public class MultiRowInsertDSL<T> implements Buildable<MultiRowInsertModel<T>> {
@@ -102,6 +103,11 @@ public MultiRowInsertDSL<T> toStringConstant(String constant) {
             columnMappings.add(StringConstantMapping.of(column, constant));
             return MultiRowInsertDSL.this;
         }
+
+        public MultiRowInsertDSL<T> toRow() {
+            columnMappings.add(RowMapping.of(column));
+            return MultiRowInsertDSL.this;
+        }
     }
 
     public static class Builder<T> extends BatchInsertDSL.AbstractBuilder<T, Builder<T>> {

src/main/java/org/mybatis/dynamic/sql/insert/render/AbstractMultiRowValuePhraseVisitor.java

@@ -18,9 +18,11 @@
 import org.mybatis.dynamic.sql.SqlColumn;
 import org.mybatis.dynamic.sql.render.RenderingStrategy;
 import org.mybatis.dynamic.sql.util.ConstantMapping;
+import org.mybatis.dynamic.sql.util.Messages;
 import org.mybatis.dynamic.sql.util.MultiRowInsertMappingVisitor;
 import org.mybatis.dynamic.sql.util.NullMapping;
 import org.mybatis.dynamic.sql.util.PropertyMapping;
+import org.mybatis.dynamic.sql.util.RowMapping;
 import org.mybatis.dynamic.sql.util.StringConstantMapping;
 
 public abstract class AbstractMultiRowValuePhraseVisitor
@@ -62,5 +64,11 @@ public FieldAndValueAndParameters visit(PropertyMapping mapping) {
                 .build();
     }
 
+    @Override
+    public FieldAndValueAndParameters visit(RowMapping mapping) {
+        // TODO - fix error number
+        throw new UnsupportedOperationException(Messages.getInternalErrorString(99));
+    }
+
     abstract String calculateJdbcPlaceholder(SqlColumn<?> column, String parameterName);
 }

src/main/java/org/mybatis/dynamic/sql/insert/render/MultiRowValuePhraseVisitor.java

@@ -17,6 +17,10 @@
 
 import org.mybatis.dynamic.sql.SqlColumn;
 import org.mybatis.dynamic.sql.render.RenderingStrategy;
+import org.mybatis.dynamic.sql.util.Messages;
+import org.mybatis.dynamic.sql.util.RowMapping;
+
+import java.util.Optional;
 
 public class MultiRowValuePhraseVisitor extends AbstractMultiRowValuePhraseVisitor {
 
@@ -29,4 +33,10 @@ String calculateJdbcPlaceholder(SqlColumn<?> column, String parameterName) {
         return column.renderingStrategy().orElse(renderingStrategy)
                 .getMultiRowFormattedJdbcPlaceholder(column, prefix, parameterName);
     }
+
+    @Override
+    public FieldAndValueAndParameters visit(RowMapping mapping) {
+        // TODO - fix error number, and move this somewhere up in the hierarchy
+        throw new UnsupportedOperationException(Messages.getInternalErrorString(99));
+    }
 }

src/main/java/org/mybatis/dynamic/sql/insert/render/ValuePhraseVisitor.java

@@ -21,9 +21,11 @@
 import org.mybatis.dynamic.sql.render.RenderingStrategy;
 import org.mybatis.dynamic.sql.util.ConstantMapping;
 import org.mybatis.dynamic.sql.util.InsertMappingVisitor;
+import org.mybatis.dynamic.sql.util.Messages;
 import org.mybatis.dynamic.sql.util.NullMapping;
 import org.mybatis.dynamic.sql.util.PropertyMapping;
 import org.mybatis.dynamic.sql.util.PropertyWhenPresentMapping;
+import org.mybatis.dynamic.sql.util.RowMapping;
 import org.mybatis.dynamic.sql.util.StringConstantMapping;
 
 public class ValuePhraseVisitor extends InsertMappingVisitor<Optional<FieldAndValueAndParameters>> {

src/main/java/org/mybatis/dynamic/sql/util/ColumnMappingVisitor.java

@@ -49,5 +49,7 @@ public interface ColumnMappingVisitor<R> {
 
     R visit(PropertyWhenPresentMapping mapping);
 
-    R visit(ColumnToColumnMapping columnMapping);
+    R visit(ColumnToColumnMapping mapping);
+
+    R visit(RowMapping mapping);
 }

src/main/java/org/mybatis/dynamic/sql/util/GeneralInsertMappingVisitor.java

@@ -35,4 +35,10 @@ public final R visit(PropertyWhenPresentMapping mapping) {
     public final R visit(ColumnToColumnMapping columnMapping) {
         throw new UnsupportedOperationException(Messages.getInternalErrorString(4));
     }
+
+    @Override
+    public final R visit(RowMapping mapping) {
+        // TODO - fix error number
+        throw new UnsupportedOperationException(Messages.getInternalErrorString(99));
+    }
 }

src/main/java/org/mybatis/dynamic/sql/util/InsertMappingVisitor.java

@@ -15,6 +15,10 @@
  */
 package org.mybatis.dynamic.sql.util;
 
+import org.mybatis.dynamic.sql.insert.render.FieldAndValueAndParameters;
+
+import java.util.Optional;
+
 public abstract class InsertMappingVisitor<R> implements ColumnMappingVisitor<R> {
     @Override
     public final <T> R visit(ValueMapping<T> mapping) {
@@ -40,4 +44,10 @@ public final R visit(SelectMapping mapping) {
     public final R visit(ColumnToColumnMapping columnMapping) {
         throw new UnsupportedOperationException(Messages.getInternalErrorString(9));
     }
+
+    @Override
+    public R visit(RowMapping mapping) {
+        // TODO - fix error number
+        throw new UnsupportedOperationException(Messages.getInternalErrorString(99));
+    }
 }

src/main/java/org/mybatis/dynamic/sql/util/RowMapping.java

@@ -0,0 +1,33 @@
+/*
+ *    Copyright 2016-2023 the original author or authors.
+ *
+ *    Licensed under the Apache License, Version 2.0 (the "License");
+ *    you may not use this file except in compliance with the License.
+ *    You may obtain a copy of the License at
+ *
+ *       https://www.apache.org/licenses/LICENSE-2.0
+ *
+ *    Unless required by applicable law or agreed to in writing, software
+ *    distributed under the License is distributed on an "AS IS" BASIS,
+ *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *    See the License for the specific language governing permissions and
+ *    limitations under the License.
+ */
+package org.mybatis.dynamic.sql.util;
+
+import org.mybatis.dynamic.sql.SqlColumn;
+
+public class RowMapping extends AbstractColumnMapping {
+    private RowMapping(SqlColumn<?> column) {
+        super(column);
+    }
+
+    public static RowMapping of(SqlColumn<?> column) {
+        return new RowMapping(column);
+    }
+
+    @Override
+    public <R> R accept(ColumnMappingVisitor<R> visitor) {
+        return visitor.visit(this);
+    }
+}

src/main/java/org/mybatis/dynamic/sql/util/UpdateMappingVisitor.java

@@ -25,4 +25,10 @@ public final R visit(PropertyMapping mapping) {
     public final R visit(PropertyWhenPresentMapping mapping) {
         throw new UnsupportedOperationException(Messages.getInternalErrorString(11));
     }
+
+    @Override
+    public final R visit(RowMapping mapping) {
+        // TODO - fix error number
+        throw new UnsupportedOperationException(Messages.getInternalErrorString(99));
+    }
 }

src/test/java/org/mybatis/dynamic/sql/insert/MapToRowTest.java

@@ -0,0 +1,70 @@
+package org.mybatis.dynamic.sql.insert;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.mybatis.dynamic.sql.SqlBuilder.insert;
+import static org.mybatis.dynamic.sql.SqlBuilder.insertMultiple;
+
+import java.sql.JDBCType;
+import java.util.ArrayList;
+import java.util.List;
+
+import org.junit.jupiter.api.Test;
+import org.mybatis.dynamic.sql.SqlColumn;
+import org.mybatis.dynamic.sql.SqlTable;
+import org.mybatis.dynamic.sql.insert.render.InsertStatementProvider;
+import org.mybatis.dynamic.sql.insert.render.MultiRowInsertStatementProvider;
+import org.mybatis.dynamic.sql.render.RenderingStrategies;
+class MapToRowTest {
+    private static final SqlTable foo = SqlTable.of("foo");
+    private static final SqlColumn<Integer> id1 = foo.column("id1", JDBCType.INTEGER);
+    private static final SqlColumn<Integer> id2 = foo.column("id2", JDBCType.INTEGER);
+
+    @Test
+    void testBasicInsertMultipleWithMyBatis() {
+        class Record {
+            public Record() {}
+
+            public Record(Integer id1, Integer id2) {
+                this.id1 = id1;
+                this.id2 = id2;
+            }
+
+            public Integer id1;
+            public Integer id2;
+        }
+
+        List<Record> records = new ArrayList<>();
+        records.add(new Record(33, 1));
+        records.add(new Record(33, 2));
+        records.add(new Record(33, 3));
+
+
+        MultiRowInsertStatementProvider<Record> insertStatement = insertMultiple(records)
+                .into(foo)
+                .map(id1).toConstant("22")
+                .map(id2).toProperty("id2")
+                .build()
+                .render(RenderingStrategies.MYBATIS3);
+
+        String expected = "insert into foo (id1, id2) values (22, #{records[0].id2,jdbcType=INTEGER}), (22, #{records[1].id2,jdbcType=INTEGER}), (22, #{records[2].id2,jdbcType=INTEGER})";
+        assertThat(insertStatement.getInsertStatement()).isEqualTo(expected);
+    }
+
+    @Test
+    void testBasicInsertMultipleRowMappingWithMyBatis() {
+        List<Integer> integers = new ArrayList<>();
+        integers.add(1);
+        integers.add(2);
+        integers.add(3);
+
+        MultiRowInsertStatementProvider<Integer> insertStatement = insertMultiple(integers)
+                .into(foo)
+                .map(id1).toConstant("22")
+                .map(id2).toRow()
+                .build()
+                .render(RenderingStrategies.MYBATIS3);
+
+        String expected = "insert into foo (id1, id2) values (22, #{records[0],jdbcType=INTEGER}), (22, #{records[1],jdbcType=INTEGER}), (22, #{records[2],jdbcType=INTEGER})";
+        assertThat(insertStatement.getInsertStatement()).isEqualTo(expected);
+    }
+}