Change DataSource wrapping from BeanPostProcessor to explicit @primary bean registration

Author: dngyukm

instrumentation/spring/spring-boot-autoconfigure/src/main/java/io/opentelemetry/instrumentation/spring/autoconfigure/internal/instrumentation/jdbc/DataSourcePostProcessor.java

@@ -0,0 +1,52 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.instrumentation.spring.autoconfigure.internal.instrumentation.jdbc;
+
+import io.opentelemetry.api.OpenTelemetry;
+import io.opentelemetry.instrumentation.jdbc.datasource.JdbcTelemetry;
+import io.opentelemetry.instrumentation.spring.autoconfigure.internal.properties.InstrumentationConfigUtil;
+import io.opentelemetry.sdk.autoconfigure.spi.ConfigProperties;
+import javax.sql.DataSource;
+
+final class JdbcDataSourceWrapper {
+  private JdbcDataSourceWrapper() {}
+
+  private static final Class<?> ROUTING_DATA_SOURCE_CLASS = getRoutingDataSourceClass();
+
+  private static Class<?> getRoutingDataSourceClass() {
+    try {
+      return Class.forName("org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource");
+    } catch (ClassNotFoundException exception) {
+      return null;
+    }
+  }
+
+  static boolean isRoutingDatasource(Object bean) {
+    return ROUTING_DATA_SOURCE_CLASS != null && ROUTING_DATA_SOURCE_CLASS.isInstance(bean);
+  }
+
+  static DataSource wrapIfNecessary(
+      DataSource dataSource,
+      OpenTelemetry openTelemetry,
+      ConfigProperties configProperties) {
+    if (!isRoutingDatasource(dataSource)) {
+      return JdbcTelemetry.builder(openTelemetry)
+          .setStatementSanitizationEnabled(
+              InstrumentationConfigUtil.isStatementSanitizationEnabled(
+                  configProperties,
+                  "otel.instrumentation.jdbc.statement-sanitizer.enabled"))
+          .setCaptureQueryParameters(
+              configProperties.getBoolean(
+                  "otel.instrumentation.jdbc.experimental.capture-query-parameters", false))
+          .setTransactionInstrumenterEnabled(
+              configProperties.getBoolean(
+                  "otel.instrumentation.jdbc.experimental.transaction.enabled", false))
+          .build()
+          .wrap(dataSource);
+    }
+    return dataSource;
+  }
+} 

instrumentation/spring/spring-boot-autoconfigure/src/main/java/io/opentelemetry/instrumentation/spring/autoconfigure/internal/instrumentation/jdbc/JdbcDataSourceWrapper.java

@@ -10,11 +10,13 @@
 import io.opentelemetry.sdk.autoconfigure.spi.ConfigProperties;
 import javax.sql.DataSource;
 import org.springframework.beans.factory.ObjectProvider;
+import org.springframework.beans.factory.annotation.Qualifier;
 import org.springframework.boot.autoconfigure.AutoConfiguration;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnBean;
 import org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfiguration;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
+import org.springframework.context.annotation.Primary;
 
 /**
  * This class is internal and is hence not for public use. Its APIs are unstable and can change at
@@ -30,10 +32,14 @@ public class JdbcInstrumentationAutoConfiguration {
   public JdbcInstrumentationAutoConfiguration() {}
 
   @Bean
-  // static to avoid "is not eligible for getting processed by all BeanPostProcessors" warning
-  static DataSourcePostProcessor dataSourcePostProcessor(
+  @Primary
+  public DataSource otelDataSource(
+      @Qualifier("dataSource") DataSource originalDataSource,
       ObjectProvider<OpenTelemetry> openTelemetryProvider,
       ObjectProvider<ConfigProperties> configPropertiesProvider) {
-    return new DataSourcePostProcessor(openTelemetryProvider, configPropertiesProvider);
+    return JdbcDataSourceWrapper.wrapIfNecessary(
+        originalDataSource,
+        openTelemetryProvider.getObject(),
+        configPropertiesProvider.getObject());
   }
 }