diff options
author | Danny van Bruggen <hexagonaal@gmail.com> | 2017-03-10 22:13:23 +0100 |
---|---|---|
committer | Danny van Bruggen <hexagonaal@gmail.com> | 2017-03-10 22:13:23 +0100 |
commit | af62675f681782a2b648e5fb2f2e1e137c996f4f (patch) | |
tree | 614646316faaad0d0b6eefdcc7002f84912a4369 /javaparser-core-generators/src/main/java/com/github/javaparser/generator/core | |
parent | 82f08468d6b957cffe459628eb7f0c1d60ee9cd8 (diff) | |
download | javaparser-af62675f681782a2b648e5fb2f2e1e137c996f4f.tar.gz |
Generalize method
Diffstat (limited to 'javaparser-core-generators/src/main/java/com/github/javaparser/generator/core')
-rw-r--r-- | javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/RemoveMethodGenerator.java | 18 |
1 files changed, 9 insertions, 9 deletions
diff --git a/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/RemoveMethodGenerator.java b/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/RemoveMethodGenerator.java index 907428a2a..edc358b43 100644 --- a/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/RemoveMethodGenerator.java +++ b/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/RemoveMethodGenerator.java @@ -71,18 +71,18 @@ public class RemoveMethodGenerator extends NodeGenerator { private MethodDeclaration getOrCreateMethod(ClassOrInterfaceDeclaration nodeCoid, Supplier<MethodDeclaration> newMethodSupplier, - String methodName, String... methodParameterTypes) { - final List<MethodDeclaration> removeMethods = nodeCoid.getMethodsBySignature(methodName, methodParameterTypes); - final MethodDeclaration removeMethod; - if (removeMethods.isEmpty()) { - removeMethod = newMethodSupplier.get(); - nodeCoid.addMember(removeMethod); - } else if (removeMethods.size() == 1) { - removeMethod = removeMethods.get(0); + String methodName, String... parameterTypes) { + final List<MethodDeclaration> methods = nodeCoid.getMethodsBySignature(methodName, parameterTypes); + final MethodDeclaration method; + if (methods.isEmpty()) { + method = newMethodSupplier.get(); + nodeCoid.addMember(method); + } else if (methods.size() == 1) { + method = methods.get(0); } else { throw new AssertionError(f("Found more than one method while expecting only one.")); } - return removeMethod; + return method; } private String attributeCheck(PropertyMetaModel property, String removeAttributeMethodName) { |