mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-30 03:22:14 +00:00
d36f52502e
It's still hackish and dirty, but server and client compies. Server starts, but throwes meaningless exception on any query. Client seems to be working fine. Linux compilation might (but shouldn't) be broken (not tested).
817 lines
24 KiB
C++
817 lines
24 KiB
C++
#pragma once
|
||
|
||
#include <cstring>
|
||
#include <cstdio>
|
||
#include <limits>
|
||
#include <algorithm>
|
||
|
||
#include <common/Common.h>
|
||
#include <common/DateLUT.h>
|
||
#include <common/find_first_symbols.h>
|
||
|
||
#include <mysqlxx/Row.h>
|
||
#include <mysqlxx/Null.h>
|
||
|
||
#include <DB/Core/Types.h>
|
||
#include <DB/Common/Exception.h>
|
||
#include <DB/Common/StringUtils.h>
|
||
#include <DB/Core/StringRef.h>
|
||
|
||
#include <DB/IO/WriteBuffer.h>
|
||
#include <DB/IO/WriteIntText.h>
|
||
#include <DB/IO/VarInt.h>
|
||
#include <DB/IO/WriteBufferFromString.h>
|
||
#include <DB/IO/DoubleConverter.h>
|
||
#include <city.h>
|
||
|
||
|
||
namespace DB
|
||
{
|
||
|
||
namespace ErrorCodes
|
||
{
|
||
extern const int CANNOT_PRINT_FLOAT_OR_DOUBLE_NUMBER;
|
||
}
|
||
|
||
/// Функции-помошники для форматированной записи
|
||
|
||
inline void writeChar(char x, WriteBuffer & buf)
|
||
{
|
||
buf.nextIfAtEnd();
|
||
*buf.position() = x;
|
||
++buf.position();
|
||
}
|
||
|
||
|
||
/// Запись POD-типа в native формате
|
||
template <typename T>
|
||
inline void writePODBinary(const T & x, WriteBuffer & buf)
|
||
{
|
||
buf.write(reinterpret_cast<const char *>(&x), sizeof(x));
|
||
}
|
||
|
||
template <typename T>
|
||
inline void writeIntBinary(const T & x, WriteBuffer & buf)
|
||
{
|
||
writePODBinary(x, buf);
|
||
}
|
||
|
||
template <typename T>
|
||
inline void writeFloatBinary(const T & x, WriteBuffer & buf)
|
||
{
|
||
writePODBinary(x, buf);
|
||
}
|
||
|
||
|
||
inline void writeStringBinary(const std::string & s, WriteBuffer & buf)
|
||
{
|
||
writeVarUInt(s.size(), buf);
|
||
buf.write(s.data(), s.size());
|
||
}
|
||
|
||
inline void writeStringBinary(const char * s, WriteBuffer & buf)
|
||
{
|
||
writeVarUInt(strlen(s), buf);
|
||
buf.write(s, strlen(s));
|
||
}
|
||
|
||
inline void writeStringBinary(const StringRef & s, WriteBuffer & buf)
|
||
{
|
||
writeVarUInt(s.size, buf);
|
||
buf.write(s.data, s.size);
|
||
}
|
||
|
||
|
||
template <typename T>
|
||
void writeVectorBinary(const std::vector<T> & v, WriteBuffer & buf)
|
||
{
|
||
writeVarUInt(v.size(), buf);
|
||
|
||
for (typename std::vector<T>::const_iterator it = v.begin(); it != v.end(); ++it)
|
||
writeBinary(*it, buf);
|
||
}
|
||
|
||
|
||
inline void writeBoolText(bool x, WriteBuffer & buf)
|
||
{
|
||
writeChar(x ? '1' : '0', buf);
|
||
}
|
||
|
||
|
||
inline void writeFloatText(double x, WriteBuffer & buf)
|
||
{
|
||
DoubleConverter<false>::BufferType buffer;
|
||
double_conversion::StringBuilder builder{buffer, sizeof(buffer)};
|
||
|
||
const auto result = DoubleConverter<false>::instance().ToShortest(x, &builder);
|
||
|
||
if (!result)
|
||
throw Exception("Cannot print double number", ErrorCodes::CANNOT_PRINT_FLOAT_OR_DOUBLE_NUMBER);
|
||
|
||
buf.write(buffer, builder.position());
|
||
}
|
||
|
||
inline void writeFloatText(float x, WriteBuffer & buf)
|
||
{
|
||
DoubleConverter<false>::BufferType buffer;
|
||
double_conversion::StringBuilder builder{buffer, sizeof(buffer)};
|
||
|
||
const auto result = DoubleConverter<false>::instance().ToShortestSingle(x, &builder);
|
||
|
||
if (!result)
|
||
throw Exception("Cannot print float number", ErrorCodes::CANNOT_PRINT_FLOAT_OR_DOUBLE_NUMBER);
|
||
|
||
buf.write(buffer, builder.position());
|
||
}
|
||
|
||
|
||
inline void writeString(const String & s, WriteBuffer & buf)
|
||
{
|
||
buf.write(s.data(), s.size());
|
||
}
|
||
|
||
inline void writeString(const char * data, size_t size, WriteBuffer & buf)
|
||
{
|
||
buf.write(data, size);
|
||
}
|
||
|
||
inline void writeString(const StringRef & ref, WriteBuffer & buf)
|
||
{
|
||
writeString(ref.data, ref.size, buf);
|
||
}
|
||
|
||
|
||
/** Пишет С-строку без создания временного объекта. Если строка - литерал, то strlen выполняется на этапе компиляции.
|
||
* Используйте, когда строка - литерал.
|
||
*/
|
||
#define writeCString(s, buf) \
|
||
(buf).write((s), strlen(s))
|
||
|
||
/** Пишет строку для использования в формате JSON:
|
||
* - строка выводится в двойных кавычках
|
||
* - эскейпится символ прямого слеша '/'
|
||
* - байты из диапазона 0x00-0x1F кроме '\b', '\f', '\n', '\r', '\t' эскейпятся как \u00XX
|
||
* - кодовые точки U+2028 и U+2029 (последовательности байт в UTF-8: e2 80 a8, e2 80 a9) эскейпятся как \u2028 и \u2029
|
||
* - предполагается, что строка в кодировке UTF-8, невалидный UTF-8 не обрабатывается
|
||
* - не-ASCII символы остаются как есть
|
||
*/
|
||
inline void writeJSONString(const char * begin, const char * end, WriteBuffer & buf)
|
||
{
|
||
writeChar('"', buf);
|
||
for (const char * it = begin; it != end; ++it)
|
||
{
|
||
switch (*it)
|
||
{
|
||
case '\b':
|
||
writeChar('\\', buf);
|
||
writeChar('b', buf);
|
||
break;
|
||
case '\f':
|
||
writeChar('\\', buf);
|
||
writeChar('f', buf);
|
||
break;
|
||
case '\n':
|
||
writeChar('\\', buf);
|
||
writeChar('n', buf);
|
||
break;
|
||
case '\r':
|
||
writeChar('\\', buf);
|
||
writeChar('r', buf);
|
||
break;
|
||
case '\t':
|
||
writeChar('\\', buf);
|
||
writeChar('t', buf);
|
||
break;
|
||
case '\\':
|
||
writeChar('\\', buf);
|
||
writeChar('\\', buf);
|
||
break;
|
||
case '/':
|
||
writeChar('\\', buf);
|
||
writeChar('/', buf);
|
||
break;
|
||
case '"':
|
||
writeChar('\\', buf);
|
||
writeChar('"', buf);
|
||
break;
|
||
default:
|
||
if (0x00 <= *it && *it <= 0x1F)
|
||
{
|
||
char higher_half = (*it) >> 4;
|
||
char lower_half = (*it) & 0xF;
|
||
|
||
writeCString("\\u00", buf);
|
||
writeChar('0' + higher_half, buf);
|
||
|
||
if (0 <= lower_half && lower_half <= 9)
|
||
writeChar('0' + lower_half, buf);
|
||
else
|
||
writeChar('A' + lower_half - 10, buf);
|
||
}
|
||
else if (end - it >= 3 && it[0] == '\xE2' && it[1] == '\x80' && (it[2] == '\xA8' || it[2] == '\xA9'))
|
||
{
|
||
if (it[2] == '\xA8')
|
||
writeCString("\\u2028", buf);
|
||
if (it[2] == '\xA9')
|
||
writeCString("\\u2029", buf);
|
||
}
|
||
else
|
||
writeChar(*it, buf);
|
||
}
|
||
}
|
||
writeChar('"', buf);
|
||
}
|
||
|
||
|
||
template <char c>
|
||
void writeAnyEscapedString(const char * begin, const char * end, WriteBuffer & buf)
|
||
{
|
||
const char * pos = begin;
|
||
while (true)
|
||
{
|
||
/// Специально будем эскейпить больше символов, чем минимально необходимо.
|
||
const char * next_pos = find_first_symbols<'\b', '\f', '\n', '\r', '\t', '\0', '\\', c>(pos, end);
|
||
|
||
if (next_pos == end)
|
||
{
|
||
buf.write(pos, next_pos - pos);
|
||
break;
|
||
}
|
||
else
|
||
{
|
||
buf.write(pos, next_pos - pos);
|
||
pos = next_pos;
|
||
switch (*pos)
|
||
{
|
||
case '\b':
|
||
writeChar('\\', buf);
|
||
writeChar('b', buf);
|
||
break;
|
||
case '\f':
|
||
writeChar('\\', buf);
|
||
writeChar('f', buf);
|
||
break;
|
||
case '\n':
|
||
writeChar('\\', buf);
|
||
writeChar('n', buf);
|
||
break;
|
||
case '\r':
|
||
writeChar('\\', buf);
|
||
writeChar('r', buf);
|
||
break;
|
||
case '\t':
|
||
writeChar('\\', buf);
|
||
writeChar('t', buf);
|
||
break;
|
||
case '\0':
|
||
writeChar('\\', buf);
|
||
writeChar('0', buf);
|
||
break;
|
||
case '\\':
|
||
writeChar('\\', buf);
|
||
writeChar('\\', buf);
|
||
break;
|
||
case c:
|
||
writeChar('\\', buf);
|
||
writeChar(c, buf);
|
||
break;
|
||
default:
|
||
writeChar(*pos, buf);
|
||
}
|
||
++pos;
|
||
}
|
||
}
|
||
}
|
||
|
||
|
||
inline void writeJSONString(const String & s, WriteBuffer & buf)
|
||
{
|
||
writeJSONString(s.data(), s.data() + s.size(), buf);
|
||
}
|
||
|
||
|
||
inline void writeJSONString(const StringRef & ref, WriteBuffer & buf)
|
||
{
|
||
writeJSONString(ref.data, ref.data + ref.size, buf);
|
||
}
|
||
|
||
|
||
template <char c>
|
||
void writeAnyEscapedString(const String & s, WriteBuffer & buf)
|
||
{
|
||
writeAnyEscapedString<c>(s.data(), s.data() + s.size(), buf);
|
||
}
|
||
|
||
|
||
inline void writeEscapedString(const char * str, size_t size, WriteBuffer & buf)
|
||
{
|
||
writeAnyEscapedString<'\''>(str, str + size, buf);
|
||
}
|
||
|
||
|
||
inline void writeEscapedString(const String & s, WriteBuffer & buf)
|
||
{
|
||
writeEscapedString(s.data(), s.size(), buf);
|
||
}
|
||
|
||
|
||
inline void writeEscapedString(const StringRef & ref, WriteBuffer & buf)
|
||
{
|
||
writeEscapedString(ref.data, ref.size, buf);
|
||
}
|
||
|
||
|
||
template <char c>
|
||
void writeAnyQuotedString(const char * begin, const char * end, WriteBuffer & buf)
|
||
{
|
||
writeChar(c, buf);
|
||
writeAnyEscapedString<c>(begin, end, buf);
|
||
writeChar(c, buf);
|
||
}
|
||
|
||
|
||
|
||
template <char c>
|
||
void writeAnyQuotedString(const String & s, WriteBuffer & buf)
|
||
{
|
||
writeAnyQuotedString<c>(s.data(), s.data() + s.size(), buf);
|
||
}
|
||
|
||
|
||
template <char c>
|
||
void writeAnyQuotedString(const StringRef & ref, WriteBuffer & buf)
|
||
{
|
||
writeAnyQuotedString<c>(ref.data, ref.data + ref.size, buf);
|
||
}
|
||
|
||
|
||
inline void writeQuotedString(const String & s, WriteBuffer & buf)
|
||
{
|
||
writeAnyQuotedString<'\''>(s, buf);
|
||
}
|
||
|
||
|
||
inline void writeQuotedString(const StringRef & ref, WriteBuffer & buf)
|
||
{
|
||
writeAnyQuotedString<'\''>(ref, buf);
|
||
}
|
||
|
||
inline void writeDoubleQuotedString(const String & s, WriteBuffer & buf)
|
||
{
|
||
writeAnyQuotedString<'"'>(s, buf);
|
||
}
|
||
|
||
/// Выводит строку в обратных кавычках, как идентификатор в MySQL.
|
||
inline void writeBackQuotedString(const String & s, WriteBuffer & buf)
|
||
{
|
||
writeAnyQuotedString<'`'>(s, buf);
|
||
}
|
||
|
||
/// То же самое, но обратные кавычки применяются только при наличии символов, не подходящих для идентификатора без обратных кавычек.
|
||
inline void writeProbablyBackQuotedString(const String & s, WriteBuffer & buf)
|
||
{
|
||
if (s.empty() || !isWordCharASCII(s[0]))
|
||
writeBackQuotedString(s, buf);
|
||
else
|
||
{
|
||
const char * pos = s.data() + 1;
|
||
const char * end = s.data() + s.size();
|
||
for (; pos < end; ++pos)
|
||
if (!isWordCharASCII(*pos))
|
||
break;
|
||
if (pos != end)
|
||
writeBackQuotedString(s, buf);
|
||
else
|
||
writeString(s, buf);
|
||
}
|
||
}
|
||
|
||
|
||
/** Выводит строку в для формата CSV.
|
||
* Правила:
|
||
* - строка выводится в кавычках;
|
||
* - кавычка внутри строки выводится как две кавычки подряд.
|
||
*/
|
||
template <char quote = '"'>
|
||
void writeCSVString(const char * begin, const char * end, WriteBuffer & buf)
|
||
{
|
||
writeChar(quote, buf);
|
||
|
||
const char * pos = begin;
|
||
while (true)
|
||
{
|
||
const char * next_pos = find_first_symbols<quote>(pos, end);
|
||
|
||
if (next_pos == end)
|
||
{
|
||
buf.write(pos, end - pos);
|
||
break;
|
||
}
|
||
else /// Кавычка.
|
||
{
|
||
++next_pos;
|
||
buf.write(pos, next_pos - pos);
|
||
writeChar(quote, buf);
|
||
}
|
||
|
||
pos = next_pos;
|
||
}
|
||
|
||
writeChar(quote, buf);
|
||
}
|
||
|
||
template <char quote = '"'>
|
||
void writeCSVString(const String & s, WriteBuffer & buf)
|
||
{
|
||
writeCSVString<quote>(s.data(), s.data() + s.size(), buf);
|
||
}
|
||
|
||
template <char quote = '"'>
|
||
void writeCSVString(const StringRef & s, WriteBuffer & buf)
|
||
{
|
||
writeCSVString<quote>(s.data, s.data + s.size, buf);
|
||
}
|
||
|
||
|
||
/// Запись строки в текстовый узел в XML (не в атрибут - иначе нужно больше эскейпинга).
|
||
inline void writeXMLString(const char * begin, const char * end, WriteBuffer & buf)
|
||
{
|
||
const char * pos = begin;
|
||
while (true)
|
||
{
|
||
/// NOTE Возможно, для некоторых парсеров XML нужно ещё эскейпить нулевой байт и некоторые control characters.
|
||
const char * next_pos = find_first_symbols<'<', '&'>(pos, end);
|
||
|
||
if (next_pos == end)
|
||
{
|
||
buf.write(pos, end - pos);
|
||
break;
|
||
}
|
||
else if (*next_pos == '<')
|
||
{
|
||
buf.write(pos, next_pos - pos);
|
||
++next_pos;
|
||
writeCString("<", buf);
|
||
}
|
||
else if (*next_pos == '&')
|
||
{
|
||
buf.write(pos, next_pos - pos);
|
||
++next_pos;
|
||
writeCString("&", buf);
|
||
}
|
||
|
||
pos = next_pos;
|
||
}
|
||
}
|
||
|
||
inline void writeXMLString(const String & s, WriteBuffer & buf)
|
||
{
|
||
writeXMLString(s.data(), s.data() + s.size(), buf);
|
||
}
|
||
|
||
inline void writeXMLString(const StringRef & s, WriteBuffer & buf)
|
||
{
|
||
writeXMLString(s.data, s.data + s.size, buf);
|
||
}
|
||
|
||
|
||
/// в формате YYYY-MM-DD
|
||
inline void writeDateText(DayNum_t date, WriteBuffer & buf)
|
||
{
|
||
char s[10] = {'0', '0', '0', '0', '-', '0', '0', '-', '0', '0'};
|
||
|
||
if (unlikely(date > DATE_LUT_MAX_DAY_NUM || date == 0))
|
||
{
|
||
buf.write(s, 10);
|
||
return;
|
||
}
|
||
|
||
const auto & values = DateLUT::instance().getValues(date);
|
||
|
||
s[0] += values.year / 1000;
|
||
s[1] += (values.year / 100) % 10;
|
||
s[2] += (values.year / 10) % 10;
|
||
s[3] += values.year % 10;
|
||
s[5] += values.month / 10;
|
||
s[6] += values.month % 10;
|
||
s[8] += values.day_of_month / 10;
|
||
s[9] += values.day_of_month % 10;
|
||
|
||
buf.write(s, 10);
|
||
}
|
||
|
||
inline void writeDateText(LocalDate date, WriteBuffer & buf)
|
||
{
|
||
char s[10] = {'0', '0', '0', '0', '-', '0', '0', '-', '0', '0'};
|
||
|
||
s[0] += date.year() / 1000;
|
||
s[1] += (date.year() / 100) % 10;
|
||
s[2] += (date.year() / 10) % 10;
|
||
s[3] += date.year() % 10;
|
||
s[5] += date.month() / 10;
|
||
s[6] += date.month() % 10;
|
||
s[8] += date.day() / 10;
|
||
s[9] += date.day() % 10;
|
||
|
||
buf.write(s, 10);
|
||
}
|
||
|
||
|
||
/// в формате YYYY-MM-DD HH:MM:SS, согласно текущему часовому поясу
|
||
template <char date_delimeter = '-', char time_delimeter = ':'>
|
||
inline void writeDateTimeText(time_t datetime, WriteBuffer & buf)
|
||
{
|
||
char s[19] = {'0', '0', '0', '0', date_delimeter, '0', '0', date_delimeter, '0', '0', ' ', '0', '0', time_delimeter, '0', '0', time_delimeter, '0', '0'};
|
||
|
||
if (unlikely(datetime > DATE_LUT_MAX || datetime == 0))
|
||
{
|
||
buf.write(s, 19);
|
||
return;
|
||
}
|
||
|
||
const auto & date_lut = DateLUT::instance();
|
||
const auto & values = date_lut.getValues(datetime);
|
||
|
||
s[0] += values.year / 1000;
|
||
s[1] += (values.year / 100) % 10;
|
||
s[2] += (values.year / 10) % 10;
|
||
s[3] += values.year % 10;
|
||
s[5] += values.month / 10;
|
||
s[6] += values.month % 10;
|
||
s[8] += values.day_of_month / 10;
|
||
s[9] += values.day_of_month % 10;
|
||
|
||
UInt8 hour = date_lut.toHourInaccurate(datetime);
|
||
UInt8 minute = date_lut.toMinuteInaccurate(datetime);
|
||
UInt8 second = date_lut.toSecondInaccurate(datetime);
|
||
|
||
s[11] += hour / 10;
|
||
s[12] += hour % 10;
|
||
s[14] += minute / 10;
|
||
s[15] += minute % 10;
|
||
s[17] += second / 10;
|
||
s[18] += second % 10;
|
||
|
||
buf.write(s, 19);
|
||
}
|
||
|
||
template <char date_delimeter = '-', char time_delimeter = ':'>
|
||
inline void writeDateTimeText(LocalDateTime datetime, WriteBuffer & buf)
|
||
{
|
||
char s[19] = {'0', '0', '0', '0', date_delimeter, '0', '0', date_delimeter, '0', '0', ' ', '0', '0', time_delimeter, '0', '0', time_delimeter, '0', '0'};
|
||
|
||
s[0] += datetime.year() / 1000;
|
||
s[1] += (datetime.year() / 100) % 10;
|
||
s[2] += (datetime.year() / 10) % 10;
|
||
s[3] += datetime.year() % 10;
|
||
s[5] += datetime.month() / 10;
|
||
s[6] += datetime.month() % 10;
|
||
s[8] += datetime.day() / 10;
|
||
s[9] += datetime.day() % 10;
|
||
|
||
s[11] += datetime.hour() / 10;
|
||
s[12] += datetime.hour() % 10;
|
||
s[14] += datetime.minute() / 10;
|
||
s[15] += datetime.minute() % 10;
|
||
s[17] += datetime.second() / 10;
|
||
s[18] += datetime.second() % 10;
|
||
|
||
buf.write(s, 19);
|
||
}
|
||
|
||
|
||
/// Вывести mysqlxx::Row в tab-separated виде
|
||
inline void writeEscapedRow(const mysqlxx::Row & row, WriteBuffer & buf)
|
||
{
|
||
for (size_t i = 0; i < row.size(); ++i)
|
||
{
|
||
if (i != 0)
|
||
buf.write('\t');
|
||
|
||
if (unlikely(row[i].isNull()))
|
||
{
|
||
buf.write("\\N", 2);
|
||
continue;
|
||
}
|
||
|
||
writeAnyEscapedString<'\''>(row[i].data(), row[i].data() + row[i].length(), buf);
|
||
}
|
||
}
|
||
|
||
|
||
/// Методы вывода в бинарном виде
|
||
inline void writeBinary(const UInt8 & x, WriteBuffer & buf) { writePODBinary(x, buf); }
|
||
inline void writeBinary(const UInt16 & x, WriteBuffer & buf) { writePODBinary(x, buf); }
|
||
inline void writeBinary(const UInt32 & x, WriteBuffer & buf) { writePODBinary(x, buf); }
|
||
inline void writeBinary(const UInt64 & x, WriteBuffer & buf) { writePODBinary(x, buf); }
|
||
inline void writeBinary(const Int8 & x, WriteBuffer & buf) { writePODBinary(x, buf); }
|
||
inline void writeBinary(const Int16 & x, WriteBuffer & buf) { writePODBinary(x, buf); }
|
||
inline void writeBinary(const Int32 & x, WriteBuffer & buf) { writePODBinary(x, buf); }
|
||
inline void writeBinary(const Int64 & x, WriteBuffer & buf) { writePODBinary(x, buf); }
|
||
#ifdef __APPLE__
|
||
inline void writeBinary(const int64_t & x, WriteBuffer & buf) { writePODBinary(x, buf); }
|
||
inline void writeBinary(const uint64_t & x, WriteBuffer & buf) { writePODBinary(x, buf); }
|
||
#endif
|
||
inline void writeBinary(const Float32 & x, WriteBuffer & buf) { writePODBinary(x, buf); }
|
||
inline void writeBinary(const Float64 & x, WriteBuffer & buf) { writePODBinary(x, buf); }
|
||
inline void writeBinary(const String & x, WriteBuffer & buf) { writeStringBinary(x, buf); }
|
||
inline void writeBinary(const bool & x, WriteBuffer & buf) { writePODBinary(x, buf); }
|
||
inline void writeBinary(const uint128 & x, WriteBuffer & buf) { writePODBinary(x, buf); }
|
||
|
||
inline void writeBinary(const VisitID_t & x, WriteBuffer & buf) { writePODBinary(static_cast<const UInt64 &>(x), buf); }
|
||
inline void writeBinary(const LocalDate & x, WriteBuffer & buf) { writePODBinary(x, buf); }
|
||
inline void writeBinary(const LocalDateTime & x, WriteBuffer & buf) { writePODBinary(x, buf); }
|
||
|
||
|
||
/// Методы для вывода значения в текстовом виде для tab-separated формата.
|
||
inline void writeText(const UInt8 & x, WriteBuffer & buf) { writeIntText(x, buf); }
|
||
inline void writeText(const UInt16 & x, WriteBuffer & buf) { writeIntText(x, buf); }
|
||
inline void writeText(const UInt32 & x, WriteBuffer & buf) { writeIntText(x, buf); }
|
||
inline void writeText(const UInt64 & x, WriteBuffer & buf) { writeIntText(x, buf); }
|
||
inline void writeText(const Int8 & x, WriteBuffer & buf) { writeIntText(x, buf); }
|
||
inline void writeText(const Int16 & x, WriteBuffer & buf) { writeIntText(x, buf); }
|
||
inline void writeText(const Int32 & x, WriteBuffer & buf) { writeIntText(x, buf); }
|
||
inline void writeText(const Int64 & x, WriteBuffer & buf) { writeIntText(x, buf); }
|
||
#ifdef __APPLE__
|
||
inline void writeText(const int64_t & x, WriteBuffer & buf) { writeIntText(x, buf); }
|
||
inline void writeText(const uint64_t & x, WriteBuffer & buf) { writeIntText(x, buf); }
|
||
#endif
|
||
inline void writeText(const Float32 & x, WriteBuffer & buf) { writeFloatText(x, buf); }
|
||
inline void writeText(const Float64 & x, WriteBuffer & buf) { writeFloatText(x, buf); }
|
||
inline void writeText(const String & x, WriteBuffer & buf) { writeEscapedString(x, buf); }
|
||
inline void writeText(const bool & x, WriteBuffer & buf) { writeBoolText(x, buf); }
|
||
/// в отличие от метода для std::string
|
||
/// здесь предполагается, что x null-terminated строка.
|
||
inline void writeText(const char * x, WriteBuffer & buf) { writeEscapedString(x, strlen(x), buf); }
|
||
inline void writeText(const char * x, size_t size, WriteBuffer & buf) { writeEscapedString(x, size, buf); }
|
||
|
||
inline void writeText(const VisitID_t & x, WriteBuffer & buf) { writeIntText(static_cast<const UInt64 &>(x), buf); }
|
||
inline void writeText(const LocalDate & x, WriteBuffer & buf) { writeDateText(x, buf); }
|
||
inline void writeText(const LocalDateTime & x, WriteBuffer & buf) { writeDateTimeText(x, buf); }
|
||
|
||
template<typename T>
|
||
inline void writeText(const mysqlxx::Null<T> & x, WriteBuffer & buf)
|
||
{
|
||
if (x.isNull())
|
||
writeCString("\\N", buf);
|
||
else
|
||
writeText(static_cast<const T &>(x), buf);
|
||
}
|
||
|
||
|
||
/// Строки, даты, даты-с-временем - в одинарных кавычках с C-style эскейпингом. Числа - без.
|
||
inline void writeQuoted(const UInt8 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeQuoted(const UInt16 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeQuoted(const UInt32 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeQuoted(const UInt64 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeQuoted(const Int8 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeQuoted(const Int16 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeQuoted(const Int32 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeQuoted(const Int64 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeQuoted(const Float32 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeQuoted(const Float64 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeQuoted(const String & x, WriteBuffer & buf) { writeQuotedString(x, buf); }
|
||
inline void writeQuoted(const bool & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
|
||
inline void writeQuoted(const VisitID_t & x, WriteBuffer & buf)
|
||
{
|
||
writeIntText(static_cast<const UInt64 &>(x), buf);
|
||
}
|
||
|
||
inline void writeQuoted(const LocalDate & x, WriteBuffer & buf)
|
||
{
|
||
writeChar('\'', buf);
|
||
writeDateText(x, buf);
|
||
writeChar('\'', buf);
|
||
}
|
||
|
||
inline void writeQuoted(const LocalDateTime & x, WriteBuffer & buf)
|
||
{
|
||
writeChar('\'', buf);
|
||
writeDateTimeText(x, buf);
|
||
writeChar('\'', buf);
|
||
}
|
||
|
||
template <typename T>
|
||
inline void writeQuoted(const mysqlxx::Null<T> & x, WriteBuffer & buf)
|
||
{
|
||
if (x.isNull())
|
||
writeCString("NULL", buf);
|
||
else
|
||
writeText(static_cast<const T &>(x), buf);
|
||
}
|
||
|
||
|
||
/// Строки, даты, даты-с-временем - в двойных кавычках с C-style эскейпингом. Числа - без.
|
||
inline void writeDoubleQuoted(const UInt8 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeDoubleQuoted(const UInt16 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeDoubleQuoted(const UInt32 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeDoubleQuoted(const UInt64 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeDoubleQuoted(const Int8 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeDoubleQuoted(const Int16 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeDoubleQuoted(const Int32 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeDoubleQuoted(const Int64 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeDoubleQuoted(const Float32 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeDoubleQuoted(const Float64 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeDoubleQuoted(const String & x, WriteBuffer & buf) { writeDoubleQuotedString(x, buf); }
|
||
inline void writeDoubleQuoted(const bool & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
|
||
inline void writeDoubleQuoted(const VisitID_t & x, WriteBuffer & buf)
|
||
{
|
||
writeIntText(static_cast<const UInt64 &>(x), buf);
|
||
}
|
||
|
||
inline void writeDoubleQuoted(const LocalDate & x, WriteBuffer & buf)
|
||
{
|
||
writeChar('"', buf);
|
||
writeDateText(x, buf);
|
||
writeChar('"', buf);
|
||
}
|
||
|
||
inline void writeDoubleQuoted(const LocalDateTime & x, WriteBuffer & buf)
|
||
{
|
||
writeChar('"', buf);
|
||
writeDateTimeText(x, buf);
|
||
writeChar('"', buf);
|
||
}
|
||
|
||
|
||
/// Строки - в двойных кавычках и с CSV-эскейпингом; даты, даты-с-временем - в двойных кавычках. Числа - без.
|
||
inline void writeCSV(const UInt8 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeCSV(const UInt16 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeCSV(const UInt32 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeCSV(const UInt64 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeCSV(const Int8 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeCSV(const Int16 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeCSV(const Int32 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeCSV(const Int64 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeCSV(const Float32 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeCSV(const Float64 & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeCSV(const String & x, WriteBuffer & buf) { writeCSVString<>(x, buf); }
|
||
inline void writeCSV(const bool & x, WriteBuffer & buf) { writeText(x, buf); }
|
||
inline void writeCSV(const VisitID_t & x, WriteBuffer & buf) { writeDoubleQuoted(x, buf); }
|
||
inline void writeCSV(const LocalDate & x, WriteBuffer & buf) { writeDoubleQuoted(x, buf); }
|
||
inline void writeCSV(const LocalDateTime & x, WriteBuffer & buf) { writeDoubleQuoted(x, buf); }
|
||
|
||
|
||
template <typename T>
|
||
void writeBinary(const std::vector<T> & x, WriteBuffer & buf)
|
||
{
|
||
size_t size = x.size();
|
||
writeVarUInt(size, buf);
|
||
for (size_t i = 0; i < size; ++i)
|
||
writeBinary(x[i], buf);
|
||
}
|
||
|
||
template <typename T>
|
||
void writeQuoted(const std::vector<T> & x, WriteBuffer & buf)
|
||
{
|
||
writeChar('[', buf);
|
||
for (size_t i = 0, size = x.size(); i < size; ++i)
|
||
{
|
||
if (i != 0)
|
||
writeChar(',', buf);
|
||
writeQuoted(x[i], buf);
|
||
}
|
||
writeChar(']', buf);
|
||
}
|
||
|
||
template <typename T>
|
||
void writeDoubleQuoted(const std::vector<T> & x, WriteBuffer & buf)
|
||
{
|
||
writeChar('[', buf);
|
||
for (size_t i = 0, size = x.size(); i < size; ++i)
|
||
{
|
||
if (i != 0)
|
||
writeChar(',', buf);
|
||
writeDoubleQuoted(x[i], buf);
|
||
}
|
||
writeChar(']', buf);
|
||
}
|
||
|
||
template <typename T>
|
||
void writeText(const std::vector<T> & x, WriteBuffer & buf)
|
||
{
|
||
writeQuoted(x, buf);
|
||
}
|
||
|
||
|
||
|
||
/// Сериализация эксепшена (чтобы его можно было передать по сети)
|
||
void writeException(const Exception & e, WriteBuffer & buf);
|
||
|
||
|
||
/// Простой для использования метод преобразования чего-либо в строку в текстовом виде.
|
||
template <typename T>
|
||
inline String toString(const T & x)
|
||
{
|
||
String res;
|
||
{
|
||
WriteBufferFromString buf(res);
|
||
writeText(x, buf);
|
||
}
|
||
return res;
|
||
}
|
||
|
||
}
|