问题描述
我正在使用 JavaParser 解析以下代码的内容并识别语句的顺序:
class X {
void x() {
int x = 1;
if (x>4) {
x=21;
} else {
if (x>1) {
x=3;
} else {
x=212;
}
}
if (x==4) {
x=3;
}
}}
我用来解析上面例子的代码:
private static void statements(BlockStmt block,State beforeState,State afterState) {
List<State> states = new ArrayList<>();
for (Statement statement : block.getStatements()) {
states.add(new State(statement));
}
beforeState.nextStates.add(states.get(0));
for (int i = 0; i < states.size(); i++) {
State currentState = states.get(i);
State nextState;
if (i == states.size() - 1) {
nextState = afterState;
} else {
nextState = states.get(i + 1);
}
currentState.statement.accept(new VoidVisitorWithDefaults<Void>() {
@Override
public void visit(ExpressionStmt n,Void arg) {
currentState.nextStates.add(nextState);
}
@Override
public void visit(IfStmt n,Void arg) {
statements(n.getThenStmt().asBlockStmt(),currentState,nextState);
n.getElseStmt().ifPresent(elseStmt ->
statements(elseStmt.asBlockStmt(),nextState));
}
},null);
} }
我还想做的是确定当前语句是否是表达式语句(即赋值),如果是,则查找它是否是 If/Else 语句的 THEN 块中包含的最后一个语句。在上面的示例中,这将是语句 x=3;
。但是,我不确定如何实现这一点。
到目前为止,我的想法是这样的:
ExpressionStmt exprStmt = (ExpressionStmt) currentState.statement;
Node exprStmtParent = exprStmt.getParent();
List<Node> exprStmtSibilings = exprStmtParent.getChildNodes();
Node lastNode = exprStmtSiblings.get(exprStmtSiblings.size() - 1);
boolean exprStmtIsLast = lastNode == ifStmt;
适用于 if 语句但不适用于赋值。关于如何解决这个问题有什么建议吗?
解决方法
这个答案可能为时已晚,但这是您解决问题的方法。
如果 n.getThenStmt() 的结果是 BlockStmt,您可以对其调用 getStatements,返回一个 NodeList,然后在列表上调用 getLast。
使用equals方法比较节点更安全。
访问者需要调用 super.visit 方法来完全访问节点(表达式或语句)。
下面是一个简化的例子。
public void test() {
String str = "class X {\r\n" +
" void x() {\r\n" +
" int x = 1;\r\n" +
" if (x>4) {\r\n" +
" x=21;\r\n" +
" } else {\r\n" +
" if (x>1) {\r\n" +
" x=3;\r\n" +
" } else {\r\n" +
" x=212;\r\n" +
" } \r\n" +
" } \r\n" +
" if (x==4) { \r\n" +
" x=3;\r\n" +
" }\r\n" +
"}}";
CompilationUnit cu = StaticJavaParser.parse(str);
VoidVisitor<Void> visitor = new VoidVisitorAdapter<Void>() {
ExpressionStmt currentStmt;
@Override
public void visit(ExpressionStmt n,Void arg) {
currentStmt = n;
super.visit(n,arg);
}
@Override
public void visit(IfStmt n,Void arg) {
Statement then = n.getThenStmt();
if (then instanceof BlockStmt) {
BlockStmt blockThenStmt = then.asBlockStmt();
super.visit(blockThenStmt,arg); // <-- visit statements in thenStmt
Optional<Statement> oStmt = blockThenStmt.getStatements().getLast();
if (oStmt.isPresent() && oStmt.get().equals(currentStmt)) {
System.out.println(String.format("--> %s expression is the last of %s",currentStmt.toString(),n.toString()));
}
}
super.visit(n,arg); // <-- visit nested ifStmt
}
};
cu.accept(visitor,null);
}