665f69b80d
Soon, we will want to convert mutation fragments into json inside the scylla codebase, not just in tools. To avoid scylla-core code having to include tools/ (and link against it), move the low-level json utilities into mutation/.
117 lines
4.0 KiB
C++
117 lines
4.0 KiB
C++
/*
|
|
* Copyright (C) 2023-present ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* SPDX-License-Identifier: AGPL-3.0-or-later
|
|
*/
|
|
|
|
#include "dht/i_partitioner.hh"
|
|
#include "schema/schema_fwd.hh"
|
|
#include "utils/rjson.hh"
|
|
|
|
// has to be below the utils/rjson.hh include
|
|
#include <rapidjson/ostreamwrapper.h>
|
|
|
|
/*
|
|
* Utilities for converting mutations, mutation-fragments and their parts into json.
|
|
*/
|
|
|
|
class atomic_cell_or_collection;
|
|
class atomic_cell_view;
|
|
class counter_cell_view;
|
|
class row;
|
|
class row_marker;
|
|
class tombstone;
|
|
struct collection_mutation_view_description;
|
|
|
|
namespace mutation_json {
|
|
|
|
class json_writer {
|
|
using stream = rapidjson::BasicOStreamWrapper<std::ostream>;
|
|
using writer = rapidjson::Writer<stream, rjson::encoding, rjson::encoding, rjson::allocator>;
|
|
|
|
stream _stream;
|
|
writer _writer;
|
|
|
|
public:
|
|
json_writer(std::ostream& os = std::cout) : _stream(os), _writer(_stream)
|
|
{ }
|
|
|
|
writer& rjson_writer() { return _writer; }
|
|
|
|
// following the rapidjson method names here
|
|
bool Null() { return _writer.Null(); }
|
|
bool Bool(bool b) { return _writer.Bool(b); }
|
|
bool Int(int i) { return _writer.Int(i); }
|
|
bool Uint(unsigned i) { return _writer.Uint(i); }
|
|
bool Int64(int64_t i) { return _writer.Int64(i); }
|
|
bool Uint64(uint64_t i) { return _writer.Uint64(i); }
|
|
bool Double(double d) { return _writer.Double(d); }
|
|
bool RawNumber(std::string_view str) { return _writer.RawNumber(str.data(), str.size(), false); }
|
|
bool String(std::string_view str) { return _writer.String(str.data(), str.size(), false); }
|
|
bool StartObject() { return _writer.StartObject(); }
|
|
bool Key(std::string_view str) { return _writer.Key(str.data(), str.size(), false); }
|
|
bool EndObject(rapidjson::SizeType memberCount = 0) { return _writer.EndObject(memberCount); }
|
|
bool StartArray() { return _writer.StartArray(); }
|
|
bool EndArray(rapidjson::SizeType elementCount = 0) { return _writer.EndArray(elementCount); }
|
|
|
|
// scylla-specific extensions (still following rapidjson naming scheme for consistency)
|
|
template <typename T>
|
|
void AsString(const T& obj) {
|
|
String(fmt::format("{}", obj));
|
|
}
|
|
// partition or clustering key
|
|
template <typename KeyType>
|
|
void DataKey(const schema& schema, const KeyType& key, std::optional<dht::token> token = {}) {
|
|
StartObject();
|
|
if (token) {
|
|
Key("token");
|
|
AsString(*token);
|
|
}
|
|
Key("raw");
|
|
String(to_hex(key.representation()));
|
|
Key("value");
|
|
AsString(key.with_schema(schema));
|
|
EndObject();
|
|
}
|
|
void StartStream() {
|
|
StartObject();
|
|
Key("sstables");
|
|
StartObject();
|
|
}
|
|
void EndStream() {
|
|
EndObject();
|
|
EndObject();
|
|
}
|
|
};
|
|
|
|
class mutation_partition_json_writer {
|
|
const schema& _schema;
|
|
json_writer _writer;
|
|
|
|
private:
|
|
void write_each_collection_cell(const collection_mutation_view_description& mv, data_type type, std::function<void(atomic_cell_view, data_type)> func);
|
|
|
|
public:
|
|
explicit mutation_partition_json_writer(const schema& s, std::ostream& os = std::cout)
|
|
: _schema(s), _writer(os) {}
|
|
|
|
const schema& schema() const { return _schema; }
|
|
json_writer& writer() { return _writer; }
|
|
|
|
sstring to_string(gc_clock::time_point tp);
|
|
void write_atomic_cell_value(const atomic_cell_view& cell, data_type type);
|
|
void write_collection_value(const collection_mutation_view_description& mv, data_type type);
|
|
void write(gc_clock::duration ttl, gc_clock::time_point expiry);
|
|
void write(const tombstone& t);
|
|
void write(const row_marker& m);
|
|
void write(counter_cell_view cv);
|
|
void write(const atomic_cell_view& cell, data_type type, bool include_value = true);
|
|
void write(const collection_mutation_view_description& mv, data_type type, bool include_value = true);
|
|
void write(const atomic_cell_or_collection& cell, const column_definition& cdef, bool include_value = true);
|
|
void write(const row& r, column_kind kind, bool include_value = true);
|
|
};
|
|
|
|
} // namespace mutation_json
|