64 lines
1.8 KiB
C++
64 lines
1.8 KiB
C++
|
#include <string>
|
||
|
|
|
||
|
|
#include "communication/bolt/client.hpp"
|
||
|
|
#include "communication/bolt/v1/decoder/decoded_value.hpp"
|
||
|
|
|
||
|
|
namespace {
|
||
|
|
|
||
|
|
void PrintJsonDecodedValue(std::ostream &os,
|
||
|
|
const communication::bolt::DecodedValue &value) {
|
||
|
|
using communication::bolt::DecodedValue;
|
||
|
|
switch (value.type()) {
|
||
|
|
case DecodedValue::Type::Null:
|
||
|
|
os << "null";
|
||
|
|
break;
|
||
|
|
case DecodedValue::Type::Bool:
|
||
|
|
os << (value.ValueBool() ? "true" : "false");
|
||
|
|
break;
|
||
|
|
case DecodedValue::Type::Int:
|
||
|
|
os << value.ValueInt();
|
||
|
|
break;
|
||
|
|
case DecodedValue::Type::Double:
|
||
|
|
os << value.ValueDouble();
|
||
|
|
break;
|
||
|
|
case DecodedValue::Type::String:
|
||
|
|
os << "\"" << value.ValueString() << "\"";
|
||
|
|
break;
|
||
|
|
case DecodedValue::Type::List:
|
||
|
|
os << "[";
|
||
|
|
PrintIterable(os, value.ValueList(), ", ",
|
||
|
|
[](auto &stream, const auto &item) {
|
||
|
|
PrintJsonDecodedValue(stream, item);
|
||
|
|
});
|
||
|
|
os << "]";
|
||
|
|
break;
|
||
|
|
case DecodedValue::Type::Map:
|
||
|
|
os << "{";
|
||
|
|
PrintIterable(os, value.ValueMap(), ", ",
|
||
|
|
[](auto &stream, const auto &pair) {
|
||
|
|
PrintJsonDecodedValue(stream, {pair.first});
|
||
|
|
stream << ": ";
|
||
|
|
PrintJsonDecodedValue(stream, pair.second);
|
||
|
|
});
|
||
|
|
os << "}";
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
std::terminate();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
template <typename SocketT>
|
||
|
|
communication::bolt::QueryData ExecuteNTimesTillSuccess(
|
||
|
|
communication::bolt::Client<SocketT> &client, const std::string &query,
|
||
|
|
int times) {
|
||
|
|
for (int i = 0; i < times; ++i) {
|
||
|
|
try {
|
||
|
|
auto ret = client.Execute(query, {});
|
||
|
|
return ret;
|
||
|
|
} catch (const communication::bolt::ClientQueryException &e) {
|
||
|
|
}
|
||
|
|
}
|
||
|
|
throw communication::bolt::ClientQueryException();
|
||
|
|
}
|
||
|
|
}
|