Wrap SQLException in Abstract DBSource

Author: psainics

database-commons/src/main/java/io/cdap/plugin/db/source/AbstractDBSource.java

@@ -25,6 +25,9 @@
 import io.cdap.cdap.api.data.format.StructuredRecord;
 import io.cdap.cdap.api.data.schema.Schema;
 import io.cdap.cdap.api.dataset.lib.KeyValue;
+import io.cdap.cdap.api.exception.ErrorCategory;
+import io.cdap.cdap.api.exception.ErrorType;
+import io.cdap.cdap.api.exception.ErrorUtils;
 import io.cdap.cdap.api.plugin.PluginConfig;
 import io.cdap.cdap.etl.api.Emitter;
 import io.cdap.cdap.etl.api.FailureCollector;
@@ -196,7 +199,11 @@ private Schema loadSchemaFromDB(Class<? extends Driver> driverClass)
 
     } catch (SQLException e) {
       // wrap exception to ensure SQLException-child instances not exposed to contexts without jdbc driver in classpath
-      throw new SQLException(e.getMessage(), e.getSQLState(), e.getErrorCode());
+      String errorMessageWithDetails = String.format("Error occurred while trying to get schema from database." +
+        "Error message: '%s'. Error code: '%s'. SQLState: '%s'", e.getMessage(), e.getErrorCode(), e.getSQLState());
+      throw ErrorUtils.getProgramFailureException(new ErrorCategory(ErrorCategory.ErrorCategoryEnum.PLUGIN),
+        e.getMessage(), errorMessageWithDetails, ErrorType.USER, false, new SQLException(e.getMessage(),
+          e.getSQLState(), e.getErrorCode()));
     } finally {
       driverCleanup.destroy();
     }