...我希望 API 直接修改 java 文件/创建它的修改副本。有没有办法做到这一点?
JavaParser 是一个 API,它允许您读取 Java 文件、修改它们并以字符串的形式获取结果。
更具体地说,JavaParser 解析文件并构建 AST(抽象语法树)。然后,您可以使用 API 修改表示您的源代码的 JavaParser AST 并检索 AST 的字符串表示。
我已经有一个 .java 文件,想在其中的方法中添加几行代码。
这是一个使用 JavaParser 在方法体末尾添加一行并打印结果的示例:
import com.github.javaparser.ast.CompilationUnit;
import com.github.javaparser.ast.body.ClassOrInterfaceDeclaration;
import com.github.javaparser.ast.body.MethodDeclaration;
import java.io.File;
import java.io.FileNotFoundException;
import java.util.Optional;
public class Main {
public static void someMethod() {
// Lines will be added here.
}
public static void main( String[] args ) throws FileNotFoundException {
String newStatement = "System.out.println(\"Hello world!\");";
File myClassSourceFile = new File( "Main.java" );
JavaParser parser = new JavaParser();
ParseResult<CompilationUnit> pr = parser.parse( myClassSourceFile );
Optional<CompilationUnit> ocu = pr.getResult();
if( ocu.isPresent() ) {
CompilationUnit cu = ocu.get();
ClassOrInterfaceDeclaration decl = cu.getClassByName( "Main" ).get();
MethodDeclaration method = decl.getMethods().get( 0 );
method.getBody().ifPresent( ( b ) -> b.addStatement( newStatement ) );
}
// Print out the resulting Java source code.
System.out.println( pr.toString() );
}
}
CompilationUnit - 来自 JavaParser 的 javadoc,“这个类代表整个编译单元。每个 java 文件表示一个编译单元。”
在您的代码中,将Option.get() 调用替换为适当的处理方式。
将方法日志记录添加到命令行中给出的类名的示例:
public class Main {
public static void main( final String[] args ) throws FileNotFoundException {
final File sourceFile = new File( args[ 0 ] );
final JavaParser parser = new JavaParser();
final ParseResult<CompilationUnit> pr = parser.parse( sourceFile );
final Optional<CompilationUnit> ocu = pr.getResult();
if( ocu.isPresent() ) {
final CompilationUnit cu = ocu.get();
final List<TypeDeclaration<?>> types = cu.getTypes();
for( final TypeDeclaration<?> type : types ) {
final List<MethodDeclaration> methods = type.getMethods();
for( final MethodDeclaration method : methods ) {
final Optional<BlockStmt> body = method.getBody();
final String m = format( "%s::%s( %s )",
type.getNameAsString(),
method.getNameAsString(),
method.getParameters().toString() );
final String mBegan = format(
"System.out.println(\"BEGAN %s\");", m );
final String mEnded = format(
"System.out.println(\"ENDED %s\");", m );
final Statement sBegan = parseStatement( mBegan );
final Statement sEnded = parseStatement( mEnded );
body.ifPresent( ( b ) -> {
final int i = b.getStatements().size();
b.addStatement( 0, sBegan );
// Insert before any "return" statement.
b.addStatement( i, sEnded );
} );
}
System.out.println( cu.toString() );
}
}
}
}
这会将更改的源文件写入标准输出。如果将Main 文件放在core 项目的主包中,则可以构建core 项目的JAR 文件(例如mvn package)。将 JAR 文件重命名为 javaparser.jar,然后对所有 JAR 文件运行 Main:
for i in $(find . -type f -name "*.java"); do \
java -cp javaparser.jar com.github.javaparser.Main "$i" > \
"$i.jp";
done
当然,让 Java 遍历目录树会更有效率。一旦.jp 文件出现并且看起来没问题,您可以使用以下命令对它们进行整体重命名:
find . -type f -name "*jp" -size +100c -exec \
sh -c 'mv {} $(dirname {})/$(basename {} .jp)' \;
这将破坏原始格式,使其非常不适合检入存储库。某些 Java 14 语句可能无法转换为可编译的文件。 YMMV。