Skip to content

Commit f118686

Browse files
committed
MBG插件增加schema配置(catalog也可以用这个),会自动在表的注解名字前面加上 schema.tablename
1 parent 561dd13 commit f118686

File tree

2 files changed

+13
-1
lines changed

2 files changed

+13
-1
lines changed

src/main/java/tk/mybatis/mapper/generator/MapperCommentGenerator.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -146,6 +146,8 @@ public void addFieldComment(Field field, IntrospectedTable introspectedTable, In
146146
if (!column.equals(introspectedColumn.getJavaProperty())) {
147147
//@Column
148148
field.addAnnotation("@Column(name = \"" + getDelimiterName(column) + "\")");
149+
} else if (StringUtility.stringHasValue(beginningDelimiter) || StringUtility.stringHasValue(endingDelimiter)) {
150+
field.addAnnotation("@Column(name = \"" + getDelimiterName(column) + "\")");
149151
}
150152
if (introspectedColumn.isIdentity()) {
151153
if (introspectedTable.getTableConfiguration().getGeneratedKey().getRuntimeSqlStatement().equals("JDBC")) {

src/main/java/tk/mybatis/mapper/generator/MapperPlugin.java

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,8 @@ public class MapperPlugin extends PluginAdapter {
5252
private String beginningDelimiter = "";
5353
//结束的分隔符,例如mysql为`,sqlserver为]
5454
private String endingDelimiter = "";
55+
//数据库模式
56+
private String schema;
5557
//注释生成器
5658
private CommentGeneratorConfiguration commentCfg;
5759

@@ -89,10 +91,14 @@ public void setProperties(Properties properties) {
8991
this.endingDelimiter = endingDelimiter;
9092
}
9193
commentCfg.addProperty("endingDelimiter", this.endingDelimiter);
94+
String schema = this.properties.getProperty("schema");
95+
if (StringUtility.stringHasValue(schema)) {
96+
this.schema = schema;
97+
}
9298
}
9399

94100
public String getDelimiterName(String name) {
95-
return beginningDelimiter + name + endingDelimiter;
101+
return schema + "." + beginningDelimiter + name + endingDelimiter;
96102
}
97103

98104
@Override
@@ -143,6 +149,10 @@ private void processEntityClass(TopLevelClass topLevelClass, IntrospectedTable i
143149
topLevelClass.addAnnotation("@Table(name = \"" + getDelimiterName(tableName) + "\")");
144150
} else if (!topLevelClass.getType().getShortName().equalsIgnoreCase(tableName)) {
145151
topLevelClass.addAnnotation("@Table(name = \"" + getDelimiterName(tableName) + "\")");
152+
} else if (StringUtility.stringHasValue(schema)
153+
|| StringUtility.stringHasValue(beginningDelimiter)
154+
|| StringUtility.stringHasValue(endingDelimiter)) {
155+
topLevelClass.addAnnotation("@Table(name = \"" + getDelimiterName(tableName) + "\")");
146156
}
147157
}
148158

0 commit comments

Comments
 (0)