mirror of
https://github.com/simdjson/simdjson
synced 2026-06-08 17:27:07 +00:00
Daniel Lemire
9c95a48fe6
* Setting RapidJSON and yyjson to their latest version. * Let us stop dumping all of the benchmarks (it is confusing) on screen.
123 lines
3.3 KiB
C++
123 lines
3.3 KiB
C++
#pragma once
|
|
#ifdef SIMDJSON_COMPETITION_YYJSON
|
|
|
|
#include "json2msgpack.h"
|
|
|
|
namespace json2msgpack {
|
|
|
|
struct yyjson2msgpack {
|
|
inline std::string_view to_msgpack(yyjson_doc *doc, uint8_t *buf);
|
|
|
|
private:
|
|
inline void write_double(const double d) noexcept;
|
|
inline void write_byte(const uint8_t b) noexcept;
|
|
inline void write_uint32(const uint32_t w) noexcept;
|
|
inline void write_string(const char *s, size_t length) noexcept;
|
|
inline void recursive_processor(yyjson_val *obj);
|
|
|
|
uint8_t *buff{};
|
|
};
|
|
|
|
std::string_view yyjson2msgpack::to_msgpack(yyjson_doc *doc, uint8_t *buf) {
|
|
buff = buf;
|
|
yyjson_val *root = yyjson_doc_get_root(doc);
|
|
recursive_processor(root);
|
|
return std::string_view(reinterpret_cast<char *>(buf), size_t(buff - buf));
|
|
}
|
|
|
|
void yyjson2msgpack::write_string(const char *c, size_t len) noexcept {
|
|
write_byte(0xdb);
|
|
write_uint32(uint32_t(len));
|
|
::memcpy(buff, c, len);
|
|
buff += len;
|
|
}
|
|
|
|
void yyjson2msgpack::write_double(const double d) noexcept {
|
|
*buff++ = 0xcb;
|
|
::memcpy(buff, &d, sizeof(d));
|
|
buff += sizeof(d);
|
|
}
|
|
|
|
void yyjson2msgpack::write_byte(const uint8_t b) noexcept {
|
|
*buff = b;
|
|
buff++;
|
|
}
|
|
|
|
void yyjson2msgpack::write_uint32(const uint32_t w) noexcept {
|
|
::memcpy(buff, &w, sizeof(w));
|
|
buff += sizeof(w);
|
|
}
|
|
|
|
void yyjson2msgpack::recursive_processor(yyjson_val *obj) {
|
|
size_t idx, max;
|
|
yyjson_val *val;
|
|
yyjson_val *key;
|
|
switch (yyjson_get_type(obj)) {
|
|
case YYJSON_TYPE_STR:
|
|
write_string(yyjson_get_str(obj), yyjson_get_len(obj));
|
|
break;
|
|
case YYJSON_TYPE_ARR:
|
|
write_byte(0xdf);
|
|
write_uint32(uint32_t(yyjson_arr_size(obj)));
|
|
yyjson_arr_foreach(obj, idx, max, val) { recursive_processor(val); }
|
|
break;
|
|
case YYJSON_TYPE_OBJ:
|
|
write_byte(0xdd);
|
|
write_uint32(uint32_t(yyjson_obj_size(obj)));
|
|
yyjson_obj_foreach(obj, idx, max, key, val) {
|
|
write_string(yyjson_get_str(key), yyjson_get_len(key));
|
|
recursive_processor(val);
|
|
}
|
|
break;
|
|
case YYJSON_TYPE_BOOL:
|
|
write_byte(0xc2 + yyjson_get_bool(obj));
|
|
break;
|
|
case YYJSON_TYPE_NULL:
|
|
write_byte(0xc0);
|
|
break;
|
|
case YYJSON_TYPE_NUM:
|
|
switch (yyjson_get_subtype(obj)) {
|
|
case YYJSON_SUBTYPE_UINT:
|
|
write_double(double(yyjson_get_uint(obj)));
|
|
break;
|
|
case YYJSON_SUBTYPE_SINT:
|
|
write_double(double(yyjson_get_sint(obj)));
|
|
break;
|
|
case YYJSON_SUBTYPE_REAL:
|
|
write_double(yyjson_get_real(obj));
|
|
break;
|
|
default:
|
|
SIMDJSON_UNREACHABLE();
|
|
}
|
|
break;
|
|
default:
|
|
SIMDJSON_UNREACHABLE();
|
|
}
|
|
}
|
|
|
|
struct yyjson : yyjson2msgpack {
|
|
bool run(simdjson::padded_string &json, char *buffer,
|
|
std::string_view &result) {
|
|
yyjson_doc *doc = yyjson_read(json.data(), json.size(), 0);
|
|
result = to_msgpack(doc, reinterpret_cast<uint8_t*>(buffer));
|
|
return true;
|
|
}
|
|
};
|
|
|
|
BENCHMARK_TEMPLATE(json2msgpack, yyjson)->UseManualTime();
|
|
|
|
#if SIMDJSON_COMPETITION_ONDEMAND_INSITU
|
|
struct yyjson_insitu : yyjson2msgpack {
|
|
bool run(simdjson::padded_string &json, char *buffer,
|
|
std::string_view &result) {
|
|
yyjson_doc *doc =
|
|
yyjson_read_opts(json.data(), json.size(), YYJSON_READ_INSITU, 0, 0);
|
|
result = to_msgpack(doc, reinterpret_cast<uint8_t*>(buffer));
|
|
return true;
|
|
}
|
|
};
|
|
BENCHMARK_TEMPLATE(json2msgpack, yyjson_insitu)->UseManualTime();
|
|
#endif // SIMDJSON_COMPETITION_ONDEMAND_INSITU
|
|
} // namespace json2msgpack
|
|
|
|
#endif // SIMDJSON_COMPETITION_YYJSON
|