github/jooq
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[jOOQ/jOOQ#3713] [jOOQ/jOOQ#10191] Add <pojosAsScalaCaseClasses/> to the

Browse Source 
code generator to generate case classes
This commit is contained in:
Lukas Eder 2020-06-17 11:35:31 +02:00
parent f1c63cfc07
commit 7712891962
1 changed files with 25 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
jOOQ-codegen/src/main/java/org/jooq/codegen/JavaGenerator.java
View File

@ -272,6 +272,7 @@ public class JavaGenerator extends AbstractGenerator {
*/
private Set<File> directoriesNotForRemoval = new LinkedHashSet<>();
private final boolean java;
private final boolean scala;
private final boolean kotlin;
private final String tokenVoid;
@ -319,6 +320,7 @@ public class JavaGenerator extends AbstractGenerator {
this.scala = (language == SCALA);
this.kotlin = (language == KOTLIN);
this.java = (language == JAVA);
this.tokenVoid = (scala || kotlin ? "Unit" : "void");
this.fileCache = new Files();
}
@ -3600,13 +3602,13 @@ public class JavaGenerator extends AbstractGenerator {
maxLength = Math.max(maxLength, out.ref(getJavaType(column.getType(resolver(Mode.POJO)), Mode.POJO)).length());
if (scala) {
out.println("%sclass %s(", (generateImmutablePojos() ? "case " : ""), className);
out.println("%sclass %s(", (generatePojosAsScalaCaseClasses() ? "case " : ""), className);
String separator = " ";
for (TypedElementDefinition<?> column : getTypedElements(tableOrUDT)) {
out.println("%s%s%s: %s",
out.println("%s%s %s: %s",
separator,
generateImmutablePojos() ? "" : "private var ",
generateImmutablePojos() ? "val" : "var",
getStrategy().getJavaMemberName(column, Mode.POJO),
out.ref(getJavaType(column.getType(resolver(Mode.POJO)), Mode.POJO)));
@ -3622,10 +3624,10 @@ public class JavaGenerator extends AbstractGenerator {
for (TypedElementDefinition<?> column : getTypedElements(tableOrUDT)) {
final String member = getStrategy().getJavaMemberName(column, Mode.POJO);
out.println("%s%s%s%s: %s? = null",
out.println("%s%s%s %s: %s? = null",
separator,
generateInterfaces() ? "override " : "",
generateImmutablePojos() ? "val " : "var ",
generateImmutablePojos() ? "val" : "var",
member,
out.ref(getJavaType(column.getType(resolver(Mode.POJO)), Mode.POJO)));
@ -3980,13 +3982,20 @@ public class JavaGenerator extends AbstractGenerator {
}
protected void generatePojoEqualsAndHashCode(Definition tableOrUDT, JavaWriter out) {
if (scala && generatePojosAsScalaCaseClasses())
return;
if (kotlin && generatePojosAsKotlinDataClasses())
return;
if (java && generatePojosAsJavaRecordClasses())
return;
final String className = getStrategy().getJavaClassName(tableOrUDT, Mode.POJO);
out.println();
if (scala) {
out.println("override def equals(obj: Any): scala.Boolean = {");
out.println("if (this == obj)");
out.println("if (this eq obj.asInstanceOf[AnyRef])");
out.println("return true");
out.println("if (obj == null)");
out.println("return false");
@ -4003,17 +4012,15 @@ public class JavaGenerator extends AbstractGenerator {
out.println("return false");
out.println("}");
if (getJavaType(column.getType(resolver())).endsWith("[]")) {
if (getJavaType(column.getType(resolver())).endsWith("[]"))
out.println("else if (!%s.equals(%s, other.%s))", Arrays.class, columnMember, columnMember);
}
else {
else
out.println("else if (!%s.equals(other.%s))", columnMember, columnMember);
}
out.println("return false");
}
out.println("return true");
out.println("true");
out.println("}");
}
else {
@ -4094,6 +4101,13 @@ public class JavaGenerator extends AbstractGenerator {
}
protected void generatePojoToString(Definition tableOrUDT, JavaWriter out) {
if (scala && generatePojosAsScalaCaseClasses())
return;
if (kotlin && generatePojosAsKotlinDataClasses())
return;
if (java && generatePojosAsJavaRecordClasses())
return;
final String className = getStrategy().getJavaClassName(tableOrUDT, Mode.POJO);
out.println();