Coverage Report

Created: 2026-03-24 20:45

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
be/src/util/jsonb_document.h
Line
Count
Source
1
/*
2
 *  Copyright (c) 2014, Facebook, Inc.
3
 *  All rights reserved.
4
 *
5
 *  This source code is licensed under the BSD-style license found in the
6
 *  LICENSE file in the root directory of this source tree. An additional grant
7
 *  of patent rights can be found in the PATENTS file in the same directory.
8
 *
9
 */
10
11
/*
12
 * This header defines JsonbDocument, JsonbKeyValue, and various value classes
13
 * which are derived from JsonbValue, and a forward iterator for container
14
 * values - essentially everything that is related to JSONB binary data
15
 * structures.
16
 *
17
 * Implementation notes:
18
 *
19
 * None of the classes in this header file can be instantiated directly (i.e.
20
 * you cannot create a JsonbKeyValue or JsonbValue object - all constructors
21
 * are declared non-public). We use the classes as wrappers on the packed JSONB
22
 * bytes (serialized), and cast the classes (types) to the underlying packed
23
 * byte array.
24
 *
25
 * For the same reason, we cannot define any JSONB value class to be virtual,
26
 * since we never call constructors, and will not instantiate vtbl and vptrs.
27
 *
28
 * Therefore, the classes are defined as packed structures (i.e. no data
29
 * alignment and padding), and the private member variables of the classes are
30
 * defined precisely in the same order as the JSONB spec. This ensures we
31
 * access the packed JSONB bytes correctly.
32
 *
33
 * The packed structures are highly optimized for in-place operations with low
34
 * overhead. The reads (and in-place writes) are performed directly on packed
35
 * bytes. There is no memory allocation at all at runtime.
36
 *
37
 * For updates/writes of values that will expand the original JSONB size, the
38
 * write will fail, and the caller needs to handle buffer increase.
39
 *
40
 * ** Iterator **
41
 * Both ObjectVal class and ArrayVal class have iterator type that you can use
42
 * to declare an iterator on a container object to go through the key-value
43
 * pairs or value list. The iterator has both non-const and const types.
44
 *
45
 * Note: iterators are forward direction only.
46
 *
47
 * ** Query **
48
 * Querying into containers is through the member functions find (for key/value
49
 * pairs) and get (for array elements), and is in streaming style. We don't
50
 * need to read/scan the whole JSONB packed bytes in order to return results.
51
 * Once the key/index is found, we will stop search.  You can use text to query
52
 * both objects and array (for array, text will be converted to integer index),
53
 * and use index to retrieve from array. Array index is 0-based.
54
 *
55
 * ** External dictionary **
56
 * During query processing, you can also pass a call-back function, so the
57
 * search will first try to check if the key string exists in the dictionary.
58
 * If so, search will be based on the id instead of the key string.
59
 * @author Tian Xia <tianx@fb.com>
60
 * 
61
 * this file is copied from 
62
 * https://github.com/facebook/mysql-5.6/blob/fb-mysql-5.6.35/fbson/FbsonDocument.h
63
 * and modified by Doris
64
 */
65
66
#ifndef JSONB_JSONBDOCUMENT_H
67
#define JSONB_JSONBDOCUMENT_H
68
69
#include <algorithm>
70
#include <cctype>
71
#include <charconv>
72
#include <cstddef>
73
#include <cstdint>
74
#include <string>
75
#include <string_view>
76
#include <type_traits>
77
78
#include "common/compiler_util.h" // IWYU pragma: keep
79
#include "common/status.h"
80
#include "core/data_type/define_primitive_type.h"
81
#include "core/string_ref.h"
82
#include "core/types.h"
83
#include "util/string_util.h"
84
85
// #include "util/string_parser.hpp"
86
87
// Concept to check for supported decimal types
88
template <typename T>
89
concept JsonbDecimalType =
90
        std::same_as<T, doris::Decimal256> || std::same_as<T, doris::Decimal64> ||
91
        std::same_as<T, doris::Decimal128V3> || std::same_as<T, doris::Decimal32>;
92
93
namespace doris {
94
95
template <typename T>
96
constexpr bool is_pod_v = std::is_trivial_v<T> && std::is_standard_layout_v<T>;
97
98
struct JsonbStringVal;
99
struct ObjectVal;
100
struct ArrayVal;
101
struct JsonbBinaryVal;
102
struct ContainerVal;
103
104
template <JsonbDecimalType T>
105
struct JsonbDecimalVal;
106
107
using JsonbDecimal256 = JsonbDecimalVal<Decimal256>;
108
using JsonbDecimal128 = JsonbDecimalVal<Decimal128V3>;
109
using JsonbDecimal64 = JsonbDecimalVal<Decimal64>;
110
using JsonbDecimal32 = JsonbDecimalVal<Decimal32>;
111
112
template <typename T>
113
    requires std::is_integral_v<T> || std::is_floating_point_v<T>
114
struct NumberValT;
115
116
using JsonbInt8Val = NumberValT<int8_t>;
117
using JsonbInt16Val = NumberValT<int16_t>;
118
using JsonbInt32Val = NumberValT<int32_t>;
119
using JsonbInt64Val = NumberValT<int64_t>;
120
using JsonbInt128Val = NumberValT<int128_t>;
121
using JsonbDoubleVal = NumberValT<double>;
122
using JsonbFloatVal = NumberValT<float>;
123
124
template <typename T>
125
concept JsonbPodType = (std::same_as<T, JsonbStringVal> || std::same_as<T, ObjectVal> ||
126
                        std::same_as<T, ContainerVal> || std::same_as<T, ArrayVal> ||
127
                        std::same_as<T, JsonbBinaryVal> || std::same_as<T, JsonbDecimal32> ||
128
                        std::same_as<T, JsonbDecimal64> || std::same_as<T, JsonbDecimal128> ||
129
                        std::same_as<T, JsonbDecimal256> || std::same_as<T, JsonbDecimal32> ||
130
                        std::same_as<T, JsonbInt8Val> || std::same_as<T, JsonbInt16Val> ||
131
                        std::same_as<T, JsonbInt32Val> || std::same_as<T, JsonbInt64Val> ||
132
                        std::same_as<T, JsonbInt128Val> || std::same_as<T, JsonbFloatVal> ||
133
                        std::same_as<T, JsonbFloatVal> || std::same_as<T, JsonbDoubleVal>);
134
135
2.41M
#define JSONB_VER 1
136
137
using int128_t = __int128;
138
139
// forward declaration
140
struct JsonbValue;
141
142
class JsonbOutStream;
143
144
template <class OS_TYPE>
145
class JsonbWriterT;
146
147
using JsonbWriter = JsonbWriterT<JsonbOutStream>;
148
149
const int MaxNestingLevel = 100;
150
151
/*
152
 * JsonbType defines 10 primitive types and 2 container types, as described
153
 * below.
154
 * NOTE: Do NOT modify the existing values or their order in this enum.
155
 *      You may only append new entries at the end before `NUM_TYPES`.
156
 *      This enum will be used in serialized data and/or persisted data.
157
 *      Changing existing values may break backward compatibility
158
 *      with previously stored or transmitted data.
159
 *
160
 * primitive_value ::=
161
 *   0x00        //null value (0 byte)
162
 * | 0x01        //boolean true (0 byte)
163
 * | 0x02        //boolean false (0 byte)
164
 * | 0x03 int8   //char/int8 (1 byte)
165
 * | 0x04 int16  //int16 (2 bytes)
166
 * | 0x05 int32  //int32 (4 bytes)
167
 * | 0x06 int64  //int64 (8 bytes)
168
 * | 0x07 double //floating point (8 bytes)
169
 * | 0x08 string //variable length string
170
 * | 0x09 binary //variable length binary
171
 *
172
 * container ::=
173
 *   0x0A int32 key_value_list //object, int32 is the total bytes of the object
174
 * | 0x0B int32 value_list     //array, int32 is the total bytes of the array
175
 */
176
enum class JsonbType : char {
177
    T_Null = 0x00,
178
    T_True = 0x01,
179
    T_False = 0x02,
180
    T_Int8 = 0x03,
181
    T_Int16 = 0x04,
182
    T_Int32 = 0x05,
183
    T_Int64 = 0x06,
184
    T_Double = 0x07,
185
    T_String = 0x08,
186
    T_Binary = 0x09,
187
    T_Object = 0x0A,
188
    T_Array = 0x0B,
189
    T_Int128 = 0x0C,
190
    T_Float = 0x0D,
191
    T_Decimal32 = 0x0E,  // DecimalV3 only
192
    T_Decimal64 = 0x0F,  // DecimalV3 only
193
    T_Decimal128 = 0x10, // DecimalV3 only
194
    T_Decimal256 = 0x11, // DecimalV3 only
195
    NUM_TYPES,
196
};
197
198
22
inline PrimitiveType get_primitive_type_from_json_type(JsonbType json_type) {
199
22
    switch (json_type) {
200
2
    case JsonbType::T_Null:
201
2
        return TYPE_NULL;
202
2
    case JsonbType::T_True:
203
4
    case JsonbType::T_False:
204
4
        return TYPE_BOOLEAN;
205
0
    case JsonbType::T_Int8:
206
0
        return TYPE_TINYINT;
207
0
    case JsonbType::T_Int16:
208
0
        return TYPE_SMALLINT;
209
0
    case JsonbType::T_Int32:
210
0
        return TYPE_INT;
211
0
    case JsonbType::T_Int64:
212
0
        return TYPE_BIGINT;
213
0
    case JsonbType::T_Double:
214
0
        return TYPE_DOUBLE;
215
2
    case JsonbType::T_String:
216
2
        return TYPE_STRING;
217
0
    case JsonbType::T_Binary:
218
0
        return TYPE_BINARY;
219
0
    case JsonbType::T_Object:
220
0
        return TYPE_STRUCT;
221
2
    case JsonbType::T_Array:
222
2
        return TYPE_ARRAY;
223
2
    case JsonbType::T_Int128:
224
2
        return TYPE_LARGEINT;
225
2
    case JsonbType::T_Float:
226
2
        return TYPE_FLOAT;
227
2
    case JsonbType::T_Decimal32:
228
2
        return TYPE_DECIMAL32;
229
2
    case JsonbType::T_Decimal64:
230
2
        return TYPE_DECIMAL64;
231
2
    case JsonbType::T_Decimal128:
232
2
        return TYPE_DECIMAL128I;
233
2
    case JsonbType::T_Decimal256:
234
2
        return TYPE_DECIMAL256;
235
0
    default:
236
0
        throw Exception(ErrorCode::INTERNAL_ERROR, "Unsupported JsonbType: {}",
237
0
                        static_cast<int>(json_type));
238
22
    }
239
22
}
240
241
//for parse json path
242
constexpr char SCOPE = '$';
243
constexpr char BEGIN_MEMBER = '.';
244
constexpr char BEGIN_ARRAY = '[';
245
constexpr char END_ARRAY = ']';
246
constexpr char DOUBLE_QUOTE = '"';
247
constexpr char WILDCARD = '*';
248
constexpr char MINUS = '-';
249
constexpr char LAST[] = "last";
250
constexpr char ESCAPE = '\\';
251
constexpr unsigned int MEMBER_CODE = 0;
252
constexpr unsigned int ARRAY_CODE = 1;
253
254
/// A simple input stream class for the JSON path parser.
255
class Stream {
256
public:
257
    /// Creates an input stream reading from a character string.
258
    /// @param string  the input string
259
    /// @param length  the length of the input string
260
192
    Stream(const char* string, size_t length) : m_position(string), m_end(string + length) {}
261
262
    /// Returns a pointer to the current position in the stream.
263
180
    const char* position() const { return m_position; }
264
265
    /// Returns a pointer to the position just after the end of the stream.
266
0
    const char* end() const { return m_end; }
267
268
    /// Returns the number of bytes remaining in the stream.
269
3.42k
    size_t remaining() const {
270
3.42k
        assert(m_position <= m_end);
271
3.42k
        return m_end - m_position;
272
3.42k
    }
273
274
    /// Tells if the stream has been exhausted.
275
3.13k
    bool exhausted() const { return remaining() == 0; }
276
277
    /// Reads the next byte from the stream and moves the position forward.
278
192
    char read() {
279
192
        assert(!exhausted());
280
192
        return *m_position++;
281
192
    }
282
283
    /// Reads the next byte from the stream without moving the position forward.
284
1.59k
    char peek() const {
285
1.59k
        assert(!exhausted());
286
1.59k
        return *m_position;
287
1.59k
    }
288
289
    /// Moves the position to the next non-whitespace character.
290
668
    void skip_whitespace() {
291
668
        m_position = std::find_if_not(m_position, m_end, [](char c) { return std::isspace(c); });
292
668
    }
293
294
    /// Moves the position n bytes forward.
295
296
    void skip(size_t n) {
296
296
        assert(remaining() >= n);
297
296
        m_position += n;
298
296
        skip_whitespace();
299
296
    }
300
301
244
    void advance() { m_position++; }
302
303
360
    void clear_leg_ptr() { leg_ptr = nullptr; }
304
305
180
    void set_leg_ptr(char* ptr) {
306
180
        clear_leg_ptr();
307
180
        leg_ptr = ptr;
308
180
    }
309
310
244
    char* get_leg_ptr() { return leg_ptr; }
311
312
180
    void clear_leg_len() { leg_len = 0; }
313
314
244
    void add_leg_len() { leg_len++; }
315
316
360
    unsigned int get_leg_len() const { return leg_len; }
317
318
0
    void remove_escapes() {
319
0
        int new_len = 0;
320
0
        for (int i = 0; i < leg_len; i++) {
321
0
            if (leg_ptr[i] != '\\') {
322
0
                leg_ptr[new_len++] = leg_ptr[i];
323
0
            }
324
0
        }
325
0
        leg_ptr[new_len] = '\0';
326
0
        leg_len = new_len;
327
0
    }
328
329
0
    void set_has_escapes(bool has) { has_escapes = has; }
330
331
64
    bool get_has_escapes() const { return has_escapes; }
332
333
private:
334
    /// The current position in the stream.
335
    const char* m_position = nullptr;
336
337
    /// The end of the stream.
338
    const char* const m_end;
339
340
    ///path leg ptr
341
    char* leg_ptr = nullptr;
342
343
    ///path leg len
344
    unsigned int leg_len;
345
346
    ///Whether to contain escape characters
347
    bool has_escapes = false;
348
};
349
350
struct leg_info {
351
    ///path leg ptr
352
    char* leg_ptr = nullptr;
353
354
    ///path leg len
355
    unsigned int leg_len;
356
357
    ///array_index
358
    int array_index;
359
360
    ///type: 0 is member 1 is array
361
    unsigned int type;
362
363
0
    bool to_string(std::string* str) const {
364
0
        if (type == MEMBER_CODE) {
365
0
            str->push_back(BEGIN_MEMBER);
366
0
            bool contains_space = false;
367
0
            std::string tmp;
368
0
            for (auto* it = leg_ptr; it != (leg_ptr + leg_len); ++it) {
369
0
                if (std::isspace(*it)) {
370
0
                    contains_space = true;
371
0
                } else if (*it == '"' || *it == ESCAPE || *it == '\r' || *it == '\n' ||
372
0
                           *it == '\b' || *it == '\t') {
373
0
                    tmp.push_back(ESCAPE);
374
0
                }
375
0
                tmp.push_back(*it);
376
0
            }
377
0
            if (contains_space) {
378
0
                str->push_back(DOUBLE_QUOTE);
379
0
            }
380
0
            str->append(tmp);
381
0
            if (contains_space) {
382
0
                str->push_back(DOUBLE_QUOTE);
383
0
            }
384
0
            return true;
385
0
        } else if (type == ARRAY_CODE) {
386
0
            str->push_back(BEGIN_ARRAY);
387
0
            std::string int_str = std::to_string(array_index);
388
0
            str->append(int_str);
389
0
            str->push_back(END_ARRAY);
390
0
            return true;
391
0
        } else {
392
0
            return false;
393
0
        }
394
0
    }
395
};
396
397
class JsonbPath {
398
public:
399
    // parse json path
400
    static bool parsePath(Stream* stream, JsonbPath* path);
401
402
    static bool parse_array(Stream* stream, JsonbPath* path);
403
    static bool parse_member(Stream* stream, JsonbPath* path);
404
405
    //return true if json path valid else return false
406
    bool seek(const char* string, size_t length);
407
408
180
    void add_leg_to_leg_vector(std::unique_ptr<leg_info> leg) {
409
180
        leg_vector.emplace_back(leg.release());
410
180
    }
411
412
0
    void pop_leg_from_leg_vector() { leg_vector.pop_back(); }
413
414
0
    bool to_string(std::string* res) const {
415
0
        res->push_back(SCOPE);
416
0
        for (const auto& leg : leg_vector) {
417
0
            auto valid = leg->to_string(res);
418
0
            if (!valid) {
419
0
                return false;
420
0
            }
421
0
        }
422
0
        return true;
423
0
    }
424
425
370
    size_t get_leg_vector_size() const { return leg_vector.size(); }
426
427
590
    leg_info* get_leg_from_leg_vector(size_t i) const { return leg_vector[i].get(); }
428
429
0
    bool is_wildcard() const { return _is_wildcard; }
430
190
    bool is_supper_wildcard() const { return _is_supper_wildcard; }
431
432
12
    void clean() { leg_vector.clear(); }
433
434
private:
435
    std::vector<std::unique_ptr<leg_info>> leg_vector;
436
    bool _is_wildcard = false;        // whether the path is a wildcard path
437
    bool _is_supper_wildcard = false; // supper wildcard likes '$**.a' or '$**[1]'
438
};
439
440
/*
441
 * JsonbFwdIteratorT implements JSONB's iterator template.
442
 *
443
 * Note: it is an FORWARD iterator only due to the design of JSONB format.
444
 */
445
template <class Iter_Type, class Cont_Type>
446
class JsonbFwdIteratorT {
447
public:
448
    using iterator = Iter_Type;
449
    using pointer = typename std::iterator_traits<Iter_Type>::pointer;
450
    using reference = typename std::iterator_traits<Iter_Type>::reference;
451
452
    explicit JsonbFwdIteratorT() : current_(nullptr) {}
453
40.4k
    explicit JsonbFwdIteratorT(const iterator& i) : current_(i) {}
_ZN5doris17JsonbFwdIteratorTIPKNS_13JsonbKeyValueENS_9ObjectValEEC2ERKS3_
Line
Count
Source
453
40.1k
    explicit JsonbFwdIteratorT(const iterator& i) : current_(i) {}
_ZN5doris17JsonbFwdIteratorTIPKNS_10JsonbValueENS_8ArrayValEEC2ERKS3_
Line
Count
Source
453
262
    explicit JsonbFwdIteratorT(const iterator& i) : current_(i) {}
454
455
    // allow non-const to const iterator conversion (same container type)
456
    template <class Iter_Ty>
457
    JsonbFwdIteratorT(const JsonbFwdIteratorT<Iter_Ty, Cont_Type>& rhs) : current_(rhs.base()) {}
458
459
40.7k
    bool operator==(const JsonbFwdIteratorT& rhs) const { return (current_ == rhs.current_); }
_ZNK5doris17JsonbFwdIteratorTIPKNS_13JsonbKeyValueENS_9ObjectValEEeqERKS5_
Line
Count
Source
459
38.1k
    bool operator==(const JsonbFwdIteratorT& rhs) const { return (current_ == rhs.current_); }
_ZNK5doris17JsonbFwdIteratorTIPKNS_10JsonbValueENS_8ArrayValEEeqERKS5_
Line
Count
Source
459
2.65k
    bool operator==(const JsonbFwdIteratorT& rhs) const { return (current_ == rhs.current_); }
460
461
39.6k
    bool operator!=(const JsonbFwdIteratorT& rhs) const { return !operator==(rhs); }
_ZNK5doris17JsonbFwdIteratorTIPKNS_13JsonbKeyValueENS_9ObjectValEEneERKS5_
Line
Count
Source
461
37.1k
    bool operator!=(const JsonbFwdIteratorT& rhs) const { return !operator==(rhs); }
_ZNK5doris17JsonbFwdIteratorTIPKNS_10JsonbValueENS_8ArrayValEEneERKS5_
Line
Count
Source
461
2.52k
    bool operator!=(const JsonbFwdIteratorT& rhs) const { return !operator==(rhs); }
462
463
2.11k
    bool operator<(const JsonbFwdIteratorT& rhs) const { return (current_ < rhs.current_); }
464
465
    bool operator>(const JsonbFwdIteratorT& rhs) const { return !operator<(rhs); }
466
467
36.0k
    JsonbFwdIteratorT& operator++() {
468
36.0k
        current_ = (iterator)(((char*)current_) + current_->numPackedBytes());
469
36.0k
        return *this;
470
36.0k
    }
_ZN5doris17JsonbFwdIteratorTIPKNS_13JsonbKeyValueENS_9ObjectValEEppEv
Line
Count
Source
467
34.8k
    JsonbFwdIteratorT& operator++() {
468
34.8k
        current_ = (iterator)(((char*)current_) + current_->numPackedBytes());
469
34.8k
        return *this;
470
34.8k
    }
_ZN5doris17JsonbFwdIteratorTIPKNS_10JsonbValueENS_8ArrayValEEppEv
Line
Count
Source
467
1.20k
    JsonbFwdIteratorT& operator++() {
468
1.20k
        current_ = (iterator)(((char*)current_) + current_->numPackedBytes());
469
1.20k
        return *this;
470
1.20k
    }
471
472
    JsonbFwdIteratorT operator++(int) {
473
        auto tmp = *this;
474
        current_ = (iterator)(((char*)current_) + current_->numPackedBytes());
475
        return tmp;
476
    }
477
478
1.20k
    explicit operator pointer() { return current_; }
479
480
0
    reference operator*() const { return *current_; }
Unexecuted instantiation: _ZNK5doris17JsonbFwdIteratorTIPKNS_10JsonbValueENS_8ArrayValEEdeEv
Unexecuted instantiation: _ZNK5doris17JsonbFwdIteratorTIPKNS_13JsonbKeyValueENS_9ObjectValEEdeEv
481
482
57.0k
    pointer operator->() const { return current_; }
_ZNK5doris17JsonbFwdIteratorTIPKNS_13JsonbKeyValueENS_9ObjectValEEptEv
Line
Count
Source
482
57.0k
    pointer operator->() const { return current_; }
_ZNK5doris17JsonbFwdIteratorTIPKNS_10JsonbValueENS_8ArrayValEEptEv
Line
Count
Source
482
6
    pointer operator->() const { return current_; }
483
484
    iterator base() const { return current_; }
485
486
private:
487
    iterator current_;
488
};
489
using JsonbTypeUnder = std::underlying_type_t<JsonbType>;
490
491
#if defined(__clang__)
492
#pragma clang diagnostic push
493
#pragma clang diagnostic ignored "-Wzero-length-array"
494
#endif
495
#pragma pack(push, 1)
496
497
/*
498
 * JsonbDocument is the main object that accesses and queries JSONB packed
499
 * bytes. NOTE: JsonbDocument only allows object container as the top level
500
 * JSONB value. However, you can use the static method "createValue" to get any
501
 * JsonbValue object from the packed bytes.
502
 *
503
 * JsonbDocument object also dereferences to an object container value
504
 * (ObjectVal) once JSONB is loaded.
505
 *
506
 * ** Load **
507
 * JsonbDocument is usable after loading packed bytes (memory location) into
508
 * the object. We only need the header and first few bytes of the payload after
509
 * header to verify the JSONB.
510
 *
511
 * Note: creating an JsonbDocument (through createDocument) does not allocate
512
 * any memory. The document object is an efficient wrapper on the packed bytes
513
 * which is accessed directly.
514
 *
515
 * ** Query **
516
 * Query is through dereferencing into ObjectVal.
517
 */
518
class JsonbDocument {
519
public:
520
    // create an JsonbDocument object from JSONB packed bytes
521
    [[nodiscard]] static Status checkAndCreateDocument(const char* pb, size_t size,
522
                                                       const JsonbDocument** doc);
523
524
    // create an JsonbValue from JSONB packed bytes
525
    static const JsonbValue* createValue(const char* pb, size_t size);
526
527
0
    uint8_t version() const { return header_.ver_; }
528
529
52.4k
    const JsonbValue* getValue() const { return ((const JsonbValue*)payload_); }
530
531
    unsigned int numPackedBytes() const;
532
533
    const ObjectVal* operator->() const;
534
535
private:
536
    /*
537
   * JsonbHeader class defines JSONB header (internal to JsonbDocument).
538
   *
539
   * Currently it only contains version information (1-byte). We may expand the
540
   * header to include checksum of the JSONB binary for more security.
541
   */
542
    struct JsonbHeader {
543
        uint8_t ver_;
544
    } header_;
545
546
    char payload_[0];
547
};
548
549
/*
550
 * JsonbKeyValue class defines JSONB key type, as described below.
551
 *
552
 * key ::=
553
 *   0x00 int8    //1-byte dictionary id
554
 * | int8 (byte*) //int8 (>0) is the size of the key string
555
 *
556
 * value ::= primitive_value | container
557
 *
558
 * JsonbKeyValue can be either an id mapping to the key string in an external
559
 * dictionary, or it is the original key string. Whether to read an id or a
560
 * string is decided by the first byte (size).
561
 *
562
 * Note: a key object must be followed by a value object. Therefore, a key
563
 * object implicitly refers to a key-value pair, and you can get the value
564
 * object right after the key object. The function numPackedBytes hence
565
 * indicates the total size of the key-value pair, so that we will be able go
566
 * to next pair from the key.
567
 *
568
 * ** Dictionary size **
569
 * By default, the dictionary size is 255 (1-byte). Users can define
570
 * "USE_LARGE_DICT" to increase the dictionary size to 655535 (2-byte).
571
 */
572
class JsonbKeyValue {
573
public:
574
    // now we use sMaxKeyId to represent an empty key
575
    static const int sMaxKeyId = 65535;
576
    using keyid_type = uint16_t;
577
578
    static const uint8_t sMaxKeyLen = 64;
579
580
    // size of the key. 0 indicates it is stored as id
581
2.45k
    uint8_t klen() const { return size; }
582
583
    // get the key string. Note the string may not be null terminated.
584
1.26k
    const char* getKeyStr() const { return key.str_; }
585
586
18.4k
    keyid_type getKeyId() const { return key.id_; }
587
588
69.9k
    unsigned int keyPackedBytes() const {
589
69.9k
        return size ? (sizeof(size) + size) : (sizeof(size) + sizeof(keyid_type));
590
69.9k
    }
591
592
34.9k
    const JsonbValue* value() const {
593
34.9k
        return (const JsonbValue*)(((char*)this) + keyPackedBytes());
594
34.9k
    }
595
596
    // size of the total packed bytes (key+value)
597
    unsigned int numPackedBytes() const;
598
599
    uint8_t size;
600
601
    union key_ {
602
        keyid_type id_;
603
        char str_[1];
604
    } key;
605
};
606
607
struct JsonbFindResult {
608
    const JsonbValue* value = nullptr;   // found value
609
    std::unique_ptr<JsonbWriter> writer; // writer to write the value
610
    bool is_wildcard = false;            // whether the path is a wildcard path
611
};
612
613
/*
614
 * JsonbValue is the base class of all JSONB types. It contains only one member
615
 * variable - type info, which can be retrieved by member functions is[Type]()
616
 * or type().
617
 */
618
struct JsonbValue {
619
    static const uint32_t sMaxValueLen = 1 << 24; // 16M
620
621
8.45k
    bool isNull() const { return (type == JsonbType::T_Null); }
622
42
    bool isTrue() const { return (type == JsonbType::T_True); }
623
2
    bool isFalse() const { return (type == JsonbType::T_False); }
624
8
    bool isInt() const { return isInt8() || isInt16() || isInt32() || isInt64() || isInt128(); }
625
8
    bool isInt8() const { return (type == JsonbType::T_Int8); }
626
4
    bool isInt16() const { return (type == JsonbType::T_Int16); }
627
0
    bool isInt32() const { return (type == JsonbType::T_Int32); }
628
6
    bool isInt64() const { return (type == JsonbType::T_Int64); }
629
2
    bool isDouble() const { return (type == JsonbType::T_Double); }
630
2
    bool isFloat() const { return (type == JsonbType::T_Float); }
631
66
    bool isString() const { return (type == JsonbType::T_String); }
632
2.15k
    bool isBinary() const { return (type == JsonbType::T_Binary); }
633
12
    bool isObject() const { return (type == JsonbType::T_Object); }
634
14
    bool isArray() const { return (type == JsonbType::T_Array); }
635
6
    bool isInt128() const { return (type == JsonbType::T_Int128); }
636
8
    bool isDecimal() const {
637
8
        return (type == JsonbType::T_Decimal32 || type == JsonbType::T_Decimal64 ||
638
8
                type == JsonbType::T_Decimal128 || type == JsonbType::T_Decimal256);
639
8
    }
640
2
    bool isDecimal32() const { return (type == JsonbType::T_Decimal32); }
641
2
    bool isDecimal64() const { return (type == JsonbType::T_Decimal64); }
642
2
    bool isDecimal128() const { return (type == JsonbType::T_Decimal128); }
643
2
    bool isDecimal256() const { return (type == JsonbType::T_Decimal256); }
644
645
22
    PrimitiveType get_primitive_type() const { return get_primitive_type_from_json_type(type); }
646
647
0
    const char* typeName() const {
648
0
        switch (type) {
649
0
        case JsonbType::T_Null:
650
0
            return "null";
651
0
        case JsonbType::T_True:
652
0
        case JsonbType::T_False:
653
0
            return "bool";
654
0
        case JsonbType::T_Int8:
655
0
        case JsonbType::T_Int16:
656
0
        case JsonbType::T_Int32:
657
0
            return "int";
658
0
        case JsonbType::T_Int64:
659
0
            return "bigint";
660
0
        case JsonbType::T_Int128:
661
0
            return "largeint";
662
0
        case JsonbType::T_Double:
663
0
            return "double";
664
0
        case JsonbType::T_Float:
665
0
            return "float";
666
0
        case JsonbType::T_String:
667
0
            return "string";
668
0
        case JsonbType::T_Binary:
669
0
            return "binary";
670
0
        case JsonbType::T_Object:
671
0
            return "object";
672
0
        case JsonbType::T_Array:
673
0
            return "array";
674
0
        case JsonbType::T_Decimal32:
675
0
            return "Decimal32";
676
0
        case JsonbType::T_Decimal64:
677
0
            return "Decimal64";
678
0
        case JsonbType::T_Decimal128:
679
0
            return "Decimal128";
680
0
        case JsonbType::T_Decimal256:
681
0
            return "Decimal256";
682
0
        default:
683
0
            return "unknown";
684
0
        }
685
0
    }
686
687
    // size of the total packed bytes
688
    unsigned int numPackedBytes() const;
689
690
    // size of the value in bytes
691
    unsigned int size() const;
692
693
    //Get the number of jsonbvalue elements
694
    int numElements() const;
695
696
    //Whether to include the jsonbvalue rhs
697
    bool contains(const JsonbValue* rhs) const;
698
699
    // find the JSONB value by JsonbPath
700
    JsonbFindResult findValue(JsonbPath& path) const;
701
    friend class JsonbDocument;
702
703
    JsonbType type; // type info
704
705
    char payload[0]; // payload, which is the packed bytes of the value
706
707
    /**
708
    * @brief Unpacks the underlying Jsonb binary content as a pointer to type `T`.
709
    *
710
    * @tparam T A POD (Plain Old Data) type that must satisfy the `JsonbPodType` concept.
711
    *           This ensures that `T` is trivially copyable, standard-layout, and safe to
712
    *           reinterpret from raw bytes without invoking undefined behavior.
713
    *
714
    * @return A pointer to a `const T` object, interpreted from the internal buffer.
715
    *
716
    * @note The caller must ensure that the current JsonbValue actually contains data
717
    *       compatible with type `T`, otherwise the result is undefined.
718
    */
719
    template <JsonbPodType T>
720
103k
    const T* unpack() const {
721
103k
        static_assert(is_pod_v<T>, "T must be a POD type");
722
103k
        return reinterpret_cast<const T*>(payload);
723
103k
    }
_ZNK5doris10JsonbValue6unpackITkNS_12JsonbPodTypeENS_9ObjectValEEEPKT_v
Line
Count
Source
720
39.1k
    const T* unpack() const {
721
39.1k
        static_assert(is_pod_v<T>, "T must be a POD type");
722
39.1k
        return reinterpret_cast<const T*>(payload);
723
39.1k
    }
_ZNK5doris10JsonbValue6unpackITkNS_12JsonbPodTypeENS_10NumberValTIaEEEEPKT_v
Line
Count
Source
720
1.33k
    const T* unpack() const {
721
1.33k
        static_assert(is_pod_v<T>, "T must be a POD type");
722
1.33k
        return reinterpret_cast<const T*>(payload);
723
1.33k
    }
_ZNK5doris10JsonbValue6unpackITkNS_12JsonbPodTypeENS_10NumberValTIsEEEEPKT_v
Line
Count
Source
720
210
    const T* unpack() const {
721
210
        static_assert(is_pod_v<T>, "T must be a POD type");
722
210
        return reinterpret_cast<const T*>(payload);
723
210
    }
_ZNK5doris10JsonbValue6unpackITkNS_12JsonbPodTypeENS_10NumberValTIiEEEEPKT_v
Line
Count
Source
720
6.94k
    const T* unpack() const {
721
6.94k
        static_assert(is_pod_v<T>, "T must be a POD type");
722
6.94k
        return reinterpret_cast<const T*>(payload);
723
6.94k
    }
_ZNK5doris10JsonbValue6unpackITkNS_12JsonbPodTypeENS_10NumberValTIlEEEEPKT_v
Line
Count
Source
720
3.79k
    const T* unpack() const {
721
3.79k
        static_assert(is_pod_v<T>, "T must be a POD type");
722
3.79k
        return reinterpret_cast<const T*>(payload);
723
3.79k
    }
_ZNK5doris10JsonbValue6unpackITkNS_12JsonbPodTypeENS_10NumberValTInEEEEPKT_v
Line
Count
Source
720
8.33k
    const T* unpack() const {
721
8.33k
        static_assert(is_pod_v<T>, "T must be a POD type");
722
8.33k
        return reinterpret_cast<const T*>(payload);
723
8.33k
    }
_ZNK5doris10JsonbValue6unpackITkNS_12JsonbPodTypeENS_14JsonbBinaryValEEEPKT_v
Line
Count
Source
720
39.2k
    const T* unpack() const {
721
39.2k
        static_assert(is_pod_v<T>, "T must be a POD type");
722
39.2k
        return reinterpret_cast<const T*>(payload);
723
39.2k
    }
_ZNK5doris10JsonbValue6unpackITkNS_12JsonbPodTypeENS_12ContainerValEEEPKT_v
Line
Count
Source
720
3.61k
    const T* unpack() const {
721
3.61k
        static_assert(is_pod_v<T>, "T must be a POD type");
722
3.61k
        return reinterpret_cast<const T*>(payload);
723
3.61k
    }
_ZNK5doris10JsonbValue6unpackITkNS_12JsonbPodTypeENS_8ArrayValEEEPKT_v
Line
Count
Source
720
198
    const T* unpack() const {
721
198
        static_assert(is_pod_v<T>, "T must be a POD type");
722
198
        return reinterpret_cast<const T*>(payload);
723
198
    }
_ZNK5doris10JsonbValue6unpackITkNS_12JsonbPodTypeENS_10NumberValTIdEEEEPKT_v
Line
Count
Source
720
336
    const T* unpack() const {
721
336
        static_assert(is_pod_v<T>, "T must be a POD type");
722
336
        return reinterpret_cast<const T*>(payload);
723
336
    }
_ZNK5doris10JsonbValue6unpackITkNS_12JsonbPodTypeENS_10NumberValTIfEEEEPKT_v
Line
Count
Source
720
50
    const T* unpack() const {
721
50
        static_assert(is_pod_v<T>, "T must be a POD type");
722
50
        return reinterpret_cast<const T*>(payload);
723
50
    }
_ZNK5doris10JsonbValue6unpackITkNS_12JsonbPodTypeENS_14JsonbStringValEEEPKT_v
Line
Count
Source
720
502
    const T* unpack() const {
721
502
        static_assert(is_pod_v<T>, "T must be a POD type");
722
502
        return reinterpret_cast<const T*>(payload);
723
502
    }
_ZNK5doris10JsonbValue6unpackITkNS_12JsonbPodTypeENS_15JsonbDecimalValINS_7DecimalIiEEEEEEPKT_v
Line
Count
Source
720
26
    const T* unpack() const {
721
26
        static_assert(is_pod_v<T>, "T must be a POD type");
722
26
        return reinterpret_cast<const T*>(payload);
723
26
    }
_ZNK5doris10JsonbValue6unpackITkNS_12JsonbPodTypeENS_15JsonbDecimalValINS_7DecimalIlEEEEEEPKT_v
Line
Count
Source
720
26
    const T* unpack() const {
721
26
        static_assert(is_pod_v<T>, "T must be a POD type");
722
26
        return reinterpret_cast<const T*>(payload);
723
26
    }
_ZNK5doris10JsonbValue6unpackITkNS_12JsonbPodTypeENS_15JsonbDecimalValINS_12Decimal128V3EEEEEPKT_v
Line
Count
Source
720
34
    const T* unpack() const {
721
34
        static_assert(is_pod_v<T>, "T must be a POD type");
722
34
        return reinterpret_cast<const T*>(payload);
723
34
    }
_ZNK5doris10JsonbValue6unpackITkNS_12JsonbPodTypeENS_15JsonbDecimalValINS_7DecimalIN4wide7integerILm256EiEEEEEEEEPKT_v
Line
Count
Source
720
26
    const T* unpack() const {
721
26
        static_assert(is_pod_v<T>, "T must be a POD type");
722
26
        return reinterpret_cast<const T*>(payload);
723
26
    }
724
725
    // /**
726
    // * @brief Unpacks the underlying Jsonb binary content as a pointer to type `T`.
727
    // *
728
    // * @tparam T A POD (Plain Old Data) type that must satisfy the `JsonbPodType` concept.
729
    // *           This ensures that `T` is trivially copyable, standard-layout, and safe to
730
    // *           reinterpret from raw bytes without invoking undefined behavior.
731
    // *
732
    // * @return A pointer to a `T` object, interpreted from the internal buffer.
733
    // *
734
    // * @note The caller must ensure that the current JsonbValue actually contains data
735
    // *       compatible with type `T`, otherwise the result is undefined.
736
    // */
737
    // template <JsonbPodType T>
738
    // T* unpack() {
739
    //     static_assert(is_pod_v<T>, "T must be a POD type");
740
    //     return reinterpret_cast<T*>(payload);
741
    // }
742
743
    int128_t int_val() const;
744
};
745
746
// inline ObjectVal* JsonbDocument::operator->() {
747
//     return (((JsonbValue*)payload_)->unpack<ObjectVal>());
748
// }
749
750
38.1k
inline const ObjectVal* JsonbDocument::operator->() const {
751
38.1k
    return (((const JsonbValue*)payload_)->unpack<ObjectVal>());
752
38.1k
}
753
754
/*
755
 * NumerValT is the template class (derived from JsonbValue) of all number
756
 * types (integers and double).
757
 */
758
template <typename T>
759
    requires std::is_integral_v<T> || std::is_floating_point_v<T>
760
struct NumberValT {
761
public:
762
21.0k
    T val() const { return num; }
_ZNK5doris10NumberValTIaE3valEv
Line
Count
Source
762
1.33k
    T val() const { return num; }
_ZNK5doris10NumberValTIsE3valEv
Line
Count
Source
762
210
    T val() const { return num; }
_ZNK5doris10NumberValTIiE3valEv
Line
Count
Source
762
6.94k
    T val() const { return num; }
_ZNK5doris10NumberValTIlE3valEv
Line
Count
Source
762
3.79k
    T val() const { return num; }
_ZNK5doris10NumberValTInE3valEv
Line
Count
Source
762
8.33k
    T val() const { return num; }
_ZNK5doris10NumberValTIdE3valEv
Line
Count
Source
762
336
    T val() const { return num; }
_ZNK5doris10NumberValTIfE3valEv
Line
Count
Source
762
50
    T val() const { return num; }
763
764
    static unsigned int numPackedBytes() { return sizeof(JsonbValue) + sizeof(T); }
765
766
    T num;
767
};
768
769
18
inline int128_t JsonbValue::int_val() const {
770
18
    switch (type) {
771
6
    case JsonbType::T_Int8:
772
6
        return unpack<JsonbInt8Val>()->val();
773
2
    case JsonbType::T_Int16:
774
2
        return unpack<JsonbInt16Val>()->val();
775
0
    case JsonbType::T_Int32:
776
0
        return unpack<JsonbInt32Val>()->val();
777
6
    case JsonbType::T_Int64:
778
6
        return unpack<JsonbInt64Val>()->val();
779
4
    case JsonbType::T_Int128:
780
4
        return unpack<JsonbInt128Val>()->val();
781
0
    default:
782
0
        throw Exception(ErrorCode::INTERNAL_ERROR, "Invalid JSONB value type: {}",
783
0
                        static_cast<int32_t>(type));
784
18
    }
785
18
}
786
787
template <JsonbDecimalType T>
788
struct JsonbDecimalVal {
789
public:
790
    using NativeType = typename T::NativeType;
791
792
    // get the decimal value
793
52
    NativeType val() const {
794
        // to avoid memory alignment issues, we use memcpy to copy the value
795
52
        NativeType tmp;
796
52
        memcpy(&tmp, &value, sizeof(NativeType));
797
52
        return tmp;
798
52
    }
_ZNK5doris15JsonbDecimalValINS_7DecimalIiEEE3valEv
Line
Count
Source
793
12
    NativeType val() const {
794
        // to avoid memory alignment issues, we use memcpy to copy the value
795
12
        NativeType tmp;
796
12
        memcpy(&tmp, &value, sizeof(NativeType));
797
12
        return tmp;
798
12
    }
_ZNK5doris15JsonbDecimalValINS_7DecimalIlEEE3valEv
Line
Count
Source
793
12
    NativeType val() const {
794
        // to avoid memory alignment issues, we use memcpy to copy the value
795
12
        NativeType tmp;
796
12
        memcpy(&tmp, &value, sizeof(NativeType));
797
12
        return tmp;
798
12
    }
_ZNK5doris15JsonbDecimalValINS_12Decimal128V3EE3valEv
Line
Count
Source
793
16
    NativeType val() const {
794
        // to avoid memory alignment issues, we use memcpy to copy the value
795
16
        NativeType tmp;
796
16
        memcpy(&tmp, &value, sizeof(NativeType));
797
16
        return tmp;
798
16
    }
_ZNK5doris15JsonbDecimalValINS_7DecimalIN4wide7integerILm256EiEEEEE3valEv
Line
Count
Source
793
12
    NativeType val() const {
794
        // to avoid memory alignment issues, we use memcpy to copy the value
795
12
        NativeType tmp;
796
12
        memcpy(&tmp, &value, sizeof(NativeType));
797
12
        return tmp;
798
12
    }
799
800
58
    static constexpr int numPackedBytes() {
801
58
        return sizeof(JsonbValue) + sizeof(precision) + sizeof(scale) + sizeof(value);
802
58
    }
_ZN5doris15JsonbDecimalValINS_7DecimalIiEEE14numPackedBytesEv
Line
Count
Source
800
14
    static constexpr int numPackedBytes() {
801
14
        return sizeof(JsonbValue) + sizeof(precision) + sizeof(scale) + sizeof(value);
802
14
    }
_ZN5doris15JsonbDecimalValINS_7DecimalIlEEE14numPackedBytesEv
Line
Count
Source
800
14
    static constexpr int numPackedBytes() {
801
14
        return sizeof(JsonbValue) + sizeof(precision) + sizeof(scale) + sizeof(value);
802
14
    }
_ZN5doris15JsonbDecimalValINS_12Decimal128V3EE14numPackedBytesEv
Line
Count
Source
800
18
    static constexpr int numPackedBytes() {
801
18
        return sizeof(JsonbValue) + sizeof(precision) + sizeof(scale) + sizeof(value);
802
18
    }
_ZN5doris15JsonbDecimalValINS_7DecimalIN4wide7integerILm256EiEEEEE14numPackedBytesEv
Line
Count
Source
800
12
    static constexpr int numPackedBytes() {
801
12
        return sizeof(JsonbValue) + sizeof(precision) + sizeof(scale) + sizeof(value);
802
12
    }
803
804
    uint32_t precision;
805
    uint32_t scale;
806
    NativeType value;
807
};
808
809
/*
810
 * BlobVal is the base class (derived from JsonbValue) for string and binary
811
 * types. The size indicates the total bytes of the payload.
812
 */
813
struct JsonbBinaryVal {
814
public:
815
    // size of the blob payload only
816
4.23k
    unsigned int getBlobLen() const { return size; }
817
818
    // return the blob as byte array
819
9.02k
    const char* getBlob() const { return payload; }
820
821
    // size of the total packed bytes
822
30.4k
    unsigned int numPackedBytes() const { return sizeof(JsonbValue) + sizeof(size) + size; }
823
    friend class JsonbDocument;
824
825
    uint32_t size;
826
    char payload[0];
827
};
828
829
/*
830
 * String type
831
 * Note: JSONB string may not be a c-string (NULL-terminated)
832
 */
833
struct JsonbStringVal : public JsonbBinaryVal {
834
public:
835
    /*
836
    This function return the actual size of a string. Since for
837
    a string, it can be null-terminated with null paddings or it
838
    can take all the space in the payload without null in the end.
839
    So we need to check it to get the true actual length of a string.
840
  */
841
258
    size_t length() const {
842
        // It's an empty string
843
258
        if (0 == size) {
844
0
            return size;
845
0
        }
846
        // The string stored takes all the spaces in payload
847
258
        if (payload[size - 1] != 0) {
848
258
            return size;
849
258
        }
850
        // It's shorter than the size of payload
851
0
        return strnlen(payload, size);
852
258
    }
853
};
854
855
/*
856
 * ContainerVal is the base class (derived from JsonbValue) for object and
857
 * array types. The size indicates the total bytes of the payload.
858
 */
859
struct ContainerVal {
860
    // size of the container payload only
861
0
    unsigned int getContainerSize() const { return size; }
862
863
    // return the container payload as byte array
864
0
    const char* getPayload() const { return payload; }
865
866
    // size of the total packed bytes
867
3.61k
    unsigned int numPackedBytes() const { return sizeof(JsonbValue) + sizeof(size) + size; }
868
    friend class JsonbDocument;
869
870
    uint32_t size;
871
    char payload[0];
872
};
873
874
/*
875
 * Object type
876
 */
877
struct ObjectVal : public ContainerVal {
878
    using value_type = JsonbKeyValue;
879
    using pointer = value_type*;
880
    using const_pointer = const value_type*;
881
    using const_iterator = JsonbFwdIteratorT<const_pointer, ObjectVal>;
882
883
2
    const_iterator search(const char* key) const {
884
2
        if (!key) {
885
0
            return end();
886
0
        }
887
2
        return search(key, (unsigned int)strlen(key));
888
2
    }
889
890
58
    const_iterator search(const char* key, unsigned int klen) const {
891
58
        if (!key || !klen) {
892
0
            return end();
893
0
        }
894
58
        return internalSearch(key, klen);
895
58
    }
896
897
    // Get number of elements in object
898
16
    int numElem() const {
899
16
        const char* pch = payload;
900
16
        const char* fence = payload + size;
901
902
16
        unsigned int num = 0;
903
80
        while (pch < fence) {
904
64
            auto* pkey = (JsonbKeyValue*)(pch);
905
64
            ++num;
906
64
            pch += pkey->numPackedBytes();
907
64
        }
908
909
16
        assert(pch == fence);
910
911
16
        return num;
912
16
    }
913
914
    // find the JSONB value by a key string (null terminated)
915
2
    const JsonbValue* find(const char* key) const {
916
2
        if (!key) {
917
0
            return nullptr;
918
0
        }
919
2
        return find(key, (unsigned int)strlen(key));
920
2
    }
921
922
    // find the JSONB value by a key string (with length)
923
54
    const JsonbValue* find(const char* key, unsigned int klen) const {
924
54
        const_iterator kv = search(key, klen);
925
54
        if (end() == kv) {
926
4
            return nullptr;
927
4
        }
928
50
        return kv->value();
929
54
    }
930
931
3.20k
    const_iterator begin() const { return const_iterator((pointer)payload); }
932
933
36.9k
    const_iterator end() const { return const_iterator((pointer)(payload + size)); }
934
935
    std::vector<std::pair<StringRef, const JsonbValue*>> get_ordered_key_value_pairs() const;
936
937
private:
938
58
    const_iterator internalSearch(const char* key, unsigned int klen) const {
939
58
        const char* pch = payload;
940
58
        const char* fence = payload + size;
941
942
78
        while (pch < fence) {
943
74
            const auto* pkey = (const JsonbKeyValue*)(pch);
944
74
            if (klen == pkey->klen() && strncmp(key, pkey->getKeyStr(), klen) == 0) {
945
54
                return const_iterator(pkey);
946
54
            }
947
20
            pch += pkey->numPackedBytes();
948
20
        }
949
950
58
        assert(pch == fence);
951
952
4
        return end();
953
4
    }
954
};
955
956
/*
957
 * Array type
958
 */
959
struct ArrayVal : public ContainerVal {
960
    using value_type = JsonbValue;
961
    using pointer = value_type*;
962
    using const_pointer = const value_type*;
963
    using const_iterator = JsonbFwdIteratorT<const_pointer, ArrayVal>;
964
965
    // get the JSONB value at index
966
62
    const JsonbValue* get(int idx) const {
967
62
        if (idx < 0) {
968
0
            return nullptr;
969
0
        }
970
971
62
        const char* pch = payload;
972
62
        const char* fence = payload + size;
973
974
144
        while (pch < fence && idx-- > 0) {
975
82
            pch += ((const JsonbValue*)pch)->numPackedBytes();
976
82
        }
977
62
        if (idx > 0 || pch == fence) {
978
14
            return nullptr;
979
14
        }
980
981
48
        return (const JsonbValue*)pch;
982
62
    }
983
984
    // Get number of elements in array
985
16
    int numElem() const {
986
16
        const char* pch = payload;
987
16
        const char* fence = payload + size;
988
989
16
        unsigned int num = 0;
990
58
        while (pch < fence) {
991
42
            ++num;
992
42
            pch += ((const JsonbValue*)pch)->numPackedBytes();
993
42
        }
994
995
16
        assert(pch == fence);
996
997
16
        return num;
998
16
    }
999
1000
132
    const_iterator begin() const { return const_iterator((pointer)payload); }
1001
1002
130
    const_iterator end() const { return const_iterator((pointer)(payload + size)); }
1003
};
1004
1005
24
inline const JsonbValue* JsonbDocument::createValue(const char* pb, size_t size) {
1006
24
    if (!pb || size < sizeof(JsonbHeader) + sizeof(JsonbValue)) {
1007
0
        return nullptr;
1008
0
    }
1009
1010
24
    auto* doc = (JsonbDocument*)pb;
1011
24
    if (doc->header_.ver_ != JSONB_VER) {
1012
0
        return nullptr;
1013
0
    }
1014
1015
24
    const auto* val = (const JsonbValue*)doc->payload_;
1016
24
    if (size != sizeof(JsonbHeader) + val->numPackedBytes()) {
1017
0
        return nullptr;
1018
0
    }
1019
1020
24
    return val;
1021
24
}
1022
1023
0
inline unsigned int JsonbDocument::numPackedBytes() const {
1024
0
    return ((const JsonbValue*)payload_)->numPackedBytes() + sizeof(header_);
1025
0
}
1026
1027
34.9k
inline unsigned int JsonbKeyValue::numPackedBytes() const {
1028
34.9k
    unsigned int ks = keyPackedBytes();
1029
34.9k
    const auto* val = (const JsonbValue*)(((char*)this) + ks);
1030
34.9k
    return ks + val->numPackedBytes();
1031
34.9k
}
1032
1033
// Poor man's "virtual" function JsonbValue::numPackedBytes
1034
141k
inline unsigned int JsonbValue::numPackedBytes() const {
1035
141k
    switch (type) {
1036
5.50k
    case JsonbType::T_Null:
1037
24.5k
    case JsonbType::T_True:
1038
25.7k
    case JsonbType::T_False: {
1039
25.7k
        return sizeof(type);
1040
24.5k
    }
1041
1042
1.38k
    case JsonbType::T_Int8: {
1043
1.38k
        return sizeof(type) + sizeof(int8_t);
1044
24.5k
    }
1045
244
    case JsonbType::T_Int16: {
1046
244
        return sizeof(type) + sizeof(int16_t);
1047
24.5k
    }
1048
6.99k
    case JsonbType::T_Int32: {
1049
6.99k
        return sizeof(type) + sizeof(int32_t);
1050
24.5k
    }
1051
24.3k
    case JsonbType::T_Int64: {
1052
24.3k
        return sizeof(type) + sizeof(int64_t);
1053
24.5k
    }
1054
21.0k
    case JsonbType::T_Double: {
1055
21.0k
        return sizeof(type) + sizeof(double);
1056
24.5k
    }
1057
52
    case JsonbType::T_Float: {
1058
52
        return sizeof(type) + sizeof(float);
1059
24.5k
    }
1060
27.7k
    case JsonbType::T_Int128: {
1061
27.7k
        return sizeof(type) + sizeof(int128_t);
1062
24.5k
    }
1063
21.7k
    case JsonbType::T_String:
1064
30.4k
    case JsonbType::T_Binary: {
1065
30.4k
        return unpack<JsonbBinaryVal>()->numPackedBytes();
1066
21.7k
    }
1067
1068
3.31k
    case JsonbType::T_Object:
1069
3.61k
    case JsonbType::T_Array: {
1070
3.61k
        return unpack<ContainerVal>()->numPackedBytes();
1071
3.31k
    }
1072
14
    case JsonbType::T_Decimal32: {
1073
14
        return JsonbDecimal32::numPackedBytes();
1074
3.31k
    }
1075
14
    case JsonbType::T_Decimal64: {
1076
14
        return JsonbDecimal64::numPackedBytes();
1077
3.31k
    }
1078
18
    case JsonbType::T_Decimal128: {
1079
18
        return JsonbDecimal128::numPackedBytes();
1080
3.31k
    }
1081
12
    case JsonbType::T_Decimal256: {
1082
12
        return JsonbDecimal256::numPackedBytes();
1083
3.31k
    }
1084
0
    case JsonbType::NUM_TYPES:
1085
0
        break;
1086
141k
    }
1087
1088
0
    throw Exception(ErrorCode::INTERNAL_ERROR, "Invalid JSONB value type: {}",
1089
0
                    static_cast<int32_t>(type));
1090
141k
}
1091
1092
12
inline int JsonbValue::numElements() const {
1093
12
    switch (type) {
1094
0
    case JsonbType::T_Int8:
1095
0
    case JsonbType::T_Int16:
1096
0
    case JsonbType::T_Int32:
1097
0
    case JsonbType::T_Int64:
1098
0
    case JsonbType::T_Double:
1099
0
    case JsonbType::T_Float:
1100
0
    case JsonbType::T_Int128:
1101
2
    case JsonbType::T_String:
1102
2
    case JsonbType::T_Binary:
1103
4
    case JsonbType::T_Null:
1104
4
    case JsonbType::T_True:
1105
4
    case JsonbType::T_False:
1106
4
    case JsonbType::T_Decimal32:
1107
4
    case JsonbType::T_Decimal64:
1108
4
    case JsonbType::T_Decimal128:
1109
4
    case JsonbType::T_Decimal256: {
1110
4
        return 1;
1111
4
    }
1112
0
    case JsonbType::T_Object: {
1113
0
        return unpack<ObjectVal>()->numElem();
1114
4
    }
1115
8
    case JsonbType::T_Array: {
1116
8
        return unpack<ArrayVal>()->numElem();
1117
4
    }
1118
0
    case JsonbType::NUM_TYPES:
1119
0
        break;
1120
12
    }
1121
0
    throw Exception(ErrorCode::INTERNAL_ERROR, "Invalid JSONB value type: {}",
1122
0
                    static_cast<int32_t>(type));
1123
12
}
1124
1125
6
inline bool JsonbValue::contains(const JsonbValue* rhs) const {
1126
6
    switch (type) {
1127
2
    case JsonbType::T_Int8:
1128
2
    case JsonbType::T_Int16:
1129
2
    case JsonbType::T_Int32:
1130
2
    case JsonbType::T_Int64:
1131
2
    case JsonbType::T_Int128: {
1132
2
        return rhs->isInt() && this->int_val() == rhs->int_val();
1133
2
    }
1134
0
    case JsonbType::T_Double:
1135
0
    case JsonbType::T_Float: {
1136
0
        if (!rhs->isDouble() && !rhs->isFloat()) {
1137
0
            return false;
1138
0
        }
1139
0
        double left = isDouble() ? unpack<JsonbDoubleVal>()->val() : unpack<JsonbFloatVal>()->val();
1140
0
        double right = rhs->isDouble() ? rhs->unpack<JsonbDoubleVal>()->val()
1141
0
                                       : rhs->unpack<JsonbFloatVal>()->val();
1142
0
        return left == right;
1143
0
    }
1144
2
    case JsonbType::T_String:
1145
2
    case JsonbType::T_Binary: {
1146
2
        if (rhs->isString() || rhs->isBinary()) {
1147
2
            const auto* str_value1 = unpack<JsonbStringVal>();
1148
2
            const auto* str_value2 = rhs->unpack<JsonbStringVal>();
1149
2
            return str_value1->length() == str_value2->length() &&
1150
2
                   std::memcmp(str_value1->getBlob(), str_value2->getBlob(),
1151
2
                               str_value1->length()) == 0;
1152
2
        }
1153
0
        return false;
1154
2
    }
1155
2
    case JsonbType::T_Array: {
1156
2
        int lhs_num = unpack<ArrayVal>()->numElem();
1157
2
        if (rhs->isArray()) {
1158
0
            int rhs_num = rhs->unpack<ArrayVal>()->numElem();
1159
0
            if (rhs_num > lhs_num) {
1160
0
                return false;
1161
0
            }
1162
0
            int contains_num = 0;
1163
0
            for (int i = 0; i < lhs_num; ++i) {
1164
0
                for (int j = 0; j < rhs_num; ++j) {
1165
0
                    if (unpack<ArrayVal>()->get(i)->contains(rhs->unpack<ArrayVal>()->get(j))) {
1166
0
                        contains_num++;
1167
0
                        break;
1168
0
                    }
1169
0
                }
1170
0
            }
1171
0
            return contains_num == rhs_num;
1172
0
        }
1173
2
        for (int i = 0; i < lhs_num; ++i) {
1174
2
            if (unpack<ArrayVal>()->get(i)->contains(rhs)) {
1175
2
                return true;
1176
2
            }
1177
2
        }
1178
0
        return false;
1179
2
    }
1180
0
    case JsonbType::T_Object: {
1181
0
        if (rhs->isObject()) {
1182
0
            const auto* obj_value1 = unpack<ObjectVal>();
1183
0
            const auto* obj_value2 = rhs->unpack<ObjectVal>();
1184
0
            for (auto it = obj_value2->begin(); it != obj_value2->end(); ++it) {
1185
0
                const JsonbValue* value = obj_value1->find(it->getKeyStr(), it->klen());
1186
0
                if (value == nullptr || !value->contains(it->value())) {
1187
0
                    return false;
1188
0
                }
1189
0
            }
1190
0
            return true;
1191
0
        }
1192
0
        return false;
1193
0
    }
1194
0
    case JsonbType::T_Null: {
1195
0
        return rhs->isNull();
1196
0
    }
1197
0
    case JsonbType::T_True: {
1198
0
        return rhs->isTrue();
1199
0
    }
1200
0
    case JsonbType::T_False: {
1201
0
        return rhs->isFalse();
1202
0
    }
1203
0
    case JsonbType::T_Decimal32: {
1204
0
        if (rhs->isDecimal32()) {
1205
0
            return unpack<JsonbDecimal32>()->val() == rhs->unpack<JsonbDecimal32>()->val() &&
1206
0
                   unpack<JsonbDecimal32>()->precision ==
1207
0
                           rhs->unpack<JsonbDecimal32>()->precision &&
1208
0
                   unpack<JsonbDecimal32>()->scale == rhs->unpack<JsonbDecimal32>()->scale;
1209
0
        }
1210
0
        return false;
1211
0
    }
1212
0
    case JsonbType::T_Decimal64: {
1213
0
        if (rhs->isDecimal64()) {
1214
0
            return unpack<JsonbDecimal64>()->val() == rhs->unpack<JsonbDecimal64>()->val() &&
1215
0
                   unpack<JsonbDecimal64>()->precision ==
1216
0
                           rhs->unpack<JsonbDecimal64>()->precision &&
1217
0
                   unpack<JsonbDecimal64>()->scale == rhs->unpack<JsonbDecimal64>()->scale;
1218
0
        }
1219
0
        return false;
1220
0
    }
1221
0
    case JsonbType::T_Decimal128: {
1222
0
        if (rhs->isDecimal128()) {
1223
0
            return unpack<JsonbDecimal128>()->val() == rhs->unpack<JsonbDecimal128>()->val() &&
1224
0
                   unpack<JsonbDecimal128>()->precision ==
1225
0
                           rhs->unpack<JsonbDecimal128>()->precision &&
1226
0
                   unpack<JsonbDecimal128>()->scale == rhs->unpack<JsonbDecimal128>()->scale;
1227
0
        }
1228
0
        return false;
1229
0
    }
1230
0
    case JsonbType::T_Decimal256: {
1231
0
        if (rhs->isDecimal256()) {
1232
0
            return unpack<JsonbDecimal256>()->val() == rhs->unpack<JsonbDecimal256>()->val() &&
1233
0
                   unpack<JsonbDecimal256>()->precision ==
1234
0
                           rhs->unpack<JsonbDecimal256>()->precision &&
1235
0
                   unpack<JsonbDecimal256>()->scale == rhs->unpack<JsonbDecimal256>()->scale;
1236
0
        }
1237
0
        return false;
1238
0
    }
1239
0
    case JsonbType::NUM_TYPES:
1240
0
        break;
1241
6
    }
1242
1243
0
    throw Exception(ErrorCode::INTERNAL_ERROR, "Invalid JSONB value type: {}",
1244
0
                    static_cast<int32_t>(type));
1245
6
}
1246
1247
192
inline bool JsonbPath::seek(const char* key_path, size_t kp_len) {
1248
192
    while (kp_len > 0 && std::isspace(key_path[kp_len - 1])) {
1249
0
        --kp_len;
1250
0
    }
1251
1252
    //path invalid
1253
192
    if (!key_path || kp_len == 0) {
1254
0
        return false;
1255
0
    }
1256
192
    Stream stream(key_path, kp_len);
1257
192
    stream.skip_whitespace();
1258
192
    if (stream.exhausted() || stream.read() != SCOPE) {
1259
        //path invalid
1260
0
        return false;
1261
0
    }
1262
1263
372
    while (!stream.exhausted()) {
1264
180
        stream.skip_whitespace();
1265
180
        stream.clear_leg_ptr();
1266
180
        stream.clear_leg_len();
1267
1268
180
        if (!JsonbPath::parsePath(&stream, this)) {
1269
            //path invalid
1270
0
            return false;
1271
0
        }
1272
180
    }
1273
192
    return true;
1274
192
}
1275
1276
180
inline bool JsonbPath::parsePath(Stream* stream, JsonbPath* path) {
1277
    // $[0]
1278
180
    if (stream->peek() == BEGIN_ARRAY) {
1279
116
        return parse_array(stream, path);
1280
116
    }
1281
    // $.a or $.[0]
1282
64
    else if (stream->peek() == BEGIN_MEMBER) {
1283
        // advance past the .
1284
64
        stream->skip(1);
1285
1286
64
        if (stream->exhausted()) {
1287
0
            return false;
1288
0
        }
1289
1290
        // $.[0]
1291
64
        if (stream->peek() == BEGIN_ARRAY) {
1292
0
            return parse_array(stream, path);
1293
0
        }
1294
        // $.a
1295
64
        else {
1296
64
            return parse_member(stream, path);
1297
64
        }
1298
64
    } else if (stream->peek() == WILDCARD) {
1299
0
        stream->skip(1);
1300
0
        if (stream->exhausted()) {
1301
0
            return false;
1302
0
        }
1303
1304
        // $**
1305
0
        if (stream->peek() == WILDCARD) {
1306
0
            path->_is_supper_wildcard = true;
1307
0
        }
1308
1309
0
        stream->skip(1);
1310
0
        if (stream->exhausted()) {
1311
0
            return false;
1312
0
        }
1313
1314
0
        if (stream->peek() == BEGIN_ARRAY) {
1315
0
            return parse_array(stream, path);
1316
0
        } else if (stream->peek() == BEGIN_MEMBER) {
1317
            // advance past the .
1318
0
            stream->skip(1);
1319
1320
0
            if (stream->exhausted()) {
1321
0
                return false;
1322
0
            }
1323
1324
            // $.[0]
1325
0
            if (stream->peek() == BEGIN_ARRAY) {
1326
0
                return parse_array(stream, path);
1327
0
            }
1328
            // $.a
1329
0
            else {
1330
0
                return parse_member(stream, path);
1331
0
            }
1332
0
        }
1333
0
        return false;
1334
0
    } else {
1335
0
        return false; //invalid json path
1336
0
    }
1337
180
}
1338
1339
116
inline bool JsonbPath::parse_array(Stream* stream, JsonbPath* path) {
1340
116
    assert(stream->peek() == BEGIN_ARRAY);
1341
116
    stream->skip(1);
1342
116
    if (stream->exhausted()) {
1343
0
        return false;
1344
0
    }
1345
1346
116
    if (stream->peek() == WILDCARD) {
1347
        // Called by function_jsonb.cpp, the variables passed in originate from a mutable block;
1348
        // using const_cast is acceptable.
1349
0
        stream->set_leg_ptr(const_cast<char*>(stream->position()));
1350
0
        stream->add_leg_len();
1351
0
        stream->skip(1);
1352
0
        if (stream->exhausted()) {
1353
0
            return false;
1354
0
        }
1355
1356
0
        if (stream->peek() == END_ARRAY) {
1357
0
            std::unique_ptr<leg_info> leg(
1358
0
                    new leg_info(stream->get_leg_ptr(), stream->get_leg_len(), 0, ARRAY_CODE));
1359
0
            path->add_leg_to_leg_vector(std::move(leg));
1360
0
            stream->skip(1);
1361
0
            path->_is_wildcard = true;
1362
0
            return true;
1363
0
        } else {
1364
0
            return false;
1365
0
        }
1366
0
    }
1367
1368
    // Called by function_jsonb.cpp, the variables passed in originate from a mutable block;
1369
    // using const_cast is acceptable.
1370
116
    stream->set_leg_ptr(const_cast<char*>(stream->position()));
1371
1372
232
    for (; !stream->exhausted() && stream->peek() != END_ARRAY; stream->advance()) {
1373
116
        stream->add_leg_len();
1374
116
    }
1375
1376
116
    if (stream->exhausted() || stream->peek() != END_ARRAY) {
1377
0
        return false;
1378
116
    } else {
1379
116
        stream->skip(1);
1380
116
    }
1381
1382
    //parse array index to int
1383
1384
116
    std::string_view idx_string(stream->get_leg_ptr(), stream->get_leg_len());
1385
116
    int index = 0;
1386
1387
116
    if (stream->get_leg_len() >= 4 &&
1388
116
        std::equal(LAST, LAST + 4, stream->get_leg_ptr(),
1389
0
                   [](char c1, char c2) { return std::tolower(c1) == std::tolower(c2); })) {
1390
0
        auto pos = idx_string.find(MINUS);
1391
1392
0
        if (pos != std::string::npos) {
1393
0
            for (size_t i = 4; i < pos; ++i) {
1394
0
                if (std::isspace(idx_string[i])) {
1395
0
                    continue;
1396
0
                } else {
1397
                    // leading zeroes are not allowed
1398
0
                    LOG(WARNING) << "Non-space char in idx_string: '" << idx_string << "'";
1399
0
                    return false;
1400
0
                }
1401
0
            }
1402
0
            idx_string = idx_string.substr(pos + 1);
1403
0
            idx_string = trim(idx_string);
1404
1405
0
            auto result = std::from_chars(idx_string.data(), idx_string.data() + idx_string.size(),
1406
0
                                          index);
1407
0
            if (result.ec != std::errc()) {
1408
0
                LOG(WARNING) << "Invalid index in JSON path: '" << idx_string << "'";
1409
0
                return false;
1410
0
            }
1411
1412
0
        } else if (stream->get_leg_len() > 4) {
1413
0
            return false;
1414
0
        }
1415
1416
0
        std::unique_ptr<leg_info> leg(new leg_info(nullptr, 0, -index - 1, ARRAY_CODE));
1417
0
        path->add_leg_to_leg_vector(std::move(leg));
1418
1419
0
        return true;
1420
0
    }
1421
1422
116
    auto result = std::from_chars(idx_string.data(), idx_string.data() + idx_string.size(), index);
1423
1424
116
    if (result.ec != std::errc()) {
1425
0
        return false;
1426
0
    }
1427
1428
116
    std::unique_ptr<leg_info> leg(new leg_info(nullptr, 0, index, ARRAY_CODE));
1429
116
    path->add_leg_to_leg_vector(std::move(leg));
1430
1431
116
    return true;
1432
116
}
1433
1434
64
inline bool JsonbPath::parse_member(Stream* stream, JsonbPath* path) {
1435
64
    if (stream->exhausted()) {
1436
0
        return false;
1437
0
    }
1438
1439
64
    if (stream->peek() == WILDCARD) {
1440
        // Called by function_jsonb.cpp, the variables passed in originate from a mutable block;
1441
        // using const_cast is acceptable.
1442
0
        stream->set_leg_ptr(const_cast<char*>(stream->position()));
1443
0
        stream->add_leg_len();
1444
0
        stream->skip(1);
1445
0
        std::unique_ptr<leg_info> leg(
1446
0
                new leg_info(stream->get_leg_ptr(), stream->get_leg_len(), 0, MEMBER_CODE));
1447
0
        path->add_leg_to_leg_vector(std::move(leg));
1448
0
        path->_is_wildcard = true;
1449
0
        return true;
1450
0
    }
1451
1452
    // Called by function_jsonb.cpp, the variables passed in originate from a mutable block;
1453
    // using const_cast is acceptable.
1454
64
    stream->set_leg_ptr(const_cast<char*>(stream->position()));
1455
1456
64
    const char* left_quotation_marks = nullptr;
1457
64
    const char* right_quotation_marks = nullptr;
1458
1459
192
    for (; !stream->exhausted(); stream->advance()) {
1460
        // Only accept space characters quoted by double quotes.
1461
128
        if (std::isspace(stream->peek()) && left_quotation_marks == nullptr) {
1462
0
            return false;
1463
128
        } else if (stream->peek() == ESCAPE) {
1464
0
            stream->add_leg_len();
1465
0
            stream->skip(1);
1466
0
            stream->add_leg_len();
1467
0
            stream->set_has_escapes(true);
1468
0
            if (stream->exhausted()) {
1469
0
                return false;
1470
0
            }
1471
0
            continue;
1472
128
        } else if (stream->peek() == DOUBLE_QUOTE) {
1473
0
            if (left_quotation_marks == nullptr) {
1474
0
                left_quotation_marks = stream->position();
1475
                // Called by function_jsonb.cpp, the variables passed in originate from a mutable block;
1476
                // using const_cast is acceptable.
1477
0
                stream->set_leg_ptr(const_cast<char*>(++left_quotation_marks));
1478
0
                continue;
1479
0
            } else {
1480
0
                right_quotation_marks = stream->position();
1481
0
                stream->skip(1);
1482
0
                break;
1483
0
            }
1484
128
        } else if (stream->peek() == BEGIN_MEMBER || stream->peek() == BEGIN_ARRAY) {
1485
0
            if (left_quotation_marks == nullptr) {
1486
0
                break;
1487
0
            }
1488
0
        }
1489
1490
128
        stream->add_leg_len();
1491
128
    }
1492
1493
64
    if ((left_quotation_marks != nullptr && right_quotation_marks == nullptr) ||
1494
64
        stream->get_leg_ptr() == nullptr || stream->get_leg_len() == 0) {
1495
0
        return false; //invalid json path
1496
0
    }
1497
1498
64
    if (stream->get_has_escapes()) {
1499
0
        stream->remove_escapes();
1500
0
    }
1501
1502
64
    std::unique_ptr<leg_info> leg(
1503
64
            new leg_info(stream->get_leg_ptr(), stream->get_leg_len(), 0, MEMBER_CODE));
1504
64
    path->add_leg_to_leg_vector(std::move(leg));
1505
1506
64
    return true;
1507
64
}
1508
1509
static_assert(is_pod_v<JsonbDocument>, "JsonbDocument must be standard layout and trivial");
1510
static_assert(is_pod_v<JsonbValue>, "JsonbValue must be standard layout and trivial");
1511
static_assert(is_pod_v<JsonbDecimal32>, "JsonbDecimal32 must be standard layout and trivial");
1512
static_assert(is_pod_v<JsonbDecimal64>, "JsonbDecimal64 must be standard layout and trivial");
1513
static_assert(is_pod_v<JsonbDecimal128>, "JsonbDecimal128 must be standard layout and trivial");
1514
static_assert(is_pod_v<JsonbDecimal256>, "JsonbDecimal256 must be standard layout and trivial");
1515
static_assert(is_pod_v<JsonbInt8Val>, "JsonbInt8Val must be standard layout and trivial");
1516
static_assert(is_pod_v<JsonbInt32Val>, "JsonbInt32Val must be standard layout and trivial");
1517
static_assert(is_pod_v<JsonbInt64Val>, "JsonbInt64Val must be standard layout and trivial");
1518
static_assert(is_pod_v<JsonbInt128Val>, "JsonbInt128Val must be standard layout and trivial");
1519
static_assert(is_pod_v<JsonbDoubleVal>, "JsonbDoubleVal must be standard layout and trivial");
1520
static_assert(is_pod_v<JsonbFloatVal>, "JsonbFloatVal must be standard layout and trivial");
1521
static_assert(is_pod_v<JsonbBinaryVal>, "JsonbBinaryVal must be standard layout and trivial");
1522
static_assert(is_pod_v<ContainerVal>, "ContainerVal must be standard layout and trivial");
1523
1524
#define ASSERT_DECIMAL_LAYOUT(type)                \
1525
    static_assert(offsetof(type, precision) == 0); \
1526
    static_assert(offsetof(type, scale) == 4);     \
1527
    static_assert(offsetof(type, value) == 8);
1528
1529
ASSERT_DECIMAL_LAYOUT(JsonbDecimal32)
1530
ASSERT_DECIMAL_LAYOUT(JsonbDecimal64)
1531
ASSERT_DECIMAL_LAYOUT(JsonbDecimal128)
1532
ASSERT_DECIMAL_LAYOUT(JsonbDecimal256)
1533
1534
#define ASSERT_NUMERIC_LAYOUT(type) static_assert(offsetof(type, num) == 0);
1535
1536
ASSERT_NUMERIC_LAYOUT(JsonbInt8Val)
1537
ASSERT_NUMERIC_LAYOUT(JsonbInt32Val)
1538
ASSERT_NUMERIC_LAYOUT(JsonbInt64Val)
1539
ASSERT_NUMERIC_LAYOUT(JsonbInt128Val)
1540
ASSERT_NUMERIC_LAYOUT(JsonbDoubleVal)
1541
1542
static_assert(offsetof(JsonbBinaryVal, size) == 0);
1543
static_assert(offsetof(JsonbBinaryVal, payload) == 4);
1544
1545
static_assert(offsetof(ContainerVal, size) == 0);
1546
static_assert(offsetof(ContainerVal, payload) == 4);
1547
1548
#pragma pack(pop)
1549
#if defined(__clang__)
1550
#pragma clang diagnostic pop
1551
#endif
1552
} // namespace doris
1553
1554
#endif // JSONB_JSONBDOCUMENT_H