Modifying Methods with JavaTemplate

This tutorial will demonstrate how the JavaTemplate can be used to manipulate and change a method declaration using rewrite's refactoring capabilities. You will build a recipe that modifies the setCustomerInfo() method in the following class:
package com.yourorg;
​
public abstract class Customer {
    private Date dateOfBirth;
    private String firstName;
    private String lastName;
​
    public abstract void setCustomerInfo(String lastName);
}
The recipe and associated visitor will:
Define a method body and remove the abstract modifier
Add two additional parameters to the method declaration
Add two additional statements to the method declaration's body
The resulting refactored class will look like this:
package com.yourorg;
​
import java.util.Date;
​
public abstract class Customer {
    private Date dateOfBirth;
    private String firstName;
    private String lastName;
​
    public void setCustomerInfo(Date dateOfBirth, String firstName, String lastName){
        this.dateOfBirth = dateOfBirth;
        this.firstName = firstName;
        this.lastName = lastName;
    }
}
This guide assumes you've already set up your Recipe Development Environment.
Define a Recipe & Visitor
Create a new class that extends org.openrewrite.Recipe. This recipe will act as a wrapper around a visitor we will build to target the method com.yourorg.Customer.setCustomerInfo()and this recipe does not require any additional configuration properties. The visitor is defined as an inner class to the recipe because it is not intended to be used outside the context of the recipe.
package org.openrewrite.samples;
​
import org.openrewrite.ExecutionContext;
import org.openrewrite.Recipe;
import org.openrewrite.TreeVisitor;
import org.openrewrite.java.JavaIsoVisitor;
import org.openrewrite.java.MethodMatcher;
import org.openrewrite.java.tree.J;
import org.openrewrite.java.tree.J.MethodDeclaration;
​
public class ExpandCustomerInfo extends Recipe {
​
    @Override
    public String getDisplayName() {
        return "Expand Customer Info";
    }
​
    @Override
    public String getDescription() {
        return "Expand the `CustomerInfo` class with new fields.";
    }
    
    // Rewrite provides a managed environment in which it discovers, instantiates, and wires configuration into Recipes.
    // This recipe has no configuration and delegates to its visitor when it is run.
    @Override
    protected JavaIsoVisitor<ExecutionContext> getVisitor() {
        return new ExpandCustomerInfoVisitor();
    }
​
    private class ExpandCustomerInfoVisitor extends JavaIsoVisitor<ExecutionContext> {
​
        // Used to identify the method declaration that will be refactored
        private final MethodMatcher methodMatcher = new MethodMatcher("com.yourorg.Customer setCustomerInfo(String)");
​
        @Override
        public MethodDeclaration visitMethodDeclaration(MethodDeclaration m, ExecutionContext c) {
            if (!methodMatcher.matches(m.getType())) {
                return m;
            }
            //TODO - Implement Refactoring operations on the matching "setCustomerInfo()" method declaration.
​
            return m;
        }
    }
}
Add a Method Body to setCustomerInfo()
Within the ExpandCustomerInfoVisitor, we will add logic to remove the abstract modifier from the method and use the JavaTemplate to add a method body to declaration. This is accomplished by defining a JavaTemplate that represents the method body as an instance variable within the ExpandCustomerInfoVisitor:
    private class ExpandCustomerInfoVisitor extends JavaIsoVisitor<ExecutionContext> {
        ...
        // Template used to add a method body to "setCustomerInfo()" method declaration.
        private final JavaTemplate addMethodBodyTemplate = template("")
                .build();
         ...
    }
When using a template to replace a method's body, the template must include the open and closing curly braces.
The method is then replaced with a copy that remove the abstract modifier and the template is then used within the visitMethodDeclaration() method to replace the method body:
public MethodDeclaration visitMethodDeclaration(MethodDeclaration method, P p) {
    ...
    // Remove the abstract modifier from the method.
    m = m.withModifiers(m.getModifiers().stream()
            .filter(mod -> mod.getType() != Type.Abstract)
            .collect(Collectors.toList()));
​
    // Add a method body
    m = m.withTemplate(addMethodBodyTemplate, m.getCoordinates().replaceBody());
​
    return m;
}
Add Parameters to setCustomerInfo()
Next we will use JavaTemplate to add two additional parameters to the method declaration. Here we use an interpolation signifier #{} to leave a space for the existing argument so that it is preserved when we replace the method's argument list. 
private class ExpandCustomerInfoVisitor extends JavaIsoVisitor<ExecutionContext> {
    ...
    // Template used to insert two additional parameters into the "setCustomerInfo()" method declaration.
    private final JavaTemplate addMethodParametersTemplate = template("Date dateOfBirth, String firstName, #{}")
            .imports("java.util.Date")
            .build();
    ...
}
NOTE: Because a new type, java.util.Date, is being introduced to the template, the type must be added when building the template. This ensures that any generated elements will have correct type attribution.
The template is is used to replace the method declaration using withTemplate(). Our code must also ensure that java.util.Date is added as an import to the compilation unit.
public MethodDeclaration visitMethodDeclaration(MethodDeclaration method, P p) {
    ...
    // Add two parameters to the method declaration by inserting them in from of the first argument.
    m = m.withTemplate(addMethodParametersTemplate,
            m.getCoordinates().replaceParameters(),
            m.getParameters().get(0));
​
    // Add an import for java.util.Date to this compilation unit's list of imports.
    maybeAddImport("java.util.Date");
​
    return m;
}
Add Additional Statements to setCustomerInfo()
We'll use another template to add the assignment statements to the method body.
    private class ExpandCustomerInfoVisitor extends JavaIsoVisitor<ExecutionContext> {
        ...
        // Template used to add initializing statements to the method body
        private final JavaTemplate addStatementsTemplate = template(
                "this.dateOfBirth = dateOfBirth;\n" +
                        "this.firstName = firstName;\n" +
                        "this.lastName = lastName;\n")
                .build();
         ...
    }
And this template is used to insert those statements into the method declaration's body:
public MethodDeclaration visitMethodDeclaration(MethodDeclaration method, P p) {
    ...
    // Safe to assert since we just added a body to the method
    assert m.getBody() != null;
​
    // Add the assignment statements to the method body
    m = m.withTemplate(addStatementsTemplate, m.getBody().getCoordinates().lastStatement());
​
    return m;
}
Testing The Recipe
After following the steps above, the final recipe will look like the following:
package org.openrewrite.samples;
​
import java.util.stream.Collectors;
​
import org.openrewrite.ExecutionContext;
import org.openrewrite.Recipe;
import org.openrewrite.java.JavaIsoVisitor;
import org.openrewrite.java.JavaTemplate;
import org.openrewrite.java.MethodMatcher;
import org.openrewrite.java.tree.J;
import org.openrewrite.java.tree.J.MethodDeclaration;
import org.openrewrite.java.tree.J.Modifier.Type;
​
public class ExpandCustomerInfo extends Recipe {
    @Override
    public String getDisplayName() {
        return "Expand Customer Info";
    }
​
    @Override
    public String getDescription() {
        return "Expand the `CustomerInfo` class with new fields.";
    }
​
    // Rewrite provides a managed environment in which it discovers, instantiates, and wires configuration into Recipes.
    // This recipe has no configuration and when it is executed, it will delegate to its visitor.
    @Override
    protected ExpandCustomerInfoVisitor getVisitor() {
        return new ExpandCustomerInfoVisitor();
    }
​
    private class ExpandCustomerInfoVisitor extends JavaIsoVisitor<ExecutionContext> {
​
        // Used to identify the method declaration that will be refactored
        private final MethodMatcher methodMatcher = new MethodMatcher("com.yourorg.Customer setCustomerInfo(String)");
​
        // Template used to add a method body to "setCustomerInfo()" method declaration.
        private final JavaTemplate addMethodBodyTemplate = template("")
                .build();
​
        // Template used to insert two additional parameters into the "setCustomerInfo()" method declaration.
        private final JavaTemplate addMethodParametersTemplate = template("Date dateOfBirth, String firstName, #{}")
                .imports("java.util.Date")
                .build();
​
        // Template used to add initializing statements to the method body
        private final JavaTemplate addStatementsTemplate = template(
                "this.dateOfBirth = dateOfBirth;\n" +
                        "this.firstName = firstName;\n" +
                        "this.lastName = lastName;\n")
                .build();
​
        @Override
        public MethodDeclaration visitMethodDeclaration(MethodDeclaration m, ExecutionContext c) {
            if (!methodMatcher.matches(m.getType())) {
                return m;
            }
            // Remove the abstract modifier from the method.
            m = m.withModifiers(m.getModifiers().stream()
                    .filter(mod -> mod.getType() != Type.Abstract)
                    .collect(Collectors.toList()));
​
            // Add a method body
            m = m.withTemplate(addMethodBodyTemplate, m.getCoordinates().replaceBody());
​
            // Add two parameters to the method declaration by inserting them in from of the first argument.
            m = m.withTemplate(addMethodParametersTemplate,
                    m.getCoordinates().replaceParameters(),
                    m.getParameters().get(0));
​
            // Add an import for java.util.Date to this compilation unit's list of imports.
            maybeAddImport("java.util.Date");
​
            // Safe to assert since we just added a body to the method
            assert m.getBody() != null;
​
            // Add the assignment statements to the method body
            m = m.withTemplate(addStatementsTemplate, m.getBody().getCoordinates().lastStatement());
​
            return m;
        }
    }
}
​
Rewrite provides testing infrastructure for recipes via the rewrite-test module. To create automated tests of this recipe we use the Kotlin language, mostly for convenient access to multi-line Strings, with JUnit 5. To assert that the recipe makes the expected changes to our test class, we will create a new test class that implements the JavaRecipeTest interface and useassertChanged ensure the recipe is making the expected changes to the test class.
package org.openrewrite.samples;
​
import org.openrewrite.java.JavaParser
import org.junit.jupiter.api.Test
import org.openrewrite.java.JavaRecipeTest
​
class ExpandCustomerInfoTest : JavaRecipeTest {
​
    override val recipe = ExpandCustomerInfo()
​
    @Test
    fun testExapandCustomerInfo() = assertChanged(
        before = """
            package com.yourorg;
​
            public abstract class Customer {
                private Date dateOfBirth;
                private String firstName;
                private String lastName;
​
                public abstract void setCustomerInfo(String lastName);
            }
        """,
        after = """
            package com.yourorg;
            
            import java.util.Date;
            
            public abstract class Customer {
                private Date dateOfBirth;
                private String firstName;
                private String lastName;
            
                public void setCustomerInfo(Date dateOfBirth, String firstName, String lastName){
                    this.dateOfBirth = dateOfBirth;
                    this.firstName = firstName;
                    this.lastName = lastName;
                }
            }
        """
    )
}
​
Tutorials & Guides - Previous
Writing a Java Refactoring Recipe
Next - Tutorials & Guides
Refactoring with Declarative YAML Recipes
Last updated 2 weeks ago
Modifying Methods with JavaTemplate

Define a Recipe & Visitor

Add a Method Body to `setCustomerInfo()`

Add Parameters to `setCustomerInfo()`

Add Additional Statements to `setCustomerInfo()`

Testing The Recipe
