mirrors/scylladb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

schema: add implementation of computing token column

Browse Source 
Computed column of 'token' type can now have its value computed.
This commit is contained in:
Piotr Sarna
2019-03-22 15:14:01 +01:00
parent 491b7a817f
commit 65bf6d34fe
2 changed files with 36 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
column_computation.hh
View File

@@ -52,3 +52,12 @@ public:
virtual bytes serialize() const = 0;
virtual bytes_opt compute_value(const schema& schema, const partition_key& key, const clustering_row& row) const = 0;
};
class token_column_computation : public column_computation {
public:
virtual column_computation_ptr clone() const override {
return std::make_unique<token_column_computation>(*this);
}
virtual bytes serialize() const override;
virtual bytes_opt compute_value(const schema& schema, const partition_key& key, const clustering_row& row) const override;
};

28
schema.cc
View File

@@ -1253,7 +1253,33 @@ raw_view_info::raw_view_info(utils::UUID base_id, sstring base_name, bool includ
{ }
column_computation_ptr column_computation::deserialize(bytes_view raw) {
throw std::runtime_error("Incorrect column computation value");
return deserialize(json::to_json_value(sstring(raw.begin(), raw.end())));
}
column_computation_ptr column_computation::deserialize(const Json::Value& parsed) {
if (!parsed.isObject()) {
throw std::runtime_error(format("Invalid column computation value: {}", parsed.toStyledString()));
}
Json::Value type_json = parsed.get("type", Json::Value());
if (!type_json.isString()) {
throw std::runtime_error(format("Type {} is not convertible to string", type_json.toStyledString()));
}
sstring type = type_json.asString();
if (type == "token") {
return std::make_unique<token_column_computation>();
}
throw std::runtime_error(format("Incorrect column computation type {} found when parsing {}", type, parsed.toStyledString()));
}
bytes token_column_computation::serialize() const {
Json::Value serialized(Json::objectValue);
serialized["type"] = Json::Value("token");
return to_bytes(json::to_sstring(serialized));
}
bytes_opt token_column_computation::compute_value(const schema& schema, const partition_key& key, const clustering_row& row) const {
dht::i_partitioner& partitioner = dht::global_partitioner();
return partitioner.token_to_bytes(partitioner.get_token(schema, key));
}
bool operator==(const raw_view_info& x, const raw_view_info& y) {