Improve logging when jdbc is shaded (#1093)

## Description
<!-- Provide a brief summary of the changes made and the issue they aim
to address.-->
Improve logging when jdbc is shaded
## Testing
<!-- Describe how the changes have been tested-->
Unit tests + manually in benchmarking
## Additional Notes to the Reviewer
<!-- Share any additional context or insights that may help the reviewer
understand the changes better. This could include challenges faced,
limitations, or compromises made during the development process.
Also, mention any areas of the code that you would like the reviewer to
focus on specifically. -->

Signed-off-by: Vikrant Puppala <vikrant.puppala@databricks.com>

Author: vikrantpuppala

NEXT_CHANGELOG.md

@@ -7,6 +7,6 @@
 ### Updated
 
 ### Fixed
-
+- Fixed logging to respect params when the driver is shaded.
 ---
 *Note: When making changes, please add your change under the appropriate section with a brief description.*

src/main/java/com/databricks/jdbc/log/JulLogger.java

@@ -233,6 +233,19 @@ private static String getPackagePrefix() {
     if (prefix != null && !prefix.isEmpty()) {
       return prefix;
     }
+    // Auto-detect the actual package prefix by using the current class
+    // This handles shaded JARs where the package might be relocated
+    String actualPackageName = JulLogger.class.getPackage().getName();
+    // Extract the root prefix (e.g., "jdbc.shaded.v1.0.12.OSS.com.databricks.jdbc"
+    // or "com.databricks.jdbc")
+    if (actualPackageName.contains(DEFAULT_PACKAGE_PREFIX)) {
+      // Find the actual prefix including any shading prefix
+      int defaultPrefixIndex = actualPackageName.indexOf(DEFAULT_PACKAGE_PREFIX);
+      if (defaultPrefixIndex > 0) {
+        // Include everything up to and including the default prefix
+        return actualPackageName.substring(0, defaultPrefixIndex + DEFAULT_PACKAGE_PREFIX.length());
+      }
+    }
     return DEFAULT_PACKAGE_PREFIX;
   }
 
@@ -241,6 +254,17 @@ private static String getDriverPackagePrefix() {
     if (StringUtils.isNotEmpty(prefix)) {
       return prefix;
     }
+    // Auto-detect the actual driver package prefix
+    // Try to determine if the driver package is also shaded
+    String mainPackagePrefix = getPackagePrefix();
+    // Check if we have a shaded prefix
+    if (!mainPackagePrefix.equals(DEFAULT_PACKAGE_PREFIX)
+        && mainPackagePrefix.contains(DEFAULT_PACKAGE_PREFIX)) {
+      // Extract the shading prefix and apply it to the driver package
+      String shadingPrefix =
+          mainPackagePrefix.substring(0, mainPackagePrefix.indexOf(DEFAULT_PACKAGE_PREFIX));
+      return shadingPrefix + DEFAULT_DRIVER_PACKAGE_PREFIX;
+    }
     return DEFAULT_DRIVER_PACKAGE_PREFIX;
   }
 

src/test/java/com/databricks/jdbc/log/JulLoggerTest.java

@@ -183,4 +183,74 @@ private String[] methodCallingLogger() {
   private String[] info() {
     return JulLogger.getCaller();
   }
+
+  @Test
+  void testPackagePrefixDetectionForUnshadedJar() {
+    // Test that package prefix detection works for unshaded JARs
+    String packagePrefix = JulLogger.PARENT_CLASS_PREFIX;
+    // For unshaded JARs, it should be the default prefix
+    assertTrue(
+        packagePrefix.endsWith("com.databricks.jdbc"),
+        "Package prefix should end with com.databricks.jdbc, got: " + packagePrefix);
+  }
+
+  @Test
+  void testPackagePrefixDetectionForShadedJar() {
+    // Test that the package prefix includes any shading prefix
+    // The JulLogger should be able to detect its own package correctly
+    String actualPackageName = JulLogger.class.getPackage().getName();
+    String packagePrefix = JulLogger.PARENT_CLASS_PREFIX;
+
+    // Verify that the package prefix matches the actual runtime package structure
+    assertTrue(
+        actualPackageName.startsWith(packagePrefix),
+        String.format(
+            "Actual package '%s' should start with detected prefix '%s'",
+            actualPackageName, packagePrefix));
+
+    // If the jar is shaded, the prefix should include the shading prefix
+    if (actualPackageName.contains(".") && !actualPackageName.startsWith("com.databricks.jdbc")) {
+      // This means we have a shaded jar
+      assertTrue(
+          packagePrefix.contains("com.databricks.jdbc"),
+          "Shaded package prefix should still contain com.databricks.jdbc");
+    }
+  }
+
+  @Test
+  void testInitLoggerConfiguresCorrectPackageForShadedJar() throws IOException {
+    // This test verifies that when initLogger is called, it configures loggers
+    // for the correct package prefix (whether shaded or unshaded)
+    JulLogger.initLogger(Level.OFF, JulLogger.STDOUT, 1024, 1);
+
+    // Get the configured logger
+    Logger jdbcLogger = Logger.getLogger(JulLogger.PARENT_CLASS_PREFIX);
+
+    // Verify it was configured with the correct level
+    assertEquals(
+        Level.OFF,
+        jdbcLogger.getLevel(),
+        "Logger should be configured with Level.OFF to suppress all logs");
+
+    // Verify that a logger created with the actual package name inherits the settings
+    String testLoggerName =
+        JulLogger.class.getPackage().getName() + ".api.impl.ExecutionResultFactory";
+    Logger testLogger = Logger.getLogger(testLoggerName);
+
+    // The effective level should be OFF (inherited from parent)
+    Level effectiveLevel = testLogger.getLevel();
+    // If the local level is null, it inherits from parent
+    if (effectiveLevel == null) {
+      Logger parent = testLogger.getParent();
+      while (parent != null && effectiveLevel == null) {
+        effectiveLevel = parent.getLevel();
+        parent = parent.getParent();
+      }
+    }
+
+    assertEquals(
+        Level.OFF,
+        effectiveLevel,
+        "Child loggers should inherit Level.OFF from properly configured parent logger");
+  }
 }