andreacavalli/data-generator
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Update

Browse Source
This commit is contained in:
Andrea Cavalli 2023-01-22 12:17:40 +01:00
parent 17116a2bd3
commit 9113ae8bf8
1 changed files with 15 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
data-generator-plugin/src/main/java/it/cavallium/data/generator/plugin/classgen/GenCurrentVersion.java
View File

@ -52,24 +52,31 @@ public class GenCurrentVersion extends ClassGenerator {
.addCode("return version == VERSION.getVersion();").build(); .addCode("return version == VERSION.getVersion();").build();
currentVersionClass.addMethod(isLatestVersionMethod); currentVersionClass.addMethod(isLatestVersionMethod);
} }
// Get super type classes method // Get super type classes method and static field
{ {
var returnType = ParameterizedTypeName.get(ClassName.get(Set.class),
ParameterizedTypeName.get(ClassName.get(Class.class),
WildcardTypeName.subtypeOf(ClassName.get(currentVersionPackage, "IType"))));
var superTypesField = FieldSpec.builder(returnType, "SUPER_TYPE_CLASSES", Modifier.PRIVATE, Modifier.STATIC, Modifier.FINAL);
var getSuperTypeClasses = MethodSpec.methodBuilder("getSuperTypeClasses").addModifiers(Modifier.PUBLIC) var getSuperTypeClasses = MethodSpec.methodBuilder("getSuperTypeClasses").addModifiers(Modifier.PUBLIC)
.addModifiers(Modifier.FINAL).addModifiers(Modifier.STATIC) .addModifiers(Modifier.FINAL).addModifiers(Modifier.STATIC)
.returns(ParameterizedTypeName.get(ClassName.get(Set.class), .returns(returnType);
ParameterizedTypeName.get(ClassName.get(Class.class),
WildcardTypeName.subtypeOf(ClassName.get(currentVersionPackage, "IType"))))) var superTypesInitializerField = CodeBlock.builder();
.addCode("return $T.of(\n", Set.class); superTypesInitializerField.add("$T.of(\n", Set.class);
AtomicBoolean isFirst = new AtomicBoolean(true); AtomicBoolean isFirst = new AtomicBoolean(true);
dataModel.getSuperTypesComputed(dataModel.getCurrentVersion()).forEach(superType -> { dataModel.getSuperTypesComputed(dataModel.getCurrentVersion()).forEach(superType -> {
if (!isFirst.getAndSet(false)) { if (!isFirst.getAndSet(false)) {
getSuperTypeClasses.addCode(",\n"); superTypesInitializerField.add(",\n");
} }
getSuperTypeClasses.addCode("$T.class", superTypesInitializerField.add("$T.class",
ClassName.get(dataModel.getVersion(superType).getDataPackage(basePackageName), superType.getName()) ClassName.get(dataModel.getVersion(superType).getDataPackage(basePackageName), superType.getName())
); );
}); });
getSuperTypeClasses.addCode("\n);"); superTypesInitializerField.add("\n);");
superTypesField.initializer(superTypesInitializerField.build());
getSuperTypeClasses.addStatement("return SUPER_TYPE_CLASSES");
currentVersionClass.addField(superTypesField.build());
currentVersionClass.addMethod(getSuperTypeClasses.build()); currentVersionClass.addMethod(getSuperTypeClasses.build());
} }
// Get super type subtypes classes method // Get super type subtypes classes method