MBG插件增加caseSensitive默认false，当数据库表名区分大小写时，可以将该属性设置为true

abel533 · abel533 · commit 12bdc77f182e · 2015-04-08T14:19:26.000+08:00
diff --git a/src/main/java/com/github/abel533/generator/MapperPlugin.java b/src/main/java/com/github/abel533/generator/MapperPlugin.java
@@ -47,6 +47,7 @@
  */
 public class MapperPlugin extends PluginAdapter {
     private Set<String> mappers = new HashSet<String>();
+    private boolean caseSensitive = false;
 
     @Override
     public void setContext(Context context) {
@@ -68,6 +69,10 @@ public void setProperties(Properties properties) {
         } else {
             throw new RuntimeException("Mapper插件缺少必要的mappers属性!");
         }
+        String caseSensitive = this.properties.getProperty("caseSensitive");
+        if (StringUtility.stringHasValue(caseSensitive)) {
+            this.caseSensitive = caseSensitive.equalsIgnoreCase("TRUE");
+        }
     }
 
     @Override
@@ -113,7 +118,10 @@ private void processEntityClass(TopLevelClass topLevelClass, IntrospectedTable i
                     + tableName
                     + context.getEndingDelimiter();
         }
-        if (!topLevelClass.getType().getShortName().equalsIgnoreCase(tableName)) {
+        //是否忽略大小写，对于区分大小写的数据库，会有用
+        if (caseSensitive && !topLevelClass.getType().getShortName().equals(tableName)) {
+            topLevelClass.addAnnotation("@Table(name = \"" + tableName + "\")");
+        } else if (!topLevelClass.getType().getShortName().equalsIgnoreCase(tableName)) {
             topLevelClass.addAnnotation("@Table(name = \"" + tableName + "\")");
         }
     }
diff --git a/src/test/resources/generator/generatorConfig.xml b/src/test/resources/generator/generatorConfig.xml
@@ -34,6 +34,7 @@
     
     <plugin type="com.github.abel533.generator.MapperPlugin">
       <property name="mappers" value="com.github.abel533.mapper.Mapper,com.github.abel533.hsqldb.HsqldbMapper"/>
+      <property name="caseSensitive" value="true"/>
     </plugin>
 
     <jdbcConnection driverClass="com.mysql.jdbc.Driver"
diff --git a/wiki/mapper/5.UseMBG.md b/wiki/mapper/5.UseMBG.md
@@ -65,6 +65,8 @@ Java代码很容易，和文档中的一样：
 	    
 	    <plugin type="com.github.abel533.generator.MapperPlugin">
 	      <property name="mappers" value="com.github.abel533.mapper.Mapper"/>
+	      <!-- caseSensitive默认false，当数据库表名区分大小写时，可以将该属性设置为true -->
+	      <property name="caseSensitive" value="true"/>
 	    </plugin>
 	
 	    <jdbcConnection driverClass="com.mysql.jdbc.Driver"

Original file line number	Diff line number	Diff line change
`@@ -47,6 +47,7 @@`
`47`	`47`	`*/`
`48`	`48`	`public class MapperPlugin extends PluginAdapter {`
`49`	`49`	`private Set<String> mappers = new HashSet<String>();`
	`50`	`+ private boolean caseSensitive = false;`
`50`	`51`
`51`	`52`	`@Override`
`52`	`53`	`public void setContext(Context context) {`
`@@ -68,6 +69,10 @@ public void setProperties(Properties properties) {`
`68`	`69`	`} else {`
`69`	`70`	`throw new RuntimeException("Mapper插件缺少必要的mappers属性!");`
`70`	`71`	`}`
	`72`	`+ String caseSensitive = this.properties.getProperty("caseSensitive");`
	`73`	`+ if (StringUtility.stringHasValue(caseSensitive)) {`
	`74`	`+ this.caseSensitive = caseSensitive.equalsIgnoreCase("TRUE");`
	`75`	`+ }`
`71`	`76`	`}`
`72`	`77`
`73`	`78`	`@Override`
`@@ -113,7 +118,10 @@ private void processEntityClass(TopLevelClass topLevelClass, IntrospectedTable i`
`113`	`118`	`+ tableName`
`114`	`119`	`+ context.getEndingDelimiter();`
`115`	`120`	`}`
`116`		`- if (!topLevelClass.getType().getShortName().equalsIgnoreCase(tableName)) {`
	`121`	`+ //是否忽略大小写，对于区分大小写的数据库，会有用`
	`122`	`+ if (caseSensitive && !topLevelClass.getType().getShortName().equals(tableName)) {`
	`123`	`+ topLevelClass.addAnnotation("@Table(name = \"" + tableName + "\")");`
	`124`	`+ } else if (!topLevelClass.getType().getShortName().equalsIgnoreCase(tableName)) {`
`117`	`125`	`topLevelClass.addAnnotation("@Table(name = \"" + tableName + "\")");`
`118`	`126`	`}`
`119`	`127`	`}`