Merge branch 'dev' into test
						commit
						71a605718c
					
				| @ -0,0 +1,28 @@ | |||||||
|  | package com.glxp.api.config; | ||||||
|  | 
 | ||||||
|  | import lombok.Getter; | ||||||
|  | 
 | ||||||
|  | /** | ||||||
|  |  * MybatisPlus自定义SQL方法枚举 | ||||||
|  |  */ | ||||||
|  | @Getter | ||||||
|  | public enum CustomerSqlMethod { | ||||||
|  |     /** | ||||||
|  |      * 插入 | ||||||
|  |      */ | ||||||
|  |     INSERT_IGNORE_ONE("insertIgnore", "插入一条数据(选择字段插入),如果中已经存在相同的记录,则忽略当前新数据", "<script>\nINSERT IGNORE INTO %s %s VALUES %s\n</script>"), | ||||||
|  |     /** | ||||||
|  |      * 替换 | ||||||
|  |      */ | ||||||
|  |     REPLACE_ONE("replace", "替换一条数据(选择字段插入),存在则替换,不存在则插入", "<script>\nREPLACE INTO %s %s VALUES %s\n</script>"); | ||||||
|  | 
 | ||||||
|  |     private final String method; | ||||||
|  |     private final String desc; | ||||||
|  |     private final String sql; | ||||||
|  | 
 | ||||||
|  |     CustomerSqlMethod(String method, String desc, String sql) { | ||||||
|  |         this.method = method; | ||||||
|  |         this.desc = desc; | ||||||
|  |         this.sql = sql; | ||||||
|  |     } | ||||||
|  | } | ||||||
| @ -0,0 +1,77 @@ | |||||||
|  | package com.glxp.api.config; | ||||||
|  | 
 | ||||||
|  | import com.baomidou.mybatisplus.annotation.IdType; | ||||||
|  | import com.baomidou.mybatisplus.core.enums.SqlMethod; | ||||||
|  | import com.baomidou.mybatisplus.core.injector.AbstractMethod; | ||||||
|  | import com.baomidou.mybatisplus.core.metadata.TableFieldInfo; | ||||||
|  | import com.baomidou.mybatisplus.core.metadata.TableInfo; | ||||||
|  | import com.baomidou.mybatisplus.core.metadata.TableInfoHelper; | ||||||
|  | import com.baomidou.mybatisplus.core.toolkit.StringUtils; | ||||||
|  | import com.baomidou.mybatisplus.core.toolkit.sql.SqlScriptUtils; | ||||||
|  | import lombok.Setter; | ||||||
|  | import lombok.experimental.Accessors; | ||||||
|  | import org.apache.ibatis.executor.keygen.Jdbc3KeyGenerator; | ||||||
|  | import org.apache.ibatis.executor.keygen.KeyGenerator; | ||||||
|  | import org.apache.ibatis.executor.keygen.NoKeyGenerator; | ||||||
|  | import org.apache.ibatis.mapping.MappedStatement; | ||||||
|  | import org.apache.ibatis.mapping.SqlSource; | ||||||
|  | 
 | ||||||
|  | import java.util.List; | ||||||
|  | import java.util.function.Predicate; | ||||||
|  | 
 | ||||||
|  | /** | ||||||
|  |  * 批量新增数据,自选字段 insert ignore | ||||||
|  |  */ | ||||||
|  | public class InsertIgnoreBatch extends AbstractMethod { | ||||||
|  |     /** | ||||||
|  |      * mapper 对应的方法名 | ||||||
|  |      */ | ||||||
|  |     private static final String MAPPER_METHOD = "insertIgnoreBatch"; | ||||||
|  | 
 | ||||||
|  |     /** | ||||||
|  |      * 字段筛选条件 | ||||||
|  |      */ | ||||||
|  |     @Setter | ||||||
|  |     @Accessors(chain = true) | ||||||
|  |     private Predicate<TableFieldInfo> predicate; | ||||||
|  | 
 | ||||||
|  |     @Override | ||||||
|  |     public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) { | ||||||
|  |         KeyGenerator keyGenerator = new NoKeyGenerator(); | ||||||
|  |         CustomerSqlMethod sqlMethod = CustomerSqlMethod.INSERT_IGNORE_ONE; | ||||||
|  |         List<TableFieldInfo> fieldList = tableInfo.getFieldList(); | ||||||
|  |         String insertSqlColumn = tableInfo.getKeyInsertSqlColumn(false, true) + | ||||||
|  |                 this.filterTableFieldInfo(fieldList, predicate, TableFieldInfo::getInsertSqlColumn, EMPTY); | ||||||
|  |         String columnScript = LEFT_BRACKET + insertSqlColumn.substring(0, insertSqlColumn.length() - 1) + RIGHT_BRACKET; | ||||||
|  |         String insertSqlProperty = tableInfo.getKeyInsertSqlProperty(false, ENTITY_DOT, true) + | ||||||
|  |                 this.filterTableFieldInfo(fieldList, predicate, i -> i.getInsertSqlProperty(ENTITY_DOT), EMPTY); | ||||||
|  |         insertSqlProperty = LEFT_BRACKET + insertSqlProperty.substring(0, insertSqlProperty.length() - 1) + RIGHT_BRACKET; | ||||||
|  |         String valuesScript = SqlScriptUtils.convertForeach(insertSqlProperty, "list", null, ENTITY, COMMA); | ||||||
|  |         String keyProperty = null; | ||||||
|  |         String keyColumn = null; | ||||||
|  |         // 表包含主键处理逻辑,如果不包含主键当普通字段处理
 | ||||||
|  |         if (tableInfo.havePK()) { | ||||||
|  |             if (tableInfo.getIdType() == IdType.AUTO) { | ||||||
|  |                 /* 自增主键 */ | ||||||
|  |                 keyGenerator = new Jdbc3KeyGenerator(); | ||||||
|  |                 keyProperty = tableInfo.getKeyProperty(); | ||||||
|  |                 keyColumn = tableInfo.getKeyColumn(); | ||||||
|  |             } else { | ||||||
|  |                 if (null != tableInfo.getKeySequence()) { | ||||||
|  |                     keyGenerator = TableInfoHelper.genKeyGenerator(MAPPER_METHOD, tableInfo, builderAssistant); | ||||||
|  |                     keyProperty = tableInfo.getKeyProperty(); | ||||||
|  |                     keyColumn = tableInfo.getKeyColumn(); | ||||||
|  |                 } | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |         String sql = String.format(sqlMethod.getSql(), tableInfo.getTableName(), columnScript, valuesScript); | ||||||
|  |         SqlSource sqlSource = languageDriver.createSqlSource(configuration, sql, modelClass); | ||||||
|  |         return this.addInsertMappedStatement(mapperClass, modelClass, MAPPER_METHOD, sqlSource, keyGenerator, keyProperty, keyColumn); | ||||||
|  |     } | ||||||
|  | 
 | ||||||
|  |     @Override | ||||||
|  |     public String getMethod(SqlMethod sqlMethod) { | ||||||
|  |         // 自定义 mapper 方法名
 | ||||||
|  |         return MAPPER_METHOD; | ||||||
|  |     } | ||||||
|  | } | ||||||
					Loading…
					
					
				
		Reference in New Issue