From 2b88b81d497a7643237cc0eb3303b58bb75f41af Mon Sep 17 00:00:00 2001 From: hewei Date: Wed, 17 May 2017 18:31:12 +0800 Subject: [PATCH] =?UTF-8?q?=E7=A7=BB=E9=99=A4=E4=B8=80=E4=BA=9B=E8=BF=87?= =?UTF-8?q?=E6=97=B6=E7=9A=84=E6=8F=92=E4=BB=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../plugins/BatchInsertOldPlugin.java | 302 ------------------ .../generator/plugins/BatchInsertPlugin.java | 3 - .../plugins/CriteriaBuilderPlugin.java | 39 --- 3 files changed, 344 deletions(-) delete mode 100644 src/main/java/com/itfsw/mybatis/generator/plugins/BatchInsertOldPlugin.java delete mode 100644 src/main/java/com/itfsw/mybatis/generator/plugins/CriteriaBuilderPlugin.java diff --git a/src/main/java/com/itfsw/mybatis/generator/plugins/BatchInsertOldPlugin.java b/src/main/java/com/itfsw/mybatis/generator/plugins/BatchInsertOldPlugin.java deleted file mode 100644 index 4f4c008..0000000 --- a/src/main/java/com/itfsw/mybatis/generator/plugins/BatchInsertOldPlugin.java +++ /dev/null @@ -1,302 +0,0 @@ -/* - * Copyright (c) 2017. - * - * 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 - * - * http://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 com.itfsw.mybatis.generator.plugins; - -import com.itfsw.mybatis.generator.plugins.utils.BasePlugin; -import com.itfsw.mybatis.generator.plugins.utils.XmlElementGeneratorTools; -import org.mybatis.generator.api.IntrospectedColumn; -import org.mybatis.generator.api.IntrospectedTable; -import org.mybatis.generator.api.dom.java.*; -import org.mybatis.generator.api.dom.xml.Attribute; -import org.mybatis.generator.api.dom.xml.Document; -import org.mybatis.generator.api.dom.xml.TextElement; -import org.mybatis.generator.api.dom.xml.XmlElement; -import org.mybatis.generator.codegen.mybatis3.ListUtilities; -import org.mybatis.generator.codegen.mybatis3.MyBatis3FormattingUtilities; -import org.mybatis.generator.config.Context; -import org.mybatis.generator.config.PluginConfiguration; - -import java.util.ArrayList; -import java.util.List; - -/** - * --------------------------------------------------------------------------- - * 批量插入插件 - * --------------------------------------------------------------------------- - * @author: hewei - * @time:2017/1/13 9:33 - * --------------------------------------------------------------------------- - */ -@Deprecated -public class BatchInsertOldPlugin extends BasePlugin { - public static final String METHOD_BATCH_INSERT = "batchInsert"; // 方法名 - private boolean hasModelBuilderPlugin; // 是否配置了ModelBuilderPlugin插件 - - /** - * {@inheritDoc} - */ - @Override - public boolean validate(List warnings) { - // 插件使用前提是数据库为MySQL或者SQLserver，因为返回主键使用了JDBC的getGenereatedKeys方法获取主键 - if ("com.mysql.jdbc.Driver".equalsIgnoreCase(this.getContext().getJdbcConnectionConfiguration().getDriverClass()) == false - && "com.microsoft.jdbc.sqlserver.SQLServer".equalsIgnoreCase(this.getContext().getJdbcConnectionConfiguration().getDriverClass()) == false - && "com.microsoft.sqlserver.jdbc.SQLServerDriver".equalsIgnoreCase(this.getContext().getJdbcConnectionConfiguration().getDriverClass()) == false){ - logger.warn("itfsw:插件" + this.getClass().getTypeName() + "插件使用前提是数据库为MySQL或者SQLserver，因为返回主键使用了JDBC的getGenereatedKeys方法获取主键！"); - return false; - } - - logger.warn("itfsw:插件" + this.getClass().getTypeName() + "插件已过期，不再维护请使用com.itfsw.mybatis.generator.plugins.BatchInsertPlugin区分batchInsert和batchInsertSelective！"); - - return super.validate(warnings); - } - - /** - * 初始化 - * 具体执行顺序 http://www.mybatis.org/generator/reference/pluggingIn.html - * - * @param introspectedTable - * @return - */ - @Override - public void initialized(IntrospectedTable introspectedTable) { - // 插件使用前提是使用了ModelColumnPlugin插件 - Context ctx = getContext(); - // 利用反射获取pluginConfigurations属性 - try { - java.lang.reflect.Field field = Context.class.getDeclaredField("pluginConfigurations"); - field.setAccessible(true); - List list = (List) field.get(ctx); - // 检查是否配置了ModelColumnPlugin插件 - for (PluginConfiguration config: list) { - if (ModelColumnPlugin.class.getName().equals(config.getConfigurationType())){ - this.hasModelBuilderPlugin = true; - } - } - } catch (Exception e) { - logger.error(e.getLocalizedMessage()); - } - } - - /** - * Java Client Methods 生成 - * 具体执行顺序 http://www.mybatis.org/generator/reference/pluggingIn.html - * - * @param interfaze - * @param topLevelClass - * @param introspectedTable - * @return - */ - @Override - public boolean clientGenerated(Interface interfaze, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { - if (hasModelBuilderPlugin){ - // 方法生成 - Method method = new Method(METHOD_BATCH_INSERT); - // 返回值类型 - method.setReturnType(FullyQualifiedJavaType.getIntInstance()); - // 添加参数 - FullyQualifiedJavaType type = FullyQualifiedJavaType.getNewListInstance(); - type.addTypeArgument(introspectedTable.getRules().calculateAllFieldsClass()); - method.addParameter(new Parameter(type, "list", "@Param(\"list\")")); - FullyQualifiedJavaType type1 = new FullyQualifiedJavaType(introspectedTable.getRules().calculateAllFieldsClass().getShortName()+"."+ModelColumnPlugin.ENUM_NAME); - method.addParameter(new Parameter(type1, "insertColumns", "@Param(\"insertColumns\")", true)); - // 添加方法说明 - commentGenerator.addGeneralMethodComment(method, introspectedTable); - - // interface 增加方法 - interfaze.addMethod(method); - logger.debug("itfsw(批量插入插件):" + interfaze.getType().getShortName() + "增加batchInsert方法。"); - } else { - // 方法生成 - Method method = new Method(METHOD_BATCH_INSERT); - // 方法可见性 interface会忽略 - // method.setVisibility(JavaVisibility.PUBLIC); - // 返回值类型 - method.setReturnType(FullyQualifiedJavaType.getIntInstance()); - // 添加参数 - FullyQualifiedJavaType type = FullyQualifiedJavaType.getNewListInstance(); - type.addTypeArgument(introspectedTable.getRules().calculateAllFieldsClass()); - method.addParameter(new Parameter(type, "list", "@Param(\"list\")")); - // 添加方法说明 - commentGenerator.addGeneralMethodComment(method, introspectedTable); - - // interface 增加方法 - interfaze.addMethod(method); - logger.debug("itfsw(批量插入插件):" + interfaze.getType().getShortName() + "增加batchInsert方法。"); - } - - return true; - } - - /** - * SQL Map Methods 生成 - * 具体执行顺序 http://www.mybatis.org/generator/reference/pluggingIn.html - * - * @param document - * @param introspectedTable - * @return - */ - @Override - public boolean sqlMapDocumentGenerated(Document document, IntrospectedTable introspectedTable) { - if (hasModelBuilderPlugin){ - XmlElement element = new XmlElement("insert"); - element.addAttribute(new Attribute("id", METHOD_BATCH_INSERT)); - // 参数类型 - element.addAttribute(new Attribute("parameterType", "map")); - // 添加注释(!!!必须添加注释，overwrite覆盖生成时，@see XmlFileMergerJaxp.isGeneratedNode会去判断注释中是否存在OLD_ELEMENT_TAGS中的一点，例子：@mbg.generated) - commentGenerator.addComment(element); - - // 使用JDBC的getGenereatedKeys方法获取主键并赋值到keyProperty设置的领域模型属性中。所以只支持MYSQL和SQLServer - XmlElementGeneratorTools.useGeneratedKeys(element, introspectedTable); - - // choose 节点 - XmlElement choose = new XmlElement("choose"); - // when 节点 - XmlElement when = new XmlElement("when"); - when.addAttribute(new Attribute("test", "insertColumns.length != 0")); - when.addElement(new TextElement("insert into "+introspectedTable.getFullyQualifiedTableNameAtRuntime()+" (")); - - XmlElement foreachInsertColumns = new XmlElement("foreach"); - foreachInsertColumns.addAttribute(new Attribute("collection", "insertColumns")); - foreachInsertColumns.addAttribute(new Attribute("item", "column")); - foreachInsertColumns.addAttribute(new Attribute("separator", ",")); - foreachInsertColumns.addElement(new TextElement("${column.value}")); - - when.addElement(foreachInsertColumns); - - when.addElement(new TextElement(")")); - - // values - when.addElement(new TextElement("values")); - - // foreach values - XmlElement foreachValues = new XmlElement("foreach"); - foreachValues.addAttribute(new Attribute("collection", "list")); - foreachValues.addAttribute(new Attribute("item", "item")); - foreachValues.addAttribute(new Attribute("separator", ",")); - - foreachValues.addElement(new TextElement("(")); - - // foreach 所有插入的列，比较是否存在 - XmlElement foreachInsertColumnsCheck = new XmlElement("foreach"); - foreachInsertColumnsCheck.addAttribute(new Attribute("collection", "insertColumns")); - foreachInsertColumnsCheck.addAttribute(new Attribute("item", "column")); - foreachInsertColumnsCheck.addAttribute(new Attribute("separator", ",")); - - // 所有表字段 - List columns = ListUtilities.removeIdentityAndGeneratedAlwaysColumns(introspectedTable.getAllColumns()); - for (int i = 0; i < columns.size(); i++) { - IntrospectedColumn introspectedColumn = columns.get(i); - XmlElement check = new XmlElement("if"); - check.addAttribute(new Attribute("test", "'"+introspectedColumn.getActualColumnName()+"' == column.value")); - check.addElement(new TextElement(MyBatis3FormattingUtilities.getParameterClause(introspectedColumn, "item."))); - - foreachInsertColumnsCheck.addElement(check); - } - foreachValues.addElement(foreachInsertColumnsCheck); - - foreachValues.addElement(new TextElement(")")); - - when.addElement(foreachValues); - choose.addElement(when); - - // 普通不带insertColumns参数实现 - XmlElement otherwise = new XmlElement("otherwise"); - this.addNormalBatchInsertXml(otherwise, introspectedTable); - choose.addElement(otherwise); - - element.addElement(choose); - if (context.getPlugins().sqlMapInsertElementGenerated(element, introspectedTable)) { - document.getRootElement().addElement(element); - logger.debug("itfsw(批量插入插件):" + introspectedTable.getMyBatis3XmlMapperFileName() + "增加batchInsert实现方法。"); - } - } else { - XmlElement element = new XmlElement("insert"); //$NON-NLS-1$ - - element.addAttribute(new Attribute("id", METHOD_BATCH_INSERT)); //$NON-NLS-1$ - - // 参数类型 - element.addAttribute(new Attribute("parameterType", FullyQualifiedJavaType.getNewListInstance().getFullyQualifiedName())); - - // 添加注释(!!!必须添加注释，overwrite覆盖生成时，@see XmlFileMergerJaxp.isGeneratedNode会去判断注释中是否存在OLD_ELEMENT_TAGS中的一点，例子：@mbg.generated) - commentGenerator.addComment(element); - - // 使用JDBC的getGenereatedKeys方法获取主键并赋值到keyProperty设置的领域模型属性中。所以只支持MYSQL和SQLServer - XmlElementGeneratorTools.useGeneratedKeys(element, introspectedTable); - - // 普通插入语句 - this.addNormalBatchInsertXml(element, introspectedTable); - - document.getRootElement().addElement(element); - logger.debug("itfsw(批量插入插件):" + introspectedTable.getMyBatis3XmlMapperFileName() + "增加batchInsert实现方法。"); - } - return true; - } - - /** - * 普通不带insertColumns的插入语句Xml实现 - * - * @param element - * @param introspectedTable - */ - private void addNormalBatchInsertXml(XmlElement element, IntrospectedTable introspectedTable) { - - StringBuilder insertClause = new StringBuilder(); - StringBuilder valuesClause = new StringBuilder(); - - insertClause.append("insert into "); //$NON-NLS-1$ - insertClause.append(introspectedTable.getFullyQualifiedTableNameAtRuntime()); - insertClause.append(" ("); //$NON-NLS-1$ - - valuesClause.append(" ("); //$NON-NLS-1$ - - List valuesClauses = new ArrayList(); - List columns = ListUtilities.removeIdentityAndGeneratedAlwaysColumns(introspectedTable.getAllColumns()); - for (int i = 0; i < columns.size(); i++) { - IntrospectedColumn introspectedColumn = columns.get(i); - - insertClause.append(MyBatis3FormattingUtilities.getEscapedColumnName(introspectedColumn)); - - // 生成foreach下插入values - valuesClause.append(MyBatis3FormattingUtilities.getParameterClause(introspectedColumn, "item.")); - if (i + 1 < columns.size()) { - insertClause.append(", "); //$NON-NLS-1$ - valuesClause.append(", "); //$NON-NLS-1$ - } - } - - insertClause.append(')'); - element.addElement(new TextElement(insertClause.toString())); - - valuesClause.append(')'); - valuesClauses.add(valuesClause.toString()); - - - // 添加foreach节点 - XmlElement foreachElement = new XmlElement("foreach"); - foreachElement.addAttribute(new Attribute("collection", "list")); - foreachElement.addAttribute(new Attribute("item", "item")); - foreachElement.addAttribute(new Attribute("separator", ",")); - - for (String clause : valuesClauses) { - foreachElement.addElement(new TextElement(clause)); - } - - // values 构建 - element.addElement(new TextElement("values")); - element.addElement(foreachElement); - } -} diff --git a/src/main/java/com/itfsw/mybatis/generator/plugins/BatchInsertPlugin.java b/src/main/java/com/itfsw/mybatis/generator/plugins/BatchInsertPlugin.java index d8a623a..f5cdd65 100644 --- a/src/main/java/com/itfsw/mybatis/generator/plugins/BatchInsertPlugin.java +++ b/src/main/java/com/itfsw/mybatis/generator/plugins/BatchInsertPlugin.java @@ -65,9 +65,6 @@ public class BatchInsertPlugin extends BasePlugin { return false; } - // TODO - logger.warn("itfsw:插件" + this.getClass().getTypeName() + "插件如之前使用1.0.5-的插件可继续使用com.itfsw.mybatis.generator.plugins.BatchInsertOldPlugin或者修改batchInsert(@Param(\"list\") List list, @Param(\"insertColumns\") Tb.Column ... selective)调用为batchInsertSelective！"); - return super.validate(warnings); } diff --git a/src/main/java/com/itfsw/mybatis/generator/plugins/CriteriaBuilderPlugin.java b/src/main/java/com/itfsw/mybatis/generator/plugins/CriteriaBuilderPlugin.java deleted file mode 100644 index cde50af..0000000 --- a/src/main/java/com/itfsw/mybatis/generator/plugins/CriteriaBuilderPlugin.java +++ /dev/null @@ -1,39 +0,0 @@ -/* - * Copyright (c) 2017. - * - * 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 - * - * http://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 com.itfsw.mybatis.generator.plugins; - -import java.util.List; - -/** - * --------------------------------------------------------------------------- - * 增加Criteria Builder方法 - * --------------------------------------------------------------------------- - * @author: hewei - * @time:2016/12/28 14:56 - * --------------------------------------------------------------------------- - */ -@Deprecated -public class CriteriaBuilderPlugin extends ExampleEnhancedPlugin{ - /** - * {@inheritDoc} - */ - @Override - public boolean validate(List warnings) { - logger.warn("itfsw:插件"+this.getClass().getTypeName()+"插件已经过时，请使用com.itfsw.mybatis.generator.plugins.ExampleEnhancedPlugin插件替换！"); - return super.validate(warnings); - } -}