2017-04-01 09:19:00 +00:00
|
|
|
#include <IO/WriteBuffer.h>
|
|
|
|
#include <IO/WriteHelpers.h>
|
|
|
|
#include <Core/Block.h>
|
2018-06-10 19:22:49 +00:00
|
|
|
#include <Formats/ODBCDriverBlockOutputStream.h>
|
|
|
|
#include <Formats/FormatFactory.h>
|
2016-01-13 02:20:12 +00:00
|
|
|
|
2017-01-21 04:24:28 +00:00
|
|
|
|
2016-01-13 02:20:12 +00:00
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
2018-06-08 01:51:55 +00:00
|
|
|
ODBCDriverBlockOutputStream::ODBCDriverBlockOutputStream(WriteBuffer & out_, const Block & header_, const FormatSettings & format_settings)
|
|
|
|
: out(out_), header(header_), format_settings(format_settings)
|
2017-04-19 13:42:58 +00:00
|
|
|
{
|
|
|
|
}
|
2016-01-13 02:20:12 +00:00
|
|
|
|
2016-08-13 01:57:35 +00:00
|
|
|
void ODBCDriverBlockOutputStream::flush()
|
|
|
|
{
|
2017-04-01 07:20:54 +00:00
|
|
|
out.next();
|
2016-08-13 01:57:35 +00:00
|
|
|
}
|
2016-01-13 02:20:12 +00:00
|
|
|
|
|
|
|
void ODBCDriverBlockOutputStream::write(const Block & block)
|
|
|
|
{
|
2017-04-19 13:42:58 +00:00
|
|
|
const size_t rows = block.rows();
|
|
|
|
const size_t columns = block.columns();
|
2017-04-01 07:20:54 +00:00
|
|
|
String text_value;
|
2017-04-19 13:42:58 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
for (size_t i = 0; i < rows; ++i)
|
|
|
|
{
|
|
|
|
for (size_t j = 0; j < columns; ++j)
|
|
|
|
{
|
|
|
|
text_value.resize(0);
|
|
|
|
const ColumnWithTypeAndName & col = block.getByPosition(j);
|
|
|
|
|
|
|
|
{
|
|
|
|
WriteBufferFromString text_out(text_value);
|
2018-12-13 13:41:47 +00:00
|
|
|
col.type->serializeAsText(*col.column, i, text_out, format_settings);
|
2017-04-01 07:20:54 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
writeStringBinary(text_value, out);
|
|
|
|
}
|
|
|
|
}
|
2016-01-13 02:20:12 +00:00
|
|
|
}
|
|
|
|
|
2017-04-19 13:42:58 +00:00
|
|
|
void ODBCDriverBlockOutputStream::writePrefix()
|
|
|
|
{
|
2018-02-19 00:45:32 +00:00
|
|
|
const size_t columns = header.columns();
|
2017-04-19 13:42:58 +00:00
|
|
|
|
|
|
|
/// Number of columns.
|
|
|
|
writeVarUInt(columns, out);
|
|
|
|
|
|
|
|
/// Names and types of columns.
|
|
|
|
for (size_t i = 0; i < columns; ++i)
|
|
|
|
{
|
2018-02-19 00:45:32 +00:00
|
|
|
const ColumnWithTypeAndName & col = header.getByPosition(i);
|
2017-04-19 13:42:58 +00:00
|
|
|
|
|
|
|
writeStringBinary(col.name, out);
|
|
|
|
writeStringBinary(col.type->getName(), out);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-06-10 19:22:49 +00:00
|
|
|
|
|
|
|
void registerOutputFormatODBCDriver(FormatFactory & factory)
|
|
|
|
{
|
|
|
|
factory.registerOutputFormat("ODBCDriver", [](
|
|
|
|
WriteBuffer & buf,
|
|
|
|
const Block & sample,
|
|
|
|
const Context &,
|
|
|
|
const FormatSettings & format_settings)
|
|
|
|
{
|
|
|
|
return std::make_shared<ODBCDriverBlockOutputStream>(buf, sample, format_settings);
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
2016-01-13 02:20:12 +00:00
|
|
|
}
|