FMWK-587 Return null for inapplicable URLs in driver connect

aerospike · Oct 30, 2024 · c4a0e7e · c4a0e7e
1 parent 5fb1b8c
commit c4a0e7e
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/src/main/java/com/aerospike/jdbc/AerospikeDriver.java b/src/main/java/com/aerospike/jdbc/AerospikeDriver.java
@@ -21,6 +21,7 @@
 public class AerospikeDriver implements Driver {
 
     private static final Logger logger = Logger.getLogger("com.aerospike.jdbc");
+    private static final String AEROSPIKE_URL_PREFIX = "jdbc:aerospike:";
 
     static {
         try {
@@ -33,12 +34,19 @@ public class AerospikeDriver implements Driver {
         Log.setCallback(asLoggerCallback);
     }
 
-    public Connection connect(String url, Properties info) {
+    public Connection connect(String url, Properties info) throws SQLException {
+        if (url == null) {
+            throw new SQLException("url is null");
+        }
+        if (!url.startsWith(AEROSPIKE_URL_PREFIX)) {
+            return null;
+        }
+
         return new AerospikeConnection(url, info);
     }
 
     public boolean acceptsURL(String url) {
-        return Objects.nonNull(url) && url.startsWith("jdbc:aerospike:");
+        return Objects.nonNull(url) && url.startsWith(AEROSPIKE_URL_PREFIX);
     }
 
     public DriverPropertyInfo[] getPropertyInfo(String url, Properties info) {

diff --git a/src/test/java/com/aerospike/jdbc/ParseJdbcUrlTest.java b/src/test/java/com/aerospike/jdbc/ParseJdbcUrlTest.java
@@ -11,6 +11,7 @@
 
 import static org.testng.Assert.assertEquals;
 import static org.testng.Assert.assertFalse;
+import static org.testng.Assert.assertNull;
 import static org.testng.Assert.assertTrue;
 
 public class ParseJdbcUrlTest {
@@ -70,6 +71,14 @@ public void testParseUrlParameters() throws Exception {
         assertEquals(config.getDriverPolicy().getRecordSetTimeoutMs(), 7000);
     }
 
+    @Test
+    public void testInapplicableUrl() throws Exception {
+        AerospikeDriver driver = (AerospikeDriver) Class.forName("com.aerospike.jdbc.AerospikeDriver")
+                .newInstance();
+        String url = "jdbc:postgresql://localhost:5432/sample";
+        assertNull(driver.connect(url, new Properties()));
+    }
+
     private void assertTotalTimeoutAll(IAerospikeClient client, int timeout) {
         assertEquals(client.getReadPolicyDefault().totalTimeout, timeout);
         assertEquals(client.getWritePolicyDefault().totalTimeout, timeout);