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

Update sources generator

Browse Source
This commit is contained in:
Andrea Cavalli 2021-10-28 12:02:05 +02:00
parent 285f25e6a7
commit e692b9a846
1 changed files with 22 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
src/main/java/it/cavallium/data/generator/SourcesGenerator.java
View File

@ -380,6 +380,7 @@ public class SourcesGenerator {
} }
// UpgradeDataToLatestVersion2 Method // UpgradeDataToLatestVersion2 Method
{ {
var versionsClassName = ClassName.get(joinPackage(basePackageName, ""), "Versions");
var upgradeDataToLatestVersion2MethodBuilder = MethodSpec.methodBuilder("upgradeDataToLatestVersion") var upgradeDataToLatestVersion2MethodBuilder = MethodSpec.methodBuilder("upgradeDataToLatestVersion")
.addModifiers(Modifier.PUBLIC).addModifiers(Modifier.STATIC).addModifiers(Modifier.FINAL).addTypeVariable(TypeVariableName.get("T")) .addModifiers(Modifier.PUBLIC).addModifiers(Modifier.STATIC).addModifiers(Modifier.FINAL).addTypeVariable(TypeVariableName.get("T"))
.addTypeVariable(TypeVariableName.get("U", ClassName .addTypeVariable(TypeVariableName.get("U", ClassName
@ -387,38 +388,42 @@ public class SourcesGenerator {
"IBasicType"))).returns(TypeVariableName.get("U")) "IBasicType"))).returns(TypeVariableName.get("U"))
.addParameter(ParameterSpec.builder(TypeName.INT, "oldVersion").build()) .addParameter(ParameterSpec.builder(TypeName.INT, "oldVersion").build())
.addParameter(ParameterSpec.builder(TypeVariableName.get("T"), "oldData").build()) .addParameter(ParameterSpec.builder(TypeVariableName.get("T"), "oldData").build())
.addException(IOException.class).beginControlFlow("switch (oldVersion)"); .addException(IOException.class)
AtomicInteger seqNumber = new AtomicInteger(0); .addStatement("int intermediateVersion = oldVersion")
.addStatement("$T intermediateData = oldData", Object.class)
.beginControlFlow("while (true)")
.beginControlFlow("switch (oldVersion)");
for (Entry<String, VersionConfiguration> entry : configuration.versions.entrySet()) { for (Entry<String, VersionConfiguration> entry : configuration.versions.entrySet()) {
String version = entry.getKey(); String version = entry.getKey();
VersionConfiguration versionConfiguration = entry.getValue(); VersionConfiguration versionConfiguration = entry.getValue();
// Add a case in which the data version deserializes the serialized data and upgrades it // Add a case in which the data version deserializes the serialized data and upgrades it
upgradeDataToLatestVersion2MethodBuilder.beginControlFlow("case $T." + getVersionVarName(version) + ":", upgradeDataToLatestVersion2MethodBuilder.beginControlFlow("case $T." + getVersionVarName(version) + ":",
ClassName.get(joinPackage(basePackageName, ""), "Versions") versionsClassName
); );
if (version.equalsIgnoreCase(configuration.currentVersion)) { if (version.equalsIgnoreCase(configuration.currentVersion)) {
// This is the latest version, don't upgrade. // This is the latest version, don't upgrade.
upgradeDataToLatestVersion2MethodBuilder.addStatement("return ($T) oldData", TypeVariableName.get("U")); upgradeDataToLatestVersion2MethodBuilder.addStatement("return ($T) intermediateData", TypeVariableName.get("U"));
} else { } else {
// Upgrade // Upgrade
upgradeDataToLatestVersion2MethodBuilder.addStatement( upgradeDataToLatestVersion2MethodBuilder
"var upgradedData" + seqNumber.incrementAndGet() + " = " .addStatement(
+ getVersionPackage(configuration.currentVersion, basePackageName, version) "intermediateData = " + getVersionPackage(configuration.currentVersion, basePackageName, version)
+ ".Version.upgradeToNextVersion(($T) oldData)", + ".Version.upgradeToNextVersion(($T) intermediateData)",
ClassName.get(joinPackage(getVersionPackage(configuration.currentVersion, basePackageName, version), ClassName.get(joinPackage(getVersionPackage(configuration.currentVersion, basePackageName, version),
"data" "data"
), "IBasicType") ), "IBasicType")
); )
upgradeDataToLatestVersion2MethodBuilder.addStatement( .addStatement("intermediateVersion = $T."
"return upgradeDataToLatestVersion(Versions." + getVersionVarName(findNextVersion(configuration, + getVersionVarName(findNextVersion(configuration, version).orElseThrow()), versionsClassName)
version .addStatement("break");
).orElseThrow()) + ", upgradedData" + seqNumber.get() + ")");
} }
upgradeDataToLatestVersion2MethodBuilder.endControlFlow(); upgradeDataToLatestVersion2MethodBuilder.endControlFlow();
} }
var upgradeDataToLatestVersion2Method = upgradeDataToLatestVersion2MethodBuilder.beginControlFlow("default:") var upgradeDataToLatestVersion2Method = upgradeDataToLatestVersion2MethodBuilder.beginControlFlow("default:")
.addStatement("throw new $T(\"Unknown version: \" + oldVersion)", IOException.class).endControlFlow() .addStatement("throw new $T(\"Unknown version: \" + oldVersion)", IOException.class).endControlFlow()
.endControlFlow().build(); .endControlFlow()
.endControlFlow()
.build();
currentVersionClass.addMethod(upgradeDataToLatestVersion2Method); currentVersionClass.addMethod(upgradeDataToLatestVersion2Method);
} }
// Save the resulting class in the main package // Save the resulting class in the main package