| <?xml version='1.0' encoding='UTF-8' standalone='no'?> |
| <doxygen xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:noNamespaceSchemaLocation="compound.xsd" version="1.9.8" xml:lang="en-US"> |
| <compounddef id="data__writer__test_8cc" kind="file" language="C++"> |
| <compoundname>data_writer_test.cc</compoundname> |
| <includes refid="data__writer_8h" local="yes">iceberg/data/data_writer.h</includes> |
| <includes local="no">arrow/array.h</includes> |
| <includes local="no">arrow/c/bridge.h</includes> |
| <includes local="no">arrow/json/from_string.h</includes> |
| <includes local="no">gmock/gmock.h</includes> |
| <includes local="no">gtest/gtest.h</includes> |
| <includes refid="arrow__fs__file__io__internal_8h" local="yes">iceberg/arrow/arrow_fs_file_io_internal.h</includes> |
| <includes refid="avro__register_8h" local="yes">iceberg/avro/avro_register.h</includes> |
| <includes refid="file__format_8h" local="yes">iceberg/file_format.h</includes> |
| <includes refid="manifest__entry_8h" local="yes">iceberg/manifest/manifest_entry.h</includes> |
| <includes refid="parquet__register_8h" local="yes">iceberg/parquet/parquet_register.h</includes> |
| <includes refid="partition__spec_8h" local="yes">iceberg/partition_spec.h</includes> |
| <includes refid="partition__values_8h" local="yes">iceberg/row/partition_values.h</includes> |
| <includes refid="schema_8h" local="yes">iceberg/schema.h</includes> |
| <includes refid="schema__field_8h" local="yes">iceberg/schema_field.h</includes> |
| <includes refid="schema__internal_8h" local="yes">iceberg/schema_internal.h</includes> |
| <includes refid="matchers_8h" local="yes">iceberg/test/matchers.h</includes> |
| <includes refid="type_8h" local="yes">iceberg/type.h</includes> |
| <includes refid="macros_8h" local="yes">iceberg/util/macros.h</includes> |
| <incdepgraph> |
| <node id="53"> |
| <label>iceberg/arrow/arrow_fs_file_io_internal.h</label> |
| <link refid="arrow__fs__file__io__internal_8h_source"/> |
| <childnode refid="4" relation="include"> |
| </childnode> |
| <childnode refid="54" relation="include"> |
| </childnode> |
| <childnode refid="55" relation="include"> |
| </childnode> |
| <childnode refid="56" relation="include"> |
| </childnode> |
| </node> |
| <node id="8"> |
| <label>iceberg/arrow_c_data.h</label> |
| <link refid="arrow__c__data_8h"/> |
| <childnode refid="3" relation="include"> |
| </childnode> |
| </node> |
| <node id="57"> |
| <label>iceberg/avro/avro_register.h</label> |
| <link refid="avro__register_8h"/> |
| <childnode refid="56" relation="include"> |
| </childnode> |
| </node> |
| <node id="2"> |
| <label>iceberg/data/data_writer.h</label> |
| <link refid="data__writer_8h"/> |
| <childnode refid="3" relation="include"> |
| </childnode> |
| <childnode refid="4" relation="include"> |
| </childnode> |
| <childnode refid="5" relation="include"> |
| </childnode> |
| <childnode refid="6" relation="include"> |
| </childnode> |
| <childnode refid="7" relation="include"> |
| </childnode> |
| <childnode refid="8" relation="include"> |
| </childnode> |
| <childnode refid="9" relation="include"> |
| </childnode> |
| <childnode refid="16" relation="include"> |
| </childnode> |
| <childnode refid="11" relation="include"> |
| </childnode> |
| <childnode refid="12" relation="include"> |
| </childnode> |
| <childnode refid="26" relation="include"> |
| </childnode> |
| <childnode refid="15" relation="include"> |
| </childnode> |
| </node> |
| <node id="9"> |
| <label>iceberg/data/writer.h</label> |
| <link refid="writer_8h"/> |
| <childnode refid="3" relation="include"> |
| </childnode> |
| <childnode refid="4" relation="include"> |
| </childnode> |
| <childnode refid="10" relation="include"> |
| </childnode> |
| <childnode refid="8" relation="include"> |
| </childnode> |
| <childnode refid="11" relation="include"> |
| </childnode> |
| <childnode refid="12" relation="include"> |
| </childnode> |
| <childnode refid="15" relation="include"> |
| </childnode> |
| </node> |
| <node id="41"> |
| <label>iceberg/exception.h</label> |
| <link refid="exception_8h"/> |
| <childnode refid="42" relation="include"> |
| </childnode> |
| <childnode refid="11" relation="include"> |
| </childnode> |
| </node> |
| <node id="29"> |
| <label>iceberg/expression/literal.h</label> |
| <link refid="literal_8h_source"/> |
| <childnode refid="30" relation="include"> |
| </childnode> |
| <childnode refid="4" relation="include"> |
| </childnode> |
| <childnode refid="6" relation="include"> |
| </childnode> |
| <childnode refid="31" relation="include"> |
| </childnode> |
| <childnode refid="10" relation="include"> |
| </childnode> |
| <childnode refid="11" relation="include"> |
| </childnode> |
| <childnode refid="12" relation="include"> |
| </childnode> |
| <childnode refid="32" relation="include"> |
| </childnode> |
| <childnode refid="43" relation="include"> |
| </childnode> |
| <childnode refid="35" relation="include"> |
| </childnode> |
| <childnode refid="45" relation="include"> |
| </childnode> |
| <childnode refid="46" relation="include"> |
| </childnode> |
| </node> |
| <node id="16"> |
| <label>iceberg/file_format.h</label> |
| <link refid="file__format_8h"/> |
| <childnode refid="17" relation="include"> |
| </childnode> |
| <childnode refid="18" relation="include"> |
| </childnode> |
| <childnode refid="11" relation="include"> |
| </childnode> |
| <childnode refid="12" relation="include"> |
| </childnode> |
| <childnode refid="19" relation="include"> |
| </childnode> |
| </node> |
| <node id="55"> |
| <label>iceberg/file_io.h</label> |
| <link refid="file__io_8h_source"/> |
| <childnode refid="5" relation="include"> |
| </childnode> |
| <childnode refid="6" relation="include"> |
| </childnode> |
| <childnode refid="17" relation="include"> |
| </childnode> |
| <childnode refid="11" relation="include"> |
| </childnode> |
| <childnode refid="12" relation="include"> |
| </childnode> |
| </node> |
| <node id="56"> |
| <label>iceberg/iceberg_bundle_export.h</label> |
| <link refid="iceberg__bundle__export_8h_source"/> |
| </node> |
| <node id="11"> |
| <label>iceberg/iceberg_export.h</label> |
| <link refid="iceberg__export_8h_source"/> |
| </node> |
| <node id="58"> |
| <label>iceberg/manifest/manifest_entry.h</label> |
| <link refid="manifest__entry_8h"/> |
| <childnode refid="3" relation="include"> |
| </childnode> |
| <childnode refid="59" relation="include"> |
| </childnode> |
| <childnode refid="4" relation="include"> |
| </childnode> |
| <childnode refid="5" relation="include"> |
| </childnode> |
| <childnode refid="6" relation="include"> |
| </childnode> |
| <childnode refid="10" relation="include"> |
| </childnode> |
| <childnode refid="16" relation="include"> |
| </childnode> |
| <childnode refid="11" relation="include"> |
| </childnode> |
| <childnode refid="12" relation="include"> |
| </childnode> |
| <childnode refid="26" relation="include"> |
| </childnode> |
| <childnode refid="34" relation="include"> |
| </childnode> |
| <childnode refid="32" relation="include"> |
| </childnode> |
| </node> |
| <node id="60"> |
| <label>iceberg/parquet/parquet_register.h</label> |
| <link refid="parquet__register_8h"/> |
| <childnode refid="56" relation="include"> |
| </childnode> |
| </node> |
| <node id="62"> |
| <label>iceberg/partition_field.h</label> |
| <link refid="partition__field_8h"/> |
| <childnode refid="3" relation="include"> |
| </childnode> |
| <childnode refid="4" relation="include"> |
| </childnode> |
| <childnode refid="6" relation="include"> |
| </childnode> |
| <childnode refid="17" relation="include"> |
| </childnode> |
| <childnode refid="11" relation="include"> |
| </childnode> |
| <childnode refid="15" relation="include"> |
| </childnode> |
| <childnode refid="35" relation="include"> |
| </childnode> |
| </node> |
| <node id="61"> |
| <label>iceberg/partition_spec.h</label> |
| <link refid="partition__spec_8h"/> |
| <childnode refid="3" relation="include"> |
| </childnode> |
| <childnode refid="4" relation="include"> |
| </childnode> |
| <childnode refid="5" relation="include"> |
| </childnode> |
| <childnode refid="28" relation="include"> |
| </childnode> |
| <childnode refid="6" relation="include"> |
| </childnode> |
| <childnode refid="7" relation="include"> |
| </childnode> |
| <childnode refid="10" relation="include"> |
| </childnode> |
| <childnode refid="11" relation="include"> |
| </childnode> |
| <childnode refid="62" relation="include"> |
| </childnode> |
| <childnode refid="12" relation="include"> |
| </childnode> |
| <childnode refid="15" relation="include"> |
| </childnode> |
| <childnode refid="35" relation="include"> |
| </childnode> |
| <childnode refid="36" relation="include"> |
| </childnode> |
| </node> |
| <node id="12"> |
| <label>iceberg/result.h</label> |
| <link refid="result_8h_source"/> |
| <childnode refid="13" relation="include"> |
| </childnode> |
| <childnode refid="14" relation="include"> |
| </childnode> |
| <childnode refid="6" relation="include"> |
| </childnode> |
| <childnode refid="11" relation="include"> |
| </childnode> |
| </node> |
| <node id="26"> |
| <label>iceberg/row/partition_values.h</label> |
| <link refid="partition__values_8h"/> |
| <childnode refid="27" relation="include"> |
| </childnode> |
| <childnode refid="28" relation="include"> |
| </childnode> |
| <childnode refid="18" relation="include"> |
| </childnode> |
| <childnode refid="29" relation="include"> |
| </childnode> |
| <childnode refid="11" relation="include"> |
| </childnode> |
| <childnode refid="47" relation="include"> |
| </childnode> |
| </node> |
| <node id="47"> |
| <label>iceberg/row/struct_like.h</label> |
| <link refid="struct__like_8h"/> |
| <childnode refid="27" relation="include"> |
| </childnode> |
| <childnode refid="4" relation="include"> |
| </childnode> |
| <childnode refid="28" relation="include"> |
| </childnode> |
| <childnode refid="17" relation="include"> |
| </childnode> |
| <childnode refid="31" relation="include"> |
| </childnode> |
| <childnode refid="29" relation="include"> |
| </childnode> |
| <childnode refid="12" relation="include"> |
| </childnode> |
| <childnode refid="15" relation="include"> |
| </childnode> |
| <childnode refid="43" relation="include"> |
| </childnode> |
| </node> |
| <node id="63"> |
| <label>iceberg/schema.h</label> |
| <link refid="schema_8h"/> |
| <childnode refid="3" relation="include"> |
| </childnode> |
| <childnode refid="5" relation="include"> |
| </childnode> |
| <childnode refid="6" relation="include"> |
| </childnode> |
| <childnode refid="7" relation="include"> |
| </childnode> |
| <childnode refid="64" relation="include"> |
| </childnode> |
| <childnode refid="10" relation="include"> |
| </childnode> |
| <childnode refid="11" relation="include"> |
| </childnode> |
| <childnode refid="12" relation="include"> |
| </childnode> |
| <childnode refid="34" relation="include"> |
| </childnode> |
| <childnode refid="32" relation="include"> |
| </childnode> |
| <childnode refid="36" relation="include"> |
| </childnode> |
| <childnode refid="19" relation="include"> |
| </childnode> |
| </node> |
| <node id="34"> |
| <label>iceberg/schema_field.h</label> |
| <link refid="schema__field_8h"/> |
| <childnode refid="3" relation="include"> |
| </childnode> |
| <childnode refid="4" relation="include"> |
| </childnode> |
| <childnode refid="6" relation="include"> |
| </childnode> |
| <childnode refid="17" relation="include"> |
| </childnode> |
| <childnode refid="11" relation="include"> |
| </childnode> |
| <childnode refid="12" relation="include"> |
| </childnode> |
| <childnode refid="15" relation="include"> |
| </childnode> |
| <childnode refid="35" relation="include"> |
| </childnode> |
| </node> |
| <node id="65"> |
| <label>iceberg/schema_internal.h</label> |
| <link refid="schema__internal_8h_source"/> |
| <childnode refid="4" relation="include"> |
| </childnode> |
| <childnode refid="5" relation="include"> |
| </childnode> |
| <childnode refid="66" relation="include"> |
| </childnode> |
| <childnode refid="11" relation="include"> |
| </childnode> |
| <childnode refid="12" relation="include"> |
| </childnode> |
| <childnode refid="15" relation="include"> |
| </childnode> |
| </node> |
| <node id="1"> |
| <label>iceberg/test/data_writer_test.cc</label> |
| <link refid="data__writer__test_8cc"/> |
| <childnode refid="2" relation="include"> |
| </childnode> |
| <childnode refid="48" relation="include"> |
| </childnode> |
| <childnode refid="49" relation="include"> |
| </childnode> |
| <childnode refid="50" relation="include"> |
| </childnode> |
| <childnode refid="51" relation="include"> |
| </childnode> |
| <childnode refid="52" relation="include"> |
| </childnode> |
| <childnode refid="53" relation="include"> |
| </childnode> |
| <childnode refid="57" relation="include"> |
| </childnode> |
| <childnode refid="16" relation="include"> |
| </childnode> |
| <childnode refid="58" relation="include"> |
| </childnode> |
| <childnode refid="60" relation="include"> |
| </childnode> |
| <childnode refid="61" relation="include"> |
| </childnode> |
| <childnode refid="26" relation="include"> |
| </childnode> |
| <childnode refid="63" relation="include"> |
| </childnode> |
| <childnode refid="34" relation="include"> |
| </childnode> |
| <childnode refid="65" relation="include"> |
| </childnode> |
| <childnode refid="67" relation="include"> |
| </childnode> |
| <childnode refid="32" relation="include"> |
| </childnode> |
| <childnode refid="39" relation="include"> |
| </childnode> |
| </node> |
| <node id="67"> |
| <label>iceberg/test/matchers.h</label> |
| <link refid="matchers_8h_source"/> |
| <childnode refid="51" relation="include"> |
| </childnode> |
| <childnode refid="52" relation="include"> |
| </childnode> |
| <childnode refid="12" relation="include"> |
| </childnode> |
| <childnode refid="39" relation="include"> |
| </childnode> |
| </node> |
| <node id="32"> |
| <label>iceberg/type.h</label> |
| <link refid="type_8h"/> |
| <childnode refid="33" relation="include"> |
| </childnode> |
| <childnode refid="3" relation="include"> |
| </childnode> |
| <childnode refid="4" relation="include"> |
| </childnode> |
| <childnode refid="5" relation="include"> |
| </childnode> |
| <childnode refid="28" relation="include"> |
| </childnode> |
| <childnode refid="6" relation="include"> |
| </childnode> |
| <childnode refid="7" relation="include"> |
| </childnode> |
| <childnode refid="10" relation="include"> |
| </childnode> |
| <childnode refid="11" relation="include"> |
| </childnode> |
| <childnode refid="12" relation="include"> |
| </childnode> |
| <childnode refid="34" relation="include"> |
| </childnode> |
| <childnode refid="35" relation="include"> |
| </childnode> |
| <childnode refid="36" relation="include"> |
| </childnode> |
| </node> |
| <node id="15"> |
| <label>iceberg/type_fwd.h</label> |
| <link refid="type__fwd_8h"/> |
| </node> |
| <node id="43"> |
| <label>iceberg/util/decimal.h</label> |
| <link refid="decimal_8h"/> |
| <childnode refid="33" relation="include"> |
| </childnode> |
| <childnode refid="30" relation="include"> |
| </childnode> |
| <childnode refid="3" relation="include"> |
| </childnode> |
| <childnode refid="44" relation="include"> |
| </childnode> |
| <childnode refid="6" relation="include"> |
| </childnode> |
| <childnode refid="17" relation="include"> |
| </childnode> |
| <childnode refid="24" relation="include"> |
| </childnode> |
| <childnode refid="10" relation="include"> |
| </childnode> |
| <childnode refid="11" relation="include"> |
| </childnode> |
| <childnode refid="12" relation="include"> |
| </childnode> |
| <childnode refid="35" relation="include"> |
| </childnode> |
| <childnode refid="45" relation="include"> |
| </childnode> |
| </node> |
| <node id="35"> |
| <label>iceberg/util/formattable.h</label> |
| <link refid="formattable_8h"/> |
| <childnode refid="6" relation="include"> |
| </childnode> |
| <childnode refid="11" relation="include"> |
| </childnode> |
| </node> |
| <node id="45"> |
| <label>iceberg/util/int128.h</label> |
| <link refid="int128_8h"/> |
| </node> |
| <node id="36"> |
| <label>iceberg/util/lazy.h</label> |
| <link refid="lazy_8h"/> |
| <childnode refid="37" relation="include"> |
| </childnode> |
| <childnode refid="27" relation="include"> |
| </childnode> |
| <childnode refid="38" relation="include"> |
| </childnode> |
| <childnode refid="12" relation="include"> |
| </childnode> |
| <childnode refid="39" relation="include"> |
| </childnode> |
| </node> |
| <node id="39"> |
| <label>iceberg/util/macros.h</label> |
| <link refid="macros_8h_source"/> |
| <childnode refid="40" relation="include"> |
| </childnode> |
| <childnode refid="41" relation="include"> |
| </childnode> |
| <childnode refid="12" relation="include"> |
| </childnode> |
| </node> |
| <node id="19"> |
| <label>iceberg/util/string_util.h</label> |
| <link refid="string__util_8h_source"/> |
| <childnode refid="20" relation="include"> |
| </childnode> |
| <childnode refid="21" relation="include"> |
| </childnode> |
| <childnode refid="22" relation="include"> |
| </childnode> |
| <childnode refid="23" relation="include"> |
| </childnode> |
| <childnode refid="6" relation="include"> |
| </childnode> |
| <childnode refid="17" relation="include"> |
| </childnode> |
| <childnode refid="24" relation="include"> |
| </childnode> |
| <childnode refid="25" relation="include"> |
| </childnode> |
| <childnode refid="18" relation="include"> |
| </childnode> |
| <childnode refid="11" relation="include"> |
| </childnode> |
| <childnode refid="12" relation="include"> |
| </childnode> |
| </node> |
| <node id="46"> |
| <label>iceberg/util/uuid.h</label> |
| <link refid="uuid_8h"/> |
| <childnode refid="33" relation="include"> |
| </childnode> |
| <childnode refid="3" relation="include"> |
| </childnode> |
| <childnode refid="28" relation="include"> |
| </childnode> |
| <childnode refid="17" relation="include"> |
| </childnode> |
| <childnode refid="11" relation="include"> |
| </childnode> |
| <childnode refid="12" relation="include"> |
| </childnode> |
| <childnode refid="35" relation="include"> |
| </childnode> |
| </node> |
| <node id="20"> |
| <label>algorithm</label> |
| </node> |
| <node id="33"> |
| <label>array</label> |
| </node> |
| <node id="48"> |
| <label>arrow/array.h</label> |
| </node> |
| <node id="49"> |
| <label>arrow/c/bridge.h</label> |
| </node> |
| <node id="54"> |
| <label>arrow/filesystem/filesystem.h</label> |
| </node> |
| <node id="50"> |
| <label>arrow/json/from_string.h</label> |
| </node> |
| <node id="40"> |
| <label>cassert</label> |
| </node> |
| <node id="21"> |
| <label>cerrno</label> |
| </node> |
| <node id="22"> |
| <label>charconv</label> |
| </node> |
| <node id="30"> |
| <label>compare</label> |
| </node> |
| <node id="37"> |
| <label>concepts</label> |
| </node> |
| <node id="3"> |
| <label>cstdint</label> |
| </node> |
| <node id="13"> |
| <label>expected</label> |
| </node> |
| <node id="14"> |
| <label>format</label> |
| </node> |
| <node id="27"> |
| <label>functional</label> |
| </node> |
| <node id="51"> |
| <label>gmock/gmock.h</label> |
| </node> |
| <node id="52"> |
| <label>gtest/gtest.h</label> |
| </node> |
| <node id="44"> |
| <label>iosfwd</label> |
| </node> |
| <node id="59"> |
| <label>map</label> |
| </node> |
| <node id="4"> |
| <label>memory</label> |
| </node> |
| <node id="38"> |
| <label>mutex</label> |
| </node> |
| <node id="66"> |
| <label>nanoarrow/nanoarrow.h</label> |
| </node> |
| <node id="5"> |
| <label>optional</label> |
| </node> |
| <node id="23"> |
| <label>ranges</label> |
| </node> |
| <node id="28"> |
| <label>span</label> |
| </node> |
| <node id="42"> |
| <label>stdexcept</label> |
| </node> |
| <node id="6"> |
| <label>string</label> |
| </node> |
| <node id="17"> |
| <label>string_view</label> |
| </node> |
| <node id="24"> |
| <label>type_traits</label> |
| </node> |
| <node id="25"> |
| <label>typeinfo</label> |
| </node> |
| <node id="7"> |
| <label>unordered_map</label> |
| </node> |
| <node id="64"> |
| <label>unordered_set</label> |
| </node> |
| <node id="18"> |
| <label>utility</label> |
| </node> |
| <node id="31"> |
| <label>variant</label> |
| </node> |
| <node id="10"> |
| <label>vector</label> |
| </node> |
| </incdepgraph> |
| <innerclass refid="classiceberg_1_1DataWriterTest" prot="public">iceberg::DataWriterTest</innerclass> |
| <innerclass refid="classiceberg_1_1DataWriterFormatTest" prot="public">iceberg::DataWriterFormatTest</innerclass> |
| <innernamespace refid="namespaceiceberg">iceberg</innernamespace> |
| <briefdescription> |
| </briefdescription> |
| <detaileddescription> |
| </detaileddescription> |
| <programlisting> |
| <codeline lineno="1"><highlight class="comment">/*</highlight></codeline> |
| <codeline lineno="2"><highlight class="comment"><sp/>*<sp/>Licensed<sp/>to<sp/>the<sp/>Apache<sp/>Software<sp/>Foundation<sp/>(ASF)<sp/>under<sp/>one</highlight></codeline> |
| <codeline lineno="3"><highlight class="comment"><sp/>*<sp/>or<sp/>more<sp/>contributor<sp/>license<sp/>agreements.<sp/><sp/>See<sp/>the<sp/>NOTICE<sp/>file</highlight></codeline> |
| <codeline lineno="4"><highlight class="comment"><sp/>*<sp/>distributed<sp/>with<sp/>this<sp/>work<sp/>for<sp/>additional<sp/>information</highlight></codeline> |
| <codeline lineno="5"><highlight class="comment"><sp/>*<sp/>regarding<sp/>copyright<sp/>ownership.<sp/><sp/>The<sp/>ASF<sp/>licenses<sp/>this<sp/>file</highlight></codeline> |
| <codeline lineno="6"><highlight class="comment"><sp/>*<sp/>to<sp/>you<sp/>under<sp/>the<sp/>Apache<sp/>License,<sp/>Version<sp/>2.0<sp/>(the</highlight></codeline> |
| <codeline lineno="7"><highlight class="comment"><sp/>*<sp/>"License");<sp/>you<sp/>may<sp/>not<sp/>use<sp/>this<sp/>file<sp/>except<sp/>in<sp/>compliance</highlight></codeline> |
| <codeline lineno="8"><highlight class="comment"><sp/>*<sp/>with<sp/>the<sp/>License.<sp/><sp/>You<sp/>may<sp/>obtain<sp/>a<sp/>copy<sp/>of<sp/>the<sp/>License<sp/>at</highlight></codeline> |
| <codeline lineno="9"><highlight class="comment"><sp/>*</highlight></codeline> |
| <codeline lineno="10"><highlight class="comment"><sp/>*<sp/><sp/><sp/>http://www.apache.org/licenses/LICENSE-2.0</highlight></codeline> |
| <codeline lineno="11"><highlight class="comment"><sp/>*</highlight></codeline> |
| <codeline lineno="12"><highlight class="comment"><sp/>*<sp/>Unless<sp/>required<sp/>by<sp/>applicable<sp/>law<sp/>or<sp/>agreed<sp/>to<sp/>in<sp/>writing,</highlight></codeline> |
| <codeline lineno="13"><highlight class="comment"><sp/>*<sp/>software<sp/>distributed<sp/>under<sp/>the<sp/>License<sp/>is<sp/>distributed<sp/>on<sp/>an</highlight></codeline> |
| <codeline lineno="14"><highlight class="comment"><sp/>*<sp/>"AS<sp/>IS"<sp/>BASIS,<sp/>WITHOUT<sp/>WARRANTIES<sp/>OR<sp/>CONDITIONS<sp/>OF<sp/>ANY</highlight></codeline> |
| <codeline lineno="15"><highlight class="comment"><sp/>*<sp/>KIND,<sp/>either<sp/>express<sp/>or<sp/>implied.<sp/><sp/>See<sp/>the<sp/>License<sp/>for<sp/>the</highlight></codeline> |
| <codeline lineno="16"><highlight class="comment"><sp/>*<sp/>specific<sp/>language<sp/>governing<sp/>permissions<sp/>and<sp/>limitations</highlight></codeline> |
| <codeline lineno="17"><highlight class="comment"><sp/>*<sp/>under<sp/>the<sp/>License.</highlight></codeline> |
| <codeline lineno="18"><highlight class="comment"><sp/>*/</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="19"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="20"><highlight class="normal"></highlight><highlight class="preprocessor">#include<sp/>"<ref refid="data__writer_8h" kindref="compound">iceberg/data/data_writer.h</ref>"</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="21"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="22"><highlight class="normal"></highlight><highlight class="preprocessor">#include<sp/><arrow/array.h></highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="23"><highlight class="normal"></highlight><highlight class="preprocessor">#include<sp/><arrow/c/bridge.h></highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="24"><highlight class="normal"></highlight><highlight class="preprocessor">#include<sp/><arrow/json/from_string.h></highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="25"><highlight class="normal"></highlight><highlight class="preprocessor">#include<sp/><gmock/gmock.h></highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="26"><highlight class="normal"></highlight><highlight class="preprocessor">#include<sp/><gtest/gtest.h></highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="27"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="28"><highlight class="normal"></highlight><highlight class="preprocessor">#include<sp/>"iceberg/arrow/arrow_fs_file_io_internal.h"</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="29"><highlight class="normal"></highlight><highlight class="preprocessor">#include<sp/>"<ref refid="avro__register_8h" kindref="compound">iceberg/avro/avro_register.h</ref>"</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="30"><highlight class="normal"></highlight><highlight class="preprocessor">#include<sp/>"<ref refid="file__format_8h" kindref="compound">iceberg/file_format.h</ref>"</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="31"><highlight class="normal"></highlight><highlight class="preprocessor">#include<sp/>"<ref refid="manifest__entry_8h" kindref="compound">iceberg/manifest/manifest_entry.h</ref>"</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="32"><highlight class="normal"></highlight><highlight class="preprocessor">#include<sp/>"<ref refid="parquet__register_8h" kindref="compound">iceberg/parquet/parquet_register.h</ref>"</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="33"><highlight class="normal"></highlight><highlight class="preprocessor">#include<sp/>"<ref refid="partition__spec_8h" kindref="compound">iceberg/partition_spec.h</ref>"</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="34"><highlight class="normal"></highlight><highlight class="preprocessor">#include<sp/>"<ref refid="partition__values_8h" kindref="compound">iceberg/row/partition_values.h</ref>"</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="35"><highlight class="normal"></highlight><highlight class="preprocessor">#include<sp/>"<ref refid="schema_8h" kindref="compound">iceberg/schema.h</ref>"</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="36"><highlight class="normal"></highlight><highlight class="preprocessor">#include<sp/>"<ref refid="schema__field_8h" kindref="compound">iceberg/schema_field.h</ref>"</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="37"><highlight class="normal"></highlight><highlight class="preprocessor">#include<sp/>"iceberg/schema_internal.h"</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="38"><highlight class="normal"></highlight><highlight class="preprocessor">#include<sp/>"iceberg/test/matchers.h"</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="39"><highlight class="normal"></highlight><highlight class="preprocessor">#include<sp/>"<ref refid="type_8h" kindref="compound">iceberg/type.h</ref>"</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="40"><highlight class="normal"></highlight><highlight class="preprocessor">#include<sp/>"iceberg/util/macros.h"</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="41"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="42"><highlight class="normal"></highlight><highlight class="keyword">namespace<sp/></highlight><highlight class="normal">iceberg<sp/>{</highlight></codeline> |
| <codeline lineno="43"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="44"><highlight class="normal">using<sp/>::testing::HasSubstr;</highlight></codeline> |
| <codeline lineno="45"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="46" refid="classiceberg_1_1DataWriterTest" refkind="compound"><highlight class="normal"></highlight><highlight class="keyword">class<sp/></highlight><highlight class="normal"><ref refid="classiceberg_1_1DataWriterTest" kindref="compound">DataWriterTest</ref><sp/>:<sp/></highlight><highlight class="keyword">public</highlight><highlight class="normal"><sp/>::testing::Test<sp/>{</highlight></codeline> |
| <codeline lineno="47"><highlight class="normal"><sp/></highlight><highlight class="keyword">protected</highlight><highlight class="normal">:</highlight></codeline> |
| <codeline lineno="48"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">static</highlight><highlight class="normal"><sp/></highlight><highlight class="keywordtype">void</highlight><highlight class="normal"><sp/>SetUpTestSuite()<sp/>{</highlight></codeline> |
| <codeline lineno="49"><highlight class="normal"><sp/><sp/><sp/><sp/>parquet::RegisterAll();</highlight></codeline> |
| <codeline lineno="50"><highlight class="normal"><sp/><sp/><sp/><sp/>avro::RegisterAll();</highlight></codeline> |
| <codeline lineno="51"><highlight class="normal"><sp/><sp/>}</highlight></codeline> |
| <codeline lineno="52"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="53"><highlight class="normal"><sp/><sp/></highlight><highlight class="keywordtype">void</highlight><highlight class="normal"><sp/>SetUp()</highlight><highlight class="keyword"><sp/>override<sp/></highlight><highlight class="normal">{</highlight></codeline> |
| <codeline lineno="54"><highlight class="normal"><sp/><sp/><sp/><sp/>file_io_<sp/>=<sp/><ref refid="classiceberg_1_1arrow_1_1ArrowFileSystemFileIO_1a9f75861e2aeb0c393526e947419d5386" kindref="member">arrow::ArrowFileSystemFileIO::MakeMockFileIO</ref>();</highlight></codeline> |
| <codeline lineno="55"><highlight class="normal"><sp/><sp/><sp/><sp/>schema_<sp/>=<sp/>std::make_shared<Schema>(</highlight></codeline> |
| <codeline lineno="56"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/>std::vector<SchemaField>{<ref refid="classiceberg_1_1SchemaField_1a4ad7baa4dedd37e2f699a2f32deef942" kindref="member">SchemaField::MakeRequired</ref>(1,<sp/></highlight><highlight class="stringliteral">"id"</highlight><highlight class="normal">,<sp/><ref refid="group__type-factories_1ga91f0688c58d168d755610a2b5a1a222e" kindref="member">int32</ref>()),</highlight></codeline> |
| <codeline lineno="57"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><ref refid="classiceberg_1_1SchemaField_1a0c8382f6e164b76f45e4b4c26028a872" kindref="member">SchemaField::MakeOptional</ref>(2,<sp/></highlight><highlight class="stringliteral">"name"</highlight><highlight class="normal">,<sp/></highlight><highlight class="keywordtype">string</highlight><highlight class="normal">())});</highlight></codeline> |
| <codeline lineno="58"><highlight class="normal"><sp/><sp/><sp/><sp/>partition_spec_<sp/>=<sp/><ref refid="classiceberg_1_1PartitionSpec_1a2955a97e29cf0637fa5275b031b198dc" kindref="member">PartitionSpec::Unpartitioned</ref>();</highlight></codeline> |
| <codeline lineno="59"><highlight class="normal"><sp/><sp/>}</highlight></codeline> |
| <codeline lineno="60"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="61"><highlight class="normal"><sp/><sp/><ref refid="structiceberg_1_1DataWriterOptions" kindref="compound">DataWriterOptions</ref><sp/>MakeDefaultOptions(</highlight></codeline> |
| <codeline lineno="62"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/>std::optional<int32_t><sp/>sort_order_id<sp/>=<sp/>std::nullopt,</highlight></codeline> |
| <codeline lineno="63"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/><ref refid="classiceberg_1_1PartitionValues" kindref="compound">PartitionValues</ref><sp/>partition<sp/>=<sp/><ref refid="classiceberg_1_1PartitionValues" kindref="compound">PartitionValues</ref>{})<sp/>{</highlight></codeline> |
| <codeline lineno="64"><highlight class="normal"><sp/><sp/><sp/><sp/></highlight><highlight class="keywordflow">return</highlight><highlight class="normal"><sp/><ref refid="structiceberg_1_1DataWriterOptions" kindref="compound">DataWriterOptions</ref>{</highlight></codeline> |
| <codeline lineno="65"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/>.path<sp/>=<sp/></highlight><highlight class="stringliteral">"test_data.parquet"</highlight><highlight class="normal">,</highlight></codeline> |
| <codeline lineno="66"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/>.schema<sp/>=<sp/>schema_,</highlight></codeline> |
| <codeline lineno="67"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/>.spec<sp/>=<sp/>partition_spec_,</highlight></codeline> |
| <codeline lineno="68"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/>.partition<sp/>=<sp/>std::move(partition),</highlight></codeline> |
| <codeline lineno="69"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/>.format<sp/>=<sp/>FileFormatType::kParquet,</highlight></codeline> |
| <codeline lineno="70"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/>.io<sp/>=<sp/>file_io_,</highlight></codeline> |
| <codeline lineno="71"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/>.sort_order_id<sp/>=<sp/>sort_order_id,</highlight></codeline> |
| <codeline lineno="72"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/>.properties<sp/>=<sp/>{{</highlight><highlight class="stringliteral">"write.parquet.compression-codec"</highlight><highlight class="normal">,<sp/></highlight><highlight class="stringliteral">"uncompressed"</highlight><highlight class="normal">}},</highlight></codeline> |
| <codeline lineno="73"><highlight class="normal"><sp/><sp/><sp/><sp/>};</highlight></codeline> |
| <codeline lineno="74"><highlight class="normal"><sp/><sp/>}</highlight></codeline> |
| <codeline lineno="75"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="76"><highlight class="normal"><sp/><sp/>std::shared_ptr<::arrow::Array><sp/>CreateTestData()<sp/>{</highlight></codeline> |
| <codeline lineno="77"><highlight class="normal"><sp/><sp/><sp/><sp/><ref refid="structArrowSchema" kindref="compound">ArrowSchema</ref><sp/>arrow_c_schema;</highlight></codeline> |
| <codeline lineno="78"><highlight class="normal"><sp/><sp/><sp/><sp/>ICEBERG_THROW_NOT_OK(ToArrowSchema(*schema_,<sp/>&arrow_c_schema));</highlight></codeline> |
| <codeline lineno="79"><highlight class="normal"><sp/><sp/><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>arrow_schema<sp/>=<sp/>::arrow::ImportType(&arrow_c_schema).ValueOrDie();</highlight></codeline> |
| <codeline lineno="80"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="81"><highlight class="normal"><sp/><sp/><sp/><sp/>return<sp/>::arrow::json::ArrayFromJSONString(</highlight></codeline> |
| <codeline lineno="82"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/>::arrow::struct_(arrow_schema->fields()),</highlight></codeline> |
| <codeline lineno="83"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/>R</highlight><highlight class="stringliteral">"([[1,<sp/>"Alice"],<sp/>[2,<sp/>"Bob"],<sp/>[3,<sp/>"Charlie"]])")</highlight></codeline> |
| <codeline lineno="84"><highlight class="stringliteral"><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/>.ValueOrDie();</highlight></codeline> |
| <codeline lineno="85"><highlight class="stringliteral"><sp/><sp/>}</highlight></codeline> |
| <codeline lineno="86"><highlight class="stringliteral"></highlight></codeline> |
| <codeline lineno="87"><highlight class="stringliteral"><sp/><sp/></highlight><highlight class="keywordtype">void</highlight><highlight class="normal"><sp/>WriteTestDataToWriter(<ref refid="classiceberg_1_1DataWriter" kindref="compound">DataWriter</ref>*<sp/>writer)<sp/>{</highlight></codeline> |
| <codeline lineno="88"><highlight class="normal"><sp/><sp/><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>test_data<sp/>=<sp/>CreateTestData();</highlight></codeline> |
| <codeline lineno="89"><highlight class="normal"><sp/><sp/><sp/><sp/><ref refid="structArrowArray" kindref="compound">ArrowArray</ref><sp/>arrow_array;</highlight></codeline> |
| <codeline lineno="90"><highlight class="normal"><sp/><sp/><sp/><sp/>ASSERT_TRUE(::arrow::ExportArray(*test_data,<sp/>&arrow_array).ok());</highlight></codeline> |
| <codeline lineno="91"><highlight class="normal"><sp/><sp/><sp/><sp/>ASSERT_THAT(writer-><ref refid="classiceberg_1_1DataWriter_1a6a72a9a4f9193b389e0465fe4120656d" kindref="member">Write</ref>(&arrow_array),<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="92"><highlight class="normal"><sp/><sp/>}</highlight></codeline> |
| <codeline lineno="93"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="94"><highlight class="normal"><sp/><sp/>std::shared_ptr<FileIO><sp/>file_io_;</highlight></codeline> |
| <codeline lineno="95"><highlight class="normal"><sp/><sp/>std::shared_ptr<Schema><sp/>schema_;</highlight></codeline> |
| <codeline lineno="96"><highlight class="normal"><sp/><sp/>std::shared_ptr<PartitionSpec><sp/>partition_spec_;</highlight></codeline> |
| <codeline lineno="97"><highlight class="normal">};</highlight></codeline> |
| <codeline lineno="98"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="99" refid="classiceberg_1_1DataWriterFormatTest" refkind="compound"><highlight class="normal"></highlight><highlight class="keyword">class<sp/></highlight><highlight class="normal"><ref refid="classiceberg_1_1DataWriterFormatTest" kindref="compound">DataWriterFormatTest</ref></highlight></codeline> |
| <codeline lineno="100"><highlight class="normal"><sp/><sp/><sp/><sp/>:<sp/></highlight><highlight class="keyword">public</highlight><highlight class="normal"><sp/><ref refid="classiceberg_1_1DataWriterTest" kindref="compound">DataWriterTest</ref>,</highlight></codeline> |
| <codeline lineno="101"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/></highlight><highlight class="keyword">public</highlight><highlight class="normal"><sp/>::testing::WithParamInterface<std::pair<FileFormatType,<sp/>std::string>><sp/>{};</highlight></codeline> |
| <codeline lineno="102"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="103"><highlight class="normal">TEST_P(<ref refid="classiceberg_1_1DataWriterFormatTest" kindref="compound">DataWriterFormatTest</ref>,<sp/>CreateWithFormat)<sp/>{</highlight></codeline> |
| <codeline lineno="104"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>[format,<sp/>path]<sp/>=<sp/>GetParam();</highlight></codeline> |
| <codeline lineno="105"><highlight class="normal"><sp/><sp/><ref refid="structiceberg_1_1DataWriterOptions" kindref="compound">DataWriterOptions</ref><sp/>options{</highlight></codeline> |
| <codeline lineno="106"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/>.path<sp/>=<sp/>path,</highlight></codeline> |
| <codeline lineno="107"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/>.schema<sp/>=<sp/>schema_,</highlight></codeline> |
| <codeline lineno="108"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/>.spec<sp/>=<sp/>partition_spec_,</highlight></codeline> |
| <codeline lineno="109"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/>.partition<sp/>=<sp/><ref refid="classiceberg_1_1PartitionValues" kindref="compound">PartitionValues</ref>{},</highlight></codeline> |
| <codeline lineno="110"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/>.format<sp/>=<sp/>format,</highlight></codeline> |
| <codeline lineno="111"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/>.io<sp/>=<sp/>file_io_,</highlight></codeline> |
| <codeline lineno="112"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/>.properties<sp/>=</highlight></codeline> |
| <codeline lineno="113"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/>format<sp/>==<sp/>FileFormatType::kParquet</highlight></codeline> |
| <codeline lineno="114"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/>?<sp/>std::unordered_map<std::string,</highlight></codeline> |
| <codeline lineno="115"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/>std::string>{{</highlight><highlight class="stringliteral">"write.parquet.compression-codec"</highlight><highlight class="normal">,</highlight></codeline> |
| <codeline lineno="116"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/></highlight><highlight class="stringliteral">"uncompressed"</highlight><highlight class="normal">}}</highlight></codeline> |
| <codeline lineno="117"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/>:<sp/>std::unordered_map<std::string,<sp/>std::string>{},</highlight></codeline> |
| <codeline lineno="118"><highlight class="normal"><sp/><sp/>};</highlight></codeline> |
| <codeline lineno="119"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="120"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>writer_result<sp/>=<sp/><ref refid="classiceberg_1_1DataWriter_1a75a3d11a5978812e66da37875f54fb98" kindref="member">DataWriter::Make</ref>(options);</highlight></codeline> |
| <codeline lineno="121"><highlight class="normal"><sp/><sp/>ASSERT_THAT(writer_result,<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="122"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>writer<sp/>=<sp/>std::move(writer_result.value());</highlight></codeline> |
| <codeline lineno="123"><highlight class="normal"><sp/><sp/>ASSERT_NE(writer,<sp/></highlight><highlight class="keyword">nullptr</highlight><highlight class="normal">);</highlight></codeline> |
| <codeline lineno="124"><highlight class="normal">}</highlight></codeline> |
| <codeline lineno="125"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="126"><highlight class="normal">INSTANTIATE_TEST_SUITE_P(</highlight></codeline> |
| <codeline lineno="127"><highlight class="normal"><sp/><sp/><sp/><sp/>FormatTypes,<sp/>DataWriterFormatTest,</highlight></codeline> |
| <codeline lineno="128"><highlight class="normal"><sp/><sp/><sp/><sp/>::testing::Values(std::make_pair(FileFormatType::kParquet,<sp/></highlight><highlight class="stringliteral">"test_data.parquet"</highlight><highlight class="normal">),</highlight></codeline> |
| <codeline lineno="129"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/>std::make_pair(FileFormatType::kAvro,<sp/></highlight><highlight class="stringliteral">"test_data.avro"</highlight><highlight class="normal">)));</highlight></codeline> |
| <codeline lineno="130"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="131"><highlight class="normal">TEST_F(DataWriterTest,<sp/>WriteAndClose)<sp/>{</highlight></codeline> |
| <codeline lineno="132"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>writer_result<sp/>=<sp/><ref refid="classiceberg_1_1DataWriter_1a75a3d11a5978812e66da37875f54fb98" kindref="member">DataWriter::Make</ref>(MakeDefaultOptions());</highlight></codeline> |
| <codeline lineno="133"><highlight class="normal"><sp/><sp/>ASSERT_THAT(writer_result,<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="134"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>writer<sp/>=<sp/>std::move(writer_result.value());</highlight></codeline> |
| <codeline lineno="135"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="136"><highlight class="normal"><sp/><sp/></highlight><highlight class="comment">//<sp/>Write<sp/>data</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="137"><highlight class="normal"><sp/><sp/>WriteTestDataToWriter(writer.get());</highlight></codeline> |
| <codeline lineno="138"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="139"><highlight class="normal"><sp/><sp/></highlight><highlight class="comment">//<sp/>Length<sp/>should<sp/>be<sp/>greater<sp/>than<sp/>0<sp/>after<sp/>write</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="140"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>length_result<sp/>=<sp/>writer->Length();</highlight></codeline> |
| <codeline lineno="141"><highlight class="normal"><sp/><sp/>ASSERT_THAT(length_result,<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="142"><highlight class="normal"><sp/><sp/>EXPECT_GT(length_result.value(),<sp/>0);</highlight></codeline> |
| <codeline lineno="143"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="144"><highlight class="normal"><sp/><sp/></highlight><highlight class="comment">//<sp/>Close</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="145"><highlight class="normal"><sp/><sp/>ASSERT_THAT(writer->Close(),<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="146"><highlight class="normal">}</highlight></codeline> |
| <codeline lineno="147"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="148"><highlight class="normal">TEST_F(DataWriterTest,<sp/>MetadataAfterClose)<sp/>{</highlight></codeline> |
| <codeline lineno="149"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>writer_result<sp/>=<sp/><ref refid="classiceberg_1_1DataWriter_1a75a3d11a5978812e66da37875f54fb98" kindref="member">DataWriter::Make</ref>(MakeDefaultOptions());</highlight></codeline> |
| <codeline lineno="150"><highlight class="normal"><sp/><sp/>ASSERT_THAT(writer_result,<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="151"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>writer<sp/>=<sp/>std::move(writer_result.value());</highlight></codeline> |
| <codeline lineno="152"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="153"><highlight class="normal"><sp/><sp/>WriteTestDataToWriter(writer.get());</highlight></codeline> |
| <codeline lineno="154"><highlight class="normal"><sp/><sp/>ASSERT_THAT(writer->Close(),<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="155"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="156"><highlight class="normal"><sp/><sp/></highlight><highlight class="comment">//<sp/>Get<sp/>metadata</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="157"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>metadata_result<sp/>=<sp/>writer->Metadata();</highlight></codeline> |
| <codeline lineno="158"><highlight class="normal"><sp/><sp/>ASSERT_THAT(metadata_result,<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="159"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="160"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">const</highlight><highlight class="normal"><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal">&<sp/>write_result<sp/>=<sp/>metadata_result.value();</highlight></codeline> |
| <codeline lineno="161"><highlight class="normal"><sp/><sp/>ASSERT_EQ(write_result.data_files.size(),<sp/>1);</highlight></codeline> |
| <codeline lineno="162"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="163"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">const</highlight><highlight class="normal"><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal">&<sp/>data_file<sp/>=<sp/>write_result.data_files[0];</highlight></codeline> |
| <codeline lineno="164"><highlight class="normal"><sp/><sp/>EXPECT_EQ(data_file->content,<sp/>DataFile::Content::kData);</highlight></codeline> |
| <codeline lineno="165"><highlight class="normal"><sp/><sp/>EXPECT_EQ(data_file->file_path,<sp/></highlight><highlight class="stringliteral">"test_data.parquet"</highlight><highlight class="normal">);</highlight></codeline> |
| <codeline lineno="166"><highlight class="normal"><sp/><sp/>EXPECT_EQ(data_file->file_format,<sp/>FileFormatType::kParquet);</highlight></codeline> |
| <codeline lineno="167"><highlight class="normal"><sp/><sp/>EXPECT_GT(data_file->file_size_in_bytes,<sp/>0);</highlight></codeline> |
| <codeline lineno="168"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="169"><highlight class="normal"><sp/><sp/></highlight><highlight class="comment">//<sp/>Metrics<sp/>availability<sp/>depends<sp/>on<sp/>the<sp/>underlying<sp/>writer<sp/>implementation</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="170"><highlight class="normal"><sp/><sp/>EXPECT_GE(data_file->column_sizes.size(),<sp/>0);</highlight></codeline> |
| <codeline lineno="171"><highlight class="normal"><sp/><sp/>EXPECT_GE(data_file->value_counts.size(),<sp/>0);</highlight></codeline> |
| <codeline lineno="172"><highlight class="normal"><sp/><sp/>EXPECT_GE(data_file->null_value_counts.size(),<sp/>0);</highlight></codeline> |
| <codeline lineno="173"><highlight class="normal">}</highlight></codeline> |
| <codeline lineno="174"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="175"><highlight class="normal">TEST_F(DataWriterTest,<sp/>MetadataBeforeCloseReturnsError)<sp/>{</highlight></codeline> |
| <codeline lineno="176"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>writer_result<sp/>=<sp/><ref refid="classiceberg_1_1DataWriter_1a75a3d11a5978812e66da37875f54fb98" kindref="member">DataWriter::Make</ref>(MakeDefaultOptions());</highlight></codeline> |
| <codeline lineno="177"><highlight class="normal"><sp/><sp/>ASSERT_THAT(writer_result,<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="178"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>writer<sp/>=<sp/>std::move(writer_result.value());</highlight></codeline> |
| <codeline lineno="179"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="180"><highlight class="normal"><sp/><sp/></highlight><highlight class="comment">//<sp/>Try<sp/>to<sp/>get<sp/>metadata<sp/>before<sp/>closing</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="181"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>metadata_result<sp/>=<sp/>writer->Metadata();</highlight></codeline> |
| <codeline lineno="182"><highlight class="normal"><sp/><sp/>ASSERT_THAT(metadata_result,<sp/>IsError(ErrorKind::kValidationFailed));</highlight></codeline> |
| <codeline lineno="183"><highlight class="normal"><sp/><sp/>EXPECT_THAT(metadata_result,</highlight></codeline> |
| <codeline lineno="184"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/><sp/>HasErrorMessage(</highlight><highlight class="stringliteral">"Cannot<sp/>get<sp/>metadata<sp/>before<sp/>closing<sp/>the<sp/>writer"</highlight><highlight class="normal">));</highlight></codeline> |
| <codeline lineno="185"><highlight class="normal">}</highlight></codeline> |
| <codeline lineno="186"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="187"><highlight class="normal">TEST_F(DataWriterTest,<sp/>CloseIsIdempotent)<sp/>{</highlight></codeline> |
| <codeline lineno="188"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>writer_result<sp/>=<sp/><ref refid="classiceberg_1_1DataWriter_1a75a3d11a5978812e66da37875f54fb98" kindref="member">DataWriter::Make</ref>(MakeDefaultOptions());</highlight></codeline> |
| <codeline lineno="189"><highlight class="normal"><sp/><sp/>ASSERT_THAT(writer_result,<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="190"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>writer<sp/>=<sp/>std::move(writer_result.value());</highlight></codeline> |
| <codeline lineno="191"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="192"><highlight class="normal"><sp/><sp/>WriteTestDataToWriter(writer.get());</highlight></codeline> |
| <codeline lineno="193"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="194"><highlight class="normal"><sp/><sp/>ASSERT_THAT(writer->Close(),<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="195"><highlight class="normal"><sp/><sp/>ASSERT_THAT(writer->Close(),<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="196"><highlight class="normal"><sp/><sp/>ASSERT_THAT(writer->Close(),<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="197"><highlight class="normal">}</highlight></codeline> |
| <codeline lineno="198"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="199"><highlight class="normal">TEST_F(DataWriterTest,<sp/>SortOrderIdInMetadata)<sp/>{</highlight></codeline> |
| <codeline lineno="200"><highlight class="normal"><sp/><sp/></highlight><highlight class="comment">//<sp/>Test<sp/>with<sp/>explicit<sp/>sort<sp/>order<sp/>id</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="201"><highlight class="normal"><sp/><sp/>{</highlight></codeline> |
| <codeline lineno="202"><highlight class="normal"><sp/><sp/><sp/><sp/></highlight><highlight class="keyword">const</highlight><highlight class="normal"><sp/>int32_t<sp/>sort_order_id<sp/>=<sp/>42;</highlight></codeline> |
| <codeline lineno="203"><highlight class="normal"><sp/><sp/><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>writer_result<sp/>=<sp/><ref refid="classiceberg_1_1DataWriter_1a75a3d11a5978812e66da37875f54fb98" kindref="member">DataWriter::Make</ref>(MakeDefaultOptions(sort_order_id));</highlight></codeline> |
| <codeline lineno="204"><highlight class="normal"><sp/><sp/><sp/><sp/>ASSERT_THAT(writer_result,<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="205"><highlight class="normal"><sp/><sp/><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>writer<sp/>=<sp/>std::move(writer_result.value());</highlight></codeline> |
| <codeline lineno="206"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="207"><highlight class="normal"><sp/><sp/><sp/><sp/>WriteTestDataToWriter(writer.get());</highlight></codeline> |
| <codeline lineno="208"><highlight class="normal"><sp/><sp/><sp/><sp/>ASSERT_THAT(writer->Close(),<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="209"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="210"><highlight class="normal"><sp/><sp/><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>metadata_result<sp/>=<sp/>writer->Metadata();</highlight></codeline> |
| <codeline lineno="211"><highlight class="normal"><sp/><sp/><sp/><sp/>ASSERT_THAT(metadata_result,<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="212"><highlight class="normal"><sp/><sp/><sp/><sp/></highlight><highlight class="keyword">const</highlight><highlight class="normal"><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal">&<sp/>data_file<sp/>=<sp/>metadata_result.value().data_files[0];</highlight></codeline> |
| <codeline lineno="213"><highlight class="normal"><sp/><sp/><sp/><sp/>ASSERT_TRUE(data_file->sort_order_id.has_value());</highlight></codeline> |
| <codeline lineno="214"><highlight class="normal"><sp/><sp/><sp/><sp/>EXPECT_EQ(data_file->sort_order_id.value(),<sp/>sort_order_id);</highlight></codeline> |
| <codeline lineno="215"><highlight class="normal"><sp/><sp/>}</highlight></codeline> |
| <codeline lineno="216"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="217"><highlight class="normal"><sp/><sp/></highlight><highlight class="comment">//<sp/>Test<sp/>without<sp/>sort<sp/>order<sp/>id<sp/>(should<sp/>be<sp/>nullopt)</highlight><highlight class="normal"></highlight></codeline> |
| <codeline lineno="218"><highlight class="normal"><sp/><sp/>{</highlight></codeline> |
| <codeline lineno="219"><highlight class="normal"><sp/><sp/><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>writer_result<sp/>=<sp/><ref refid="classiceberg_1_1DataWriter_1a75a3d11a5978812e66da37875f54fb98" kindref="member">DataWriter::Make</ref>(MakeDefaultOptions());</highlight></codeline> |
| <codeline lineno="220"><highlight class="normal"><sp/><sp/><sp/><sp/>ASSERT_THAT(writer_result,<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="221"><highlight class="normal"><sp/><sp/><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>writer<sp/>=<sp/>std::move(writer_result.value());</highlight></codeline> |
| <codeline lineno="222"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="223"><highlight class="normal"><sp/><sp/><sp/><sp/>WriteTestDataToWriter(writer.get());</highlight></codeline> |
| <codeline lineno="224"><highlight class="normal"><sp/><sp/><sp/><sp/>ASSERT_THAT(writer->Close(),<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="225"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="226"><highlight class="normal"><sp/><sp/><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>metadata_result<sp/>=<sp/>writer->Metadata();</highlight></codeline> |
| <codeline lineno="227"><highlight class="normal"><sp/><sp/><sp/><sp/>ASSERT_THAT(metadata_result,<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="228"><highlight class="normal"><sp/><sp/><sp/><sp/></highlight><highlight class="keyword">const</highlight><highlight class="normal"><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal">&<sp/>data_file<sp/>=<sp/>metadata_result.value().data_files[0];</highlight></codeline> |
| <codeline lineno="229"><highlight class="normal"><sp/><sp/><sp/><sp/>EXPECT_FALSE(data_file->sort_order_id.has_value());</highlight></codeline> |
| <codeline lineno="230"><highlight class="normal"><sp/><sp/>}</highlight></codeline> |
| <codeline lineno="231"><highlight class="normal">}</highlight></codeline> |
| <codeline lineno="232"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="233"><highlight class="normal">TEST_F(DataWriterTest,<sp/>PartitionValuesPreserved)<sp/>{</highlight></codeline> |
| <codeline lineno="234"><highlight class="normal"><sp/><sp/>PartitionValues<sp/>partition_values({Literal::Int(42),<sp/>Literal::String(</highlight><highlight class="stringliteral">"test"</highlight><highlight class="normal">)});</highlight></codeline> |
| <codeline lineno="235"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="236"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>writer_result<sp/>=</highlight></codeline> |
| <codeline lineno="237"><highlight class="normal"><sp/><sp/><sp/><sp/><sp/><sp/><ref refid="classiceberg_1_1DataWriter_1a75a3d11a5978812e66da37875f54fb98" kindref="member">DataWriter::Make</ref>(MakeDefaultOptions(std::nullopt,<sp/>partition_values));</highlight></codeline> |
| <codeline lineno="238"><highlight class="normal"><sp/><sp/>ASSERT_THAT(writer_result,<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="239"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>writer<sp/>=<sp/>std::move(writer_result.value());</highlight></codeline> |
| <codeline lineno="240"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="241"><highlight class="normal"><sp/><sp/>WriteTestDataToWriter(writer.get());</highlight></codeline> |
| <codeline lineno="242"><highlight class="normal"><sp/><sp/>ASSERT_THAT(writer->Close(),<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="243"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="244"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>metadata_result<sp/>=<sp/>writer->Metadata();</highlight></codeline> |
| <codeline lineno="245"><highlight class="normal"><sp/><sp/>ASSERT_THAT(metadata_result,<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="246"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">const</highlight><highlight class="normal"><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal">&<sp/>data_file<sp/>=<sp/>metadata_result.value().data_files[0];</highlight></codeline> |
| <codeline lineno="247"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="248"><highlight class="normal"><sp/><sp/>EXPECT_EQ(data_file->partition.num_fields(),<sp/>partition_values.num_fields());</highlight></codeline> |
| <codeline lineno="249"><highlight class="normal"><sp/><sp/>EXPECT_EQ(data_file->partition.num_fields(),<sp/>2);</highlight></codeline> |
| <codeline lineno="250"><highlight class="normal">}</highlight></codeline> |
| <codeline lineno="251"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="252"><highlight class="normal">TEST_F(DataWriterTest,<sp/>WriteMultipleBatches)<sp/>{</highlight></codeline> |
| <codeline lineno="253"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>writer_result<sp/>=<sp/><ref refid="classiceberg_1_1DataWriter_1a75a3d11a5978812e66da37875f54fb98" kindref="member">DataWriter::Make</ref>(MakeDefaultOptions());</highlight></codeline> |
| <codeline lineno="254"><highlight class="normal"><sp/><sp/>ASSERT_THAT(writer_result,<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="255"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>writer<sp/>=<sp/>std::move(writer_result.value());</highlight></codeline> |
| <codeline lineno="256"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="257"><highlight class="normal"><sp/><sp/>WriteTestDataToWriter(writer.get());</highlight></codeline> |
| <codeline lineno="258"><highlight class="normal"><sp/><sp/>WriteTestDataToWriter(writer.get());</highlight></codeline> |
| <codeline lineno="259"><highlight class="normal"><sp/><sp/>ASSERT_THAT(writer->Close(),<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="260"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="261"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal"><sp/>metadata_result<sp/>=<sp/>writer->Metadata();</highlight></codeline> |
| <codeline lineno="262"><highlight class="normal"><sp/><sp/>ASSERT_THAT(metadata_result,<sp/>IsOk());</highlight></codeline> |
| <codeline lineno="263"><highlight class="normal"><sp/><sp/></highlight><highlight class="keyword">const</highlight><highlight class="normal"><sp/></highlight><highlight class="keyword">auto</highlight><highlight class="normal">&<sp/>data_file<sp/>=<sp/>metadata_result.value().data_files[0];</highlight></codeline> |
| <codeline lineno="264"><highlight class="normal"><sp/><sp/>EXPECT_GT(data_file->file_size_in_bytes,<sp/>0);</highlight></codeline> |
| <codeline lineno="265"><highlight class="normal">}</highlight></codeline> |
| <codeline lineno="266"><highlight class="normal"></highlight></codeline> |
| <codeline lineno="267"><highlight class="normal">}<sp/><sp/></highlight><highlight class="comment">//<sp/>namespace<sp/>iceberg</highlight><highlight class="normal"></highlight></codeline> |
| </programlisting> |
| <location file="iceberg/test/data_writer_test.cc"/> |
| </compounddef> |
| </doxygen> |
