mirror of
https://github.com/nlohmann/json.git
synced 2026-09-01 14:07:14 +00:00
Reserve output capacity up front for binary serialization
The vector-returning to_cbor/to_msgpack/to_ubjson/to_bjdata/to_bson grew the output buffer purely by geometric reallocation. Reserving an estimate up front avoids the early reallocations, which is the dominant per-byte cost for array/object-heavy output. The estimate (binary_reserve_hint) is deliberately conservative and safe against untrusted input: it consults only the top-level element count (O(1), no walk of the DOM), guards the multiplication against overflow, and clamps the result to a fixed 1 MiB ceiling, so a large or hostile DOM can never force an oversized allocation here. The buffer still grows geometrically past the hint, so an underestimate only costs a few later reallocations; scalars/strings/binary are written in one shot and get no hint. Reserving capacity does not change the bytes produced. Throughput (g++/clang -O3, vs the previous commit): cbor int array +10% / +13% cbor object array +20% / +38% Output is byte-for-byte identical to develop across the binary differential corpus. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01XAYM1qhSA2FDaDcGfPW3fG Signed-off-by: Niels Lohmann <mail@nlohmann.me>
This commit is contained in:
co-authored by
Claude Opus 4.8
parent
09e5e85ba5
commit
1d51b00830
@@ -39,6 +39,35 @@ enum class bjdata_version_t
|
||||
// binary writer //
|
||||
///////////////////
|
||||
|
||||
/*!
|
||||
@brief conservative capacity hint for binary serialization into a std::vector
|
||||
|
||||
Returns an approximate number of bytes to reserve up front so that serializing
|
||||
an array/object of many elements does not repeatedly reallocate the output
|
||||
buffer. Only the top-level element count is consulted (O(1), no walk of the
|
||||
DOM), and the result is clamped to a fixed ceiling: a large or untrusted DOM can
|
||||
therefore never trigger an oversized allocation here, and the multiplication
|
||||
cannot overflow. The buffer still grows geometrically beyond the hint, so a hint
|
||||
that is too small only costs a few later reallocations. A single scalar, string,
|
||||
or binary value is written in one shot and needs no hint.
|
||||
*/
|
||||
template<typename BasicJsonType>
|
||||
std::size_t binary_reserve_hint(const BasicJsonType& j)
|
||||
{
|
||||
constexpr std::size_t max_hint = static_cast<std::size_t>(1) << 20; // 1 MiB
|
||||
if (j.is_array() || j.is_object())
|
||||
{
|
||||
const std::size_t elements = j.size();
|
||||
// guard the multiplication against overflow and cap the reservation
|
||||
if (elements > max_hint / 4)
|
||||
{
|
||||
return max_hint;
|
||||
}
|
||||
return (elements * 4) + 2;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
/*!
|
||||
@brief serialization to CBOR and MessagePack values
|
||||
*/
|
||||
|
||||
Reference in New Issue
Block a user