小編精選 - 技術文章翻譯 · 09月11日

OpenRewrite 進階技術：利用訊息實現複雜邏輯

標題：進階 OpenRewrite 技術：使用訊息實現複雜邏輯

描述：如何在 AST 的多個層面上實現協調行動...

標籤：java、openrewrite、訊息傳遞

封面圖片：https://images.pexels.com/photos/260551/pexels-photo-260551.jpeg

規範網址：https://jtama.github.io/posts/technique-avanc-ed-openrewrite-utiliser-les-messages-pour-impl-menter-des-logiques-complexes/

系列：Openrewrite

已發布：真實

圖片來源：Pixabay

我已經在本系列的第一篇文章中介紹過 Openrewrite。如果你還沒讀過，我需要複習一下，我建議你收藏這篇文章，以後再回來看。這樣可以嗎？那就繼續吧。

上一篇文章討論的所有用例都涉及一個只需要在LST的一個且僅一個層級上找到資訊的配方。由於一圖勝過千言，而且我感覺我沒有表達清楚，因此這裡有一個可靠的 AST 範例：

AST 簡化範例

到目前為止，範例涉及對方法呼叫的修改，或例如類別重命名。

在上圖中，每個節點對應一個 AST 元素（這裡顯然簡化了）。正如我們已經看到的， OpenRewrite使用訪客模式來遍歷樹。更準確地說，食譜實現了存取者，其介面與所操作的語言相對應。

在這種情況下，操作是在Java上執行的，因此使用的訪客將是JavaIsoVisitor 。

是的，但是如果我的用例需要辨識方法呼叫以對類別中包含它的方法的聲明採取行動怎麼辦？

是的，我明白，這個問題也讓我輾轉反側了好幾天。不過，你看，OpenRewrite 背後的人也是。

因此，他們為我們提供了一個非常令人信服且相當安全的訊息傳遞系統。

Philipidès 最新訊息

用例

假設有人不明智，不，別追問，我不會告訴你是誰。好吧，是我。那麼，假設一下，我剛才說的是，有人不明智地在一個庫中開發了一種記錄方法處理過程的方法：

import static com.github.jtama.toxic.timer.logStart;
import static com.github.jtama.toxic.timer.logEnd;
import static com.github.jtama.acme.Process.longRunningMethod;

public class ManualGearCar {

    @Deprecated<4>
    public void drift(String param) {
        logStart();<1>
        longRunningMethod(param);<2>
        logEnd();<3>
    }
}

記錄方法執行的開始
進行治療
記錄方法執行的結束
稍後會感興趣的隨機註釋

經過深入研究，我們發現了***micrometer***及其註解io.micrometer.core.annotation.Timed的存在。

因此，作為開發人員，我將辨識Timer#logStart()方法的呼叫，如果找到它：

刪除它
如果存在Timer#logEnd()方法呼叫，則刪除它
在drift方法宣告中加入@Timed註解

這正是我們將在食譜中實現的邏輯。

執行

本例中，我們僅介紹訪客模式的實現，而非完整的方案。不過，一切程式碼均可在GitHub 倉庫中找到。

private static class ReplaceCompareVisitor extends JavaIsoVisitor<ExecutionContext> {

        private final MethodMatcher logStartInvocaMatcher = new MethodMatcher("com.github.jtama.toxic.Timer logStart()");<1>
        private final MethodMatcher logEndInvocaMatcher = new MethodMatcher("com.github.jtama.toxic.Timer logEnd()");
        private final AnnotationMatcher logStartMatcher = new AnnotationMatcher("@io.micrometer.core.annotation.Timed");
        private final JavaTemplate annotationTemplate = JavaTemplate.builder("@Timed")
                .imports("io.micrometer.core.annotation.Timed")
                .javaParser(
                        JavaParser.fromJavaVersion()
                            .classpath(JavaParser.runtimeClasspath()))
                .build();

        public ReplaceCompareVisitor() {
            maybeRemoveImport("com.github.jtama.toxic.Timer");
        }

        @Override
        public J.MethodDeclaration visitMethodDeclaration(J.MethodDeclaration method, ExecutionContext ctx) {<3>
            J.MethodDeclaration md = super.visitMethodDeclaration(method, ctx);
            Cursor cursor = getCursor();
            if (cursor.getMessage("appendAnnotation", false)) {
                if (md.getLeadingAnnotations().stream()
                        .noneMatch(logStartMatcher::matches)) {
                    maybeAddImport("io.micrometer.core.annotation.Timed");
                    md = annotationTemplate.apply(cursor, method.getCoordinates().addAnnotation(Comparator.comparing(J.Annotation::getSimpleName)));
                }
            }
            return md;
        }

        @Override
        public J.MethodInvocation visitMethodInvocation(J.MethodInvocation method, ExecutionContext ctx) {<2>
            J.MethodInvocation mi = super.visitMethodInvocation(method, ctx);
            if (logStartInvocaMatcher.matches(mi) || logEndInvocaMatcher.matches(mi)) {
                getCursor().putMessageOnFirstEnclosing(J.MethodDeclaration.class, "appendAnnotation", true);
                return null;
            }
            return mi;
        }
    }