exception handling for jdbc

Author: Trianz-Akshay

athena-jdbc/src/main/java/com/amazonaws/athena/connectors/jdbc/connection/GenericJdbcConnectionFactory.java

@@ -29,6 +29,7 @@
 import java.nio.charset.StandardCharsets;
 import java.sql.Connection;
 import java.sql.DriverManager;
+import java.sql.SQLException;
 import java.util.Map;
 import java.util.Properties;
 import java.util.regex.Matcher;
@@ -88,7 +89,19 @@ public Connection getConnection(final CredentialsProvider credentialsProvider)
         Class.forName(databaseConnectionInfo.getDriverClassName()).newInstance();
 
         // create connection
-        return DriverManager.getConnection(derivedJdbcString, this.jdbcProperties);
+        Connection connection = null;
+        try {
+            connection = DriverManager.getConnection(derivedJdbcString, this.jdbcProperties);
+        }
+        catch (SQLException e) {
+            if (e.getMessage().contains("Name or service not known")) {
+                throw new AthenaConnectorException(e.getMessage(), new ErrorDetails().withErrorCode(FederationSourceErrorCode.InvalidInputException.toString()));
+            }
+            else if (e.getMessage().contains("Incorrect username or password was specified.")) {
+                throw new AthenaConnectorException(e.getMessage(), new ErrorDetails().withErrorCode(FederationSourceErrorCode.InvalidCredentialsException.toString()));
+            }
+        }
+        return connection;
     }
 
     private String encodeValue(String value)

athena-jdbc/src/main/java/com/amazonaws/athena/connectors/jdbc/manager/JdbcMetadataHandler.java

@@ -168,6 +168,9 @@ private Set<String> listDatabaseNames(final Connection jdbcConnection)
             }
             return schemaNames.build();
         }
+        catch (RuntimeException ex) {
+            throw new AthenaConnectorException("Invalid credentials was specified", new ErrorDetails().withErrorCode(FederationSourceErrorCode.InvalidCredentialsException.toString()));
+        }
     }
 
     @Override