Call class loading callbacks from onComplete
This should make things more clear
@@ -270,13 +270,7 @@ public class AgentInstaller {
final String typeName,
final ClassLoader classLoader,
final JavaModule javaModule,
- final boolean b) {
- for (final Map.Entry<String, Runnable> entry : classLoadCallbacks.entrySet()) {
- if (entry.getKey().equals(typeName)) {
- entry.getValue().run();
- }
- }
- }
+ final boolean b) {}
@Override
public void onTransformation(
@@ -303,10 +297,16 @@ public class AgentInstaller {
@Override
public void onComplete(
- final String s,
+ final String typeName,
final ClassLoader classLoader,
final JavaModule javaModule,
- final boolean b) {}
+ final boolean b) {
+ for (final Map.Entry<String, Runnable> entry : classLoadCallbacks.entrySet()) {
+ if (entry.getKey().equals(typeName)) {
+ entry.getValue().run();
+ }
+ }
+ }
}
private AgentInstaller() {}
