Resolve "No suitable driver found" error in DB-based SPM

Fixes the runtime error:

    Caused by: ConnectionException: SQLException: No suitable driver found for jdbc:mysql://...

The issue occurs when both MariaDB and MySQL JDBC drivers are present on the classpath, causing conflicts in driver resolution. This fix ensures the correct driver is selected at runtime.

Author: imjalpreet

presto-db-session-property-manager/src/main/java/com/facebook/presto/session/db/DbSessionPropertyManagerConfig.java

@@ -23,6 +23,7 @@
 public class DbSessionPropertyManagerConfig
 {
     private String configDbUrl;
+    private String jdbcDriverName = "com.mysql.jdbc.Driver";
     private Duration specsRefreshPeriod = new Duration(10, SECONDS);
 
     @NotNull
@@ -38,6 +39,19 @@ public DbSessionPropertyManagerConfig setConfigDbUrl(String configDbUrl)
         return this;
     }
 
+    @NotNull
+    public String getJdbcDriverName()
+    {
+        return jdbcDriverName;
+    }
+
+    @Config("session-property-manager.db.driver-name")
+    public DbSessionPropertyManagerConfig setJdbcDriverName(String jdbcDriverName)
+    {
+        this.jdbcDriverName = jdbcDriverName;
+        return this;
+    }
+
     @NotNull
     @MinDuration("1ms")
     public Duration getSpecsRefreshPeriod()

presto-db-session-property-manager/src/main/java/com/facebook/presto/session/db/SessionPropertiesDaoProvider.java

@@ -33,6 +33,14 @@ public SessionPropertiesDaoProvider(DbSessionPropertyManagerConfig config)
     {
         requireNonNull(config, "config is null");
         requireNonNull(config.getConfigDbUrl(), "db url is null");
+
+        try {
+            Class.forName(config.getJdbcDriverName());
+        }
+        catch (ClassNotFoundException e) {
+            throw new RuntimeException("JDBC driver class not found: " + config.getJdbcDriverName(), e);
+        }
+
         this.dao = Jdbi.create(() -> DriverManager.getConnection(config.getConfigDbUrl()))
                 .installPlugin(new SqlObjectPlugin())
                 .onDemand(SessionPropertiesDao.class);

presto-db-session-property-manager/src/test/java/com/facebook/presto/session/db/TestDbSessionPropertyManagerConfig.java

@@ -32,6 +32,7 @@ public void testDefaults()
     {
         assertRecordedDefaults(recordDefaults(DbSessionPropertyManagerConfig.class)
                 .setConfigDbUrl(null)
+                .setJdbcDriverName("com.mysql.jdbc.Driver")
                 .setSpecsRefreshPeriod(new Duration(10, SECONDS)));
     }
 
@@ -40,11 +41,13 @@ public void testExplicitPropertyMappings()
     {
         Map<String, String> properties = new ImmutableMap.Builder<String, String>()
                 .put("session-property-manager.db.url", "foo")
+                .put("session-property-manager.db.driver-name", "org.mariadb.jdbc.Driver")
                 .put("session-property-manager.db.refresh-period", "50s")
                 .build();
 
         DbSessionPropertyManagerConfig expected = new DbSessionPropertyManagerConfig()
                 .setConfigDbUrl("foo")
+                .setJdbcDriverName("org.mariadb.jdbc.Driver")
                 .setSpecsRefreshPeriod(new Duration(50, TimeUnit.SECONDS));
 
         assertFullMapping(properties, expected);