mirror of
https://github.com/typesense/typesense.git
synced 2025-05-19 21:22:25 +08:00
680 lines
28 KiB
C++
680 lines
28 KiB
C++
#include <store.h>
|
|
#include "field.h"
|
|
|
|
Option<bool> filter::parse_geopoint_filter_value(std::string& raw_value,
|
|
const std::string& format_err_msg,
|
|
std::string& processed_filter_val,
|
|
NUM_COMPARATOR& num_comparator) {
|
|
|
|
num_comparator = LESS_THAN_EQUALS;
|
|
|
|
if(!(raw_value[0] == '(' && raw_value[raw_value.size() - 1] == ')')) {
|
|
return Option<bool>(400, format_err_msg);
|
|
}
|
|
|
|
std::vector<std::string> filter_values;
|
|
auto raw_val_without_paran = raw_value.substr(1, raw_value.size() - 2);
|
|
StringUtils::split(raw_val_without_paran, filter_values, ",");
|
|
|
|
// we will end up with: "10.45 34.56 2 km" or "10.45 34.56 2mi" or a geo polygon
|
|
|
|
if(filter_values.size() < 3) {
|
|
return Option<bool>(400, format_err_msg);
|
|
}
|
|
|
|
// do validation: format should match either a point + radius or polygon
|
|
|
|
size_t num_floats = 0;
|
|
for(const auto& fvalue: filter_values) {
|
|
if(StringUtils::is_float(fvalue)) {
|
|
num_floats++;
|
|
}
|
|
}
|
|
|
|
bool is_polygon = (num_floats == filter_values.size());
|
|
if(!is_polygon) {
|
|
// we have to ensure that this is a point + radius match
|
|
if(!StringUtils::is_float(filter_values[0]) || !StringUtils::is_float(filter_values[1])) {
|
|
return Option<bool>(400, format_err_msg);
|
|
}
|
|
}
|
|
|
|
if(is_polygon) {
|
|
processed_filter_val = raw_val_without_paran;
|
|
} else {
|
|
// point + radius
|
|
// filter_values[2] is distance, get the unit, validate it and split on that
|
|
if(filter_values[2].size() < 2) {
|
|
return Option<bool>(400, "Unit must be either `km` or `mi`.");
|
|
}
|
|
|
|
std::string unit = filter_values[2].substr(filter_values[2].size()-2, 2);
|
|
|
|
if(unit != "km" && unit != "mi") {
|
|
return Option<bool>(400, "Unit must be either `km` or `mi`.");
|
|
}
|
|
|
|
std::vector<std::string> dist_values;
|
|
StringUtils::split(filter_values[2], dist_values, unit);
|
|
|
|
if(dist_values.size() != 1) {
|
|
return Option<bool>(400, format_err_msg);
|
|
}
|
|
|
|
if(!StringUtils::is_float(dist_values[0])) {
|
|
return Option<bool>(400, format_err_msg);
|
|
}
|
|
|
|
processed_filter_val = filter_values[0] + ", " + filter_values[1] + ", " + // co-ords
|
|
dist_values[0] + ", " + unit; // X km
|
|
}
|
|
|
|
return Option<bool>(true);
|
|
}
|
|
|
|
Option<bool> filter::parse_filter_query(const string& simple_filter_query,
|
|
const tsl::htrie_map<char, field>& search_schema,
|
|
const Store* store,
|
|
const std::string& doc_id_prefix,
|
|
std::vector<filter>& filters) {
|
|
|
|
std::vector<filter> exclude_filters; // to ensure that they go last in the list of filters
|
|
|
|
std::vector<std::string> filter_blocks;
|
|
StringUtils::split(simple_filter_query, filter_blocks, "&&");
|
|
|
|
for(const std::string & filter_block: filter_blocks) {
|
|
// split into [field_name, value]
|
|
size_t found_index = filter_block.find(':');
|
|
|
|
if(found_index == std::string::npos) {
|
|
return Option<bool>(400, "Could not parse the filter query.");
|
|
}
|
|
|
|
std::string&& field_name = filter_block.substr(0, found_index);
|
|
StringUtils::trim(field_name);
|
|
|
|
if(field_name == "id") {
|
|
filter id_filter;
|
|
std::string&& raw_value = filter_block.substr(found_index+1, std::string::npos);
|
|
StringUtils::trim(raw_value);
|
|
std::string empty_filter_err = "Error with filter field `id`: Filter value cannot be empty.";
|
|
|
|
if(raw_value.empty()) {
|
|
return Option<bool>(400, empty_filter_err);
|
|
}
|
|
|
|
id_filter = {field_name, {}, {}};
|
|
|
|
NUM_COMPARATOR id_comparator = EQUALS;
|
|
size_t filter_value_index = 0;
|
|
|
|
if(raw_value[0] == '=') {
|
|
id_comparator = EQUALS;
|
|
while(++filter_value_index < raw_value.size() && raw_value[filter_value_index] == ' ');
|
|
} else if(raw_value.size() >= 2 && raw_value[0] == '!' && raw_value[1] == '=') {
|
|
return Option<bool>(400, "Not equals filtering is not supported on the `id` field.");
|
|
}
|
|
|
|
if(filter_value_index != 0) {
|
|
raw_value = raw_value.substr(filter_value_index);
|
|
}
|
|
|
|
if(raw_value.empty()) {
|
|
return Option<bool>(400, empty_filter_err);
|
|
}
|
|
|
|
if(raw_value[0] == '[' && raw_value[raw_value.size() - 1] == ']') {
|
|
std::vector<std::string> doc_ids;
|
|
StringUtils::split_to_values(raw_value.substr(1, raw_value.size() - 2), doc_ids);
|
|
|
|
for(std::string& doc_id: doc_ids) {
|
|
// we have to convert the doc_id to seq id
|
|
std::string seq_id_str;
|
|
StoreStatus seq_id_status = store->get(doc_id_prefix + doc_id, seq_id_str);
|
|
|
|
if(seq_id_status != StoreStatus::FOUND) {
|
|
continue;
|
|
}
|
|
|
|
id_filter.values.push_back(seq_id_str);
|
|
id_filter.comparators.push_back(id_comparator);
|
|
}
|
|
} else {
|
|
std::vector<std::string> doc_ids;
|
|
StringUtils::split_to_values(raw_value, doc_ids); // to handle backticks
|
|
|
|
std::string seq_id_str;
|
|
StoreStatus seq_id_status = store->get(doc_id_prefix + doc_ids[0], seq_id_str);
|
|
if(seq_id_status == StoreStatus::FOUND) {
|
|
id_filter.values.push_back(seq_id_str);
|
|
id_filter.comparators.push_back(id_comparator);
|
|
}
|
|
}
|
|
|
|
filters.push_back(id_filter);
|
|
|
|
continue;
|
|
}
|
|
|
|
if(search_schema.count(field_name) == 0) {
|
|
return Option<bool>(404, "Could not find a filter field named `" + field_name + "` in the schema.");
|
|
}
|
|
|
|
field _field = search_schema.at(field_name);
|
|
std::string&& raw_value = filter_block.substr(found_index+1, std::string::npos);
|
|
StringUtils::trim(raw_value);
|
|
filter f;
|
|
|
|
// skip past optional `:=` operator, which has no meaning for non-string fields
|
|
if(!_field.is_string() && raw_value[0] == '=') {
|
|
size_t filter_value_index = 0;
|
|
while(raw_value[++filter_value_index] == ' ');
|
|
raw_value = raw_value.substr(filter_value_index);
|
|
}
|
|
|
|
if(_field.is_integer() || _field.is_float()) {
|
|
// could be a single value or a list
|
|
if(raw_value[0] == '[' && raw_value[raw_value.size() - 1] == ']') {
|
|
std::vector<std::string> filter_values;
|
|
StringUtils::split(raw_value.substr(1, raw_value.size() - 2), filter_values, ",");
|
|
|
|
f = {field_name, {}, {}};
|
|
|
|
for(std::string & filter_value: filter_values) {
|
|
Option<NUM_COMPARATOR> op_comparator = filter::extract_num_comparator(filter_value);
|
|
if(!op_comparator.ok()) {
|
|
return Option<bool>(400, "Error with filter field `" + _field.name + "`: " + op_comparator.error());
|
|
}
|
|
|
|
if(op_comparator.get() == RANGE_INCLUSIVE) {
|
|
// split the value around range operator to extract bounds
|
|
std::vector<std::string> range_values;
|
|
StringUtils::split(filter_value, range_values, filter::RANGE_OPERATOR());
|
|
for(const std::string& range_value: range_values) {
|
|
auto validate_op = filter::validate_numerical_filter_value(_field, range_value);
|
|
if(!validate_op.ok()) {
|
|
return validate_op;
|
|
}
|
|
|
|
f.values.push_back(range_value);
|
|
f.comparators.push_back(op_comparator.get());
|
|
}
|
|
} else {
|
|
auto validate_op = filter::validate_numerical_filter_value(_field, filter_value);
|
|
if(!validate_op.ok()) {
|
|
return validate_op;
|
|
}
|
|
|
|
f.values.push_back(filter_value);
|
|
f.comparators.push_back(op_comparator.get());
|
|
}
|
|
}
|
|
|
|
} else {
|
|
Option<NUM_COMPARATOR> op_comparator = filter::extract_num_comparator(raw_value);
|
|
if(!op_comparator.ok()) {
|
|
return Option<bool>(400, "Error with filter field `" + _field.name + "`: " + op_comparator.error());
|
|
}
|
|
|
|
if(op_comparator.get() == RANGE_INCLUSIVE) {
|
|
// split the value around range operator to extract bounds
|
|
std::vector<std::string> range_values;
|
|
StringUtils::split(raw_value, range_values, filter::RANGE_OPERATOR());
|
|
|
|
f.field_name = field_name;
|
|
for(const std::string& range_value: range_values) {
|
|
auto validate_op = filter::validate_numerical_filter_value(_field, range_value);
|
|
if(!validate_op.ok()) {
|
|
return validate_op;
|
|
}
|
|
|
|
f.values.push_back(range_value);
|
|
f.comparators.push_back(op_comparator.get());
|
|
}
|
|
} else {
|
|
auto validate_op = filter::validate_numerical_filter_value(_field, raw_value);
|
|
if(!validate_op.ok()) {
|
|
return validate_op;
|
|
}
|
|
f = {field_name, {raw_value}, {op_comparator.get()}};
|
|
}
|
|
}
|
|
} else if(_field.is_bool()) {
|
|
NUM_COMPARATOR bool_comparator = EQUALS;
|
|
size_t filter_value_index = 0;
|
|
|
|
if(raw_value[0] == '=') {
|
|
bool_comparator = EQUALS;
|
|
while(++filter_value_index < raw_value.size() && raw_value[filter_value_index] == ' ');
|
|
} else if(raw_value.size() >= 2 && raw_value[0] == '!' && raw_value[1] == '=') {
|
|
bool_comparator = NOT_EQUALS;
|
|
filter_value_index++;
|
|
while(++filter_value_index < raw_value.size() && raw_value[filter_value_index] == ' ');
|
|
}
|
|
|
|
if(filter_value_index != 0) {
|
|
raw_value = raw_value.substr(filter_value_index);
|
|
}
|
|
|
|
if(filter_value_index == raw_value.size()) {
|
|
return Option<bool>(400, "Error with filter field `" + _field.name +
|
|
"`: Filter value cannot be empty.");
|
|
}
|
|
|
|
if(raw_value[0] == '[' && raw_value[raw_value.size() - 1] == ']') {
|
|
std::vector<std::string> filter_values;
|
|
StringUtils::split(raw_value.substr(1, raw_value.size() - 2), filter_values, ",");
|
|
f = {field_name, {}, {}};
|
|
|
|
for(std::string & filter_value: filter_values) {
|
|
if(filter_value != "true" && filter_value != "false") {
|
|
return Option<bool>(400, "Values of filter field `" + _field.name +
|
|
"`: must be `true` or `false`.");
|
|
}
|
|
|
|
filter_value = (filter_value == "true") ? "1" : "0";
|
|
f.values.push_back(filter_value);
|
|
f.comparators.push_back(bool_comparator);
|
|
}
|
|
} else {
|
|
if(raw_value != "true" && raw_value != "false") {
|
|
return Option<bool>(400, "Value of filter field `" + _field.name + "` must be `true` or `false`.");
|
|
}
|
|
|
|
std::string bool_value = (raw_value == "true") ? "1" : "0";
|
|
f = {field_name, {bool_value}, {bool_comparator}};
|
|
}
|
|
|
|
} else if(_field.is_geopoint()) {
|
|
f = {field_name, {}, {}};
|
|
|
|
const std::string& format_err_msg = "Value of filter field `" + _field.name +
|
|
"`: must be in the `(-44.50, 170.29, 0.75 km)` or "
|
|
"(56.33, -65.97, 23.82, -127.82) format.";
|
|
|
|
NUM_COMPARATOR num_comparator;
|
|
|
|
// could be a single value or a list
|
|
if(raw_value[0] == '[' && raw_value[raw_value.size() - 1] == ']') {
|
|
std::vector<std::string> filter_values;
|
|
StringUtils::split(raw_value.substr(1, raw_value.size() - 2), filter_values, "),");
|
|
|
|
for(std::string& filter_value: filter_values) {
|
|
filter_value += ")";
|
|
std::string processed_filter_val;
|
|
auto parse_op = parse_geopoint_filter_value(filter_value, format_err_msg, processed_filter_val, num_comparator);
|
|
|
|
if(!parse_op.ok()) {
|
|
return parse_op;
|
|
}
|
|
|
|
f.values.push_back(processed_filter_val);
|
|
f.comparators.push_back(num_comparator);
|
|
}
|
|
} else {
|
|
// single value, e.g. (10.45, 34.56, 2 km)
|
|
std::string processed_filter_val;
|
|
auto parse_op = parse_geopoint_filter_value(raw_value, format_err_msg, processed_filter_val, num_comparator);
|
|
|
|
if(!parse_op.ok()) {
|
|
return parse_op;
|
|
}
|
|
|
|
f.values.push_back(processed_filter_val);
|
|
f.comparators.push_back(num_comparator);
|
|
}
|
|
|
|
} else if(_field.is_string()) {
|
|
size_t filter_value_index = 0;
|
|
NUM_COMPARATOR str_comparator = CONTAINS;
|
|
|
|
if(raw_value[0] == '=') {
|
|
// string filter should be evaluated in strict "equals" mode
|
|
str_comparator = EQUALS;
|
|
while(++filter_value_index < raw_value.size() && raw_value[filter_value_index] == ' ');
|
|
} else if(raw_value.size() >= 2 && raw_value[0] == '!' && raw_value[1] == '=') {
|
|
str_comparator = NOT_EQUALS;
|
|
filter_value_index++;
|
|
while(++filter_value_index < raw_value.size() && raw_value[filter_value_index] == ' ');
|
|
}
|
|
|
|
if(filter_value_index == raw_value.size()) {
|
|
return Option<bool>(400, "Error with filter field `" + _field.name +
|
|
"`: Filter value cannot be empty.");
|
|
}
|
|
|
|
if(raw_value[filter_value_index] == '[' && raw_value[raw_value.size() - 1] == ']') {
|
|
std::vector<std::string> filter_values;
|
|
StringUtils::split_to_values(raw_value.substr(filter_value_index+1, raw_value.size() - filter_value_index - 2), filter_values);
|
|
f = {field_name, filter_values, {str_comparator}};
|
|
} else {
|
|
f = {field_name, {raw_value.substr(filter_value_index)}, {str_comparator}};
|
|
}
|
|
} else {
|
|
return Option<bool>(400, "Error with filter field `" + _field.name +
|
|
"`: Unidentified field data type, see docs for supported data types.");
|
|
}
|
|
|
|
if(!f.comparators.empty() && f.comparators.front() == NOT_EQUALS) {
|
|
exclude_filters.push_back(f);
|
|
} else {
|
|
filters.push_back(f);
|
|
}
|
|
}
|
|
|
|
filters.insert( filters.end(), exclude_filters.begin(), exclude_filters.end() );
|
|
|
|
return Option<bool>(true);
|
|
}
|
|
|
|
Option<bool> field::json_field_to_field(nlohmann::json& field_json, std::vector<field>& the_fields,
|
|
string& fallback_field_type, size_t& num_auto_detect_fields) {
|
|
|
|
if(field_json["name"] == "id") {
|
|
// No field should exist with the name "id" as it is reserved for internal use
|
|
// We cannot throw an error here anymore since that will break backward compatibility!
|
|
LOG(WARNING) << "Collection schema cannot contain a field with name `id`. Ignoring field.";
|
|
return Option<bool>(true);
|
|
}
|
|
|
|
if(!field_json.is_object() ||
|
|
field_json.count(fields::name) == 0 || field_json.count(fields::type) == 0 ||
|
|
!field_json.at(fields::name).is_string() || !field_json.at(fields::type).is_string()) {
|
|
|
|
return Option<bool>(400, "Wrong format for `fields`. It should be an array of objects containing "
|
|
"`name`, `type`, `optional` and `facet` properties.");
|
|
}
|
|
|
|
if(field_json.count("drop") != 0) {
|
|
return Option<bool>(400, std::string("Invalid property `drop` on field `") +
|
|
field_json[fields::name].get<std::string>() + std::string("`: it is allowed only "
|
|
"during schema update."));
|
|
}
|
|
|
|
if(field_json.count(fields::facet) != 0 && !field_json.at(fields::facet).is_boolean()) {
|
|
return Option<bool>(400, std::string("The `facet` property of the field `") +
|
|
field_json[fields::name].get<std::string>() + std::string("` should be a boolean."));
|
|
}
|
|
|
|
if(field_json.count(fields::optional) != 0 && !field_json.at(fields::optional).is_boolean()) {
|
|
return Option<bool>(400, std::string("The `optional` property of the field `") +
|
|
field_json[fields::name].get<std::string>() + std::string("` should be a boolean."));
|
|
}
|
|
|
|
if(field_json.count(fields::index) != 0 && !field_json.at(fields::index).is_boolean()) {
|
|
return Option<bool>(400, std::string("The `index` property of the field `") +
|
|
field_json[fields::name].get<std::string>() + std::string("` should be a boolean."));
|
|
}
|
|
|
|
if(field_json.count(fields::sort) != 0 && !field_json.at(fields::sort).is_boolean()) {
|
|
return Option<bool>(400, std::string("The `sort` property of the field `") +
|
|
field_json[fields::name].get<std::string>() + std::string("` should be a boolean."));
|
|
}
|
|
|
|
if(field_json.count(fields::infix) != 0 && !field_json.at(fields::infix).is_boolean()) {
|
|
return Option<bool>(400, std::string("The `infix` property of the field `") +
|
|
field_json[fields::name].get<std::string>() + std::string("` should be a boolean."));
|
|
}
|
|
|
|
if(field_json.count(fields::locale) != 0){
|
|
if(!field_json.at(fields::locale).is_string()) {
|
|
return Option<bool>(400, std::string("The `locale` property of the field `") +
|
|
field_json[fields::name].get<std::string>() + std::string("` should be a string."));
|
|
}
|
|
|
|
if(!field_json[fields::locale].get<std::string>().empty() &&
|
|
field_json[fields::locale].get<std::string>().size() != 2) {
|
|
return Option<bool>(400, std::string("The `locale` value of the field `") +
|
|
field_json[fields::name].get<std::string>() + std::string("` is not valid."));
|
|
}
|
|
}
|
|
|
|
if(field_json["name"] == ".*") {
|
|
if(field_json.count(fields::facet) == 0) {
|
|
field_json[fields::facet] = false;
|
|
}
|
|
|
|
if(field_json.count(fields::optional) == 0) {
|
|
field_json[fields::optional] = true;
|
|
}
|
|
|
|
if(field_json.count(fields::index) == 0) {
|
|
field_json[fields::index] = true;
|
|
}
|
|
|
|
if(field_json.count(fields::locale) == 0) {
|
|
field_json[fields::locale] = "";
|
|
}
|
|
|
|
if(field_json.count(fields::sort) == 0) {
|
|
field_json[fields::sort] = false;
|
|
}
|
|
|
|
if(field_json.count(fields::infix) == 0) {
|
|
field_json[fields::infix] = false;
|
|
}
|
|
|
|
if(field_json[fields::optional] == false) {
|
|
return Option<bool>(400, "Field `.*` must be an optional field.");
|
|
}
|
|
|
|
if(field_json[fields::facet] == true) {
|
|
return Option<bool>(400, "Field `.*` cannot be a facet field.");
|
|
}
|
|
|
|
if(field_json[fields::index] == false) {
|
|
return Option<bool>(400, "Field `.*` must be an index field.");
|
|
}
|
|
|
|
field fallback_field(field_json["name"], field_json["type"], field_json["facet"],
|
|
field_json["optional"], field_json[fields::index], field_json[fields::locale],
|
|
field_json[fields::sort], field_json[fields::infix]);
|
|
|
|
if(fallback_field.has_valid_type()) {
|
|
fallback_field_type = fallback_field.type;
|
|
num_auto_detect_fields++;
|
|
} else {
|
|
return Option<bool>(400, "The `type` of field `.*` is invalid.");
|
|
}
|
|
|
|
the_fields.emplace_back(fallback_field);
|
|
return Option<bool>(true);
|
|
}
|
|
|
|
if(field_json.count(fields::facet) == 0) {
|
|
field_json[fields::facet] = false;
|
|
}
|
|
|
|
if(field_json.count(fields::index) == 0) {
|
|
field_json[fields::index] = true;
|
|
}
|
|
|
|
if(field_json.count(fields::locale) == 0) {
|
|
field_json[fields::locale] = "";
|
|
}
|
|
|
|
if(field_json.count(fields::sort) == 0) {
|
|
if(field_json["type"] == field_types::INT32 || field_json["type"] == field_types::INT64 ||
|
|
field_json["type"] == field_types::FLOAT || field_json["type"] == field_types::BOOL ||
|
|
field_json["type"] == field_types::GEOPOINT || field_json["type"] == field_types::GEOPOINT_ARRAY) {
|
|
field_json[fields::sort] = true;
|
|
} else {
|
|
field_json[fields::sort] = false;
|
|
}
|
|
}
|
|
|
|
if(field_json.count(fields::infix) == 0) {
|
|
field_json[fields::infix] = false;
|
|
}
|
|
|
|
if(field_json.count(fields::optional) == 0) {
|
|
// dynamic type fields are always optional
|
|
bool is_dynamic = field::is_dynamic_type(field_json[fields::name], field_json[fields::type]);
|
|
field_json[fields::optional] = is_dynamic;
|
|
}
|
|
|
|
bool is_obj = field_json[fields::type] == field_types::OBJECT || field_json[fields::type] == field_types::OBJECT_ARRAY;
|
|
bool is_regexp_name = field_json[fields::name].get<std::string>().find(".*") != std::string::npos;
|
|
|
|
if(is_obj || (!is_regexp_name && field_json[fields::name].get<std::string>().find('.') != std::string::npos)) {
|
|
field_json[fields::nested] = true;
|
|
field_json[fields::nested_array] = field::VAL_UNKNOWN; // unknown, will be resolved during read
|
|
} else {
|
|
field_json[fields::nested] = false;
|
|
field_json[fields::nested_array] = 0;
|
|
}
|
|
|
|
the_fields.emplace_back(
|
|
field(field_json[fields::name], field_json[fields::type], field_json[fields::facet],
|
|
field_json[fields::optional], field_json[fields::index], field_json[fields::locale],
|
|
field_json[fields::sort], field_json[fields::infix], field_json[fields::nested],
|
|
field_json[fields::nested_array])
|
|
);
|
|
|
|
return Option<bool>(true);
|
|
}
|
|
|
|
bool field::flatten_obj(nlohmann::json& doc, nlohmann::json& value, bool has_array, bool has_obj_array,
|
|
const std::string& flat_name, std::vector<field>& flattened_fields) {
|
|
if(value.is_object()) {
|
|
has_obj_array = has_array;
|
|
for(const auto& kv: value.items()) {
|
|
flatten_obj(doc, kv.value(), has_array, has_obj_array, flat_name + "." + kv.key(), flattened_fields);
|
|
}
|
|
} else if(value.is_array()) {
|
|
for(const auto& kv: value.items()) {
|
|
flatten_obj(doc, kv.value(), true, has_obj_array, flat_name, flattened_fields);
|
|
}
|
|
} else {
|
|
// must be a primitive
|
|
if(has_array) {
|
|
doc[flat_name].push_back(value);
|
|
} else {
|
|
doc[flat_name] = value;
|
|
}
|
|
|
|
std::string detected_type;
|
|
if(!field::get_type(value, detected_type)) {
|
|
return false;
|
|
}
|
|
|
|
if(std::isalnum(detected_type.back()) && has_array) {
|
|
// convert singular type to multi valued type
|
|
detected_type += "[]";
|
|
}
|
|
|
|
field flattened_field(flat_name, detected_type, false, true);
|
|
flattened_field.nested = true;
|
|
flattened_field.nested_array = has_obj_array;
|
|
flattened_fields.push_back(flattened_field);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool field::flatten_field(nlohmann::json& doc, nlohmann::json& obj, const field& the_field,
|
|
std::vector<std::string>& path_parts, size_t path_index,
|
|
bool has_array, bool has_obj_array, std::vector<field>& flattened_fields) {
|
|
if(path_index == path_parts.size()) {
|
|
// end of path: check if obj matches expected type
|
|
std::string detected_type;
|
|
if(!field::get_type(obj, detected_type)) {
|
|
return false;
|
|
}
|
|
|
|
if(std::isalnum(detected_type.back()) && has_array) {
|
|
// convert singular type to multi valued type
|
|
detected_type += "[]";
|
|
}
|
|
|
|
has_obj_array = has_obj_array || ((detected_type == field_types::OBJECT) && has_array);
|
|
|
|
if(detected_type == the_field.type) {
|
|
if(the_field.is_object()) {
|
|
flatten_obj(doc, obj, has_array, has_obj_array, the_field.name, flattened_fields);
|
|
} else {
|
|
if(has_array) {
|
|
doc[the_field.name].push_back(obj);
|
|
} else {
|
|
doc[the_field.name] = obj;
|
|
}
|
|
|
|
field flattened_field(the_field.name, detected_type, false, true);
|
|
flattened_field.nested = (path_index > 1);
|
|
flattened_field.nested_array = has_obj_array;
|
|
flattened_fields.push_back(flattened_field);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
// handle differences in detection of numerical types
|
|
bool is_numericaly_valid = (detected_type == field_types::INT64 && (the_field.type == field_types::INT32 ||
|
|
the_field.type == field_types::FLOAT)) ||
|
|
(detected_type == field_types::INT64_ARRAY &&
|
|
(the_field.type == field_types::INT32_ARRAY ||
|
|
the_field.type == field_types::FLOAT_ARRAY));
|
|
|
|
if(is_numericaly_valid) {
|
|
if(has_array) {
|
|
doc[the_field.name].push_back(obj);
|
|
} else {
|
|
doc[the_field.name] = obj;
|
|
}
|
|
|
|
field flattened_field(the_field.name, the_field.type, false, true);
|
|
flattened_field.nested = (path_index > 1);
|
|
flattened_field.nested_array = has_obj_array;
|
|
flattened_fields.push_back(flattened_field);
|
|
return true;
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
const std::string& fragment = path_parts[path_index];
|
|
const auto& it = obj.find(fragment);
|
|
|
|
if(it != obj.end()) {
|
|
if(it.value().is_array()) {
|
|
has_array = true;
|
|
bool resolved = false;
|
|
for(auto& ele: it.value()) {
|
|
has_obj_array = has_obj_array || ele.is_object();
|
|
resolved |= flatten_field(doc, ele, the_field, path_parts, path_index + 1, has_array, has_obj_array, flattened_fields);
|
|
}
|
|
return resolved;
|
|
} else {
|
|
return flatten_field(doc, it.value(), the_field, path_parts, path_index + 1, has_array, has_obj_array, flattened_fields);
|
|
}
|
|
} {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
Option<bool> field::flatten_doc(nlohmann::json& document,
|
|
const std::vector<field>& nested_fields,
|
|
std::vector<field>& flattened_fields) {
|
|
|
|
for(auto& nested_field: nested_fields) {
|
|
std::vector<std::string> field_parts;
|
|
StringUtils::split(nested_field.name, field_parts, ".");
|
|
|
|
bool resolved = flatten_field(document, document, nested_field, field_parts, 0, false, false, flattened_fields);
|
|
if(!resolved) {
|
|
return Option<bool>(400, "Field `" + nested_field.name + "` was not found or has an incorrect type.");
|
|
}
|
|
}
|
|
|
|
std::sort(flattened_fields.begin(), flattened_fields.end());
|
|
flattened_fields.erase(std::unique(flattened_fields.begin(), flattened_fields.end()), flattened_fields.end());
|
|
|
|
document[".flat"] = nlohmann::json::array();
|
|
for(auto& f: flattened_fields) {
|
|
document[".flat"].push_back(f.name);
|
|
}
|
|
|
|
return Option<bool>(true);
|
|
}
|