From df364d6da821d44c87acaceac13acc3461d7f6fe Mon Sep 17 00:00:00 2001
From: Thomas Groh <tgroh@google.com>
Date: Thu, 30 Mar 2017 17:54:29 -0700
Subject: [PATCH] Improve PTransformMatcher ToStrings

---
 .../core/construction/PTransformMatchers.java | 45 +++++++++++++++++++
 1 file changed, 45 insertions(+)

diff --git a/runners/core-construction-java/src/main/java/org/apache/beam/runners/core/construction/PTransformMatchers.java b/runners/core-construction-java/src/main/java/org/apache/beam/runners/core/construction/PTransformMatchers.java
index d5a91a79ce16..f4ae5779af99 100644
--- a/runners/core-construction-java/src/main/java/org/apache/beam/runners/core/construction/PTransformMatchers.java
+++ b/runners/core-construction-java/src/main/java/org/apache/beam/runners/core/construction/PTransformMatchers.java
@@ -17,6 +17,7 @@
  */
 package org.apache.beam.runners.core.construction;
 
+import com.google.common.base.MoreObjects;
 import java.util.HashSet;
 import java.util.Set;
 import org.apache.beam.sdk.annotations.Experimental;
@@ -67,6 +68,13 @@ private EqualClassPTransformMatcher(Class<? extends PTransform> clazz) {
     public boolean matches(AppliedPTransform<?, ?, ?> application) {
       return application.getTransform().getClass().equals(clazz);
     }
+
+    @Override
+    public String toString() {
+      return MoreObjects.toStringHelper(EqualClassPTransformMatcher.class)
+          .add("class", clazz)
+          .toString();
+    }
   }
 
   /**
@@ -85,6 +93,11 @@ public boolean matches(AppliedPTransform<?, ?, ?> application) {
         }
         return false;
       }
+
+      @Override
+      public String toString() {
+        return MoreObjects.toStringHelper("SplittableParDoSingleMatcher").toString();
+      }
     };
   }
 
@@ -105,6 +118,11 @@ public boolean matches(AppliedPTransform<?, ?, ?> application) {
         }
         return false;
       }
+
+      @Override
+      public String toString() {
+        return MoreObjects.toStringHelper("StateOrTimerParDoSingleMatcher").toString();
+      }
     };
   }
 
@@ -124,6 +142,11 @@ public boolean matches(AppliedPTransform<?, ?, ?> application) {
         }
         return false;
       }
+
+      @Override
+      public String toString() {
+        return MoreObjects.toStringHelper("SplittableParDoMultiMatcher").toString();
+      }
     };
   }
 
@@ -144,6 +167,11 @@ public boolean matches(AppliedPTransform<?, ?, ?> application) {
         }
         return false;
       }
+
+      @Override
+      public String toString() {
+        return MoreObjects.toStringHelper("StateOrTimerParDoMultiMatcher").toString();
+      }
     };
   }
 
@@ -165,6 +193,13 @@ public boolean matches(AppliedPTransform<?, ?, ?> application) {
         }
         return fnType.equals(fn.getClass());
       }
+
+      @Override
+      public String toString() {
+        return MoreObjects.toStringHelper("ParDoWithFnTypeMatcher")
+            .add("fnType", fnType)
+            .toString();
+      }
     };
   }
 
@@ -179,6 +214,11 @@ public boolean matches(AppliedPTransform<?, ?, ?> application) {
         return (application.getTransform() instanceof Flatten.PCollections)
             && application.getInputs().isEmpty();
       }
+
+      @Override
+      public String toString() {
+        return MoreObjects.toStringHelper("EmptyFlattenMatcher").toString();
+      }
     };
   }
 
@@ -201,6 +241,11 @@ public boolean matches(AppliedPTransform<?, ?, ?> application) {
         }
         return false;
       }
+
+      @Override
+      public String toString() {
+        return MoreObjects.toStringHelper("FlattenWithDuplicateInputsMatcher").toString();
+      }
     };
   }