【Mybatis】Mybatis-Generator-Postgresql(批量更新)方法插件

系列

1. 【Mybatis】Mybatis-Generator-Postgresql返回主键插件
2. 【Mybatis】Mybatis-Generator-Oracle返回主键插件
3. 【Mybatis】Mybatis-Generator-batchDelete(批量删除)方法插件
4. 【Mybatis】Mybatis-Generator-Postgresql&Mysql(批量新增)方法插件
5. 【Mybatis】Mybatis-Generator-Oracle(批量新增)方法插件
6. 【Mybatis】Mybatis-Generator-Mysql(批量更新)方法插件
7. 【Mybatis】Mybatis-Generator-Postgresql(批量更新)方法插件
8. 【Mybatis】Mybatis-Generator-Oracle(批量更新)方法插件
9. 【Mybatis】Mybatis-Generator-Tool 生成Java和Xml的Method工具类

前言

PG批量更新插件，核心贴出来，有需要的看我的GITHUB，准备更新微服务BLOG

github上针对mybatis-generator还有很多不同类型的插件，我实在懒的看，而且太多复杂的功能，我简单点搞自己需要的。

装备

1. maven

   <dependencies>
       
       <dependency>
           <groupId>org.mybatisgroupId>
           <artifactId>mybatisartifactId>
           <version>3.4.6version>
       dependency>
   
       
       <dependency>
           <groupId>org.mybatis.generatorgroupId>
           <artifactId>mybatis-generator-coreartifactId>
           <version>1.3.7version>
       dependency>
   dependencies>
   

Core-Code

sql

update t_hui_table from (values (1,'name'),(2,'name2') ) as temp (id,name) where t_hui_table.id = temp.id

mybatis-implement

<update id="batchUpdate" parameterType="java.util.List">
    update T_HUI_ORDER
    <set>
       ORDER_NAME = temp.ORDER_NAME,
       PRODUCT_ID = temp.PRODUCT_ID,
       BUY_QUANTITY = temp.BUY_QUANTITY,
       CREATED_TIME = temp.CREATED_TIME
    set>
    from (values
    <foreach collection="recordList" index="index" item="item" separator=",">
      (
      #{item.orderId,jdbcType=VARCHAR},#{item.orderName,jdbcType=VARCHAR},#{item.productId,jdbcType=VARCHAR},#{item.buyQuantity,jdbcType=DECIMAL},to_timestamp(#{item.createdTime,jdbcType=TIMESTAMP},'yyyy-MM-dd hh24:mi:ss')
      )
    foreach>
    ) as temp (ORDER_ID,ORDER_NAME,PRODUCT_ID,BUY_QUANTITY,CREATED_TIME) where T_HUI_ORDER.ORDER_ID=temp.ORDER_ID;
  update>

plugin-code

/**
 * PostgreBatchUpdatePlugin
 * 

 * Description:
 * 

 * Creation Time: 2018/12/15 1:12.
 *
 * @author HuWeihui
 */
public class PostgreBatchUpdatePlugin extends PluginAdapter {

    private final static String BATCH_UPDATE = "batchUpdate";

    private final static String PARAMETER_NAME = "recordList";

    @Override
    public boolean validate(List<String> list) {
        return true;
    }

    @Override
    public boolean clientGenerated(Interface interfaze, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
        if (introspectedTable.getTargetRuntime().equals(IntrospectedTable.TargetRuntime.MYBATIS3)) {
            MethodGeneratorTool.defaultBatchInsertOrUpdateMethodGen(MethodGeneratorTool.UPDATE,interfaze,introspectedTable,context);
        }
        return super.clientGenerated(interfaze, topLevelClass, introspectedTable);
    }

    @Override
    public boolean sqlMapDocumentGenerated(Document document, IntrospectedTable introspectedTable) {
        if (introspectedTable.getTargetRuntime().equals(IntrospectedTable.TargetRuntime.MYBATIS3)) {
            addSqlMapper(document, introspectedTable);
        }
        return super.sqlMapDocumentGenerated(document, introspectedTable);
    }


    private void addSqlMapper(Document document, IntrospectedTable introspectedTable){
        String tableName = introspectedTable.getFullyQualifiedTableNameAtRuntime();
        List<IntrospectedColumn> columnList = introspectedTable.getAllColumns();

        String primaryKeyName = introspectedTable.getPrimaryKeyColumns().get(0).getActualColumnName();
        //primaryKey的JAVA名字
        String primaryKeyJavaName = introspectedTable.getPrimaryKeyColumns().get(0).getJavaProperty();

        XmlElement updateElement = SqlMapperGeneratorTool.baseElementGenerator(SqlMapperGeneratorTool.UPDATE,
                BATCH_UPDATE,
                FullyQualifiedJavaType.getNewListInstance());

        XmlElement foreachElement = SqlMapperGeneratorTool.baseForeachElementGenerator(PARAMETER_NAME,
                "item",
                "index",
                ",");

        String baseSql = String.format("update %s", tableName);

        updateElement.addElement(new TextElement(baseSql));

        XmlElement setElement = new XmlElement("set");


        StringBuilder columnInfo = new StringBuilder();
        StringBuilder valuesInfo = new StringBuilder();

        StringBuilder columnInfoTotal = new StringBuilder();
        for (int i = 0; i < columnList.size(); i++) {

            IntrospectedColumn introspectedColumn = columnList.get(i);

            columnInfo.append(introspectedColumn.getActualColumnName());
            columnInfoTotal.append(introspectedColumn.getActualColumnName());
            if (introspectedColumn.getFullyQualifiedJavaType().equals(FullyQualifiedJavaType.getDateInstance())){
                valuesInfo.append("to_timestamp(");
                valuesInfo.append(MyBatis3FormattingUtilities.getParameterClause(introspectedColumn, "item."));
                valuesInfo.append(",'yyyy-MM-dd hh24:mi:ss')");
            }else {
                valuesInfo.append(MyBatis3FormattingUtilities.getParameterClause(introspectedColumn, "item."));
            }

            String setSql = String.format(" %s = %s," ,columnInfo,"temp."+columnInfo);
            if (i == (columnList.size() - 1)) {
                setSql = setSql.substring(0, setSql.length() - 1);
            }
            if (!introspectedColumn.isIdentity() && !introspectedColumn.getActualColumnName().equals(primaryKeyName)) {
                setElement.addElement(new TextElement(setSql));
            }

            if (i != (columnList.size() - 1)) {
                valuesInfo.append(",");
                columnInfo.append(",");
                columnInfoTotal.append(",");
            }

            columnInfo.delete(0, valuesInfo.length());
        }

        foreachElement.addElement(new TextElement("("));

        foreachElement.addElement(new TextElement(valuesInfo.toString()));

        foreachElement.addElement(new TextElement(")"));

        updateElement.addElement(setElement);

        updateElement.addElement(new TextElement("from (values"));

        updateElement.addElement(foreachElement);

        updateElement.addElement(new TextElement(String.format(") as temp (%s) where %s.%s=temp.%s;",columnInfoTotal,tableName,primaryKeyName,primaryKeyName)));

        //3.parent Add
        document.getRootElement().addElement(updateElement);
    }
}

plugin-common-tool

插件通用方法/工具类，上面生成方法基于了一个工具类，下面博客有源码，真正需要的可以看看github

9. 【Mybatis】Mybatis-Generator-Tool 生成Java和Xml的Method工具类

使用方法

generatorConfig.xml

<plugin type="com.hui.mybatis.plugins.PostgreBatchUpdatePlugin"/>

pom.xml

<build>
        <plugins>
            <plugin>
                <groupId>org.mybatis.generatorgroupId>
                <artifactId>mybatis-generator-maven-pluginartifactId>
                <version>${mybatis.generator}version>
                <configuration>
                    
                    <configurationFile>
                        ${basedir}/src/main/resources/generator/generatorConfig.xml
                    configurationFile>
                    
                    <verbose>trueverbose>
                    
                    <overwrite>trueoverwrite>
                configuration>
                <dependencies>
                    
                    <dependency>
                        <groupId>org.mybatis.generatorgroupId>
                        <artifactId>mybatis-generator-coreartifactId>
                        <version>${mybatis.generator}version>
                    dependency>

                    
                    <dependency>
                        <groupId>com.hui.mybatis.pluginsgroupId>
                        <artifactId>hui-mybatis-pluginsartifactId>
                        <version>0.0.1-SNAPSHOTversion>
                    dependency>

                    
                    <dependency>
                        <groupId>mysqlgroupId>
                        <artifactId>mysql-connector-javaartifactId>
                        <version>5.1.44version>
                    dependency>

                    
                    <dependency>
                        <groupId>com.oraclegroupId>
                        <artifactId>ojdbc6artifactId>
                        <version>11.1.0.7.0version>
                    dependency>

                    
                    <dependency>
                        <groupId>org.postgresqlgroupId>
                        <artifactId>postgresqlartifactId>
                        <version>${postgresql.version}version>
                    dependency>

                dependencies>
            plugin>
        plugins>
    build>

Github

https://github.com/ithuhui/hui-mybatis-plugins/tree/master/src/main/java/com/hui/mybatis/plugins

作者

 作者：HuHui
 转载：欢迎一起讨论web和大数据问题,转载请注明作者和原文链接，感谢