From 27fd2e268c2c711bdf2dda4512a06e6d44d56579 Mon Sep 17 00:00:00 2001 From: Arina Ielchiieva Date: Sat, 27 Jul 2019 17:16:05 +0300 Subject: [PATCH] Add forTable method for Avro WriteBuilder --- core/src/main/java/org/apache/iceberg/avro/Avro.java | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/core/src/main/java/org/apache/iceberg/avro/Avro.java b/core/src/main/java/org/apache/iceberg/avro/Avro.java index 65ba9d833cef..a658c9e6d5fc 100644 --- a/core/src/main/java/org/apache/iceberg/avro/Avro.java +++ b/core/src/main/java/org/apache/iceberg/avro/Avro.java @@ -34,6 +34,7 @@ import org.apache.avro.io.DatumWriter; import org.apache.avro.specific.SpecificData; import org.apache.iceberg.SchemaParser; +import org.apache.iceberg.Table; import org.apache.iceberg.io.FileAppender; import org.apache.iceberg.io.InputFile; import org.apache.iceberg.io.OutputFile; @@ -89,6 +90,12 @@ private WriteBuilder(OutputFile file) { this.file = file; } + public WriteBuilder forTable(Table table) { + schema(table.schema()); + setAll(table.properties()); + return this; + } + public WriteBuilder schema(org.apache.iceberg.Schema newSchema) { this.schema = newSchema; return this;