添加不生成Example相关代码的功能

anealo · Aug 13, 2017 · 72e4e72 · 72e4e72
1 parent b2347aa
commit 72e4e72
Show file tree

Hide file tree

Showing 6 changed files with 21 additions and 77 deletions.
diff --git a/src/main/java/com/zzg/mybatis/generator/bridge/MybatisGeneratorBridge.java b/src/main/java/com/zzg/mybatis/generator/bridge/MybatisGeneratorBridge.java
@@ -63,6 +63,12 @@ public void generate() throws Exception {
         TableConfiguration tableConfig = new TableConfiguration(context);
         tableConfig.setTableName(generatorConfig.getTableName());
         tableConfig.setDomainObjectName(generatorConfig.getDomainObjectName());
+        if(!generatorConfig.isUseExampe()) {
+            tableConfig.setUpdateByExampleStatementEnabled(false);
+            tableConfig.setCountByExampleStatementEnabled(false);
+            tableConfig.setDeleteByExampleStatementEnabled(false);
+            tableConfig.setSelectByExampleStatementEnabled(false);
+        }
 
         // 针对 postgresql 单独配置
         if (DbType.valueOf(selectedDatabaseConfig.getDbType()).getDriverClass() == "org.postgresql.Driver") {
@@ -110,6 +116,7 @@ public void generate() throws Exception {
         daoConfig.setTargetPackage(generatorConfig.getDaoPackage());
         daoConfig.setTargetProject(generatorConfig.getProjectFolder() + "/" + generatorConfig.getDaoTargetFolder());
 
+
         context.setId("myid");
         context.addTableConfiguration(tableConfig);
         context.setJdbcConnectionConfiguration(jdbcConfig);

diff --git a/src/main/java/com/zzg/mybatis/generator/controller/MainUIController.java b/src/main/java/com/zzg/mybatis/generator/controller/MainUIController.java
@@ -77,6 +77,8 @@ public class MainUIController extends BaseFXController {
     @FXML
     private CheckBox useActualColumnNamesCheckbox;
     @FXML
+    private CheckBox useExample;
+    @FXML
     private TreeView<String> leftDBTree;
     // Current selected databaseConfig
     private DatabaseConfig selectedDatabaseConfig;
@@ -303,6 +305,7 @@ public GeneratorConfig getGeneratorConfigFromUI() {
         generatorConfig.setNeedToStringHashcodeEquals(needToStringHashcodeEquals.isSelected());
         generatorConfig.setAnnotation(annotationCheckBox.isSelected());
         generatorConfig.setUseActualColumnNames(useActualColumnNamesCheckbox.isSelected());
+        generatorConfig.setUseExampe(useExample.isSelected());
         return generatorConfig;
     }
 

diff --git a/src/main/java/com/zzg/mybatis/generator/model/GeneratorConfig.java b/src/main/java/com/zzg/mybatis/generator/model/GeneratorConfig.java
@@ -46,8 +46,18 @@ public class GeneratorConfig {
 
 	private boolean useActualColumnNames;
 
+	private boolean useExampe;
+
 	private String generateKeys;
 
+	public boolean isUseExampe() {
+		return useExampe;
+	}
+
+	public void setUseExampe(boolean useExampe) {
+		this.useExampe = useExampe;
+	}
+
 	public String getName() {
 		return name;
 	}

diff --git a/src/main/resources/fxml/MainUI.fxml b/src/main/resources/fxml/MainUI.fxml
@@ -186,6 +186,7 @@
                                        <children>
                                           <CheckBox fx:id="annotationCheckBox" mnemonicParsing="false" selected="false" text="生成JPA注解" />
                                           <CheckBox fx:id="useActualColumnNamesCheckbox" mnemonicParsing="false" selected="false" text="使用实际的列名" />
+                                           <CheckBox fx:id="useExample" mnemonicParsing="false" selected="false" text="使用Example" />
                                        </children>
                                     </HBox>
                                  </children>

diff --git a/src/test/java/com/zzg/mybatis/generator/util/ConfigHelperTest.java b/src/test/java/com/zzg/mybatis/generator/util/ConfigHelperTest.java
diff --git a/src/test/java/com/zzg/mybatis/generator/util/StringUtilTest.java b/src/test/java/com/zzg/mybatis/generator/util/StringUtilTest.java