Add support to return stageName in stageConfigurer

cdapio · Sep 16, 2024 · 0b97ce6 · 0b97ce6
1 parent 3c6b832
commit 0b97ce6
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;
       }