Error Management AbstractDBAction

Author: psainics

database-commons/src/main/java/io/cdap/plugin/db/action/AbstractDBAction.java

@@ -16,6 +16,7 @@
 
 package io.cdap.plugin.db.action;
 
+import io.cdap.cdap.etl.api.FailureCollector;
 import io.cdap.cdap.etl.api.PipelineConfigurer;
 import io.cdap.cdap.etl.api.action.Action;
 import io.cdap.cdap.etl.api.action.ActionContext;
@@ -40,7 +41,13 @@ public AbstractDBAction(QueryConfig config, Boolean enableAutoCommit) {
   public void run(ActionContext context) throws Exception {
     Class<? extends Driver> driverClass = context.loadPluginClass(JDBC_PLUGIN_ID);
     DBRun executeQuery = new DBRun(config, driverClass, enableAutoCommit);
-    executeQuery.run();
+    try {
+      executeQuery.run();
+    } catch (Exception e) {
+      FailureCollector collector = context.getFailureCollector();
+      collector.addFailure("Unable to execute query.", e.getMessage()).withStacktrace(e.getStackTrace());
+      collector.getOrThrowException();
+    }
   }
 
   @Override