基于Java的fluent-mybatis语法增强框架设计源码

package cn.org.atool.fluent.mybatis.base.provider; import cn.org.atool.fluent.mybatis.If; import cn.org.atool.fluent.mybatis.base.IEntity; import cn.org.atool.fluent.mybatis.base.crud.BaseQuery; import cn.org.atool.fluent.mybatis.base.crud.IQuery; import cn.org.atool.fluent.mybatis.base.crud.IUpdate; import cn.org.atool.fluent.mybatis.base.crud.IWrapper; import cn.org.atool.fluent.mybatis.base.entity.IMapping; import cn.org.atool.fluent.mybatis.base.entity.IRichEntity; import cn.org.atool.fluent.mybatis.base.entity.PkGeneratorKits; import cn.org.atool.fluent.mybatis.base.entity.TableId; import cn.org.atool.fluent.mybatis.base.model.FieldMapping; import cn.org.atool.fluent.mybatis.base.model.InsertList; import cn.org.atool.fluent.mybatis.base.model.SqlOp; import cn.org.atool.fluent.mybatis.base.model.UpdateDefault; import cn.org.atool.fluent.mybatis.exception.FluentMybatisException; import cn.org.atool.fluent.mybatis.mapper.MapperSql; import cn.org.atool.fluent.mybatis.segment.fragment.Column; import cn.org.atool.fluent.mybatis.segment.fragment.IFragment; import cn.org.atool.fluent.mybatis.segment.fragment.JoiningFrag; import cn.org.atool.fluent.mybatis.segment.model.WrapperData; import cn.org.atool.fluent.mybatis.utility.RefKit; import cn.org.atool.fluent.mybatis.utility.SqlProviderKit; import org.apache.ibatis.executor.keygen.Jdbc3KeyGenerator; import org.apache.ibatis.executor.keygen.KeyGenerator; import org.apache.ibatis.executor.keygen.NoKeyGenerator; import java.util.*; import java.util.function.Consumer; import java.util.stream.Collectors; import java.util.stream.Stream; import static cn.org.atool.fluent.mybatis.If.isBlank; import static cn.org.atool.fluent.mybatis.If.notBlank; import static cn.org.atool.fluent.mybatis.base.model.InsertList.el; import static cn.org.atool.fluent.mybatis.mapper.FluentConst.*; import static cn.org.atool.fluent.mybatis.mapper.MapperSql.brackets; import static cn.org.atool.fluent.mybatis.mapper.MapperSql.tmpTable; import static cn.org.atool.fluent.mybatis.segment.fragment.KeyFrag.*; import static cn.org.atool.fluent.mybatis.utility.MybatisUtil.*; import static cn.org.atool.fluent.mybatis.utility.StrConstant.*; import static java.lang.String.format; import static java.util.stream.Collectors.joining; import static java.util.stream.Collectors.toList; /** * 通用SQL构造器 * * @author darui.wu */ @SuppressWarnings({"unchecked", "rawtypes"}) public class CommonSqlKit implements SqlKit { public CommonSqlKit() { } @Override public KeyGenerator insert(StatementBuilder builder, FieldMapping primary, TableId tableId) { if (tableId == null) { return NoKeyGenerator.INSTANCE; } else if (this.isAutoKeyGenerator(tableId)) { return Jdbc3KeyGenerator.INSTANCE; } else { return builder.handleSelectKey(primary, tableId); } } @Override public KeyGenerator insertBatch(IMapping mapping, StatementBuilder builder, FieldMapping primary, TableId tableId) { if (tableId == null) { return NoKeyGenerator.INSTANCE; } else if (this.isAutoKeyGenerator(tableId)) { return Jdbc3KeyGenerator.INSTANCE; } else if (tableId.isSeqBefore(mapping.db())) { /* 使用 select insert 方式 */ return NoKeyGenerator.INSTANCE; } else { return builder.handleSelectKey(primary, tableId); } } protected boolean isAutoKeyGenerator(TableId tableId) { return tableId.auto && isBlank(tableId.seqName); } @Override public <E extends IEntity> String insertEntity(IMapping mapping, String prefix, E entity, boolean withPk) { assertNotNull(Param_Entity, entity); withPk = validateInsertEntity(mapping, entity, withPk, mapping.defaultSetter()::setInsertDefault, mapping.tableId()); MapperSql sql = new MapperSql(); sql.INSERT_INTO(dynamic(entity, mapping)); InsertList inserts = this.insertColumns(mapping, prefix, entity, withPk); sql.INSERT_COLUMNS(mapping, inserts.columns); sql.VALUES(); sql.APPEND(brackets(COMMA_SPACE, inserts.values)); return sql.toString(); } /** * 单个插入时, insert字段和值 * * @param entity 实体实例 * @param withPk true:with id; false: without id */ private InsertList insertColumns(IMapping mapping, String prefix, IEntity entity, boolean withPk) { InsertList inserts = new InsertList(); List<FieldMapping> fields = mapping.allFields(); Map map = entity.toEntityMap(); for (FieldMapping f : fields) { if (!f.isPrimary() || withPk) { inserts.add(prefix, f, map.get(f.name), f.insert); } } return inserts; } @Override public String insertSelect(IMapping mapping, String tableName, String[] fields, IQuery query) { assertNotBlank("tableName", tableName); String columns = Stream.of(fields).map(mapping.db()::wrap).collect(joining(", ")); if (isBlank(columns)) { List<String> list = new ArrayList<>(); List<String> items = splitByComma(query.data().getSelect().get(mapping)); for (String item : items) { String[] arr = splitBySpace(item).stream().filter(If::notBlank).toArray(String[]::new); list.add(arr[arr.length - 1]); } columns = String.join(COMMA_SPACE, list); } assertNotBlank(Param_Fields, columns); assertNotNull(Param_EW, query); if (!query.data().hasSelect()) { ((BaseQuery) query).select(fields); } return joinWithSpace( INSERT_INTO.key(), tableName, brackets(columns), query.data().sql(false).get(mapping)); } @Override public <E extends IEntity> String insertBatch(IMapping mapping, Collection<E> entities, boolean withPk, TableId tableId) { MapperSql sql = new MapperSql(); List<Map> maps = this.toMaps(mapping, entities, withPk); /* 所有非空字段 */ List<FieldMapping> nonFields = this.nonFields(mapping, maps, withPk); String tableName = dynamic(entities, mapping); sql.INSERT_INTO(tableName); if (this.isSelectInsert(mapping, withPk, tableId)) { this.insertSelect(mapping, tableId, withPk, sql, maps, nonFields); } else { this.insertValues(mapping, sql, maps, nonFields); } return sql.toString(); } /** * 是否用 insert select 方式批量插入 */ protected boolean isSelectInsert(IMapping mapping, boolean withPk, TableId tableId) { return !withPk && tableId != null && tableId.isSeqBefore(mapping.db()); } /** * https://blog.csdn.net/w_y_t_/article/details/51416201 * <p> * https://www.cnblogs.com/xunux/p/4882761.html * <p> * https://blog.csdn.net/weixin_41175479/article/details/80608512 */ protected void insertSelect(IMapping mapping, TableId tableId, boolean withPk, MapperSql sql, List<Map> maps, List<FieldMapping> nonFields) { List<String> columns = new ArrayList<>(); if (!withPk && tableId != null) { columns.add(tableId.column); } nonFields.stream().map(f -> f.column).forEach(columns::add); sql.INSERT_COLUMNS(mapping, columns); sql.APPEND("SELECT"); if (!withPk && tableId != null) { String seq = isBlank(tableId.seqName) ? mapping.db().feature.getSeq() : tableId.seqName; sql.APPEND(getSeq(seq) + ","); } sql.APPEND("TMP.* FROM ("); for (int index = 0; index < maps.size(); index++) { if (index > 0) { sql.APPEND(" UNION ALL "); } sql.APPEND("(SELECT"); boolean first = true; for (FieldMapping f : nonFields) { if (f.isPrimary(