Java 中的屠龙之术：如何修改语法树？(1)

老k不爱考

于 2024-04-28 21:28:32 发布

阅读量1k

点赞数 30

分类专栏：程序员文章标签： java python 开发语言

本文链接：https://blog.csdn.net/GTAKK155GC/article/details/138291883

版权

程序员专栏收录该内容

196 篇文章 0 订阅

订阅专栏

tree.pos = pos;

return tree;

}

public JCMethodDecl MethodDef(MethodSymbol m,

Type mtype,

JCBlock body) {

return (JCMethodDecl)

new JCMethodDecl(

Modifiers(m.flags(), Annotations(m.getAnnotationMirrors())),

m.name,

Type(mtype.getReturnType()),

TypeParams(mtype.getTypeArguments()),

Params(mtype.getParameterTypes(), m),

Types(mtype.getThrownTypes()),

body,

null,

m).setPos(pos).setType(mtype);

}

返回类型restype填写null或者treeMaker.TypeIdent(TypeTag.VOID)都代表返回void类型。另外，Java 系列面试题和答案全部整理好了，微信搜索Java技术栈，在后台发送：面试，可以在线阅读。

TreeMaker.VarDef

TreeMaker.VarDef用于创建字段/变量定义语法树节点（JCVariableDecl），源码如下

public JCVariableDecl VarDef(JCModifiers mods,

Name name,

JCExpression vartype,

JCExpression init) {

JCVariableDecl tree = new JCVariableDecl(mods, name, vartype, init, null);

tree.pos = pos;

return tree;

}

public JCVariableDecl VarDef(VarSymbol v,

JCExpression init) {

return (JCVariableDecl)

new JCVariableDecl(

Modifiers(v.flags(), Annotations(v.getAnnotationMirrors())),

v.name,

Type(v.type),

init,

v).setPos(pos).setType(v.type);

}

mods：访问标志
name：参数名称
vartype：类型
init：初始化语句
v：变量符号

TreeMaker.Ident

TreeMaker.Ident用于创建标识符语法树节点（JCIdent），源码如下

public JCIdent Ident(Name name) {

JCIdent tree = new JCIdent(name, null);

tree.pos = pos;

return tree;

}

public JCIdent Ident(Symbol sym) {

return (JCIdent)new JCIdent((sym.name != names.empty)

? sym.name

: sym.flatName(), sym)

.setPos(pos)

.setType(sym.type);

}

public JCExpression Ident(JCVariableDecl param) {

return Ident(param.sym);

}

TreeMaker.Return

TreeMaker.Return用于创建return语句（JCReturn），源码如下

public JCReturn Return(JCExpression expr) {

JCReturn tree = new JCReturn(expr);

tree.pos = pos;

return tree;

}

TreeMaker.Select

TreeMaker.Select用于创建域访问/方法访问（这里的方法访问只是取到名字，方法的调用需要用TreeMaker.Apply）语法树节点（JCFieldAccess），源码如下

public JCFieldAccess Select(JCExpression selected,

Name selector)

{

JCFieldAccess tree = new JCFieldAccess(selected, selector, null);

tree.pos = pos;

return tree;

}

public JCExpression Select(JCExpression base,

Symbol sym) {

return new JCFieldAccess(base, sym.name, sym).setPos(pos).setType(sym.type);

}

selected：.运算符左边的表达式
selector：.运算符右边的表达式

下面给出一个例子，一语句生成的Java语句就是二语句

一. TreeMaker.Select(treeMaker.Ident(names.fromString(“this”)), names.fromString(“name”));

二. this.name

TreeMaker.NewClass

TreeMaker.NewClass用于创建new语句语法树节点（JCNewClass）,源码如下：

public JCNewClass NewClass(JCExpression encl,

List typeargs,

JCExpression clazz,

List args,

JCClassDecl def) {

JCNewClass tree = new JCNewClass(encl, typeargs, clazz, args, def);

tree.pos = pos;

return tree;

}

encl：不太明白此参数的含义，我看很多例子中此参数都设置为null
typeargs：参数类型列表
clazz：待创建对象的类型
args：参数列表
def：类定义

TreeMaker.Apply

TreeMaker.Apply用于创建方法调用语法树节点（JCMethodInvocation），源码如下：

public JCMethodInvocation Apply(List typeargs,

JCExpression fn,

List args) {

JCMethodInvocation tree = new JCMethodInvocation(typeargs, fn, args);

tree.pos = pos;

return tree;

}

typeargs：参数类型列表
fn：调用语句
args：参数列表

TreeMaker.Assign

TreeMaker.Assign用户创建赋值语句语法树节点（JCAssign），源码如下：

ublic JCAssign Assign(JCExpression lhs,

JCExpression rhs) {

JCAssign tree = new JCAssign(lhs, rhs);

tree.pos = pos;

return tree;

}

lhs：赋值语句左边表达式
rhs：赋值语句右边表达式

TreeMaker.Exec

TreeMaker.Exec用于创建可执行语句语法树节点（JCExpressionStatement），源码如下：

public JCExpressionStatement Exec(JCExpression expr) {

JCExpressionStatement tree = new JCExpressionStatement(expr);

tree.pos = pos;

return tree;

}

TreeMaker.Apply以及TreeMaker.Assign就需要外面包一层TreeMaker.Exec来获得一个JCExpressionStatement。分享：Spring Boot 学习笔记，这个太全了

TreeMaker.Block

TreeMaker.Block用于创建组合语句的语法树节点（JCBlock），源码如下：

public JCBlock Block(long flags,

List stats) {

JCBlock tree = new JCBlock(flags, stats);

tree.pos = pos;

return tree;

}

flags：访问标志
stats：语句列表

com.sun.tools.javac.util.List介绍

在我们操作抽象语法树的时候，有时会涉及到关于List的操作，但是这个List不是我们经常使用的java.util.List而是com.sun.tools.javac.util.List，这个List比较奇怪，是一个链式的结构，有头结点和尾节点，但是只有尾节点是一个List，这里作为了解就行了。

public class List extends AbstractCollection implements java.util.List {

public A head;

public List tail;

private static final List<?> EMPTY_LIST = new List((Object)null, (List)null) {

public List setTail(List var1) {

throw new UnsupportedOperationException();

}

public boolean isEmpty() {

return true;

}

};

List(A head, List tail) {

this.tail = tail;

this.head = head;

}

public static List nil() {

return EMPTY_LIST;

}

public List prepend(A var1) {

return new List(var1, this);

}

public List append(A var1) {

return of(var1).prependList(this);

}

public static List of(A var0) {

return new List(var0, nil());

}

public static List of(A var0, A var1) {

return new List(var0, of(var1));

}

public static List of(A var0, A var1, A var2) {

return new List(var0, of(var1, var2));

}

public static List of(A var0, A var1, A var2, A… var3) {

return new List(var0, new List(var1, new List(var2, from(var3))));

}

…

}

com.sun.tools.javac.util.ListBuffer

由于com.sun.tools.javac.util.List使用起来不方便，所以又在其上面封装了一层，这个封装类是ListBuffer，此类的操作和我们平时经常使用的java.util.List用法非常类似。

点击关注公众号，Java干货****及时送达

public class ListBuffer extends AbstractQueue {

public static ListBuffer of(T x) {

ListBuffer lb = new ListBuffer();

lb.add(x);

return lb;

}

/** The list of elements of this buffer.

private List elems;

/** A pointer pointing to the last element of ‘elems’ containing data,

* or null if the list is empty.

private List last;

/** The number of element in this buffer.

private int count;

/** Has a list been created from this buffer yet?

private boolean shared;

/** Create a new initially empty list buffer.

public ListBuffer() {

clear();

}

/** Append an element to buffer.

public ListBuffer append(A x) {

x.getClass(); // null check

if (shared) copy();

List newLast = List.of(x);

if (last != null) {

last.tail = newLast;

last = newLast;

} else {

elems = last = newLast;

}

count++;

return this;

}

…

}

com.sun.tools.javac.util.Names介绍

这个是为我们创建名称的一个工具类，无论是类、方法、参数的名称都需要通过此类来创建。它里面经常被使用到的一个方法就是fromString()，一般使用方法如下所示。

Names names = new Names()

names. fromString(“setName”);

实战演练

上面我们大概了解了如何操作抽象语法树，接下来我们就来写几个真实的案例加深理解。

变量相关

在类中我们经常操作的参数就是变量，那么如何使用抽象语法树的特性为我们操作变量呢？接下来我们就将一些对于变量的一些操作。

生成变量

例如生成private String age;这样一个变量，借用我们上面讲的VarDef方法

// 生成参数例如：private String age;

treeMaker.VarDef(treeMaker.Modifiers(Flags.PRIVATE), names.fromString(“age”), treeMaker.Ident(names.fromString(“String”)), null);

对变量赋值

例如我们想生成private String name = "BuXueWuShu"，还是利用VarDef方法

// private String name = “BuXueWuShu”

treeMaker.VarDef(treeMaker.Modifiers(Flags.PRIVATE),names.fromString(“name”),treeMaker.Ident(names.fromString(“String”)),treeMaker.Literal(“BuXueWuShu”))

两个字面量相加

例如我们生成String add = "a" + "b";，借用我们上面讲的Exec方法和Assign方法

// add = “a”+“b”

treeMaker.Exec(treeMaker.Assign(treeMaker.Ident(names.fromString(“add”)),treeMaker.Binary(JCTree.Tag.PLUS,treeMaker.Literal(“a”),treeMaker.Literal(“b”))))

+=语法

例如我们想生成add += "test"，则和上面字面量差不多。

// add+=“test”

treeMaker.Exec(treeMaker.Assignop(JCTree.Tag.PLUS_ASG, treeMaker.Ident(names.fromString(“add”)), treeMaker.Literal(“test”)))

++语法

例如想生成++i

treeMaker.Exec(treeMaker.Unary(JCTree.Tag.PREINC,treeMaker.Ident(names.fromString(“i”))))

方法相关

我们对于变量进行了操作，那么基本上都是要生成方法的，那么如何对方法进行生成和操作呢？我们接下来演示一下关于方法相关的操作方法。

无参无返回值

我们可以利用上面讲到的MethodDef方法进行生成

无参无返回值的方法生成

public void test(){

}

// 定义方法体

ListBuffer<JCTree.JCStatement> testStatement = new ListBuffer<>();

JCTree.JCBlock testBody = treeMaker.Block(0, testStatement.toList());

JCTree.JCMethodDecl test = treeMaker.MethodDef(

treeMaker.Modifiers(Flags.PUBLIC), // 方法限定值

names.fromString(“test”), // 方法名

treeMaker.Type(new Type.JCVoidType()), // 返回类型

com.sun.tools.javac.util.List.nil(),

testBody, // 方法体

null

);

有参无返回值

我们可以利用上面讲到的MethodDef方法进行生成

无参无返回值的方法生成

public void test2(String name){

name = “xxxx”;

}

ListBuffer<JCTree.JCStatement> testStatement2 = new ListBuffer<>();

testStatement2.append(treeMaker.Exec(treeMaker.Assign(treeMaker.Ident(names.fromString(“name”)),treeMaker.Literal(“xxxx”))));

JCTree.JCBlock testBody2 = treeMaker.Block(0, testStatement2.toList());

最后

如果觉得本文对你有帮助的话，不妨给我点个赞，关注一下吧！

eMaker.Block(0, testStatement.toList());

JCTree.JCMethodDecl test = treeMaker.MethodDef(

treeMaker.Modifiers(Flags.PUBLIC), // 方法限定值

names.fromString(“test”), // 方法名

treeMaker.Type(new Type.JCVoidType()), // 返回类型

com.sun.tools.javac.util.List.nil(),

testBody, // 方法体

null

);

有参无返回值

我们可以利用上面讲到的MethodDef方法进行生成

无参无返回值的方法生成

public void test2(String name){

name = “xxxx”;

}

ListBuffer<JCTree.JCStatement> testStatement2 = new ListBuffer<>();

testStatement2.append(treeMaker.Exec(treeMaker.Assign(treeMaker.Ident(names.fromString(“name”)),treeMaker.Literal(“xxxx”))));

JCTree.JCBlock testBody2 = treeMaker.Block(0, testStatement2.toList());

最后

如果觉得本文对你有帮助的话，不妨给我点个赞，关注一下吧！

[外链图片转存中…(img-NnYAQuth-1714310900777)]

[外链图片转存中…(img-Wd3iuoOn-1714310900778)]

本文已被CODING开源项目：【一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码】收录

老k不爱考

关注

30
点赞
踩
14

收藏

觉得还不错? 一键收藏
0
评论
Java 中的屠龙之术：如何修改语法树？(1)

treeMaker.Modifiers(Flags.PUBLIC), // 方法限定值names.fromString(“test”), // 方法名treeMaker.Type(new Type.JCVoidType()), // 返回类型testBody, // 方法体null[外链图片转存中…(img-NnYAQuth-1714310900777)][外链图片转存中…(img-Wd3iuoOn-1714310900778)]本文已被。
复制链接

扫一扫