mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-24 16:42:05 +00:00
67 lines
1.4 KiB
C++
67 lines
1.4 KiB
C++
#include "JSONString.h"
|
|
|
|
#include <regex>
|
|
#include <sstream>
|
|
namespace DB
|
|
{
|
|
|
|
namespace
|
|
{
|
|
std::string pad(size_t padding)
|
|
{
|
|
return std::string(padding * 4, ' ');
|
|
}
|
|
|
|
const std::regex NEW_LINE{"\n"};
|
|
}
|
|
|
|
void JSONString::set(const std::string & key, std::string value, bool wrap)
|
|
{
|
|
if (value.empty())
|
|
value = "null";
|
|
|
|
bool reserved = (value[0] == '[' || value[0] == '{' || value == "null");
|
|
if (!reserved && wrap)
|
|
value = '"' + std::regex_replace(value, NEW_LINE, "\\n") + '"';
|
|
|
|
content[key] = value;
|
|
}
|
|
|
|
void JSONString::set(const std::string & key, const std::vector<JSONString> & run_infos)
|
|
{
|
|
std::ostringstream value;
|
|
value << "[\n";
|
|
|
|
for (size_t i = 0; i < run_infos.size(); ++i)
|
|
{
|
|
value << pad(padding + 1) + run_infos[i].asString(padding + 2);
|
|
if (i != run_infos.size() - 1)
|
|
value << ',';
|
|
|
|
value << "\n";
|
|
}
|
|
|
|
value << pad(padding) << ']';
|
|
content[key] = value.str();
|
|
}
|
|
|
|
std::string JSONString::asString(size_t cur_padding) const
|
|
{
|
|
std::ostringstream repr;
|
|
repr << "{";
|
|
|
|
for (auto it = content.begin(); it != content.end(); ++it)
|
|
{
|
|
if (it != content.begin())
|
|
repr << ',';
|
|
/// construct "key": "value" string with padding
|
|
repr << "\n" << pad(cur_padding) << '"' << it->first << '"' << ": " << it->second;
|
|
}
|
|
|
|
repr << "\n" << pad(cur_padding - 1) << '}';
|
|
return repr.str();
|
|
}
|
|
|
|
|
|
}
|