jsonb_op.c File Reference

#include "postgres.h"
#include "catalog/pg_type.h"
#include "utils/fmgrprotos.h"
#include "utils/jsonb.h"

Include dependency graph for jsonb_op.c:

Go to the source code of this file.

Functions
Datum	jsonb_exists (PG_FUNCTION_ARGS)
Datum	jsonb_exists_any (PG_FUNCTION_ARGS)
Datum	jsonb_exists_all (PG_FUNCTION_ARGS)
Datum	jsonb_contains (PG_FUNCTION_ARGS)
Datum	jsonb_contained (PG_FUNCTION_ARGS)
Datum	jsonb_ne (PG_FUNCTION_ARGS)
Datum	jsonb_lt (PG_FUNCTION_ARGS)
Datum	jsonb_gt (PG_FUNCTION_ARGS)
Datum	jsonb_le (PG_FUNCTION_ARGS)
Datum	jsonb_ge (PG_FUNCTION_ARGS)
Datum	jsonb_eq (PG_FUNCTION_ARGS)
Datum	jsonb_cmp (PG_FUNCTION_ARGS)
Datum	jsonb_hash (PG_FUNCTION_ARGS)
Datum	jsonb_hash_extended (PG_FUNCTION_ARGS)

Function Documentation

jsonb_cmp()

Datum jsonb_cmp

(

PG_FUNCTION_ARGS

)

Definition at line 236 of file jsonb_op.c.

{
    Jsonb      *jba = PG_GETARG_JSONB_P(0);
    Jsonb      *jbb = PG_GETARG_JSONB_P(1);
    int         res;
 
    res = compareJsonbContainers(&jba->root, &jbb->root);
 
    PG_FREE_IF_COPY(jba, 0);
    PG_FREE_IF_COPY(jbb, 1);
    PG_RETURN_INT32(res);
}

References compareJsonbContainers(), PG_FREE_IF_COPY, PG_GETARG_JSONB_P, PG_RETURN_INT32, and Jsonb::root.

jsonb_contained()

Datum jsonb_contained

(

PG_FUNCTION_ARGS

)

Definition at line 130 of file jsonb_op.c.

{
    /* Commutator of "contains" */
    Jsonb      *tmpl = PG_GETARG_JSONB_P(0);
    Jsonb      *val = PG_GETARG_JSONB_P(1);
 
    JsonbIterator *it1,
               *it2;
 
    if (JB_ROOT_IS_OBJECT(val) != JB_ROOT_IS_OBJECT(tmpl))
        PG_RETURN_BOOL(false);
 
    it1 = JsonbIteratorInit(&val->root);
    it2 = JsonbIteratorInit(&tmpl->root);
 
    PG_RETURN_BOOL(JsonbDeepContains(&it1, &it2));
}

References JB_ROOT_IS_OBJECT, JsonbDeepContains(), JsonbIteratorInit(), PG_GETARG_JSONB_P, PG_RETURN_BOOL, Jsonb::root, and val.

jsonb_contains()

Datum jsonb_contains

(

PG_FUNCTION_ARGS

)

Definition at line 112 of file jsonb_op.c.

{
    Jsonb      *val = PG_GETARG_JSONB_P(0);
    Jsonb      *tmpl = PG_GETARG_JSONB_P(1);
 
    JsonbIterator *it1,
               *it2;
 
    if (JB_ROOT_IS_OBJECT(val) != JB_ROOT_IS_OBJECT(tmpl))
        PG_RETURN_BOOL(false);
 
    it1 = JsonbIteratorInit(&val->root);
    it2 = JsonbIteratorInit(&tmpl->root);
 
    PG_RETURN_BOOL(JsonbDeepContains(&it1, &it2));
}

References JB_ROOT_IS_OBJECT, JsonbDeepContains(), JsonbIteratorInit(), PG_GETARG_JSONB_P, PG_RETURN_BOOL, Jsonb::root, and val.

jsonb_eq()

Datum jsonb_eq

(

PG_FUNCTION_ARGS

)

Definition at line 222 of file jsonb_op.c.

{
    Jsonb      *jba = PG_GETARG_JSONB_P(0);
    Jsonb      *jbb = PG_GETARG_JSONB_P(1);
    bool        res;
 
    res = (compareJsonbContainers(&jba->root, &jbb->root) == 0);
 
    PG_FREE_IF_COPY(jba, 0);
    PG_FREE_IF_COPY(jbb, 1);
    PG_RETURN_BOOL(res);
}

References compareJsonbContainers(), PG_FREE_IF_COPY, PG_GETARG_JSONB_P, PG_RETURN_BOOL, and Jsonb::root.

jsonb_exists()

Datum jsonb_exists

(

PG_FUNCTION_ARGS

)

Definition at line 21 of file jsonb_op.c.

{
    Jsonb      *jb = PG_GETARG_JSONB_P(0);
    text       *key = PG_GETARG_TEXT_PP(1);
    JsonbValue  kval;
    JsonbValue *v = NULL;
 
    /*
     * We only match Object keys (which are naturally always Strings), or
     * string elements in arrays.  In particular, we do not match non-string
     * scalar elements.  Existence of a key/element is only considered at the
     * top level.  No recursion occurs.
     */
    kval.type = jbvString;
    kval.val.string.val = VARDATA_ANY(key);
    kval.val.string.len = VARSIZE_ANY_EXHDR(key);
 
    v = findJsonbValueFromContainer(&jb->root,
                                    JB_FOBJECT | JB_FARRAY,
                                    &kval);
 
    PG_RETURN_BOOL(v != NULL);
}

References findJsonbValueFromContainer(), JB_FARRAY, JB_FOBJECT, jbvString, sort-test::key, PG_GETARG_JSONB_P, PG_GETARG_TEXT_PP, PG_RETURN_BOOL, Jsonb::root, JsonbValue::type, JsonbValue::val, VARDATA_ANY, and VARSIZE_ANY_EXHDR.

jsonb_exists_all()

Datum jsonb_exists_all

(

PG_FUNCTION_ARGS

)

Definition at line 79 of file jsonb_op.c.

{
    Jsonb      *jb = PG_GETARG_JSONB_P(0);
    ArrayType  *keys = PG_GETARG_ARRAYTYPE_P(1);
    int         i;
    Datum      *key_datums;
    bool       *key_nulls;
    int         elem_count;
 
    deconstruct_array_builtin(keys, TEXTOID, &key_datums, &key_nulls, &elem_count);
 
    for (i = 0; i < elem_count; i++)
    {
        JsonbValue  strVal;
 
        if (key_nulls[i])
            continue;
 
        strVal.type = jbvString;
        /* We rely on the array elements not being toasted */
        strVal.val.string.val = VARDATA_ANY(key_datums[i]);
        strVal.val.string.len = VARSIZE_ANY_EXHDR(key_datums[i]);
 
        if (findJsonbValueFromContainer(&jb->root,
                                        JB_FOBJECT | JB_FARRAY,
                                        &strVal) == NULL)
            PG_RETURN_BOOL(false);
    }
 
    PG_RETURN_BOOL(true);
}

References deconstruct_array_builtin(), findJsonbValueFromContainer(), i, JB_FARRAY, JB_FOBJECT, jbvString, PG_GETARG_ARRAYTYPE_P, PG_GETARG_JSONB_P, PG_RETURN_BOOL, Jsonb::root, strVal, VARDATA_ANY, and VARSIZE_ANY_EXHDR.

jsonb_exists_any()

Datum jsonb_exists_any

(

PG_FUNCTION_ARGS

)

Definition at line 46 of file jsonb_op.c.

{
    Jsonb      *jb = PG_GETARG_JSONB_P(0);
    ArrayType  *keys = PG_GETARG_ARRAYTYPE_P(1);
    int         i;
    Datum      *key_datums;
    bool       *key_nulls;
    int         elem_count;
 
    deconstruct_array_builtin(keys, TEXTOID, &key_datums, &key_nulls, &elem_count);
 
    for (i = 0; i < elem_count; i++)
    {
        JsonbValue  strVal;
 
        if (key_nulls[i])
            continue;
 
        strVal.type = jbvString;
        /* We rely on the array elements not being toasted */
        strVal.val.string.val = VARDATA_ANY(key_datums[i]);
        strVal.val.string.len = VARSIZE_ANY_EXHDR(key_datums[i]);
 
        if (findJsonbValueFromContainer(&jb->root,
                                        JB_FOBJECT | JB_FARRAY,
                                        &strVal) != NULL)
            PG_RETURN_BOOL(true);
    }
 
    PG_RETURN_BOOL(false);
}

References deconstruct_array_builtin(), findJsonbValueFromContainer(), i, JB_FARRAY, JB_FOBJECT, jbvString, PG_GETARG_ARRAYTYPE_P, PG_GETARG_JSONB_P, PG_RETURN_BOOL, Jsonb::root, strVal, VARDATA_ANY, and VARSIZE_ANY_EXHDR.

jsonb_ge()

Datum jsonb_ge

(

PG_FUNCTION_ARGS

)

Definition at line 208 of file jsonb_op.c.

{
    Jsonb      *jba = PG_GETARG_JSONB_P(0);
    Jsonb      *jbb = PG_GETARG_JSONB_P(1);
    bool        res;
 
    res = (compareJsonbContainers(&jba->root, &jbb->root) >= 0);
 
    PG_FREE_IF_COPY(jba, 0);
    PG_FREE_IF_COPY(jbb, 1);
    PG_RETURN_BOOL(res);
}

References compareJsonbContainers(), PG_FREE_IF_COPY, PG_GETARG_JSONB_P, PG_RETURN_BOOL, and Jsonb::root.

jsonb_gt()

Datum jsonb_gt

(

PG_FUNCTION_ARGS

)

Definition at line 180 of file jsonb_op.c.

{
    Jsonb      *jba = PG_GETARG_JSONB_P(0);
    Jsonb      *jbb = PG_GETARG_JSONB_P(1);
    bool        res;
 
    res = (compareJsonbContainers(&jba->root, &jbb->root) > 0);
 
    PG_FREE_IF_COPY(jba, 0);
    PG_FREE_IF_COPY(jbb, 1);
    PG_RETURN_BOOL(res);
}

References compareJsonbContainers(), PG_FREE_IF_COPY, PG_GETARG_JSONB_P, PG_RETURN_BOOL, and Jsonb::root.

jsonb_hash()

Datum jsonb_hash

(

PG_FUNCTION_ARGS

)

Definition at line 253 of file jsonb_op.c.

{
    Jsonb      *jb = PG_GETARG_JSONB_P(0);
    JsonbIterator *it;
    JsonbValue  v;
    JsonbIteratorToken r;
    uint32      hash = 0;
 
    if (JB_ROOT_COUNT(jb) == 0)
        PG_RETURN_INT32(0);
 
    it = JsonbIteratorInit(&jb->root);
 
    while ((r = JsonbIteratorNext(&it, &v, false)) != WJB_DONE)
    {
        switch (r)
        {
                /* Rotation is left to JsonbHashScalarValue() */
            case WJB_BEGIN_ARRAY:
                hash ^= JB_FARRAY;
                break;
            case WJB_BEGIN_OBJECT:
                hash ^= JB_FOBJECT;
                break;
            case WJB_KEY:
            case WJB_VALUE:
            case WJB_ELEM:
                JsonbHashScalarValue(&v, &hash);
                break;
            case WJB_END_ARRAY:
            case WJB_END_OBJECT:
                break;
            default:
                elog(ERROR, "invalid JsonbIteratorNext rc: %d", (int) r);
        }
    }
 
    PG_FREE_IF_COPY(jb, 0);
    PG_RETURN_INT32(hash);
}

References elog, ERROR, hash(), JB_FARRAY, JB_FOBJECT, JB_ROOT_COUNT, JsonbHashScalarValue(), JsonbIteratorInit(), JsonbIteratorNext(), PG_FREE_IF_COPY, PG_GETARG_JSONB_P, PG_RETURN_INT32, Jsonb::root, WJB_BEGIN_ARRAY, WJB_BEGIN_OBJECT, WJB_DONE, WJB_ELEM, WJB_END_ARRAY, WJB_END_OBJECT, WJB_KEY, and WJB_VALUE.

jsonb_hash_extended()

Datum jsonb_hash_extended

(

PG_FUNCTION_ARGS

)

Definition at line 295 of file jsonb_op.c.

{
    Jsonb      *jb = PG_GETARG_JSONB_P(0);
    uint64      seed = PG_GETARG_INT64(1);
    JsonbIterator *it;
    JsonbValue  v;
    JsonbIteratorToken r;
    uint64      hash = 0;
 
    if (JB_ROOT_COUNT(jb) == 0)
        PG_RETURN_UINT64(seed);
 
    it = JsonbIteratorInit(&jb->root);
 
    while ((r = JsonbIteratorNext(&it, &v, false)) != WJB_DONE)
    {
        switch (r)
        {
                /* Rotation is left to JsonbHashScalarValueExtended() */
            case WJB_BEGIN_ARRAY:
                hash ^= ((uint64) JB_FARRAY) << 32 | JB_FARRAY;
                break;
            case WJB_BEGIN_OBJECT:
                hash ^= ((uint64) JB_FOBJECT) << 32 | JB_FOBJECT;
                break;
            case WJB_KEY:
            case WJB_VALUE:
            case WJB_ELEM:
                JsonbHashScalarValueExtended(&v, &hash, seed);
                break;
            case WJB_END_ARRAY:
            case WJB_END_OBJECT:
                break;
            default:
                elog(ERROR, "invalid JsonbIteratorNext rc: %d", (int) r);
        }
    }
 
    PG_FREE_IF_COPY(jb, 0);
    PG_RETURN_UINT64(hash);
}

References elog, ERROR, hash(), JB_FARRAY, JB_FOBJECT, JB_ROOT_COUNT, JsonbHashScalarValueExtended(), JsonbIteratorInit(), JsonbIteratorNext(), PG_FREE_IF_COPY, PG_GETARG_INT64, PG_GETARG_JSONB_P, PG_RETURN_UINT64, Jsonb::root, WJB_BEGIN_ARRAY, WJB_BEGIN_OBJECT, WJB_DONE, WJB_ELEM, WJB_END_ARRAY, WJB_END_OBJECT, WJB_KEY, and WJB_VALUE.

jsonb_le()

Datum jsonb_le

(

PG_FUNCTION_ARGS

)

Definition at line 194 of file jsonb_op.c.

{
    Jsonb      *jba = PG_GETARG_JSONB_P(0);
    Jsonb      *jbb = PG_GETARG_JSONB_P(1);
    bool        res;
 
    res = (compareJsonbContainers(&jba->root, &jbb->root) <= 0);
 
    PG_FREE_IF_COPY(jba, 0);
    PG_FREE_IF_COPY(jbb, 1);
    PG_RETURN_BOOL(res);
}

References compareJsonbContainers(), PG_FREE_IF_COPY, PG_GETARG_JSONB_P, PG_RETURN_BOOL, and Jsonb::root.

jsonb_lt()

Datum jsonb_lt

(

PG_FUNCTION_ARGS

)

Definition at line 166 of file jsonb_op.c.

{
    Jsonb      *jba = PG_GETARG_JSONB_P(0);
    Jsonb      *jbb = PG_GETARG_JSONB_P(1);
    bool        res;
 
    res = (compareJsonbContainers(&jba->root, &jbb->root) < 0);
 
    PG_FREE_IF_COPY(jba, 0);
    PG_FREE_IF_COPY(jbb, 1);
    PG_RETURN_BOOL(res);
}

References compareJsonbContainers(), PG_FREE_IF_COPY, PG_GETARG_JSONB_P, PG_RETURN_BOOL, and Jsonb::root.

jsonb_ne()

Datum jsonb_ne

(

PG_FUNCTION_ARGS

)

Definition at line 149 of file jsonb_op.c.

{
    Jsonb      *jba = PG_GETARG_JSONB_P(0);
    Jsonb      *jbb = PG_GETARG_JSONB_P(1);
    bool        res;
 
    res = (compareJsonbContainers(&jba->root, &jbb->root) != 0);
 
    PG_FREE_IF_COPY(jba, 0);
    PG_FREE_IF_COPY(jbb, 1);
    PG_RETURN_BOOL(res);
}

References compareJsonbContainers(), PG_FREE_IF_COPY, PG_GETARG_JSONB_P, PG_RETURN_BOOL, and Jsonb::root.