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-21 23:14:30 +01:00
parent a0eb0b2f8e
commit 74449f25a7
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
data-generator-plugin/src/main/java/it/cavallium/data/generator/plugin/classgen/GenNullableX.java
View File

@ -10,6 +10,8 @@ import it.cavallium.data.generator.TypedNullable;
import it.cavallium.data.generator.nativedata.INullable; import it.cavallium.data.generator.nativedata.INullable;
import it.cavallium.data.generator.plugin.ClassGenerator; import it.cavallium.data.generator.plugin.ClassGenerator;
import it.cavallium.data.generator.plugin.ComputedTypeBase; import it.cavallium.data.generator.plugin.ComputedTypeBase;
import it.cavallium.data.generator.plugin.ComputedTypeNullable;
import it.cavallium.data.generator.plugin.ComputedTypeNullableFixed;
import it.cavallium.data.generator.plugin.ComputedTypeNullableVersioned; import it.cavallium.data.generator.plugin.ComputedTypeNullableVersioned;
import it.cavallium.data.generator.plugin.ComputedTypeSuper; import it.cavallium.data.generator.plugin.ComputedTypeSuper;
import it.cavallium.data.generator.plugin.ComputedVersion; import it.cavallium.data.generator.plugin.ComputedVersion;
@ -35,13 +37,14 @@ public class GenNullableX extends ClassGenerator {
.getComputedTypes(version) .getComputedTypes(version)
.values() .values()
.stream() .stream()
.filter(type -> type instanceof ComputedTypeNullableVersioned) .filter(f -> f instanceof ComputedTypeNullable)
.map(type -> (ComputedTypeNullableVersioned) type) .map(f -> (ComputedTypeNullable) f)
.filter(type -> type.getVersion().equals(version)) .filter(type -> (type instanceof ComputedTypeNullableVersioned versioned
&& versioned.getVersion().equals(version)) || type instanceof ComputedTypeNullableFixed)
.map(type -> generateTypeVersioned(version, type)); .map(type -> generateTypeVersioned(version, type));
} }
private GeneratedClass generateTypeVersioned(ComputedVersion version, ComputedTypeNullableVersioned computedType) { private GeneratedClass generateTypeVersioned(ComputedVersion version, ComputedTypeNullable computedType) {
var type = (ClassName) computedType.getJTypeName(basePackageName); var type = (ClassName) computedType.getJTypeName(basePackageName);
var classBuilder = TypeSpec.recordBuilder(type.simpleName()); var classBuilder = TypeSpec.recordBuilder(type.simpleName());