diff --git a/pom.xml b/pom.xml index 997337b92..999505946 100644 --- a/pom.xml +++ b/pom.xml @@ -4,7 +4,7 @@ 4.0.0 com.mysema.codegen codegen - 0.2.5-SNAPSHOT + 0.2.6 Codegen Code generation and compilation for Java diff --git a/src/main/java/com/mysema/codegen/ScalaWriter.java b/src/main/java/com/mysema/codegen/ScalaWriter.java index 321cfeaa5..58e10de64 100644 --- a/src/main/java/com/mysema/codegen/ScalaWriter.java +++ b/src/main/java/com/mysema/codegen/ScalaWriter.java @@ -50,6 +50,8 @@ public class ScalaWriter extends AbstractCodeWriter{ private static final String PUBLIC = "public "; private static final String PUBLIC_CLASS = "class "; + + private static final String PUBLIC_OBJECT = "object "; private static final String VAR = "var "; @@ -155,8 +157,14 @@ public class ScalaWriter extends AbstractCodeWriter{ return this; } - public ScalaWriter beginClass(String header) throws IOException{ - line(PUBLIC_CLASS, header); + public ScalaWriter beginObject(String header) throws IOException { + line(PUBLIC_OBJECT, header, " {"); + goIn(); + return this; + } + + public ScalaWriter beginClass(String header) throws IOException { + line(PUBLIC_CLASS, header, " {"); goIn(); return this; } @@ -308,7 +316,11 @@ public class ScalaWriter extends AbstractCodeWriter{ } } if (rv.endsWith("[]")){ - return "Array[" + rv.substring(0, rv.length()-2) + "]"; + rv = rv.substring(0, rv.length()-2); + if (classes.contains(rv)){ + rv = rv.substring(packageName.length()+1); + } + return "Array[" + rv + "]"; }else{ return rv; }