增强distinct方法，实现链式调用

Author: hewei

src/main/java/com/itfsw/mybatis/generator/plugins/ExampleEnhancedPlugin.java

@@ -92,6 +92,9 @@ public boolean modelExampleClassGenerated(TopLevelClass topLevelClass, Introspec
         // when
         addWhenToExample(topLevelClass, introspectedTable);
 
+        // 增强链式调用(distinct)
+        addDistinctMethodToExample(topLevelClass, introspectedTable);
+
         return true;
     }
 
@@ -412,4 +415,27 @@ private void addOrderByMethodToExample(TopLevelClass topLevelClass, Introspected
         FormatTools.addMethodWithBestPosition(topLevelClass, orderByMethod1);
         logger.debug("itfsw(Example增强插件):" + topLevelClass.getType().getShortName() + "增加方法orderBy(String ... orderByClauses)");
     }
+
+    /**
+     * 增强链式调用(distinct)
+     * @param topLevelClass
+     * @param introspectedTable
+     */
+    private void addDistinctMethodToExample(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
+        // 添加orderBy
+        Method distinctMethod = JavaElementGeneratorTools.generateMethod(
+                "distinct",
+                JavaVisibility.PUBLIC,
+                topLevelClass.getType(),
+                new Parameter(FullyQualifiedJavaType.getBooleanPrimitiveInstance(), "distinct")
+        );
+        commentGenerator.addGeneralMethodComment(distinctMethod, introspectedTable);
+        distinctMethod = JavaElementGeneratorTools.generateMethodBody(
+                distinctMethod,
+                "this.setDistinct(distinct);",
+                "return this;"
+        );
+        FormatTools.addMethodWithBestPosition(topLevelClass, distinctMethod);
+        logger.debug("itfsw(Example增强插件):" + topLevelClass.getType().getShortName() + "增加方法distinct");
+    }
 }

src/test/java/com/itfsw/mybatis/generator/plugins/ExampleEnhancedPluginTest.java

@@ -88,6 +88,28 @@ public void reloadProject(SqlSession sqlSession, ClassLoader loader, String pack
         });
     }
 
+    /**
+     * 测试生成的distinct方法
+     */
+    @Test
+    public void testDistinct() throws Exception {
+        MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/ExampleEnhancedPlugin/mybatis-generator.xml");
+        tool.generate(new AbstractShellCallback() {
+            @Override
+            public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) throws Exception {
+                ObjectUtil tbMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbMapper")));
+
+                ObjectUtil tbExample = new ObjectUtil(loader, packagz + ".TbExample");
+                Object rExample = tbExample.invoke("distinct", true);
+
+                Assert.assertEquals(rExample.getClass().getTypeName(), packagz + ".TbExample");
+
+                String sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "selectByExample", tbExample.getObject());
+                Assert.assertEquals(sql, "select distinct id, field1, field2 from tb");
+            }
+        });
+    }
+
     /**
      * 测试静态 createCriteria
      * @throws Exception