sander/exceptional
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

last nicer making

Browse source
This commit is contained in:
Shautvast 2024-06-23 20:45:00 +02:00
parent 862aad4462
commit 3414c3d510
1 changed files with 16 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
agent/src/main/java/com/github/shautvast/exceptional/Agent.java
View file

@ -12,7 +12,6 @@ import java.nio.file.Files;
import java.nio.file.Path;
import java.security.ProtectionDomain;
@SuppressWarnings("ALL")
public class Agent {
private static final String MESSAGE = "--- Exceptional agent active";
@ -62,20 +61,21 @@ public class Agent {
});
}
private static void instrumentMethodWithMyExceptionLogger(ClassBuilder classBuilder, MethodModel mm) {
classBuilder.withMethod(mm.methodName(), mm.methodType(), mm.flags().flagsMask(),
methodBuilder ->
methodBuilder.transformCode(mm.code().get(), ((builder, element) -> {
if (element instanceof ReturnInstruction) {
builder.aload(0); // load `this` on the stack, ie the Throwable instance
builder.invokestatic( // call my code
ClassDesc.of("com.github.shautvast.exceptional.ExceptionLogger"), "log",
MethodTypeDesc.ofDescriptor("(Ljava/lang/Throwable;)V"));
builder.return_();
} else {
builder.with(element);
}
})
));
private static void instrumentMethodWithMyExceptionLogger(ClassBuilder classBuilder, MethodModel methodModel) {
methodModel.code().ifPresent(code ->
classBuilder.withMethod(methodModel.methodName(), methodModel.methodType(), methodModel.flags().flagsMask(),
methodBuilder ->
methodBuilder.transformCode(code, ((builder, element) -> {
if (element instanceof ReturnInstruction) {
builder.aload(0); // load `this` on the stack, ie the Throwable instance
builder.invokestatic( // call my code
ClassDesc.of("com.github.shautvast.exceptional.ExceptionLogger"), "log",
MethodTypeDesc.ofDescriptor("(Ljava/lang/Throwable;)V"));
builder.return_();
} else {
builder.with(element); // leave any other bytecode in place
}
})
)));
}
}