Merge pull request #15703 from cdapio/CDAP-21059-error-management-and…

…-classification [CDAP-21059] Add support to return stageName in stageConfigurer
cdapio · Sep 16, 2024 · 7292c8c · 7292c8c
2 parents 0600222 + 0b97ce6
commit 7292c8c
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 0 deletions.
diff --git a/...p-templates/cdap-etl/cdap-etl-api/src/main/java/io/cdap/cdap/etl/api/StageConfigurer.java b/...p-templates/cdap-etl/cdap-etl-api/src/main/java/io/cdap/cdap/etl/api/StageConfigurer.java
@@ -35,6 +35,13 @@ public interface StageConfigurer {
   @Nullable
   Schema getInputSchema();
 
+  /**
+   * Returns the name of the stage.
+   *
+   * @return the name of the stage
+   */
+  String getStageName();
+
   /**
    * set the output schema for this stage, or null if its unknown
    *

diff --git a/.../cdap-etl/cdap-etl-core/src/main/java/io/cdap/cdap/etl/common/DefaultStageConfigurer.java b/.../cdap-etl/cdap-etl-core/src/main/java/io/cdap/cdap/etl/common/DefaultStageConfigurer.java
@@ -95,6 +95,11 @@ public void setErrorSchema(@Nullable Schema errorSchema) {
     errorSchemaSet = true;
   }
 
+  @Override
+  public String getStageName() {
+    return stageName;
+  }
+
   @Override
   public FailureCollector getFailureCollector() {
     if (collector == null) {

diff --git a/...-etl/hydrator-test/src/main/java/io/cdap/cdap/etl/mock/common/MockPipelineConfigurer.java b/...-etl/hydrator-test/src/main/java/io/cdap/cdap/etl/mock/common/MockPipelineConfigurer.java
@@ -86,6 +86,11 @@ public void setErrorSchema(@Nullable Schema errorSchema) {
         // no-op
       }
 
+      @Override
+      public String getStageName() {
+        return null;
+      }
+
       public FailureCollector getFailureCollector() {
         return collector;
       }