View Javadoc
1   /*
2    *    Copyright 2016-2025 the original author or authors.
3    *
4    *    Licensed under the Apache License, Version 2.0 (the "License");
5    *    you may not use this file except in compliance with the License.
6    *    You may obtain a copy of the License at
7    *
8    *       https://www.apache.org/licenses/LICENSE-2.0
9    *
10   *    Unless required by applicable law or agreed to in writing, software
11   *    distributed under the License is distributed on an "AS IS" BASIS,
12   *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13   *    See the License for the specific language governing permissions and
14   *    limitations under the License.
15   */
16  package org.mybatis.dynamic.sql.insert.render;
17  
18  import org.mybatis.dynamic.sql.SqlColumn;
19  import org.mybatis.dynamic.sql.render.RenderingStrategy;
20  import org.mybatis.dynamic.sql.util.ConstantMapping;
21  import org.mybatis.dynamic.sql.util.MultiRowInsertMappingVisitor;
22  import org.mybatis.dynamic.sql.util.NullMapping;
23  import org.mybatis.dynamic.sql.util.PropertyMapping;
24  import org.mybatis.dynamic.sql.util.RowMapping;
25  import org.mybatis.dynamic.sql.util.StringConstantMapping;
26  import org.mybatis.dynamic.sql.util.StringUtilities;
27  
28  public class MultiRowValuePhraseVisitor extends MultiRowInsertMappingVisitor<FieldAndValueAndParameters> {
29      protected final RenderingStrategy renderingStrategy;
30      protected final String prefix;
31  
32      protected MultiRowValuePhraseVisitor(RenderingStrategy renderingStrategy, String prefix) {
33          this.renderingStrategy = renderingStrategy;
34          this.prefix = prefix;
35      }
36  
37      @Override
38      public FieldAndValueAndParameters visit(NullMapping mapping) {
39          return FieldAndValueAndParameters.withFieldName(mapping.columnName())
40                  .withValuePhrase("null") //$NON-NLS-1$
41                  .build();
42      }
43  
44      @Override
45      public FieldAndValueAndParameters visit(ConstantMapping mapping) {
46          return FieldAndValueAndParameters.withFieldName(mapping.columnName())
47                  .withValuePhrase(mapping.constant())
48                  .build();
49      }
50  
51      @Override
52      public FieldAndValueAndParameters visit(StringConstantMapping mapping) {
53          return FieldAndValueAndParameters.withFieldName(mapping.columnName())
54                  .withValuePhrase(StringUtilities.formatConstantForSQL(mapping.constant()))
55                  .build();
56      }
57  
58      @Override
59      public FieldAndValueAndParameters visit(PropertyMapping mapping) {
60          return FieldAndValueAndParameters.withFieldName(mapping.columnName())
61                  .withValuePhrase(calculateJdbcPlaceholder(mapping.column(), mapping.property()))
62                  .build();
63      }
64  
65      @Override
66      public FieldAndValueAndParameters visit(RowMapping mapping) {
67          return FieldAndValueAndParameters.withFieldName(mapping.columnName())
68                  .withValuePhrase(calculateJdbcPlaceholder(mapping.column()))
69                  .build();
70      }
71  
72      private String calculateJdbcPlaceholder(SqlColumn<?> column) {
73          return column.renderingStrategy().orElse(renderingStrategy).getRecordBasedInsertBinding(column, prefix);
74      }
75  
76      private String calculateJdbcPlaceholder(SqlColumn<?> column, String parameterName) {
77          return column.renderingStrategy().orElse(renderingStrategy)
78                  .getRecordBasedInsertBinding(column, prefix, parameterName);
79      }
80  }