python javalang如何获取ast指定类型的节点

在Python中，可以使用ast模块获取抽象语法树（AST）。要获取指定类型的节点，可以使用ast.NodeVisitor类来遍历整个AST，并在visit_*()方法中过滤需要的节点类型。例如，如果想获取所有的函数定义节点，可以实现以下代码：

import ast

class FunctionVisitor(ast.NodeVisitor):
    def __init__(self):
        self.functions = []

    def visit_FunctionDef(self, node):
        self.functions.append(node)

# 解析代码并获取函数定义节点
code = """
def func1():
    pass

def func2():
    pass
"""

tree = ast.parse(code)
visitor = FunctionVisitor()
visitor.visit(tree)

# 输出所有的函数定义节点
for func in visitor.functions:
    print(func.name)

在Java中，可以使用javalang模块获取语法树。同样地，在遍历整个语法树时，可以使用NodeVisitor类来过滤需要的节点类型。例如，如果想获取所有的方法声明节点，可以实现以下代码：

import javalang
from javalang.tree import MethodDeclaration

# 解析Java代码并获取方法声明节点
code = """
public class MyClass {
    public void method1() {}
    public void method2() {}
}

"""

tree = javalang.parse.parse(code)
methods = [node for _, node in tree.filter(MethodDeclaration)]

# 输出所有的方法声明节点
for method in methods:
    print(method.name)