ClickHouse/dbms/include/DB/DataStreams/JSONRowOutputStream.h

51 lines
1.0 KiB
C
Raw Normal View History

2013-05-15 11:18:58 +00:00
#pragma once
#include <Poco/SharedPtr.h>
#include <DB/Core/Block.h>
#include <DB/IO/WriteBuffer.h>
#include <DB/IO/WriteBufferValidUTF8.h>
#include <DB/DataStreams/IRowOutputStream.h>
2013-05-20 12:21:51 +00:00
#include <DB/DataStreams/IProfilingBlockInputStream.h>
2013-05-15 11:18:58 +00:00
namespace DB
{
/** Поток для вывода данных в формате JSON.
*/
class JSONRowOutputStream : public IRowOutputStream
{
public:
2013-05-22 14:57:43 +00:00
JSONRowOutputStream(WriteBuffer & ostr_, const Block & sample_);
2013-05-15 11:18:58 +00:00
void writeField(const Field & field);
void writeFieldDelimiter();
void writeRowStartDelimiter();
void writeRowEndDelimiter();
void writePrefix();
void writeSuffix();
2013-05-22 14:57:43 +00:00
void setRowsBeforeLimit(size_t rows_before_limit_)
{
applied_limit = true;
rows_before_limit = rows_before_limit_;
}
2013-05-15 11:18:58 +00:00
protected:
2013-05-20 12:21:51 +00:00
void writeRowsBeforeLimitAtLeast();
2013-05-15 11:18:58 +00:00
typedef std::vector<NameAndTypePair> NamesAndTypesVector;
WriteBufferValidUTF8 ostr;
2013-05-15 11:18:58 +00:00
size_t field_number;
2013-05-20 12:21:51 +00:00
size_t row_count;
2013-05-22 14:57:43 +00:00
bool applied_limit;
size_t rows_before_limit;
2013-05-20 12:21:51 +00:00
NamesAndTypesVector fields;
2013-05-15 11:18:58 +00:00
};
}