Изменить класс java.util с помощью byte-buddy-agent - PullRequest
Новая
       60

Изменить класс java.util с помощью byte-buddy-agent

0 голосов
/ 18 июня 2019

Можно ли добавить поле в классе java.util с помощью byte-buddy?

Я пытаюсь добавить поле в java.util.concurrent.FutureTask и перехватить конструктор и произвольный метод для установкии получить значение поля.Короче говоря, я пытаюсь добавить поле в FutureTask, чтобы я мог передать некоторое значение дочернему потоку из родительского потока, даже если они выполняются в пуле потоков.не возможно ли добавить поле в FutureTask?

FutureTaskTransofrmer 

@Override
    protected ElementMatcher.Junction<TypeDescription> getNarrowTypesMatcher() {
        return named("java.util.concurrent.FutureTask");
    }

    @Override
    public AgentBuilder.Transformer getTransformer() {

        return (builder, typeDescription, classLoader, module) -> {
            beforeTransformation(typeDescription, classLoader);

            return builder
                    .defineField("pit", String.class)
                    .constructor(ElementMatchers.any())
                    .intercept(Advice.to(SetPitAdvice.class))
                    .method(named("run"))
                    .intercept(Advice.to(GetPitAdvice.class))
                    ;
        };

    }

GetPitAdvice 

    @Advice.OnMethodEnter
    public static void getValues(@Advice.FieldValue(value = "pit") String pit)
            throws Exception {

        logger.info("pit in future Task {}", pit);
    }

SetPitAdvice 

@Advice.OnMethodExit
    public static void setValues(
            @Advice.FieldValue(value = "pit", readOnly = false) String pit
    )
            throws Exception {

        logger.debug("Setting pit field in FutureTask");

        pit = SimulationRequestContextHolder.get("pit");

    }

AgentBuilder 

    private static AgentBuilder createAgentBuilder(AutoEvictingCachingBinaryLocator binaryLocator) {
        final ByteBuddy byteBuddy = new ByteBuddy()
                .with(TypeValidation.of(AgentProperties.PROPERTIES.isDebugInstrumentation()))
                .with(MethodGraph.Compiler.ForDeclaredMethods.INSTANCE);

        return new AgentBuilder.Default(byteBuddy)
                .with(AgentBuilder.RedefinitionStrategy.RETRANSFORMATION)
                .with(getListener())
                .with(binaryLocator)
                .ignore(any(), ClassLoaderNameMatcher.isReflectionClassLoader())
                .or(any(), ClassLoaderNameMatcher.classLoaderWithName("org.codehaus.groovy.runtime.callsite.CallSiteClassLoader"))
                .or(any(), new IsIgnoredClassLoaderElementMatcher())
                .or(nameStartsWith("org.aspectj.")
                        .or(nameStartsWith("org.groovy."))
                        .or(nameStartsWith("com.sun."))
                        .or(nameStartsWith("com.p6spy."))
                        .or(nameStartsWith("net.bytebuddy."))
                        .or(nameStartsWith("org.slf4j.").and(not(nameStartsWith("org.slf4j.impl."))))
                        .or(nameContains("javassist"))
                        .or(nameContains(".asm."))
                        .or(nameStartsWith("com.java.agent.sims")
                        ))
//                .disableClassFormatChanges()
                .enableUnsafeBootstrapInjection()
                ;
    }

onTransformListener показывает, что класс преобразован 

11:27:24.141 [main] INFO com.java.agent.sims.ApplicationClassLoaderMatcher - Instrumenting ClassLoader null: true
11:27:24.186 [main] DEBUG com.java.agent.sims.transformers.ByteBuddyTransformer - TRANSFORM java.util.concurrent.FutureTask (FutureTaskTransformer)
11:27:24.466 [main] INFO com.java.agent.sims.instrument.TransformListener - Class modified by Byte Buddy: java.util.concurrent.FutureTask

, но ни один из моих перехватов советов не вызывается.

1 Ответ

1 голос
/ 19 июня 2019

Если класс уже загружен, это невозможно. Вместо этого вы можете внедрить класс в загрузчик классов начальной загрузки, используя API Instrumentation, и в этом классе хранится статическая карта со слабыми ключами, в которую вы помещаете значение поля каждого экземпляра.

Также обратите внимание, что классы с до загружаются загрузчиком классов начальной загрузки, который не может видеть никаких классов, содержащихся в вашем агенте, который загружен загрузчиком системных классов. По этой причине любой API, используемый по совету, должен быть введен в загрузчик начальной загрузки.

...