admin/mybatis-flex
1
0
Fork 0
mirror of https://gitee.com/mybatis-flex/mybatis-flex.git synced 2025-12-08 09:38:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: optimize TableInfo.setTableName

Browse Source
This commit is contained in:
开源海哥 2023-08-17 14:00:27 +08:00
parent 4af6863b82
commit b325882a35
4 changed files with 24 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
mybatis-flex-core/src/main/java/com/mybatisflex/core/table/TableInfo.java
View File

@ -150,7 +150,17 @@ public class TableInfo {
} }
public void setTableName(String tableName) { public void setTableName(String tableName) {
this.tableName = tableName; int indexOf = tableName.indexOf(".");
if (indexOf > 0) {
if (StringUtil.isBlank(schema)) {
this.schema = tableName.substring(0, indexOf);
this.tableName = tableName.substring(indexOf + 1);
} else {
this.tableName = tableName;
}
} else {
this.tableName = tableName;
}
} }
public Class<?> getEntityClass() { public Class<?> getEntityClass() {

2
mybatis-flex-core/src/main/java/com/mybatisflex/core/table/TableInfoFactory.java
View File

@ -134,8 +134,8 @@ public class TableInfoFactory {
//初始化表名 //初始化表名
Table table = entityClass.getAnnotation(Table.class); Table table = entityClass.getAnnotation(Table.class);
if (table != null) { if (table != null) {
tableInfo.setTableName(table.value());
tableInfo.setSchema(table.schema()); tableInfo.setSchema(table.schema());
tableInfo.setTableName(table.value());
tableInfo.setCamelToUnderline(table.camelToUnderline()); tableInfo.setCamelToUnderline(table.camelToUnderline());
if (table.onInsert().length > 0) { if (table.onInsert().length > 0) {

2
mybatis-flex-processor/src/main/java/com/mybatisflex/processor/MybatisFlexProcessor.java
View File

@ -180,8 +180,8 @@ public class MybatisFlexProcessor extends AbstractProcessor {
TableInfo tableInfo = new TableInfo(); TableInfo tableInfo = new TableInfo();
tableInfo.setEntityName(entityClass); tableInfo.setEntityName(entityClass);
tableInfo.setEntitySimpleName(entityClassName); tableInfo.setEntitySimpleName(entityClassName);
tableInfo.setTableName(table.value());
tableInfo.setSchema(table.schema()); tableInfo.setSchema(table.schema());
tableInfo.setTableName(table.value());
tableInfo.setEntityComment(elementUtils.getDocComment(entityClassElement)); tableInfo.setEntityComment(elementUtils.getDocComment(entityClassElement));
// 生成 TableDef 文件 // 生成 TableDef 文件

12
mybatis-flex-processor/src/main/java/com/mybatisflex/processor/entity/TableInfo.java
View File

@ -78,7 +78,17 @@ public class TableInfo {
} }
public void setTableName(String tableName) { public void setTableName(String tableName) {
this.tableName = tableName; int indexOf = tableName.indexOf(".");
if (indexOf > 0) {
if (schema == null || schema.trim().length() == 0) {
this.schema = tableName.substring(0, indexOf);
this.tableName = tableName.substring(indexOf + 1);
} else {
this.tableName = tableName;
}
} else {
this.tableName = tableName;
}
} }
public String getSchema() { public String getSchema() {