Improve PTransformMatcher ToStrings
Project: http://git-wip-us.apache.org/repos/asf/beam/repo Commit: http://git-wip-us.apache.org/repos/asf/beam/commit/054fbadd Tree: http://git-wip-us.apache.org/repos/asf/beam/tree/054fbadd Diff: http://git-wip-us.apache.org/repos/asf/beam/diff/054fbadd Branch: refs/heads/master Commit: 054fbadd1ad75bde1162e4b1cad4e59e8c1e9e93 Parents: 813839a Author: Thomas Groh <[email protected]> Authored: Thu Mar 30 17:54:29 2017 -0700 Committer: Thomas Groh <[email protected]> Committed: Fri Mar 31 09:38:11 2017 -0700 ---------------------------------------------------------------------- .../core/construction/PTransformMatchers.java | 45 ++++++++++++++++++++ 1 file changed, 45 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/beam/blob/054fbadd/runners/core-construction-java/src/main/java/org/apache/beam/runners/core/construction/PTransformMatchers.java ---------------------------------------------------------------------- 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 d5a91a7..f4ae577 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 @@ public class PTransformMatchers { 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 class PTransformMatchers { } return false; } + + @Override + public String toString() { + return MoreObjects.toStringHelper("SplittableParDoSingleMatcher").toString(); + } }; } @@ -105,6 +118,11 @@ public class PTransformMatchers { } return false; } + + @Override + public String toString() { + return MoreObjects.toStringHelper("StateOrTimerParDoSingleMatcher").toString(); + } }; } @@ -124,6 +142,11 @@ public class PTransformMatchers { } return false; } + + @Override + public String toString() { + return MoreObjects.toStringHelper("SplittableParDoMultiMatcher").toString(); + } }; } @@ -144,6 +167,11 @@ public class PTransformMatchers { } return false; } + + @Override + public String toString() { + return MoreObjects.toStringHelper("StateOrTimerParDoMultiMatcher").toString(); + } }; } @@ -165,6 +193,13 @@ public class PTransformMatchers { } return fnType.equals(fn.getClass()); } + + @Override + public String toString() { + return MoreObjects.toStringHelper("ParDoWithFnTypeMatcher") + .add("fnType", fnType) + .toString(); + } }; } @@ -179,6 +214,11 @@ public class PTransformMatchers { return (application.getTransform() instanceof Flatten.PCollections) && application.getInputs().isEmpty(); } + + @Override + public String toString() { + return MoreObjects.toStringHelper("EmptyFlattenMatcher").toString(); + } }; } @@ -201,6 +241,11 @@ public class PTransformMatchers { } return false; } + + @Override + public String toString() { + return MoreObjects.toStringHelper("FlattenWithDuplicateInputsMatcher").toString(); + } }; }
