admin/mybatis-flex
1
0
Fork 0
mirror of https://gitee.com/mybatis-flex/mybatis-flex.git synced 2025-12-08 01:28:24 +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

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

@ -150,8 +150,18 @@ public class TableInfo {
}
public void setTableName(String 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() {
return entityClass;

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);
if (table != null) {
tableInfo.setTableName(table.value());
tableInfo.setSchema(table.schema());
tableInfo.setTableName(table.value());
tableInfo.setCamelToUnderline(table.camelToUnderline());
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.setEntityName(entityClass);
tableInfo.setEntitySimpleName(entityClassName);
tableInfo.setTableName(table.value());
tableInfo.setSchema(table.schema());
tableInfo.setTableName(table.value());
tableInfo.setEntityComment(elementUtils.getDocComment(entityClassElement));
// 生成 TableDef 文件

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

@ -78,8 +78,18 @@ public class TableInfo {
}
public void setTableName(String 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() {
return schema;