Add withCompression to TextIO.Write

wcn3 · Jul 18, 2018 · bf02368 · bf02368
1 parent c14c975
commit bf02368
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/sdks/java/core/src/main/java/org/apache/beam/sdk/io/TextIO.java b/sdks/java/core/src/main/java/org/apache/beam/sdk/io/TextIO.java
@@ -1136,6 +1136,15 @@ public Write withWritableByteChannelFactory(
       return new Write(inner.withWritableByteChannelFactory(writableByteChannelFactory));
     }
 
+    /**
+     * See {@link TypedWrite#withCompression(Compression)}.
+     */
+    public Write withCompression(Compression compression) {
+      checkArgument(compression != null, "compression can not be null");
+      return withWritableByteChannelFactory(
+          FileBasedSink.CompressionType.fromCanonical(compression));
+    }
+
     /** See {@link TypedWrite#withWindowedWrites}. */
     public Write withWindowedWrites() {
       return new Write(inner.withWindowedWrites());