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

Customizable record builders

Browse Source
This commit is contained in:
Andrea Cavalli 2022-03-09 12:53:16 +01:00
parent 8a9122447d
commit 1a90b38ff0
3 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/main/java/it/cavallium/data/generator/MavenPlugin.java
View File

@ -21,6 +21,9 @@ public class MavenPlugin extends AbstractMojo {
@Parameter( required = true) @Parameter( required = true)
private String basePackageName; private String basePackageName;
@Parameter( required = true, defaultValue = "false")
private String useRecordBuilder;
/** /**
* @parameter default-value="${project}" * @parameter default-value="${project}"
* @required * @required
@ -38,7 +41,7 @@ public class MavenPlugin extends AbstractMojo {
Path outPath = genRecordsPath.resolve("java"); Path outPath = genRecordsPath.resolve("java");
this.project.addCompileSourceRoot(outPath.toString()); this.project.addCompileSourceRoot(outPath.toString());
sourcesGenerator.generateSources(basePackageName, outPath); sourcesGenerator.generateSources(basePackageName, outPath, Boolean.parseBoolean(useRecordBuilder));
} catch (IOException e) { } catch (IOException e) {
throw new MojoExecutionException("Exception while generating classes", e); throw new MojoExecutionException("Exception while generating classes", e);
} }

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

@ -93,8 +93,9 @@ public class SourcesGenerator {
/** /**
* @param basePackageName org.example * @param basePackageName org.example
* @param outPath path/to/output * @param outPath path/to/output
* @param useRecordBuilders if true, the data will have @RecordBuilder annotation
*/ */
public void generateSources(String basePackageName, Path outPath) throws IOException { public void generateSources(String basePackageName, Path outPath, boolean useRecordBuilders) throws IOException {
// Fix the configuration // Fix the configuration
for (Entry<String, InterfaceDataConfiguration> interfacesDatum : configuration.interfacesData.entrySet()) { for (Entry<String, InterfaceDataConfiguration> interfacesDatum : configuration.interfacesData.entrySet()) {
@ -2117,7 +2118,9 @@ public class SourcesGenerator {
typeClass.addModifiers(Modifier.STATIC); typeClass.addModifiers(Modifier.STATIC);
typeClass.addModifiers(Modifier.FINAL); typeClass.addModifiers(Modifier.FINAL);
typeClass.addSuperinterface(ClassName.get(joinPackage(versionPackage, "data"), "IBasicType")); typeClass.addSuperinterface(ClassName.get(joinPackage(versionPackage, "data"), "IBasicType"));
typeClass.addAnnotation(RecordBuilder.class); if (nextVersion.isEmpty() && useRecordBuilders) {
typeClass.addAnnotation(RecordBuilder.class);
}
var getBasicTypeMethod = MethodSpec var getBasicTypeMethod = MethodSpec
.methodBuilder("getBasicType$") .methodBuilder("getBasicType$")
.addModifiers(Modifier.PUBLIC) .addModifiers(Modifier.PUBLIC)

2
src/main/java/it/cavallium/data/generator/Standalone.java
View File

@ -7,6 +7,6 @@ public class Standalone {
public static void main(String[] args) throws IOException { public static void main(String[] args) throws IOException {
SourcesGenerator sourcesGenerator = SourcesGenerator.load(Paths.get(args[0])); SourcesGenerator sourcesGenerator = SourcesGenerator.load(Paths.get(args[0]));
sourcesGenerator.generateSources(args[1], Paths.get(args[2])); sourcesGenerator.generateSources(args[1], Paths.get(args[2]), Boolean.parseBoolean(args[3]));
} }
} }