review

Author: trask

instrumentation/jdbc/library/src/main/java/io/opentelemetry/instrumentation/jdbc/internal/parser/PostgresqlUrlParser.java

@@ -8,6 +8,7 @@
 import static io.opentelemetry.instrumentation.jdbc.internal.parser.UrlParsingUtils.splitQuery;
 
 import java.util.Map;
+import java.util.Properties;
 
 /**
  * Parses PostgreSQL JDBC connection strings.
@@ -51,8 +52,10 @@ public void parse(String jdbcUrl, ParseContext ctx) {
     // Delegate to generic parser for standard URL parsing
     GenericUrlParser.INSTANCE.parse(jdbcUrl, ctx);
 
-    // Extract schema from currentSchema URL parameter for namespace formatting
     String schema = extractCurrentSchema(jdbcUrl);
+    if (schema == null) {
+      schema = extractCurrentSchema(ctx.props());
+    }
     if (schema == null && ctx.user() != null) {
       // Fall back to user as schema if no currentSchema param
       schema = ctx.user();
@@ -73,4 +76,12 @@ private static String extractCurrentSchema(String jdbcUrl) {
     Map<String, String> params = splitQuery(jdbcUrl.substring(queryIndex + 1), "&");
     return params.get("currentschema");
   }
+
+  private static String extractCurrentSchema(Properties props) {
+    if (props == null) {
+      return null;
+    }
+    String currentSchema = props.getProperty("currentSchema");
+    return currentSchema == null || currentSchema.isEmpty() ? null : currentSchema;
+  }
 }

instrumentation/jdbc/library/src/test/java/io/opentelemetry/instrumentation/jdbc/internal/JdbcConnectionUrlParserTest.java

@@ -44,6 +44,13 @@ private static Properties stdProps() {
     return prop;
   }
 
+  private static Properties postgresProps(String user, String currentSchema) {
+    Properties prop = new Properties();
+    prop.setProperty("user", user);
+    prop.setProperty("currentSchema", currentSchema);
+    return prop;
+  }
+
   @ParameterizedTest
   @ValueSource(strings = {"", "jdbc:", "jdbc::", "bogus:string"})
   void testInvalidUrlReturnsDefault(String url) {
@@ -273,6 +280,28 @@ private static Stream<Arguments> postgresArguments() {
             .setNamespace("pgdb|myschema")
             .setName("pgdb")
             .build(),
+        // currentSchema from connection properties is used when the URL does not specify it
+        arg("jdbc:postgresql://pg.host/pgdb")
+            .setProperties(postgresProps("pguser", "propertyschema"))
+            .setShortUrl("postgresql://pg.host:5432")
+            .setSystem("postgresql")
+            .setUser("pguser")
+            .setHost("pg.host")
+            .setPort(5432)
+            .setNamespace("pgdb|propertyschema")
+            .setName("pgdb")
+            .build(),
+        // currentSchema URL param takes precedence over currentSchema property
+        arg("jdbc:postgresql://pg.host/pgdb?currentSchema=urlschema")
+            .setProperties(postgresProps("pguser", "propertyschema"))
+            .setShortUrl("postgresql://pg.host:5432")
+            .setSystem("postgresql")
+            .setUser("pguser")
+            .setHost("pg.host")
+            .setPort(5432)
+            .setNamespace("pgdb|urlschema")
+            .setName("pgdb")
+            .build(),
         // database only, no schema or user — namespace falls back to database name
         arg("jdbc:postgresql://pg.host/pgdb")
             .setShortUrl("postgresql://pg.host:5432")