升级mybatis-generator-core版本

anealo · Mar 22, 2018 · 6ebbbcb · 6ebbbcb
1 parent 184e709
commit 6ebbbcb
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 3 deletions.
diff --git a/pom.xml b/pom.xml
@@ -15,7 +15,7 @@
         <dependency>
             <groupId>org.mybatis.generator</groupId>
             <artifactId>mybatis-generator-core</artifactId>
-            <version>1.3.5</version>
+            <version>1.3.6</version>
         </dependency>
         <dependency>
             <groupId>ch.qos.logback</groupId>

diff --git a/src/main/java/com/zzg/mybatis/generator/bridge/MybatisGeneratorBridge.java b/src/main/java/com/zzg/mybatis/generator/bridge/MybatisGeneratorBridge.java
@@ -70,7 +70,11 @@ public void generate() throws Exception {
             tableConfig.setSelectByExampleStatementEnabled(false);
         }
 
-        tableConfig.setSchema(selectedDatabaseConfig.getSchema());
+	    if (DbType.MySQL.name().equals(selectedDatabaseConfig.getDbType())) {
+		    tableConfig.setSchema(selectedDatabaseConfig.getSchema());
+	    } else {
+            tableConfig.setCatalog(selectedDatabaseConfig.getSchema());
+	    }
 
         // 针对 postgresql 单独配置
         if (DbType.valueOf(selectedDatabaseConfig.getDbType()).getDriverClass() == "org.postgresql.Driver") {
@@ -100,6 +104,10 @@ public void generate() throws Exception {
 			tableConfig.addProperty("useActualColumnNames", "true");
         }
         JDBCConnectionConfiguration jdbcConfig = new JDBCConnectionConfiguration();
+        // http://www.mybatis.org/generator/usage/mysql.html
+        if (DbType.MySQL.name().equals(selectedDatabaseConfig.getDbType())) {
+	        jdbcConfig.addProperty("nullCatalogMeansCurrent", "true");
+        }
         jdbcConfig.setDriverClass(DbType.valueOf(selectedDatabaseConfig.getDbType()).getDriverClass());
         jdbcConfig.setConnectionURL(DbUtil.getConnectionUrlWithSchema(selectedDatabaseConfig));
         jdbcConfig.setUserId(selectedDatabaseConfig.getUsername());

diff --git a/src/main/java/com/zzg/mybatis/generator/plugins/DbRemarksCommentGenerator.java b/src/main/java/com/zzg/mybatis/generator/plugins/DbRemarksCommentGenerator.java
@@ -24,6 +24,7 @@
 import org.mybatis.generator.internal.util.StringUtility;
 
 import java.util.Properties;
+import java.util.Set;
 
 import static org.mybatis.generator.internal.util.StringUtility.isTrue;
 
@@ -69,7 +70,33 @@ public void addRootComment(XmlElement rootElement) {
         return;
     }
 
-    public void addConfigurationProperties(Properties properties) {
+	@Override
+	public void addGeneralMethodAnnotation(Method method, IntrospectedTable introspectedTable, Set<FullyQualifiedJavaType> set) {
+
+
+	}
+
+	@Override
+	public void addGeneralMethodAnnotation(Method method, IntrospectedTable introspectedTable, IntrospectedColumn introspectedColumn, Set<FullyQualifiedJavaType> set) {
+
+	}
+
+	@Override
+	public void addFieldAnnotation(Field field, IntrospectedTable introspectedTable, Set<FullyQualifiedJavaType> set) {
+
+	}
+
+	@Override
+	public void addFieldAnnotation(Field field, IntrospectedTable introspectedTable, IntrospectedColumn introspectedColumn, Set<FullyQualifiedJavaType> set) {
+
+	}
+
+	@Override
+	public void addClassAnnotation(InnerClass innerClass, IntrospectedTable introspectedTable, Set<FullyQualifiedJavaType> set) {
+
+	}
+
+	public void addConfigurationProperties(Properties properties) {
         this.properties.putAll(properties);
         columnRemarks = isTrue(properties
                 .getProperty("columnRemarks"));
@@ -84,6 +111,7 @@ public void addClassComment(InnerClass innerClass,
     public void addModelClassComment(TopLevelClass topLevelClass,
                                 IntrospectedTable introspectedTable) {
         topLevelClass.addJavaDocLine("/**");
+        topLevelClass.addJavaDocLine(" * " + introspectedTable.getFullyQualifiedTable().getIntrospectedTableName());
         topLevelClass.addJavaDocLine(" * @author ");
         topLevelClass.addJavaDocLine(" */");
         if(isAnnotations) {