CDAP-21024: Return null when there is no outputfield

data-integrations · May 6, 2024 · f4b8805 · f4b8805
1 parent 564f734
commit f4b8805
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);
   }