Skip to content

Commit

Permalink
支持ManyToOne OneToOne的Schema增强
Browse files Browse the repository at this point in the history
  • Loading branch information
@binking338
binking338 committed Oct 9, 2024
1 parent 25270af commit 8a61a07
Showing 1 changed file with 61 additions and 17 deletions.
78 changes: 61 additions & 17 deletions ...dd-codegen-maven-plugin/src/main/java/org/netcorepal/cap4j/ddd/codegen/GenEntityMojo.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -357,7 +357,7 @@ public String getSchemaPath() {
*/
public String getSchemaPackage() {
return SourceFileUtils.resolvePackage(getSchemaPath() + File.separator + "X.java")
.substring(basePackage.length() + 1);
.substring(StringUtils.isBlank(basePackage) ? 0 : (basePackage.length() + 1));
}

/**
Expand Down Expand Up @@ -594,6 +594,18 @@ private Map<String, Object> getIdColumn(List<Map<String, Object>> columns) {
.findFirst().orElse(null);
}

/**
* 获取指定列
*
* @param columns
* @param columnName
* @return
*/
private Map<String, Object> getColumn(List<Map<String, Object>> columns, String columnName) {
return columns.stream().filter(column -> Objects.equals(columnName, getColumnName(column)))
.findFirst().orElse(null);
}

/**
* 生成字段注释
*
Expand Down Expand Up @@ -740,8 +752,8 @@ public Map<String, Map<String, String>> resolveRelationTable(Map<String, Object>
refTableName = getReference(column);
result.putIfAbsent(tableName, new HashMap<>());
result.get(tableName).putIfAbsent(refTableName, "OneToOne;" + colName + (lazy ? ";LAZY" : ""));
result.putIfAbsent(refTableName, new HashMap<>());
result.get(refTableName).putIfAbsent(tableName, "*OneToOne;" + colName + (lazy ? ";LAZY" : ""));
// result.putIfAbsent(refTableName, new HashMap<>());
// result.get(refTableName).putIfAbsent(tableName, "*OneToOne;" + colName + (lazy ? ";LAZY" : ""));
break;
case "ManyToOne":
case "*:1":
Expand Down Expand Up @@ -1150,13 +1162,15 @@ public void writeRelationProperty(BufferedWriter out, Map<String, Object> table,
}

String relation = refInfos[0];
String joinColumn = refInfos[1];
String fetchAnnotation = ""; // fetchType.equals("LAZY") ? "" : (" @Fetch(FetchMode." + this.fetchMode + ")");


writeLine(out, "");
switch (relation) {
case "OneToMany":// 专属聚合内关系
writeLine(out, " @" + relation.replace("*", "") + "(cascade = { CascadeType.ALL }, fetch = FetchType." + fetchType + ", orphanRemoval = true)" + fetchAnnotation);
writeLine(out, " @JoinColumn(name = \"" + LEFT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + refInfos[1] + RIGHT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + "\", nullable = false)");
writeLine(out, " @JoinColumn(name = \"" + LEFT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + joinColumn + RIGHT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + "\", nullable = false)");
boolean countIsOne = countIsOne(navTable);
if (countIsOne) {
writeLine(out, " @Getter(lombok.AccessLevel.PROTECTED)");
Expand All @@ -1172,12 +1186,12 @@ public void writeRelationProperty(BufferedWriter out, Map<String, Object> table,
break;
case "*ManyToOne":
writeLine(out, " @" + relation.replace("*", "") + "(cascade = { }, fetch = FetchType." + fetchType + ")" + fetchAnnotation);
writeLine(out, " @JoinColumn(name = \"" + LEFT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + refInfos[1] + RIGHT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + "\", nullable = false, insertable = false, updatable = false)");
writeLine(out, " @JoinColumn(name = \"" + LEFT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + joinColumn + RIGHT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + "\", nullable = false, insertable = false, updatable = false)");
writeLine(out, " private " + tablePackageMap.get(entry.getKey()) + "." + getEntityJavaType(entry.getKey()) + " " + toLowerCamelCase(getEntityJavaType(entry.getKey())) + ";");
break;
case "ManyToOne":
writeLine(out, " @" + relation.replace("*", "") + "(cascade = { }, fetch = FetchType." + fetchType + ")" + fetchAnnotation);
writeLine(out, " @JoinColumn(name = \"" + LEFT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + refInfos[1] + RIGHT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + "\", nullable = false)");
writeLine(out, " @JoinColumn(name = \"" + LEFT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + joinColumn + RIGHT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + "\", nullable = false)");
writeLine(out, " private " + tablePackageMap.get(entry.getKey()) + "." + getEntityJavaType(entry.getKey()) + " " + toLowerCamelCase(getEntityJavaType(entry.getKey())) + ";");
break;
case "*OneToMany":// 当前不会用到，无法控制集合数量规模
Expand All @@ -1187,7 +1201,7 @@ public void writeRelationProperty(BufferedWriter out, Map<String, Object> table,
break;
case "OneToOne":
writeLine(out, " @" + relation.replace("*", "") + "(cascade = { }, fetch = FetchType." + fetchType + ")" + fetchAnnotation);
writeLine(out, " @JoinColumn(name = \"" + LEFT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + refInfos[1] + RIGHT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + "\", nullable = false)");
writeLine(out, " @JoinColumn(name = \"" + LEFT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + joinColumn + RIGHT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + "\", nullable = false)");
writeLine(out, " private " + tablePackageMap.get(entry.getKey()) + "." + getEntityJavaType(entry.getKey()) + " " + toLowerCamelCase(getEntityJavaType(entry.getKey())) + ";");
break;
case "*OneToOne":
Expand All @@ -1198,7 +1212,7 @@ public void writeRelationProperty(BufferedWriter out, Map<String, Object> table,
case "ManyToMany":
writeLine(out, " @" + relation.replace("*", "") + "(cascade = { }, fetch = FetchType." + fetchType + ")" + fetchAnnotation);
writeLine(out, " @JoinTable(name = \"" + LEFT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + refInfos[3] + RIGHT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + "\"" +
", joinColumns = {@JoinColumn(name = \"" + LEFT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + refInfos[1] + RIGHT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + "\", nullable = false)}" +
", joinColumns = {@JoinColumn(name = \"" + LEFT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + joinColumn + RIGHT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + "\", nullable = false)}" +
", inverseJoinColumns = {@JoinColumn(name = \"" + LEFT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + refInfos[2] + RIGHT_QUOTES_4_ID_ALIAS.replace("\"", "\\\"") + "\", nullable = false)})");
writeLine(out, " private java.util.List<" + tablePackageMap.get(entry.getKey()) + "." + getEntityJavaType(entry.getKey()) + "> " + Inflector.getInstance().pluralize(toLowerCamelCase(getEntityJavaType(entry.getKey()))) + ";");
break;
Expand Down Expand Up @@ -1496,25 +1510,53 @@ public void writeSchemaSourceFile(Map<String, Object> table, List<Map<String, Ob
: getDefaultSchemaFieldTemplateNode().clone()
).resolve(itemContext).getData();
}
putContext(tag, "FIELD_ITEMS", fieldItems, context);

String joinItems = "";
if (relations.containsKey(tableName)) {
for (Map.Entry<String, String> entry : relations.get(tableName).entrySet()) {
String[] refInfos = entry.getValue().split(";");
Map<String, String> joinContext = new HashMap<>(context);
switch (refInfos[0]) {
case "OneToMany":
case "*OneToMany":
Map<String, String> joinContext = new HashMap<>(context);
putContext(joinTag, "joinEntityPackage", tablePackageMap.get(entry.getKey()), joinContext);
putContext(joinTag, "joinEntityType", getEntityJavaType(entry.getKey()), joinContext);
putContext(joinTag, "joinEntityVars", Inflector.getInstance().pluralize(toLowerCamelCase(getEntityJavaType(entry.getKey()))), joinContext);
PathNode joinItemTemplateNode = (
if (!("abs".equalsIgnoreCase(entitySchemaOutputMode))) {
putContext(joinTag, "joinEntitySchemaPackage", concatPackage(tablePackageMap.get(entry.getKey()), DEFAULT_SCHEMA_PACKAGE) + ".", joinContext);
}
joinItems += (
(templateNodeMap.containsKey(joinTag) && templateNodeMap.get(joinTag).size() > 0)
? templateNodeMap.get(joinTag).get(templateNodeMap.get(joinTag).size() - 1)
: getDefaultSchemaJoinTemplateNode()
).clone().resolve(joinContext);
joinItems += joinItemTemplateNode.getData();
).clone().resolve(joinContext).getData();
break;
case "OneToOne":
case "ManyToOne":
putContext(joinTag, "joinEntityPackage", tablePackageMap.get(entry.getKey()), joinContext);
putContext(joinTag, "joinEntityType", getEntityJavaType(entry.getKey()), joinContext);
putContext(joinTag, "joinEntityVars", toLowerCamelCase(getEntityJavaType(entry.getKey())), joinContext);
if (!("abs".equalsIgnoreCase(entitySchemaOutputMode))) {
putContext(joinTag, "joinEntitySchemaPackage", concatPackage(tablePackageMap.get(entry.getKey()), DEFAULT_SCHEMA_PACKAGE) + ".", joinContext);
}
joinItems += (
(templateNodeMap.containsKey(joinTag) && templateNodeMap.get(joinTag).size() > 0)
? templateNodeMap.get(joinTag).get(templateNodeMap.get(joinTag).size() - 1)
: getDefaultSchemaJoinTemplateNode()
).clone().resolve(joinContext).getData();

String fieldType = tablePackageMap.get(entry.getKey()) + "." + getEntityJavaType(entry.getKey());
String fieldName = toLowerCamelCase(getEntityJavaType(entry.getKey()));
String fieldComment = generateFieldComment(getColumn(columns, refInfos[1])).stream().reduce((a, b) -> a + "\n " + b).orElse("");
Map<String, String> itemContext = new HashMap<>(context);
putContext(fieldTag, "fieldType", fieldType, itemContext);
putContext(fieldTag, "fieldName", fieldName, itemContext);
putContext(fieldTag, "fieldComment", fieldComment, itemContext);
fieldItems += (
templateNodeMap.containsKey(fieldTag) && templateNodeMap.get(fieldTag).size() > 0
? templateNodeMap.get(fieldTag).get(templateNodeMap.get(fieldTag).size() - 1)
: getDefaultSchemaFieldTemplateNode().clone()
).resolve(itemContext).getData();
break;
default:
// 暂不支持
Expand All @@ -1523,6 +1565,8 @@ public void writeSchemaSourceFile(Map<String, Object> table, List<Map<String, Ob

}
}

putContext(tag, "FIELD_ITEMS", fieldItems, context);
putContext(tag, "JOIN_ITEMS", joinItems, context);

List<TemplateNode> schemaTemplateNodes = templateNodeMap.containsKey(tag)
Expand Down Expand Up @@ -1862,10 +1906,10 @@ public TemplateNode getDefaultSchemaJoinTemplateNode() {
" * @param joinType\n" +
" * @return\n" +
" */\n" +
" public ${joinEntityType}Schema join${joinEntityType}(${SchemaBase}.JoinType joinType) {\n" +
" public ${joinEntitySchemaPackage}${joinEntityType}Schema join${joinEntityType}(${SchemaBase}.JoinType joinType) {\n" +
" JoinType type = joinType.toJpaJoinType();\n" +
" Join<${Entity}, ${joinEntityPackage}.${joinEntityType}> join = ((Root<${Entity}>) root).join(\"${joinEntityVars}\", type);\n" +
" ${joinEntityType}Schema schema = new ${joinEntityType}Schema(join, criteriaBuilder);\n" +
" ${joinEntitySchemaPackage}${joinEntityType}Schema schema = new ${joinEntitySchemaPackage}${joinEntityType}Schema(join, criteriaBuilder);\n" +
" return schema;\n" +
" }";
TemplateNode templateNode = new TemplateNode();
Expand Down Expand Up @@ -2001,7 +2045,7 @@ public TemplateNode getDefaultSchemaTemplateNode() {
templateNode.setName("${path}${SEPARATOR}" + DEFAULT_SCHEMA_PACKAGE + "${SEPARATOR}${Entity}Schema.java");
templateNode.setFormat("raw");
templateNode.setData(template);
templateNode.setConflict("skip");
templateNode.setConflict("overwrite");
return templateNode;
}

Expand Down Expand Up @@ -2258,7 +2302,7 @@ public TemplateNode getDefaultSchemaBaseTemplateNode() {
templateNode.setName("${SchemaBase}.java");
templateNode.setFormat("raw");
templateNode.setData(template);
templateNode.setConflict("skip");
templateNode.setConflict("overwrite");
return templateNode;
}
}

0 comments on commit 8a61a07

Please sign in to comment.