52 lines
1.6 KiB
C++
52 lines
1.6 KiB
C++
#pragma once
|
|
|
|
#ifdef SIMDJSON_COMPETITION_RAPIDJSON
|
|
|
|
#include "find_tweet.h"
|
|
|
|
namespace find_tweet {
|
|
|
|
using namespace rapidjson;
|
|
|
|
struct rapidjson_base {
|
|
Document doc{};
|
|
|
|
bool run(Document &root, uint64_t find_id, std::string_view &text) {
|
|
if (root.HasParseError() || !root.IsObject()) { return false; }
|
|
auto statuses = root.FindMember("statuses");
|
|
if (statuses == root.MemberEnd() || !statuses->value.IsArray()) { return false; }
|
|
for (auto &tweet : statuses->value.GetArray()) {
|
|
if (!tweet.IsObject()) { return false; }
|
|
auto id = tweet.FindMember("id");
|
|
if (id == tweet.MemberEnd() || !id->value.IsUint64()) { return false; }
|
|
if (id->value.GetUint64() == find_id) {
|
|
auto _text = tweet.FindMember("text");
|
|
if (_text == tweet.MemberEnd() || !_text->value.IsString()) { return false; }
|
|
text = { _text->value.GetString(), _text->value.GetStringLength() };
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
};
|
|
|
|
struct rapidjson : public rapidjson_base {
|
|
bool run(const padded_string &json, uint64_t find_id, std::string_view &text) {
|
|
return rapidjson_base::run(doc.Parse<kParseValidateEncodingFlag>(json.data()), find_id, text);
|
|
}
|
|
};
|
|
BENCHMARK_TEMPLATE(find_tweet, rapidjson);
|
|
|
|
struct rapidjson_insitu : public rapidjson_base {
|
|
bool run(const padded_string &json, uint64_t find_id, std::string_view &text) {
|
|
padded_string json_copy{json.data(), json.size()};
|
|
return rapidjson_base::run(doc.ParseInsitu<kParseValidateEncodingFlag>(json_copy.data()), find_id, text);
|
|
}
|
|
};
|
|
BENCHMARK_TEMPLATE(find_tweet, rapidjson_insitu);
|
|
|
|
} // namespace partial_tweets
|
|
|
|
#endif // SIMDJSON_COMPETITION_RAPIDJSON
|