Add Mysql8UpsertAssembler

Author: momosetkn

doma-core/src/main/java/org/seasar/doma/jdbc/dialect/Mysql8UpsertAssembler.java

@@ -0,0 +1,98 @@
+package org.seasar.doma.jdbc.dialect;
+
+import java.util.List;
+import org.seasar.doma.internal.jdbc.sql.PreparedSqlBuilder;
+import org.seasar.doma.jdbc.InParameter;
+import org.seasar.doma.jdbc.criteria.tuple.Tuple2;
+import org.seasar.doma.jdbc.entity.EntityPropertyType;
+import org.seasar.doma.jdbc.entity.EntityType;
+import org.seasar.doma.jdbc.query.DuplicateKeyType;
+import org.seasar.doma.jdbc.query.UpsertAssembler;
+import org.seasar.doma.jdbc.query.UpsertAssemblerContext;
+import org.seasar.doma.jdbc.query.UpsertAssemblerSupport;
+import org.seasar.doma.jdbc.query.UpsertSetValue;
+
+public class Mysql8UpsertAssembler implements UpsertAssembler {
+  private final PreparedSqlBuilder buf;
+  private final EntityType<?> entityType;
+  private final DuplicateKeyType duplicateKeyType;
+  private final UpsertAssemblerSupport upsertAssemblerSupport;
+  private final List<Tuple2<EntityPropertyType<?, ?>, InParameter<?>>> insertValues;
+  private final List<Tuple2<EntityPropertyType<?, ?>, UpsertSetValue>> setValues;
+  private final UpsertSetValue.Visitor upsertSetValueVisitor = new UpsertSetValueVisitor();
+
+  public Mysql8UpsertAssembler(UpsertAssemblerContext context) {
+    this.buf = context.buf;
+    this.entityType = context.entityType;
+    this.duplicateKeyType = context.duplicateKeyType;
+    this.insertValues = context.insertValues;
+    this.setValues = context.setValues;
+    this.upsertAssemblerSupport = new UpsertAssemblerSupport(context.naming, context.dialect);
+  }
+
+  @Override
+  public void assemble() {
+    buf.appendSql("insert");
+    if (duplicateKeyType == DuplicateKeyType.IGNORE) {
+      buf.appendSql(" ignore");
+    }
+    buf.appendSql(" into ");
+    tableNameOnly(entityType);
+    buf.appendSql(" (");
+    for (Tuple2<EntityPropertyType<?, ?>, InParameter<?>> insertValue : insertValues) {
+      column(insertValue.component1());
+      buf.appendSql(", ");
+    }
+    buf.cutBackSql(2);
+    buf.appendSql(") values (");
+    for (Tuple2<EntityPropertyType<?, ?>, InParameter<?>> insertValue : insertValues) {
+      buf.appendParameter(insertValue.component2());
+      buf.appendSql(", ");
+    }
+    buf.cutBackSql(2);
+    buf.appendSql(") as ");
+    excludeAlias();
+    if (duplicateKeyType == DuplicateKeyType.UPDATE) {
+      buf.appendSql(" on duplicate key update ");
+      for (Tuple2<EntityPropertyType<?, ?>, UpsertSetValue> setValue : setValues) {
+        column(setValue.component1());
+        buf.appendSql(" = ");
+        setValue.component2().accept(upsertSetValueVisitor);
+        buf.appendSql(", ");
+      }
+      buf.cutBackSql(2);
+    }
+  }
+
+  private void tableNameOnly(EntityType<?> entityType) {
+    String sql =
+        this.upsertAssemblerSupport.targetTable(
+            entityType, UpsertAssemblerSupport.TableNameType.NAME);
+    buf.appendSql(sql);
+  }
+
+  private void excludeAlias() {
+    String sql = this.upsertAssemblerSupport.excludeAlias();
+    buf.appendSql(sql);
+  }
+
+  private void column(EntityPropertyType<?, ?> propertyType) {
+    String sql = this.upsertAssemblerSupport.prop(propertyType);
+    buf.appendSql(sql);
+  }
+
+  class UpsertSetValueVisitor implements UpsertSetValue.Visitor {
+    @Override
+    public void visit(UpsertSetValue.Param param) {
+      buf.appendParameter(param.inParameter);
+    }
+
+    @Override
+    public void visit(UpsertSetValue.Prop prop) {
+      String sql =
+          upsertAssemblerSupport.excludeProp(
+              prop.propertyType, UpsertAssemblerSupport.ColumnNameType.NAME_ALIAS);
+      buf.appendSql(sql);
+    }
+  }
+}