✨ 完成抽象
luochuan
12 months ago
parent
9ae0250410
commit
7004c83a00
@ -0,0 +1,143 @@
|
||||
package org.alis.generator.factory;
|
||||
|
||||
import org.alis.generator.domain.model.AliasNode;
|
||||
import org.alis.generator.domain.model.Content;
|
||||
import org.alis.generator.domain.model.ContextConfig;
|
||||
import org.springframework.util.CollectionUtils;
|
||||
import org.springframework.util.StringUtils;
|
||||
|
||||
import java.text.MessageFormat;
|
||||
import java.time.LocalDateTime;
|
||||
import java.time.format.DateTimeFormatter;
|
||||
import java.util.*;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
/**
|
||||
* @author lc
|
||||
* @date 2023/11/23 18:49
|
||||
**/
|
||||
public abstract class AbstractGenerator implements Generator {
|
||||
|
||||
|
||||
@Override
|
||||
public Content generator(ContextConfig prop) {
|
||||
String source = prop.getSource().trim();
|
||||
if (!StringUtils.hasLength(source)) {
|
||||
return null;
|
||||
}
|
||||
|
||||
// 使用组合模式
|
||||
List<AliasNode> aliasNodes = new ArrayList<>();
|
||||
// 匿名内部类
|
||||
List<AliasNode> classNodes = new ArrayList<>();
|
||||
|
||||
dealHandler(prop.getSource(), aliasNodes, classNodes);
|
||||
|
||||
// merge删除对应的类
|
||||
Map<String, String> merge = merge(classNodes);
|
||||
|
||||
// 生成class 文件
|
||||
StringBuilder builder = new StringBuilder();
|
||||
|
||||
// 包路径
|
||||
builder.append("package ").append(prop.getPackagePath()).append(";").append("\n\n");
|
||||
// 注解的包
|
||||
builder.append("import lombok.Data;").append("\n");
|
||||
// 导入所有的包
|
||||
List<String> importList = new ArrayList<>();
|
||||
fillAllType(aliasNodes, importList);
|
||||
importList.stream().filter(Objects::nonNull).distinct().forEach(path -> builder.append("import ").append(path).append(";").append("\n\n"));
|
||||
|
||||
// 注解信息
|
||||
LocalDateTime now = LocalDateTime.now();
|
||||
String localDate = now.toLocalDate().format(DateTimeFormatter.ofPattern("yyyy-MM-dd"));
|
||||
String localTime = now.toLocalTime().format(DateTimeFormatter.ofPattern("HH:mm"));
|
||||
String author = "/**\n * @author {0}\n * @date {1} {2}\n **/";
|
||||
String format = MessageFormat.format(author, prop.getAuthor(), localDate, localTime);
|
||||
builder.append(format).append("\n");
|
||||
// 类名
|
||||
builder.append("@Data").append("\n");
|
||||
builder.append("public class ").append(prop.getClassName()).append("{\n");
|
||||
|
||||
// 构造函数
|
||||
builder.append(" ").append("public ").append(prop.getClassName()).append("(){}\n");
|
||||
|
||||
// 生成属性
|
||||
aliasNodes.forEach(node -> builder.append(" ").append("private ").append(merge.getOrDefault(node.getType(), node.getType())).append(" ").append(node.getName()).append(";\n"));
|
||||
|
||||
// 生成静态属性类
|
||||
classNodes.forEach(clazzNode -> {
|
||||
builder.append(" ").append("@Data").append("\n");
|
||||
builder.append(" ").append("public static class ").append(clazzNode.getAlias()).append("{\n");
|
||||
List<AliasNode> files = clazzNode.getNodes();
|
||||
files.forEach(node -> builder.append(" ").append(" ").append("private ").append(node.getType()).append(" ").append(node.getName()).append(";\n"));
|
||||
builder.append(" ").append("}\n");
|
||||
});
|
||||
builder.append("}\n");
|
||||
|
||||
|
||||
Content content = new Content();
|
||||
content.setPath(prop.getPackagePath());
|
||||
content.setSource(prop.getSource());
|
||||
content.setContent(builder.toString());
|
||||
|
||||
return content;
|
||||
}
|
||||
|
||||
protected abstract void dealHandler(String source, List<AliasNode> fieldNodes, List<AliasNode> classNodes);
|
||||
|
||||
private void fillAllType(List<AliasNode> jsonNodes, List<String> typeList) {
|
||||
if (CollectionUtils.isEmpty(jsonNodes)) {
|
||||
return;
|
||||
}
|
||||
for (AliasNode jsonNode : jsonNodes) {
|
||||
typeList.add(jsonNode.getAbsolutePath());
|
||||
fillAllType(jsonNode.getNodes(), typeList);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
protected String upFirstChar(String name) {
|
||||
String first = name.substring(0, 1);
|
||||
return name.replaceFirst(first, first.toUpperCase());
|
||||
}
|
||||
|
||||
|
||||
private Map<String, String> merge(List<AliasNode> classNodes) {
|
||||
Map<String, String> mapping = new HashMap<>();
|
||||
Map<Integer, String> hashTable = new HashMap<>();
|
||||
Iterator<AliasNode> iterator = classNodes.iterator();
|
||||
while (iterator.hasNext()) {
|
||||
AliasNode next = iterator.next();
|
||||
int hash = next.getNodes().stream().map(AliasNode::getName).sorted(String::compareTo)
|
||||
.collect(Collectors.joining()).hashCode();
|
||||
// 如果存在
|
||||
if (hashTable.containsKey(hash)) {
|
||||
mapping.put(next.getType(), hashTable.get(hash));
|
||||
if (Boolean.TRUE.equals(next.getArrayFlag())) {
|
||||
mapping.put(next.getType(), "List<" + hashTable.get(hash) + ">");
|
||||
}
|
||||
iterator.remove();
|
||||
} else {
|
||||
hashTable.put(hash, next.getType());
|
||||
}
|
||||
}
|
||||
// 如果遇到名称相同的 直接组合成大对象
|
||||
Iterator<AliasNode> duplicate = classNodes.iterator();
|
||||
Map<String, AliasNode> duplicateMap = new HashMap<>();
|
||||
while (duplicate.hasNext()) {
|
||||
AliasNode next = duplicate.next();
|
||||
if (duplicateMap.containsKey(next.getType())) {
|
||||
duplicate.remove();
|
||||
List<AliasNode> nodes = duplicateMap.get(next.getType()).getNodes();
|
||||
Set<String> collect = nodes.stream().map(AliasNode::getName).collect(Collectors.toSet());
|
||||
next.getNodes().stream().filter(no -> !collect.contains(no.getName())).forEach(nodes::add);
|
||||
} else {
|
||||
duplicateMap.put(next.getType(), next);
|
||||
}
|
||||
}
|
||||
|
||||
return mapping;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,17 @@
|
||||
package org.alis.generator.factory;
|
||||
|
||||
import org.alis.generator.domain.model.AliasNode;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* @author lc
|
||||
* @date 2023/11/23 18:43
|
||||
**/
|
||||
public class XmlGenerator extends AbstractGenerator {
|
||||
|
||||
@Override
|
||||
protected void dealHandler(String source, List<AliasNode> fieldNodes, List<AliasNode> classNodes) {
|
||||
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue