content: /*
   * Copyright The OpenTelemetry Authors
   * SPDX-License-Identifier: Apache-2.0
   */
  
  package io.opentelemetry.javaagent.instrumentation.spring.integration.v4_1;
  
  import static io.opentelemetry.javaagent.extension.matcher.AgentElementMatchers.extendsClass;
  import static io.opentelemetry.javaagent.extension.matcher.AgentElementMatchers.hasClassesNamed;
  import static net.bytebuddy.matcher.ElementMatchers.isMethod;
  import static net.bytebuddy.matcher.ElementMatchers.named;
  import static net.bytebuddy.matcher.ElementMatchers.takesArgument;
  import static org.springframework.beans.factory.support.BeanDefinitionBuilder.genericBeanDefinition;
  
  import io.opentelemetry.javaagent.extension.instrumentation.TypeInstrumentation;
  import io.opentelemetry.javaagent.extension.instrumentation.TypeTransformer;
  import net.bytebuddy.asm.Advice;
  import net.bytebuddy.description.type.TypeDescription;
  import net.bytebuddy.matcher.ElementMatcher;
  import org.springframework.beans.factory.config.BeanDefinition;
  import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
  import org.springframework.beans.factory.support.BeanDefinitionRegistry;
  import org.springframework.integration.channel.interceptor.GlobalChannelInterceptorWrapper;
  
  public class ApplicationContextInstrumentation implements TypeInstrumentation {
    @Override
    public ElementMatcher<ClassLoader> classLoaderOptimization() {
      return hasClassesNamed("org.springframework.context.support.AbstractApplicationContext");
    }
  
    @Override
    public ElementMatcher<TypeDescription> typeMatcher() {
      return extendsClass(named("org.springframework.context.support.AbstractApplicationContext"));
    }
  
    @Override
    public void transform(TypeTransformer transformer) {
      transformer.applyAdviceToMethod(
          isMethod()
              .and(named("postProcessBeanFactory"))
              .and(
                  takesArgument(
                      0,
                      named(
                          "org.springframework.beans.factory.config.ConfigurableListableBeanFactory"))),
          ApplicationContextInstrumentation.class.getName() + "$PostProcessBeanFactoryAdvice");
    }
  
    @SuppressWarnings("unused")
    public static class PostProcessBeanFactoryAdvice {
      @Advice.OnMethodEnter(suppress = Throwable.class)
      public static void onEnter(@Advice.Argument(0) ConfigurableListableBeanFactory beanFactory) {
        if (beanFactory instanceof BeanDefinitionRegistry
            && !beanFactory.containsBean("otelGlobalChannelInterceptor")) {
  
          BeanDefinition globalChannelInterceptorBean =
              genericBeanDefinition(GlobalChannelInterceptorWrapper.class)
                  .addConstructorArgValue(SpringIntegrationSingletons.interceptor())
                  .addPropertyValue("patterns", SpringIntegrationSingletons.patterns())
                  // it is important for the tracing interceptor to run first for CONSUMER spans so
                  // that they capture the whole operation and also so that users can write their own
                  // interceptors to enrich the CONSUMER span (similar to writing a servlet filter to
                  // enrich the SERVER span)
                  .addPropertyValue("order", Integer.MIN_VALUE)
                  .getBeanDefinition();
  
          ((BeanDefinitionRegistry) beanFactory)
              .registerBeanDefinition("otelGlobalChannelInterceptor", globalChannelInterceptorBean);
        }
      }
    }
  }
  