Java中如何从JCTree（com.sun.tools.javac.tree.JCTree）获取父树节点？

好问题！在Java Compiler Tree（也就是com.sun.tools.javac.tree.JCTree）的体系里，默认情况下JCTree节点本身并没有内置的父节点引用——这也是很多开发者在用TreeTranslator遍历语法树时会遇到的痛点。下面给你两种实用的方案来实现获取父节点的需求：

方案一：用栈维护遍历路径中的父节点 #

TreeTranslator的遍历是深度优先的，我们可以利用这个特性，用一个栈来跟踪当前遍历路径上的父节点：进入节点时把它压入栈，处理完子节点后弹出，这样在处理子节点时，栈顶元素就是它的父节点。

示例代码如下：

import com.sun.tools.javac.tree.JCTree;
import com.sun.tools.javac.tree.TreeTranslator;
import java.util.ArrayDeque;
import java.util.Deque;

public class MyTreeTranslator extends TreeTranslator {
    // 用双端队列实现栈，维护当前遍历的父节点路径
    private Deque<JCTree> parentStack = new ArrayDeque<>();

    @Override
    public void visitMethodDef(JCTree.JCMethodDecl jcMethodDecl) {
        // 获取当前方法节点的父节点（栈顶元素）
        JCTree parentNode = parentStack.peek();
        if (parentNode != null) {
            // 这里可以添加你的父节点处理逻辑，比如打印父节点类型
            System.out.println("当前方法的父节点类型: " + parentNode.getClass().getSimpleName());
        }

        // 将当前方法节点压入栈，作为它子节点的父节点
        parentStack.push(jcMethodDecl);
        // 必须调用super方法，否则不会遍历该方法的子节点（比如参数、方法体）
        super.visitMethodDef(jcMethodDecl);
        // 处理完子节点后，弹出当前节点，恢复栈的状态
        parentStack.pop();
    }

    // 如果需要处理其他节点类型（比如类声明），同样要维护栈的状态
    @Override
    public void visitClassDef(JCTree.JCClassDecl jcClassDecl) {
        JCTree parentNode = parentStack.peek();
        // 处理类节点的父节点逻辑...

        parentStack.push(jcClassDecl);
        super.visitClassDef(jcClassDecl);
        parentStack.pop();
    }
}

这个方案的优点是直观易懂，完全贴合TreeTranslator的遍历机制，不会引入额外的内存映射开销，适合只需要处理特定节点类型的场景。

方案二：用映射表全局记录节点与父节点的关系 #

如果你需要全局记录所有节点的父关系，或者不想为每个节点类型都重写visit方法，可以重写TreeTranslator的scan方法，用一个映射表来存储每个节点对应的父节点：

import com.sun.tools.javac.tree.JCTree;
import com.sun.tools.javac.tree.TreeTranslator;
import java.util.WeakHashMap;

public class MyTreeTranslator extends TreeTranslator {
    // 使用WeakHashMap避免内存泄漏，因为JCTree节点无需长期持有父节点引用
    private WeakHashMap<JCTree, JCTree> nodeParentMap = new WeakHashMap<>();
    private JCTree currentParent;

    @Override
    public void scan(JCTree tree) {
        if (tree != null && currentParent != null) {
            // 记录当前节点的父节点
            nodeParentMap.put(tree, currentParent);
        }
        // 保存当前父节点，处理子节点时将父节点更新为当前树节点
        JCTree oldParent = currentParent;
        currentParent = tree;
        // 调用super方法完成子节点的扫描
        super.scan(tree);
        // 恢复父节点状态
        currentParent = oldParent;
    }

    @Override
    public void visitMethodDef(JCTree.JCMethodDecl jcMethodDecl) {
        // 从映射表中获取当前方法节点的父节点
        JCTree parentNode = nodeParentMap.get(jcMethodDecl);
        if (parentNode != null) {
            // 在这里处理父节点逻辑...
        }
        // 调用super方法遍历子节点
        super.visitMethodDef(jcMethodDecl);
    }
}

这个方案的优势是通用性强，只需要重写一次scan方法就能处理所有节点的父关系记录，适合需要全局处理语法树节点的场景。

内容的提问来源于stack exchange，提问作者Guo