From d302a0f3bcb89d1a2367efb90925ab5abe70f7dd Mon Sep 17 00:00:00 2001 From: Luke Cwik Date: Wed, 7 Jun 2017 07:48:09 -0700 Subject: [PATCH] [BEAM-1231] Add missed "kind:bytes" to CloudObjectKinds/CloudObjectTranslators --- .../apache/beam/runners/dataflow/util/CloudObjectKinds.java | 1 + .../beam/runners/dataflow/util/CloudObjectTranslators.java | 5 +++-- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/runners/google-cloud-dataflow-java/src/main/java/org/apache/beam/runners/dataflow/util/CloudObjectKinds.java b/runners/google-cloud-dataflow-java/src/main/java/org/apache/beam/runners/dataflow/util/CloudObjectKinds.java index 403ade28c61e..58bc44679106 100644 --- a/runners/google-cloud-dataflow-java/src/main/java/org/apache/beam/runners/dataflow/util/CloudObjectKinds.java +++ b/runners/google-cloud-dataflow-java/src/main/java/org/apache/beam/runners/dataflow/util/CloudObjectKinds.java @@ -28,4 +28,5 @@ class CloudObjectKinds { static final String KIND_PAIR = "kind:pair"; static final String KIND_STREAM = "kind:stream"; static final String KIND_WINDOWED_VALUE = "kind:windowed_value"; + static final String KIND_BYTES = "kind:bytes"; } diff --git a/runners/google-cloud-dataflow-java/src/main/java/org/apache/beam/runners/dataflow/util/CloudObjectTranslators.java b/runners/google-cloud-dataflow-java/src/main/java/org/apache/beam/runners/dataflow/util/CloudObjectTranslators.java index 012a669431be..ad2363d64578 100644 --- a/runners/google-cloud-dataflow-java/src/main/java/org/apache/beam/runners/dataflow/util/CloudObjectTranslators.java +++ b/runners/google-cloud-dataflow-java/src/main/java/org/apache/beam/runners/dataflow/util/CloudObjectTranslators.java @@ -279,7 +279,8 @@ static CloudObjectTranslator bytes() { @Override public CloudObject toCloudObject(ByteArrayCoder target) { return addComponents( - CloudObject.forClass(target.getClass()), Collections.>emptyList()); + CloudObject.forClassName(CloudObjectKinds.KIND_BYTES), + Collections.>emptyList()); } @Override @@ -294,7 +295,7 @@ public Class getSupportedClass() { @Override public String cloudObjectClassName() { - return CloudObject.forClass(ByteArrayCoder.class).getClassName(); + return CloudObjectKinds.KIND_BYTES; } };