content: /*
   * Copyright The OpenTelemetry Authors
   * SPDX-License-Identifier: Apache-2.0
   */
  
  package io.opentelemetry.javaagent.instrumentation.awssdk.v2_2;
  
  import static io.opentelemetry.javaagent.extension.matcher.AgentElementMatchers.hasClassesNamed;
  import static net.bytebuddy.matcher.ElementMatchers.none;
  
  import com.google.auto.service.AutoService;
  import io.opentelemetry.instrumentation.awssdk.v2_2.internal.SqsAdviceBridge;
  import io.opentelemetry.javaagent.extension.instrumentation.InstrumentationModule;
  import io.opentelemetry.javaagent.extension.instrumentation.TypeInstrumentation;
  import io.opentelemetry.javaagent.extension.instrumentation.TypeTransformer;
  import java.util.ArrayList;
  import java.util.List;
  import net.bytebuddy.asm.Advice;
  import net.bytebuddy.matcher.ElementMatcher;
  
  @AutoService(InstrumentationModule.class)
  public class SqsInstrumentationModule extends AbstractAwsSdkInstrumentationModule {
  
    public SqsInstrumentationModule() {
      super("aws-sdk-2.2-sqs");
    }
  
    @Override
    public ElementMatcher.Junction<ClassLoader> classLoaderMatcher() {
      return hasClassesNamed("software.amazon.awssdk.services.sqs.SqsClient");
    }
  
    @Override
    public List<TypeInstrumentation> typeInstrumentations() {
      List<TypeInstrumentation> instrumentations = new ArrayList<>(super.typeInstrumentations());
      instrumentations.add(new DefaultSqsClientBuilderInstrumentation());
      instrumentations.add(new DefaultSqsAsyncClientBuilderInstrumentation());
      return instrumentations;
    }
  
    @Override
    public void doTransform(TypeTransformer transformer) {
      transformer.applyAdviceToMethod(
          none(), SqsInstrumentationModule.class.getName() + "$RegisterAdvice");
    }
  
    @SuppressWarnings("unused")
    public static class RegisterAdvice {
      @Advice.OnMethodExit(suppress = Throwable.class)
      public static void onExit() {
        // (indirectly) using SqsImpl class here to make sure it is available from SqsAccess
        // (injected into app classloader) and checked by Muzzle
        SqsAdviceBridge.referenceForMuzzleOnly();
      }
    }
  }
  