#pragma once #include #include #include #include #include #include namespace DB { /** A stream for outputting data in XML format. */ class XMLRowOutputFormat : public IRowOutputFormat { public: XMLRowOutputFormat(WriteBuffer & out_, const Block & header_, const RowOutputFormatParams & params, const FormatSettings & format_settings_); String getName() const override { return "XMLRowOutputFormat"; } void writeField(const IColumn & column, const IDataType & type, size_t row_num) override; void writeRowStartDelimiter() override; void writeRowEndDelimiter() override; void writePrefix() override; void writeSuffix() override; void writeLastSuffix() override; void writeMinExtreme(const Columns & columns, size_t row_num) override; void writeMaxExtreme(const Columns & columns, size_t row_num) override; void writeTotals(const Columns & columns, size_t row_num) override; void writeBeforeTotals() override; void writeAfterTotals() override; void writeBeforeExtremes() override; void writeAfterExtremes() override; void flush() override { ostr->next(); if (validating_ostr) out.next(); } void setRowsBeforeLimit(size_t rows_before_limit_) override { applied_limit = true; rows_before_limit = rows_before_limit_; } void onProgress(const Progress & value) override; String getContentType() const override { return "application/xml; charset=UTF-8"; } protected: void writeExtremesElement(const char * title, const Columns & columns, size_t row_num); void writeRowsBeforeLimitAtLeast(); void writeStatistics(); std::unique_ptr validating_ostr; /// Validates UTF-8 sequences, replaces bad sequences with replacement character. WriteBuffer * ostr; size_t field_number = 0; size_t row_count = 0; bool applied_limit = false; size_t rows_before_limit = 0; NamesAndTypes fields; Names field_tag_names; Progress progress; Stopwatch watch; const FormatSettings format_settings; }; }