[BEAM-12384] Infer typeDescriptor from coder if typeDescriptor is not…

… set explicitly.
apache · May 26, 2021 · 83bccf9 · 83bccf9
1 parent 3b705c1
commit 83bccf9
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/sdks/java/core/src/main/java/org/apache/beam/sdk/values/PCollection.java b/sdks/java/core/src/main/java/org/apache/beam/sdk/values/PCollection.java
@@ -124,7 +124,13 @@ public void finishSpecifying(PInput input, PTransform<?, ?> transform) {
    * override this to enable better {@code Coder} inference.
    */
   public @Nullable TypeDescriptor<T> getTypeDescriptor() {
-    return typeDescriptor;
+    if (typeDescriptor != null) {
+      return typeDescriptor;
+    }
+    if (coderOrFailure.coder != null) {
+      return coderOrFailure.coder.getEncodedTypeDescriptor();
+    }
+    return null;
   }
 
   /**