Add method beginOverridePublicMethod for ScalaWriter

querydsl · Sep 1, 2015 · 6d841dd · 6d841dd
1 parent 1ea7390
commit 6d841dd
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/main/java/com/mysema/codegen/ScalaWriter.java b/src/main/java/com/mysema/codegen/ScalaWriter.java
@@ -44,6 +44,8 @@ public class ScalaWriter extends AbstractCodeWriter<ScalaWriter> {
 
     private static final String DEF = "def ";
 
+    private static final String OVERRIDE_DEF = "override " + DEF;
+
     private static final String EXTENDS = " extends ";
 
     private static final String WITH = " with ";
@@ -313,6 +315,17 @@ public ScalaWriter beginPublicMethod(Type returnType, String methodName, Paramet
         return beginMethod(DEF, returnType, methodName, args);
     }
 
+    public <T> ScalaWriter beginOverridePublicMethod(Type returnType, String methodName,
+                                                     Collection<T> parameters, Function<T, Parameter> transformer)
+            throws IOException {
+        return beginMethod(OVERRIDE_DEF, returnType, methodName, transform(parameters, transformer));
+    }
+
+    public ScalaWriter beginOverridePublicMethod(Type returnType, String methodName, Parameter... args)
+            throws IOException {
+        return beginMethod(OVERRIDE_DEF, returnType, methodName, args);
+    }
+
     @Override
     public <T> ScalaWriter beginStaticMethod(Type returnType, String methodName,
             Collection<T> parameters, Function<T, Parameter> transformer) throws IOException {