Browse Source

代码生成优化表前缀判断

test
smallchill 2 years ago
parent
commit
8e1dff84f0
  1. 8
      blade-ops/blade-develop/src/main/java/org/springblade/develop/controller/ModelController.java

8
blade-ops/blade-develop/src/main/java/org/springblade/develop/controller/ModelController.java

@ -29,7 +29,6 @@ import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import lombok.AllArgsConstructor;
import org.springblade.core.boot.ctrl.BladeController;
import org.springblade.core.log.exception.ServiceException;
import org.springblade.core.mp.support.Condition;
import org.springblade.core.mp.support.Query;
import org.springblade.core.tool.api.R;
@ -209,10 +208,11 @@ public class ModelController extends BladeController {
Iterator<TableInfo> iterator = tableInfoList.stream().filter(table -> table.getName().equals(tableName)).collect(Collectors.toList()).iterator();
if (iterator.hasNext()) {
tableInfo = iterator.next();
if (!tableName.contains(StringPool.UNDERSCORE)) {
throw new ServiceException("请添加表前缀后再进行配置!");
if (tableName.contains(StringPool.UNDERSCORE)) {
tableInfo.setEntityName(tableInfo.getEntityName().replace(StringUtil.firstCharToUpper(tableName.split(StringPool.UNDERSCORE)[0]), StringPool.EMPTY));
} else {
tableInfo.setEntityName(StringUtil.firstCharToUpper(tableName));
}
tableInfo.setEntityName(tableInfo.getEntityName().replace(StringUtil.firstCharToUpper(tableName.split(StringPool.UNDERSCORE)[0]), StringPool.EMPTY));
}
return tableInfo;
}

Loading…
Cancel
Save