【fix sonar】CodeGenerateOneToMany文件

This commit is contained in:
梁琦涛
2023-03-31 17:38:21 +08:00
parent aaf93bc8f8
commit 5ce9dbee67
@@ -91,33 +91,7 @@ public class CodeGenerateOneToMany extends BaseCodeGenerate implements IGenerate
}
}
for (SubTableVo subTableVo : this.subTables) {
// 从表(局部变量)
List<ColumnVo> originalColumns;
if (subTableVo.getColums() == null || subTableVo.getColums().isEmpty()) {
originalColumns = DbReadTableUtil.listColumns(subTableVo.getTableName());
subTableVo.setColums(originalColumns);
}
if (subTableVo.getOriginalColumns() == null || subTableVo.getOriginalColumns().isEmpty()) {
originalColumns = DbReadTableUtil.listOriginalColumns(subTableVo.getTableName());
subTableVo.setOriginalColumns(originalColumns);
}
// 从表外键
String[] foreignKeys = subTableVo.getForeignKeys();
// 存储外键的集合
ArrayList<String> list = new ArrayList<>();
String[] foreignKeys1 = foreignKeys;
int length = foreignKeys.length;
for (int i = 0; i < length; ++i) {
String foreignKey = foreignKeys1[i];
list.add(DbReadTableUtil.convertForeignKeyToCamelCase(foreignKey));
}
// put外键数组
subTableVo.setForeignKeys(list.toArray(new String[0]));
subTableVo.setOriginalForeignKeys(foreignKeys);
}
setSubTableVo();
// put从表集合
hashMap.put("subTables", this.subTables);
} catch (Exception e) {
@@ -130,6 +104,36 @@ public class CodeGenerateOneToMany extends BaseCodeGenerate implements IGenerate
return hashMap;
}
private void setSubTableVo() throws Exception {
for (SubTableVo subTableVo : this.subTables) {
// 从表(局部变量)
List<ColumnVo> originalColumns;
if (subTableVo.getColums() == null || subTableVo.getColums().isEmpty()) {
originalColumns = DbReadTableUtil.listColumns(subTableVo.getTableName());
subTableVo.setColums(originalColumns);
}
if (subTableVo.getOriginalColumns() == null || subTableVo.getOriginalColumns().isEmpty()) {
originalColumns = DbReadTableUtil.listOriginalColumns(subTableVo.getTableName());
subTableVo.setOriginalColumns(originalColumns);
}
// 从表外键
String[] foreignKeys = subTableVo.getForeignKeys();
// 存储外键的集合
ArrayList<String> list = new ArrayList<>();
String[] foreignKeys1 = foreignKeys;
int length = foreignKeys.length;
for (int i = 0; i < length; ++i) {
String foreignKey = foreignKeys1[i];
list.add(DbReadTableUtil.convertForeignKeyToCamelCase(foreignKey));
}
// put外键数组
subTableVo.setForeignKeys(list.toArray(new String[0]));
subTableVo.setOriginalForeignKeys(foreignKeys);
}
}
@Override
public List<String> generateCodeFile(String stylePath) throws Exception {
String projectPath = com.jero.codegenerate.properties.CodeConfigProperties.projectPath;