fixup! Swap to use encode/decode in anonymous inner class coder and @AutoValue coder
Project: http://git-wip-us.apache.org/repos/asf/beam/repo Commit: http://git-wip-us.apache.org/repos/asf/beam/commit/2d379ddd Tree: http://git-wip-us.apache.org/repos/asf/beam/tree/2d379ddd Diff: http://git-wip-us.apache.org/repos/asf/beam/diff/2d379ddd Branch: refs/heads/master Commit: 2d379ddd5ef9b0b7a54ddc70203f37cd2763f387 Parents: 4486730 Author: Lukasz Cwik <[email protected]> Authored: Sun May 7 19:41:07 2017 -0700 Committer: Luke Cwik <[email protected]> Committed: Mon May 8 20:17:57 2017 -0700 ---------------------------------------------------------------------- .../core/construction/PCollectionsTest.java | 17 +++++++---------- 1 file changed, 7 insertions(+), 10 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/beam/blob/2d379ddd/runners/core-construction-java/src/test/java/org/apache/beam/runners/core/construction/PCollectionsTest.java ---------------------------------------------------------------------- diff --git a/runners/core-construction-java/src/test/java/org/apache/beam/runners/core/construction/PCollectionsTest.java b/runners/core-construction-java/src/test/java/org/apache/beam/runners/core/construction/PCollectionsTest.java index 2c45cbd..a114cf5 100644 --- a/runners/core-construction-java/src/test/java/org/apache/beam/runners/core/construction/PCollectionsTest.java +++ b/runners/core-construction-java/src/test/java/org/apache/beam/runners/core/construction/PCollectionsTest.java @@ -29,10 +29,7 @@ import java.io.OutputStream; import java.util.Collection; import java.util.Collections; import org.apache.beam.sdk.Pipeline; -import org.apache.beam.sdk.coders.AtomicCoder; -import org.apache.beam.sdk.coders.BigEndianLongCoder; -import org.apache.beam.sdk.coders.Coder; -import org.apache.beam.sdk.coders.CustomCoder; +import org.apache.beam.sdk.coders.*; import org.apache.beam.sdk.common.runner.v1.RunnerApi; import org.apache.beam.sdk.io.GenerateSequence; import org.apache.beam.sdk.testing.TestPipeline; @@ -130,13 +127,13 @@ public class PCollectionsTest { @AutoValue abstract static class CustomIntCoder extends CustomCoder<Integer> { @Override - public void encode(Integer value, OutputStream outStream, Context context) throws IOException { - VarInt.encode(value, outStream); + public Integer decode(InputStream inStream) throws IOException { + return VarInt.decodeInt(inStream); } @Override - public Integer decode(InputStream inStream, Context context) throws IOException { - return VarInt.decodeInt(inStream); + public void encode(Integer value, OutputStream outStream) throws IOException { + VarInt.encode(value, outStream); } } @@ -163,13 +160,13 @@ public class PCollectionsTest { @Override public void verifyDeterministic() {} @Override - public void encode(BoundedWindow value, OutputStream outStream, Context context) + public void encode(BoundedWindow value, OutputStream outStream) throws IOException { VarInt.encode(value.maxTimestamp().getMillis(), outStream); } @Override - public BoundedWindow decode(InputStream inStream, Context context) throws IOException { + public BoundedWindow decode(InputStream inStream) throws IOException { final Instant ts = new Instant(VarInt.decodeLong(inStream)); return new BoundedWindow() { @Override
