Merge pull request #714 from data-integrations/cherrypick-21024

[🍒] CDAP-21024: Return null when there is no outputfield
data-integrations · May 7, 2024 · 5711f91 · 5711f91
2 parents 4c8fafc + b9b0202
commit 5711f91
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/wrangler-core/src/main/java/io/cdap/wrangler/schema/DirectiveOutputSchemaGenerator.java b/wrangler-core/src/main/java/io/cdap/wrangler/schema/DirectiveOutputSchemaGenerator.java
@@ -103,6 +103,9 @@ private Schema generateDirectiveOutputSchema(Schema inputSchema)
         outputFields.add(Schema.Field.of(fieldName, Schema.of(Schema.Type.NULL)));
       }
     }
+    if (outputFields.isEmpty()) {
+      return null;
+    }
     return Schema.recordOf("output", outputFields);
   }