Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
31 changes: 4 additions & 27 deletions ydb/core/formats/arrow/accessor/sub_columns/json_extractors.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -39,33 +39,14 @@ TConclusionStatus TKVExtractor::DoFill(TDataBuilder& dataBuilder, std::deque<std
}

TConclusionStatus IJsonObjectExtractor::AddDataToBuilder(TDataBuilder& dataBuilder,
std::deque<std::unique_ptr<IJsonObjectExtractor>>& iterators, const TStringBuf key, NBinaryJson::TEntryCursor& value) const {
std::deque<std::unique_ptr<IJsonObjectExtractor>>& iterators, const TStringBuf key, const NBinaryJson::TEntryCursor& value) const {
std::variant<NBinaryJson::TBinaryJson, TString> res;
bool addRes = true;

if (value.GetType() == NBinaryJson::EEntryType::String) {
NJson::TJsonValue jsonValue(value.GetString());
NJsonWriter::TBuf sout;
sout.WriteJsonValue(&jsonValue);
res = NBinaryJson::SerializeToBinaryJson(sout.Str(), false);
} else if (value.GetType() == NBinaryJson::EEntryType::Number) {
const double val = value.GetNumber();
double integer;
if (modf(val, &integer)) {
res = NBinaryJson::SerializeToBinaryJson(std::to_string(val), false);
} else {
res = NBinaryJson::SerializeToBinaryJson(std::to_string((i64)integer), false);
}
} else if (value.GetType() == NBinaryJson::EEntryType::BoolFalse) {
static const TString falseString = "false";
res = NBinaryJson::SerializeToBinaryJson(falseString, false);
} else if (value.GetType() == NBinaryJson::EEntryType::BoolTrue) {
static const TString trueString = "true";
res = NBinaryJson::SerializeToBinaryJson(trueString, false);
} else if (value.GetType() == NBinaryJson::EEntryType::Container) {
if (value.GetType() == NBinaryJson::EEntryType::Container) {
auto container = value.GetContainer();
if (FirstLevelOnly || container.GetType() == NBinaryJson::EContainerType::Array) {
res = NBinaryJson::SerializeToBinaryJson(NBinaryJson::SerializeToJson(container), false);
res = NBinaryJson::SerializeToBinaryJson(value);
// TODO: add support for arrays if needed
// } else if (container.GetType() == NBinaryJson::EContainerType::Array) {
// iterators.emplace_back(std::make_unique<TArrayExtractor>(container.GetArrayIterator(), key));
Expand All @@ -76,12 +57,8 @@ TConclusionStatus IJsonObjectExtractor::AddDataToBuilder(TDataBuilder& dataBuild
} else {
return TConclusionStatus::Fail("unexpected top value scalar in container iterator");
}

} else if (value.GetType() == NBinaryJson::EEntryType::Null) {
static const TString nullString = "null";
res = NBinaryJson::SerializeToBinaryJson(nullString, false);
} else {
return TConclusionStatus::Fail("unexpected json value type: " + ::ToString((int)value.GetType()));
res = NBinaryJson::SerializeToBinaryJson(value);
}

if (addRes) {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ class IJsonObjectExtractor {
}

[[nodiscard]] TConclusionStatus AddDataToBuilder(TDataBuilder& dataBuilder, std::deque<std::unique_ptr<IJsonObjectExtractor>>& iterators,
const TStringBuf key, NBinaryJson::TEntryCursor& value) const;
const TStringBuf key, const NBinaryJson::TEntryCursor& value) const;

public:
virtual ~IJsonObjectExtractor() = default;
Expand Down
Loading