Openrewrite:如何用2个参数“`foo(a，b)”替换一个方法，并将其替换为每个``foo(A).bar(B)‘两个参数的方法？

Question

我想皈依

x.foo(a, b);

转到

x.foo(a).bar(b);

我可以很容易地匹配acme.X foo(acme.A, acme.B)，但是如何构建一个JavaTemplate来代替我呢？

当我跑的时候

@Override
protected TreeVisitor<?, ExecutionContext> getVisitor() {
    return new JavaIsoVisitor<>() {

        private final JavaTemplate template = JavaTemplate.builder(this::getCursor,
                "foo(#{any(java.lang.String)}).bar(#{any(java.lang.String)})")
                .build();

        @Override
        public J.MethodInvocation visitMethodInvocation(J.MethodInvocation method, ExecutionContext executionContext) {
            J.MethodInvocation m = super.visitMethodInvocation(method, executionContext);
            if (....matches(method)) {
                List<Expression> arguments = m.getArguments();
                m = m.withTemplate(template, m.getCoordinates().replace(), arguments.get(0), arguments.get(1));
            }
            return m;
        }
    };
}

我得到了

foo(a).bar(b);

而不是

x.foo(a).bar(b);

Answer 1

这对我起了作用(归功于帕特里克·路德福)：

private static final MethodMatcher MATCHER =
        new MethodMatcher("org.optaplanner.core.api.score.stream.ConstraintStream " +
                "penalize(java.lang.String, org.optaplanner.core.api.score.Score)");

@Override
protected TreeVisitor<?, ExecutionContext> getVisitor() {
    return new JavaIsoVisitor<>() {

        private final JavaTemplate template = JavaTemplate.builder(() -> getCursor().getParentOrThrow(),
                "#{any(org.optaplanner.core.api.score.stream.ConstraintStream)}" +
                        ".penalize(#{any(org.optaplanner.core.api.score.Score)})" +
                        ".asConstraint(#{any(java.lang.String)})"
        ).build();

        @Override
        public Expression visitExpression(Expression expression, ExecutionContext executionContext) {
            Expression e = super.visitExpression(expression, executionContext);
            if (MATCHER.matches(e)){
                J.MethodInvocation mi = (J.MethodInvocation) e;
                e = e.withTemplate(template,
                        e.getCoordinates().replace(), mi.getSelect(),
                        mi.getArguments().get(1), mi.getArguments().get(0));
            }
            return e;
        }
    };
}