Repository: beam Updated Branches: refs/heads/master a32371eb3 -> 650ee8ea6
[BEAM-1871] Re-add BitSetCoder to sdk/util for Dataflow worker. Project: http://git-wip-us.apache.org/repos/asf/beam/repo Commit: http://git-wip-us.apache.org/repos/asf/beam/commit/7d76e100 Tree: http://git-wip-us.apache.org/repos/asf/beam/tree/7d76e100 Diff: http://git-wip-us.apache.org/repos/asf/beam/diff/7d76e100 Branch: refs/heads/master Commit: 7d76e1009795157e81f0faca12202516bfe15015 Parents: 11a20ff Author: Luke Cwik <[email protected]> Authored: Wed Apr 26 15:32:06 2017 -0700 Committer: Luke Cwik <[email protected]> Committed: Wed Apr 26 17:23:19 2017 -0700 ---------------------------------------------------------------------- .../org/apache/beam/sdk/util/BitSetCoder.java | 61 ++++++++++++++++++++ 1 file changed, 61 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/beam/blob/7d76e100/sdks/java/core/src/main/java/org/apache/beam/sdk/util/BitSetCoder.java ---------------------------------------------------------------------- diff --git a/sdks/java/core/src/main/java/org/apache/beam/sdk/util/BitSetCoder.java b/sdks/java/core/src/main/java/org/apache/beam/sdk/util/BitSetCoder.java new file mode 100644 index 0000000..b0e9b5c --- /dev/null +++ b/sdks/java/core/src/main/java/org/apache/beam/sdk/util/BitSetCoder.java @@ -0,0 +1,61 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.beam.sdk.util; + +import java.io.IOException; +import java.io.InputStream; +import java.io.OutputStream; +import java.util.BitSet; +import org.apache.beam.sdk.coders.ByteArrayCoder; +import org.apache.beam.sdk.coders.CoderException; +import org.apache.beam.sdk.coders.CustomCoder; + +/** + * Coder for the BitSet used to track child-trigger finished states. + */ +@Deprecated +public class BitSetCoder extends CustomCoder<BitSet> { + + private static final BitSetCoder INSTANCE = new BitSetCoder(); + private static final ByteArrayCoder BYTE_ARRAY_CODER = ByteArrayCoder.of(); + + private BitSetCoder() {} + + public static BitSetCoder of() { + return INSTANCE; + } + + @Override + public void encode(BitSet value, OutputStream outStream, Context context) + throws CoderException, IOException { + BYTE_ARRAY_CODER.encodeAndOwn(value.toByteArray(), outStream, context); + } + + @Override + public BitSet decode(InputStream inStream, Context context) + throws CoderException, IOException { + return BitSet.valueOf(BYTE_ARRAY_CODER.decode(inStream, context)); + } + + @Override + public void verifyDeterministic() throws NonDeterministicException { + verifyDeterministic( + "BitSetCoder requires its ByteArrayCoder to be deterministic.", BYTE_ARRAY_CODER); + } +} +
