json.c File Reference

#include "postgres.h"
#include "access/htup_details.h"
#include "catalog/pg_proc.h"
#include "catalog/pg_type.h"
#include "common/hashfn.h"
#include "funcapi.h"
#include "libpq/pqformat.h"
#include "miscadmin.h"
#include "port/simd.h"
#include "utils/array.h"
#include "utils/builtins.h"
#include "utils/date.h"
#include "utils/datetime.h"
#include "utils/fmgroids.h"
#include "utils/json.h"
#include "utils/jsonfuncs.h"
#include "utils/lsyscache.h"
#include "utils/typcache.h"

Include dependency graph for json.c:

Go to the source code of this file.

Data Structures
struct	JsonUniqueHashEntry
struct	JsonUniqueStackEntry
struct	JsonUniqueParsingState
struct	JsonUniqueBuilderState
struct	JsonAggState

Macros
#define	ESCAPE_JSON_FLUSH_AFTER   512

Typedefs
typedef struct HTAB *	JsonUniqueCheckState
typedef struct JsonUniqueHashEntry	JsonUniqueHashEntry
typedef struct JsonUniqueStackEntry	JsonUniqueStackEntry
typedef struct JsonUniqueParsingState	JsonUniqueParsingState
typedef struct JsonUniqueBuilderState	JsonUniqueBuilderState
typedef struct JsonAggState	JsonAggState

Functions
static void	composite_to_json (Datum composite, StringInfo result, bool use_line_feeds)
static void	array_dim_to_json (StringInfo result, int dim, int ndims, int *dims, const Datum *vals, const bool *nulls, int *valcount, JsonTypeCategory tcategory, Oid outfuncoid, bool use_line_feeds)
static void	array_to_json_internal (Datum array, StringInfo result, bool use_line_feeds)
static void	datum_to_json_internal (Datum val, bool is_null, StringInfo result, JsonTypeCategory tcategory, Oid outfuncoid, bool key_scalar)
static void	add_json (Datum val, bool is_null, StringInfo result, Oid val_type, bool key_scalar)
static text *	catenate_stringinfo_string (StringInfo buffer, const char *addon)
Datum	json_in (PG_FUNCTION_ARGS)
Datum	json_out (PG_FUNCTION_ARGS)
Datum	json_send (PG_FUNCTION_ARGS)
Datum	json_recv (PG_FUNCTION_ARGS)
char *	JsonEncodeDateTime (char *buf, Datum value, Oid typid, const int *tzp)
Datum	array_to_json (PG_FUNCTION_ARGS)
Datum	array_to_json_pretty (PG_FUNCTION_ARGS)
Datum	row_to_json (PG_FUNCTION_ARGS)
Datum	row_to_json_pretty (PG_FUNCTION_ARGS)
bool	to_json_is_immutable (Oid typoid)
Datum	to_json (PG_FUNCTION_ARGS)
Datum	datum_to_json (Datum val, JsonTypeCategory tcategory, Oid outfuncoid)
static Datum	json_agg_transfn_worker (FunctionCallInfo fcinfo, bool absent_on_null)
Datum	json_agg_transfn (PG_FUNCTION_ARGS)
Datum	json_agg_strict_transfn (PG_FUNCTION_ARGS)
Datum	json_agg_finalfn (PG_FUNCTION_ARGS)
static uint32	json_unique_hash (const void *key, Size keysize)
static int	json_unique_hash_match (const void *key1, const void *key2, Size keysize)
static void	json_unique_check_init (JsonUniqueCheckState *cxt)
static void	json_unique_builder_init (JsonUniqueBuilderState *cxt)
static bool	json_unique_check_key (JsonUniqueCheckState *cxt, const char *key, int object_id)
static StringInfo	json_unique_builder_get_throwawaybuf (JsonUniqueBuilderState *cxt)
static Datum	json_object_agg_transfn_worker (FunctionCallInfo fcinfo, bool absent_on_null, bool unique_keys)
Datum	json_object_agg_transfn (PG_FUNCTION_ARGS)
Datum	json_object_agg_strict_transfn (PG_FUNCTION_ARGS)
Datum	json_object_agg_unique_transfn (PG_FUNCTION_ARGS)
Datum	json_object_agg_unique_strict_transfn (PG_FUNCTION_ARGS)
Datum	json_object_agg_finalfn (PG_FUNCTION_ARGS)
Datum	json_build_object_worker (int nargs, const Datum *args, const bool *nulls, const Oid *types, bool absent_on_null, bool unique_keys)
Datum	json_build_object (PG_FUNCTION_ARGS)
Datum	json_build_object_noargs (PG_FUNCTION_ARGS)
Datum	json_build_array_worker (int nargs, const Datum *args, const bool *nulls, const Oid *types, bool absent_on_null)
Datum	json_build_array (PG_FUNCTION_ARGS)
Datum	json_build_array_noargs (PG_FUNCTION_ARGS)
Datum	json_object (PG_FUNCTION_ARGS)
Datum	json_object_two_arg (PG_FUNCTION_ARGS)
static pg_attribute_always_inline void	escape_json_char (StringInfo buf, char c)
void	escape_json (StringInfo buf, const char *str)
void	escape_json_with_len (StringInfo buf, const char *str, int len)
void	escape_json_text (StringInfo buf, const text *txt)
static JsonParseErrorType	json_unique_object_start (void *_state)
static JsonParseErrorType	json_unique_object_end (void *_state)
static JsonParseErrorType	json_unique_object_field_start (void *_state, char *field, bool isnull)
bool	json_validate (text *json, bool check_unique_keys, bool throw_error)
Datum	json_typeof (PG_FUNCTION_ARGS)

Macro Definition Documentation

ESCAPE_JSON_FLUSH_AFTER

#define ESCAPE_JSON_FLUSH_AFTER   512

Definition at line 1624 of file json.c.

Typedef Documentation

JsonAggState

typedef struct JsonAggState JsonAggState

JsonUniqueBuilderState

typedef struct JsonUniqueBuilderState JsonUniqueBuilderState

JsonUniqueCheckState

typedef struct HTAB* JsonUniqueCheckState

Definition at line 42 of file json.c.

JsonUniqueHashEntry

typedef struct JsonUniqueHashEntry JsonUniqueHashEntry

JsonUniqueParsingState

typedef struct JsonUniqueParsingState JsonUniqueParsingState

JsonUniqueStackEntry

typedef struct JsonUniqueStackEntry JsonUniqueStackEntry

Function Documentation

add_json()

static void add_json ( Datum  val, bool  is_null, StringInfo  result, Oid  val_type, bool  key_scalar  )

static

Definition at line 603 of file json.c.

{
    JsonTypeCategory tcategory;
    Oid         outfuncoid;
 
    if (val_type == InvalidOid)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                 errmsg("could not determine input data type")));
 
    if (is_null)
    {
        tcategory = JSONTYPE_NULL;
        outfuncoid = InvalidOid;
    }
    else
        json_categorize_type(val_type, false,
                             &tcategory, &outfuncoid);
 
    datum_to_json_internal(val, is_null, result, tcategory, outfuncoid,
                           key_scalar);
}

References datum_to_json_internal(), ereport, errcode(), errmsg(), ERROR, InvalidOid, json_categorize_type(), JSONTYPE_NULL, and val.

Referenced by json_build_array_worker(), and json_build_object_worker().

array_dim_to_json()

static void array_dim_to_json ( StringInfo  result, int  dim, int  ndims, int *  dims, const Datum *  vals, const bool *  nulls, int *  valcount, JsonTypeCategory  tcategory, Oid  outfuncoid, bool  use_line_feeds  )

static

Definition at line 432 of file json.c.

{
    int         i;
    const char *sep;
 
    Assert(dim < ndims);
 
    sep = use_line_feeds ? ",\n " : ",";
 
    appendStringInfoChar(result, '[');
 
    for (i = 1; i <= dims[dim]; i++)
    {
        if (i > 1)
            appendStringInfoString(result, sep);
 
        if (dim + 1 == ndims)
        {
            datum_to_json_internal(vals[*valcount], nulls[*valcount],
                                   result, tcategory,
                                   outfuncoid, false);
            (*valcount)++;
        }
        else
        {
            /*
             * Do we want line feeds on inner dimensions of arrays? For now
             * we'll say no.
             */
            array_dim_to_json(result, dim + 1, ndims, dims, vals, nulls,
                              valcount, tcategory, outfuncoid, false);
        }
    }
 
    appendStringInfoChar(result, ']');
}

References appendStringInfoChar(), appendStringInfoString(), array_dim_to_json(), Assert(), datum_to_json_internal(), and i.

Referenced by array_dim_to_json(), and array_to_json_internal().

array_to_json()

Datum array_to_json

(

PG_FUNCTION_ARGS

)

Definition at line 631 of file json.c.

{
    Datum       array = PG_GETARG_DATUM(0);
    StringInfoData result;
 
    initStringInfo(&result);
 
    array_to_json_internal(array, &result, false);
 
    PG_RETURN_TEXT_P(cstring_to_text_with_len(result.data, result.len));
}

References array_to_json_internal(), cstring_to_text_with_len(), StringInfoData::data, initStringInfo(), StringInfoData::len, PG_GETARG_DATUM, and PG_RETURN_TEXT_P.

array_to_json_internal()

static void array_to_json_internal ( Datum  array, StringInfo  result, bool  use_line_feeds  )

static

Definition at line 475 of file json.c.

{
    ArrayType  *v = DatumGetArrayTypeP(array);
    Oid         element_type = ARR_ELEMTYPE(v);
    int        *dim;
    int         ndim;
    int         nitems;
    int         count = 0;
    Datum      *elements;
    bool       *nulls;
    int16       typlen;
    bool        typbyval;
    char        typalign;
    JsonTypeCategory tcategory;
    Oid         outfuncoid;
 
    ndim = ARR_NDIM(v);
    dim = ARR_DIMS(v);
    nitems = ArrayGetNItems(ndim, dim);
 
    if (nitems <= 0)
    {
        appendStringInfoString(result, "[]");
        return;
    }
 
    get_typlenbyvalalign(element_type,
                         &typlen, &typbyval, &typalign);
 
    json_categorize_type(element_type, false,
                         &tcategory, &outfuncoid);
 
    deconstruct_array(v, element_type, typlen, typbyval,
                      typalign, &elements, &nulls,
                      &nitems);
 
    array_dim_to_json(result, 0, ndim, dim, elements, nulls, &count, tcategory,
                      outfuncoid, use_line_feeds);
 
    pfree(elements);
    pfree(nulls);
}

References appendStringInfoString(), ARR_DIMS, ARR_ELEMTYPE, ARR_NDIM, array_dim_to_json(), ArrayGetNItems(), DatumGetArrayTypeP, deconstruct_array(), get_typlenbyvalalign(), json_categorize_type(), nitems, pfree(), and typalign.

Referenced by array_to_json(), array_to_json_pretty(), and datum_to_json_internal().

array_to_json_pretty()

Datum array_to_json_pretty

(

PG_FUNCTION_ARGS

)

Definition at line 647 of file json.c.

{
    Datum       array = PG_GETARG_DATUM(0);
    bool        use_line_feeds = PG_GETARG_BOOL(1);
    StringInfoData result;
 
    initStringInfo(&result);
 
    array_to_json_internal(array, &result, use_line_feeds);
 
    PG_RETURN_TEXT_P(cstring_to_text_with_len(result.data, result.len));
}

References array_to_json_internal(), cstring_to_text_with_len(), StringInfoData::data, initStringInfo(), StringInfoData::len, PG_GETARG_BOOL, PG_GETARG_DATUM, and PG_RETURN_TEXT_P.

catenate_stringinfo_string()

static text * catenate_stringinfo_string ( StringInfo  buffer, const char *  addon  )

static

Definition at line 1211 of file json.c.

{
    /* custom version of cstring_to_text_with_len */
    int         buflen = buffer->len;
    int         addlen = strlen(addon);
    text       *result = (text *) palloc(buflen + addlen + VARHDRSZ);
 
    SET_VARSIZE(result, buflen + addlen + VARHDRSZ);
    memcpy(VARDATA(result), buffer->data, buflen);
    memcpy(VARDATA(result) + buflen, addon, addlen);
 
    return result;
}

References StringInfoData::data, StringInfoData::len, palloc(), SET_VARSIZE(), VARDATA(), and VARHDRSZ.

Referenced by json_agg_finalfn(), and json_object_agg_finalfn().

composite_to_json()

static void composite_to_json ( Datum  composite, StringInfo  result, bool  use_line_feeds  )

static

Definition at line 522 of file json.c.

{
    HeapTupleHeader td;
    Oid         tupType;
    int32       tupTypmod;
    TupleDesc   tupdesc;
    HeapTupleData tmptup,
               *tuple;
    int         i;
    bool        needsep = false;
    const char *sep;
    int         seplen;
 
    /*
     * We can avoid expensive strlen() calls by precalculating the separator
     * length.
     */
    sep = use_line_feeds ? ",\n " : ",";
    seplen = use_line_feeds ? strlen(",\n ") : strlen(",");
 
    td = DatumGetHeapTupleHeader(composite);
 
    /* Extract rowtype info and find a tupdesc */
    tupType = HeapTupleHeaderGetTypeId(td);
    tupTypmod = HeapTupleHeaderGetTypMod(td);
    tupdesc = lookup_rowtype_tupdesc(tupType, tupTypmod);
 
    /* Build a temporary HeapTuple control structure */
    tmptup.t_len = HeapTupleHeaderGetDatumLength(td);
    tmptup.t_data = td;
    tuple = &tmptup;
 
    appendStringInfoChar(result, '{');
 
    for (i = 0; i < tupdesc->natts; i++)
    {
        Datum       val;
        bool        isnull;
        char       *attname;
        JsonTypeCategory tcategory;
        Oid         outfuncoid;
        Form_pg_attribute att = TupleDescAttr(tupdesc, i);
 
        if (att->attisdropped)
            continue;
 
        if (needsep)
            appendBinaryStringInfo(result, sep, seplen);
        needsep = true;
 
        attname = NameStr(att->attname);
        escape_json(result, attname);
        appendStringInfoChar(result, ':');
 
        val = heap_getattr(tuple, i + 1, tupdesc, &isnull);
 
        if (isnull)
        {
            tcategory = JSONTYPE_NULL;
            outfuncoid = InvalidOid;
        }
        else
            json_categorize_type(att->atttypid, false, &tcategory,
                                 &outfuncoid);
 
        datum_to_json_internal(val, isnull, result, tcategory, outfuncoid,
                               false);
    }
 
    appendStringInfoChar(result, '}');
    ReleaseTupleDesc(tupdesc);
}

References appendBinaryStringInfo(), appendStringInfoChar(), attname, datum_to_json_internal(), DatumGetHeapTupleHeader, escape_json(), heap_getattr(), HeapTupleHeaderGetDatumLength(), HeapTupleHeaderGetTypeId(), HeapTupleHeaderGetTypMod(), i, InvalidOid, json_categorize_type(), JSONTYPE_NULL, lookup_rowtype_tupdesc(), NameStr, TupleDescData::natts, ReleaseTupleDesc, HeapTupleData::t_data, HeapTupleData::t_len, TupleDescAttr(), and val.

Referenced by datum_to_json_internal(), row_to_json(), and row_to_json_pretty().

datum_to_json()

Datum datum_to_json	(	Datum	val,
		JsonTypeCategory	tcategory,
		Oid	outfuncoid
	)

Definition at line 764 of file json.c.

{
    StringInfoData result;
 
    initStringInfo(&result);
    datum_to_json_internal(val, false, &result, tcategory, outfuncoid,
                           false);
 
    return PointerGetDatum(cstring_to_text_with_len(result.data, result.len));
}

References cstring_to_text_with_len(), StringInfoData::data, datum_to_json_internal(), initStringInfo(), StringInfoData::len, PointerGetDatum(), and val.

Referenced by ExecEvalJsonConstructor(), and to_json().

datum_to_json_internal()

static void datum_to_json_internal ( Datum  val, bool  is_null, StringInfo  result, JsonTypeCategory  tcategory, Oid  outfuncoid, bool  key_scalar  )

static

Definition at line 180 of file json.c.

{
    char       *outputstr;
    text       *jsontext;
 
    check_stack_depth();
 
    /* callers are expected to ensure that null keys are not passed in */
    Assert(!(key_scalar && is_null));
 
    if (is_null)
    {
        appendBinaryStringInfo(result, "null", strlen("null"));
        return;
    }
 
    if (key_scalar &&
        (tcategory == JSONTYPE_ARRAY ||
         tcategory == JSONTYPE_COMPOSITE ||
         tcategory == JSONTYPE_JSON ||
         tcategory == JSONTYPE_CAST))
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                 errmsg("key value must be scalar, not array, composite, or json")));
 
    switch (tcategory)
    {
        case JSONTYPE_ARRAY:
            array_to_json_internal(val, result, false);
            break;
        case JSONTYPE_COMPOSITE:
            composite_to_json(val, result, false);
            break;
        case JSONTYPE_BOOL:
            if (key_scalar)
                appendStringInfoChar(result, '"');
            if (DatumGetBool(val))
                appendBinaryStringInfo(result, "true", strlen("true"));
            else
                appendBinaryStringInfo(result, "false", strlen("false"));
            if (key_scalar)
                appendStringInfoChar(result, '"');
            break;
        case JSONTYPE_NUMERIC:
            outputstr = OidOutputFunctionCall(outfuncoid, val);
 
            /*
             * Don't quote a non-key if it's a valid JSON number (i.e., not
             * "Infinity", "-Infinity", or "NaN").  Since we know this is a
             * numeric data type's output, we simplify and open-code the
             * validation for better performance.
             */
            if (!key_scalar &&
                ((*outputstr >= '0' && *outputstr <= '9') ||
                 (*outputstr == '-' &&
                  (outputstr[1] >= '0' && outputstr[1] <= '9'))))
                appendStringInfoString(result, outputstr);
            else
            {
                appendStringInfoChar(result, '"');
                appendStringInfoString(result, outputstr);
                appendStringInfoChar(result, '"');
            }
            pfree(outputstr);
            break;
        case JSONTYPE_DATE:
            {
                char        buf[MAXDATELEN + 1];
 
                JsonEncodeDateTime(buf, val, DATEOID, NULL);
                appendStringInfoChar(result, '"');
                appendStringInfoString(result, buf);
                appendStringInfoChar(result, '"');
            }
            break;
        case JSONTYPE_TIMESTAMP:
            {
                char        buf[MAXDATELEN + 1];
 
                JsonEncodeDateTime(buf, val, TIMESTAMPOID, NULL);
                appendStringInfoChar(result, '"');
                appendStringInfoString(result, buf);
                appendStringInfoChar(result, '"');
            }
            break;
        case JSONTYPE_TIMESTAMPTZ:
            {
                char        buf[MAXDATELEN + 1];
 
                JsonEncodeDateTime(buf, val, TIMESTAMPTZOID, NULL);
                appendStringInfoChar(result, '"');
                appendStringInfoString(result, buf);
                appendStringInfoChar(result, '"');
            }
            break;
        case JSONTYPE_JSON:
            /* JSON and JSONB output will already be escaped */
            outputstr = OidOutputFunctionCall(outfuncoid, val);
            appendStringInfoString(result, outputstr);
            pfree(outputstr);
            break;
        case JSONTYPE_CAST:
            /* outfuncoid refers to a cast function, not an output function */
            jsontext = DatumGetTextPP(OidFunctionCall1(outfuncoid, val));
            appendBinaryStringInfo(result, VARDATA_ANY(jsontext),
                                   VARSIZE_ANY_EXHDR(jsontext));
            pfree(jsontext);
            break;
        default:
            /* special-case text types to save useless palloc/memcpy cycles */
            if (outfuncoid == F_TEXTOUT || outfuncoid == F_VARCHAROUT ||
                outfuncoid == F_BPCHAROUT)
                escape_json_text(result, (text *) DatumGetPointer(val));
            else
            {
                outputstr = OidOutputFunctionCall(outfuncoid, val);
                escape_json(result, outputstr);
                pfree(outputstr);
            }
            break;
    }
}

References appendBinaryStringInfo(), appendStringInfoChar(), appendStringInfoString(), array_to_json_internal(), Assert(), buf, check_stack_depth(), composite_to_json(), DatumGetBool(), DatumGetPointer(), DatumGetTextPP, ereport, errcode(), errmsg(), ERROR, escape_json(), escape_json_text(), JsonEncodeDateTime(), JSONTYPE_ARRAY, JSONTYPE_BOOL, JSONTYPE_CAST, JSONTYPE_COMPOSITE, JSONTYPE_DATE, JSONTYPE_JSON, JSONTYPE_NUMERIC, JSONTYPE_TIMESTAMP, JSONTYPE_TIMESTAMPTZ, MAXDATELEN, OidFunctionCall1, OidOutputFunctionCall(), pfree(), val, VARDATA_ANY(), and VARSIZE_ANY_EXHDR().

Referenced by add_json(), array_dim_to_json(), composite_to_json(), datum_to_json(), json_agg_transfn_worker(), and json_object_agg_transfn_worker().

escape_json()

static void escape_json	(	StringInfo	buf,
		const char *	str
	)

Definition at line 1604 of file json.c.

{
    appendStringInfoCharMacro(buf, '"');
 
    for (; *str != '\0'; str++)
        escape_json_char(buf, *str);
 
    appendStringInfoCharMacro(buf, '"');
}

References appendStringInfoCharMacro, buf, escape_json_char(), and str.

Referenced by appendJSONKeyValue(), composite_to_json(), datum_to_json_internal(), escape_yaml(), ExplainDummyGroup(), ExplainOpenGroup(), ExplainProperty(), ExplainPropertyList(), ExplainPropertyListNested(), generate_error_response(), populate_scalar(), sn_object_field_start(), sn_scalar(), transform_string_values_object_field_start(), transformJsonTableColumn(), and write_jsonlog().

escape_json_char()

static pg_attribute_always_inline void escape_json_char ( StringInfo  buf, char  c  )

static

Definition at line 1564 of file json.c.

{
    switch (c)
    {
        case '\b':
            appendStringInfoString(buf, "\\b");
            break;
        case '\f':
            appendStringInfoString(buf, "\\f");
            break;
        case '\n':
            appendStringInfoString(buf, "\\n");
            break;
        case '\r':
            appendStringInfoString(buf, "\\r");
            break;
        case '\t':
            appendStringInfoString(buf, "\\t");
            break;
        case '"':
            appendStringInfoString(buf, "\\\"");
            break;
        case '\\':
            appendStringInfoString(buf, "\\\\");
            break;
        default:
            if ((unsigned char) c < ' ')
                appendStringInfo(buf, "\\u%04x", (int) c);
            else
                appendStringInfoCharMacro(buf, c);
            break;
    }
}

References appendStringInfo(), appendStringInfoCharMacro, appendStringInfoString(), and buf.

Referenced by escape_json(), and escape_json_with_len().

escape_json_text()

void escape_json_text	(	StringInfo	buf,
		const text *	txt
	)

Definition at line 1738 of file json.c.

{
    /* must cast away the const, unfortunately */
    text       *tunpacked = pg_detoast_datum_packed(unconstify(text *, txt));
    int         len = VARSIZE_ANY_EXHDR(tunpacked);
    char       *str;
 
    str = VARDATA_ANY(tunpacked);
 
    escape_json_with_len(buf, str, len);
 
    /* pfree any detoasted values */
    if (tunpacked != txt)
        pfree(tunpacked);
}

References buf, escape_json_with_len(), len, pfree(), pg_detoast_datum_packed(), str, unconstify, VARDATA_ANY(), and VARSIZE_ANY_EXHDR().

Referenced by datum_to_json_internal(), json_object(), json_object_two_arg(), and transform_string_values_scalar().

escape_json_with_len()

void escape_json_with_len	(	StringInfo	buf,
		const char *	str,
		int	len
	)

Definition at line 1633 of file json.c.

{
    int         vlen;
 
    Assert(len >= 0);
 
    /*
     * Since we know the minimum length we'll need to append, let's just
     * enlarge the buffer now rather than incrementally making more space when
     * we run out.  Add two extra bytes for the enclosing quotes.
     */
    enlargeStringInfo(buf, len + 2);
 
    /*
     * Figure out how many bytes to process using SIMD.  Round 'len' down to
     * the previous multiple of sizeof(Vector8), assuming that's a power-of-2.
     */
    vlen = len & (int) (~(sizeof(Vector8) - 1));
 
    appendStringInfoCharMacro(buf, '"');
 
    for (int i = 0, copypos = 0;;)
    {
        /*
         * To speed this up, try searching sizeof(Vector8) bytes at once for
         * special characters that we need to escape.  When we find one, we
         * fall out of the Vector8 loop and copy the portion we've vector
         * searched and then we process sizeof(Vector8) bytes one byte at a
         * time.  Once done, come back and try doing vector searching again.
         * We'll also process any remaining bytes at the tail end of the
         * string byte-by-byte.  This optimization assumes that most chunks of
         * sizeof(Vector8) bytes won't contain any special characters.
         */
        for (; i < vlen; i += sizeof(Vector8))
        {
            Vector8     chunk;
 
            vector8_load(&chunk, (const uint8 *) &str[i]);
 
            /*
             * Break on anything less than ' ' or if we find a '"' or '\\'.
             * Those need special handling.  That's done in the per-byte loop.
             */
            if (vector8_has_le(chunk, (unsigned char) 0x1F) ||
                vector8_has(chunk, (unsigned char) '"') ||
                vector8_has(chunk, (unsigned char) '\\'))
                break;
 
#ifdef ESCAPE_JSON_FLUSH_AFTER
 
            /*
             * Flush what's been checked so far out to the destination buffer
             * every so often to avoid having to re-read cachelines when
             * escaping large strings.
             */
            if (i - copypos >= ESCAPE_JSON_FLUSH_AFTER)
            {
                appendBinaryStringInfo(buf, &str[copypos], i - copypos);
                copypos = i;
            }
#endif
        }
 
        /*
         * Write to the destination up to the point that we've vector searched
         * so far.  Do this only when switching into per-byte mode rather than
         * once every sizeof(Vector8) bytes.
         */
        if (copypos < i)
        {
            appendBinaryStringInfo(buf, &str[copypos], i - copypos);
            copypos = i;
        }
 
        /*
         * Per-byte loop for Vector8s containing special chars and for
         * processing the tail of the string.
         */
        for (int b = 0; b < sizeof(Vector8); b++)
        {
            /* check if we've finished */
            if (i == len)
                goto done;
 
            Assert(i < len);
 
            escape_json_char(buf, str[i++]);
        }
 
        copypos = i;
        /* We're not done yet.  Try the vector search again. */
    }
 
done:
    appendStringInfoCharMacro(buf, '"');
}

References appendBinaryStringInfo(), appendStringInfoCharMacro, Assert(), b, buf, enlargeStringInfo(), escape_json_char(), ESCAPE_JSON_FLUSH_AFTER, i, len, str, vector8_has(), vector8_has_le(), and vector8_load().

Referenced by AddFileToBackupManifest(), escape_json_text(), hstore_to_json(), hstore_to_json_loose(), jsonb_put_escaped_value(), populate_scalar(), and printJsonPathItem().

json_agg_finalfn()

Datum json_agg_finalfn

(

PG_FUNCTION_ARGS

)

Definition at line 881 of file json.c.

{
    JsonAggState *state;
 
    /* cannot be called directly because of internal-type argument */
    Assert(AggCheckCallContext(fcinfo, NULL));
 
    state = PG_ARGISNULL(0) ?
        NULL :
        (JsonAggState *) PG_GETARG_POINTER(0);
 
    /* NULL result for no rows in, as is standard with aggregates */
    if (state == NULL)
        PG_RETURN_NULL();
 
    /* Else return state with appropriate array terminator added */
    PG_RETURN_TEXT_P(catenate_stringinfo_string(state->str, "]"));
}

References AggCheckCallContext(), Assert(), catenate_stringinfo_string(), PG_ARGISNULL, PG_GETARG_POINTER, PG_RETURN_NULL, and PG_RETURN_TEXT_P.

json_agg_strict_transfn()

Datum json_agg_strict_transfn

(

PG_FUNCTION_ARGS

)

Definition at line 872 of file json.c.

{
    return json_agg_transfn_worker(fcinfo, true);
}

References json_agg_transfn_worker().

json_agg_transfn()

Datum json_agg_transfn

(

PG_FUNCTION_ARGS

)

Definition at line 863 of file json.c.

{
    return json_agg_transfn_worker(fcinfo, false);
}

References json_agg_transfn_worker().

json_agg_transfn_worker()

static Datum json_agg_transfn_worker ( FunctionCallInfo  fcinfo, bool  absent_on_null  )

static

Definition at line 781 of file json.c.

{
    MemoryContext aggcontext,
                oldcontext;
    JsonAggState *state;
    Datum       val;
 
    if (!AggCheckCallContext(fcinfo, &aggcontext))
    {
        /* cannot be called directly because of internal-type argument */
        elog(ERROR, "json_agg_transfn called in non-aggregate context");
    }
 
    if (PG_ARGISNULL(0))
    {
        Oid         arg_type = get_fn_expr_argtype(fcinfo->flinfo, 1);
 
        if (arg_type == InvalidOid)
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                     errmsg("could not determine input data type")));
 
        /*
         * Make this state object in a context where it will persist for the
         * duration of the aggregate call.  MemoryContextSwitchTo is only
         * needed the first time, as the StringInfo routines make sure they
         * use the right context to enlarge the object if necessary.
         */
        oldcontext = MemoryContextSwitchTo(aggcontext);
        state = (JsonAggState *) palloc(sizeof(JsonAggState));
        state->str = makeStringInfo();
        MemoryContextSwitchTo(oldcontext);
 
        appendStringInfoChar(state->str, '[');
        json_categorize_type(arg_type, false, &state->val_category,
                             &state->val_output_func);
    }
    else
    {
        state = (JsonAggState *) PG_GETARG_POINTER(0);
    }
 
    if (absent_on_null && PG_ARGISNULL(1))
        PG_RETURN_POINTER(state);
 
    if (state->str->len > 1)
        appendStringInfoString(state->str, ", ");
 
    /* fast path for NULLs */
    if (PG_ARGISNULL(1))
    {
        datum_to_json_internal((Datum) 0, true, state->str, JSONTYPE_NULL,
                               InvalidOid, false);
        PG_RETURN_POINTER(state);
    }
 
    val = PG_GETARG_DATUM(1);
 
    /* add some whitespace if structured type and not first item */
    if (!PG_ARGISNULL(0) && state->str->len > 1 &&
        (state->val_category == JSONTYPE_ARRAY ||
         state->val_category == JSONTYPE_COMPOSITE))
    {
        appendStringInfoString(state->str, "\n ");
    }
 
    datum_to_json_internal(val, false, state->str, state->val_category,
                           state->val_output_func, false);
 
    /*
     * The transition type for json_agg() is declared to be "internal", which
     * is a pass-by-value type the same size as a pointer.  So we can safely
     * pass the JsonAggState pointer through nodeAgg.c's machinations.
     */
    PG_RETURN_POINTER(state);
}

References AggCheckCallContext(), appendStringInfoChar(), appendStringInfoString(), datum_to_json_internal(), elog, ereport, errcode(), errmsg(), ERROR, FunctionCallInfoBaseData::flinfo, get_fn_expr_argtype(), InvalidOid, json_categorize_type(), JSONTYPE_ARRAY, JSONTYPE_COMPOSITE, JSONTYPE_NULL, makeStringInfo(), MemoryContextSwitchTo(), palloc(), PG_ARGISNULL, PG_GETARG_DATUM, PG_GETARG_POINTER, PG_RETURN_POINTER, and val.

Referenced by json_agg_strict_transfn(), and json_agg_transfn().

json_build_array()

Datum json_build_array

(

PG_FUNCTION_ARGS

)

Definition at line 1376 of file json.c.

{
    Datum      *args;
    bool       *nulls;
    Oid        *types;
 
    /* build argument values to build the object */
    int         nargs = extract_variadic_args(fcinfo, 0, true,
                                              &args, &types, &nulls);
 
    if (nargs < 0)
        PG_RETURN_NULL();
 
    PG_RETURN_DATUM(json_build_array_worker(nargs, args, nulls, types, false));
}

References generate_unaccent_rules::args, extract_variadic_args(), json_build_array_worker(), PG_RETURN_DATUM, PG_RETURN_NULL, and types.

json_build_array_noargs()

Datum json_build_array_noargs

(

PG_FUNCTION_ARGS

)

Definition at line 1396 of file json.c.

{
    PG_RETURN_TEXT_P(cstring_to_text_with_len("[]", 2));
}

References cstring_to_text_with_len(), and PG_RETURN_TEXT_P.

json_build_array_worker()

Datum json_build_array_worker	(	int	nargs,
		const Datum *	args,
		const bool *	nulls,
		const Oid *	types,
		bool	absent_on_null
	)

Definition at line 1346 of file json.c.

{
    int         i;
    const char *sep = "";
    StringInfoData result;
 
    initStringInfo(&result);
 
    appendStringInfoChar(&result, '[');
 
    for (i = 0; i < nargs; i++)
    {
        if (absent_on_null && nulls[i])
            continue;
 
        appendStringInfoString(&result, sep);
        sep = ", ";
        add_json(args[i], nulls[i], &result, types[i], false);
    }
 
    appendStringInfoChar(&result, ']');
 
    return PointerGetDatum(cstring_to_text_with_len(result.data, result.len));
}

References add_json(), appendStringInfoChar(), appendStringInfoString(), generate_unaccent_rules::args, cstring_to_text_with_len(), StringInfoData::data, i, initStringInfo(), StringInfoData::len, PointerGetDatum(), and types.

Referenced by ExecEvalJsonConstructor(), and json_build_array().

json_build_object()

Datum json_build_object

(

PG_FUNCTION_ARGS

)

Definition at line 1320 of file json.c.

{
    Datum      *args;
    bool       *nulls;
    Oid        *types;
 
    /* build argument values to build the object */
    int         nargs = extract_variadic_args(fcinfo, 0, true,
                                              &args, &types, &nulls);
 
    if (nargs < 0)
        PG_RETURN_NULL();
 
    PG_RETURN_DATUM(json_build_object_worker(nargs, args, nulls, types, false, false));
}

References generate_unaccent_rules::args, extract_variadic_args(), json_build_object_worker(), PG_RETURN_DATUM, PG_RETURN_NULL, and types.

json_build_object_noargs()

Datum json_build_object_noargs

(

PG_FUNCTION_ARGS

)

Definition at line 1340 of file json.c.

{
    PG_RETURN_TEXT_P(cstring_to_text_with_len("{}", 2));
}

References cstring_to_text_with_len(), and PG_RETURN_TEXT_P.

json_build_object_worker()

Datum json_build_object_worker	(	int	nargs,
		const Datum *	args,
		const bool *	nulls,
		const Oid *	types,
		bool	absent_on_null,
		bool	unique_keys
	)

Definition at line 1226 of file json.c.

{
    int         i;
    const char *sep = "";
    StringInfo  result;
    JsonUniqueBuilderState unique_check;
 
    if (nargs % 2 != 0)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                 errmsg("argument list must have even number of elements"),
        /* translator: %s is a SQL function name */
                 errhint("The arguments of %s must consist of alternating keys and values.",
                         "json_build_object()")));
 
    result = makeStringInfo();
 
    appendStringInfoChar(result, '{');
 
    if (unique_keys)
        json_unique_builder_init(&unique_check);
 
    for (i = 0; i < nargs; i += 2)
    {
        StringInfo  out;
        bool        skip;
        int         key_offset;
 
        /* Skip null values if absent_on_null */
        skip = absent_on_null && nulls[i + 1];
 
        if (skip)
        {
            /* If key uniqueness check is needed we must save skipped keys */
            if (!unique_keys)
                continue;
 
            out = json_unique_builder_get_throwawaybuf(&unique_check);
        }
        else
        {
            appendStringInfoString(result, sep);
            sep = ", ";
            out = result;
        }
 
        /* process key */
        if (nulls[i])
            ereport(ERROR,
                    (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
                     errmsg("null value not allowed for object key")));
 
        /* save key offset before appending it */
        key_offset = out->len;
 
        add_json(args[i], false, out, types[i], true);
 
        if (unique_keys)
        {
            /*
             * check key uniqueness after key appending
             *
             * Copy the key first, instead of pointing into the buffer. It
             * will be added to the hash table, but the buffer may get
             * reallocated as we're appending more data to it. That would
             * invalidate pointers to keys in the current buffer.
             */
            const char *key = pstrdup(&out->data[key_offset]);
 
            if (!json_unique_check_key(&unique_check.check, key, 0))
                ereport(ERROR,
                        errcode(ERRCODE_DUPLICATE_JSON_OBJECT_KEY_VALUE),
                        errmsg("duplicate JSON object key value: %s", key));
 
            if (skip)
                continue;
        }
 
        appendStringInfoString(result, " : ");
 
        /* process value */
        add_json(args[i + 1], nulls[i + 1], result, types[i + 1], false);
    }
 
    appendStringInfoChar(result, '}');
 
    return PointerGetDatum(cstring_to_text_with_len(result->data, result->len));
}

References add_json(), appendStringInfoChar(), appendStringInfoString(), generate_unaccent_rules::args, JsonUniqueBuilderState::check, cstring_to_text_with_len(), StringInfoData::data, ereport, errcode(), errhint(), errmsg(), ERROR, i, json_unique_builder_get_throwawaybuf(), json_unique_builder_init(), json_unique_check_key(), sort-test::key, StringInfoData::len, makeStringInfo(), PointerGetDatum(), pstrdup(), skip, and types.

Referenced by ExecEvalJsonConstructor(), and json_build_object().

json_in()

Datum json_in

(

PG_FUNCTION_ARGS

)

Definition at line 108 of file json.c.

{
    char       *json = PG_GETARG_CSTRING(0);
    text       *result = cstring_to_text(json);
    JsonLexContext lex;
 
    /* validate it */
    makeJsonLexContext(&lex, result, false);
    if (!pg_parse_json_or_errsave(&lex, &nullSemAction, fcinfo->context))
        PG_RETURN_NULL();
 
    /* Internal representation is the same as text */
    PG_RETURN_TEXT_P(result);
}

References cstring_to_text(), makeJsonLexContext(), nullSemAction, PG_GETARG_CSTRING, pg_parse_json_or_errsave(), PG_RETURN_NULL, and PG_RETURN_TEXT_P.

json_object()

Datum json_object

(

PG_FUNCTION_ARGS

)

Definition at line 1408 of file json.c.

{
    ArrayType  *in_array = PG_GETARG_ARRAYTYPE_P(0);
    int         ndims = ARR_NDIM(in_array);
    StringInfoData result;
    Datum      *in_datums;
    bool       *in_nulls;
    int         in_count,
                count,
                i;
    text       *rval;
 
    switch (ndims)
    {
        case 0:
            PG_RETURN_DATUM(CStringGetTextDatum("{}"));
            break;
 
        case 1:
            if ((ARR_DIMS(in_array)[0]) % 2)
                ereport(ERROR,
                        (errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
                         errmsg("array must have even number of elements")));
            break;
 
        case 2:
            if ((ARR_DIMS(in_array)[1]) != 2)
                ereport(ERROR,
                        (errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
                         errmsg("array must have two columns")));
            break;
 
        default:
            ereport(ERROR,
                    (errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
                     errmsg("wrong number of array subscripts")));
    }
 
    deconstruct_array_builtin(in_array, TEXTOID, &in_datums, &in_nulls, &in_count);
 
    count = in_count / 2;
 
    initStringInfo(&result);
 
    appendStringInfoChar(&result, '{');
 
    for (i = 0; i < count; ++i)
    {
        if (in_nulls[i * 2])
            ereport(ERROR,
                    (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
                     errmsg("null value not allowed for object key")));
 
        if (i > 0)
            appendStringInfoString(&result, ", ");
        escape_json_text(&result, (text *) DatumGetPointer(in_datums[i * 2]));
        appendStringInfoString(&result, " : ");
        if (in_nulls[i * 2 + 1])
            appendStringInfoString(&result, "null");
        else
        {
            escape_json_text(&result,
                             (text *) DatumGetPointer(in_datums[i * 2 + 1]));
        }
    }
 
    appendStringInfoChar(&result, '}');
 
    pfree(in_datums);
    pfree(in_nulls);
 
    rval = cstring_to_text_with_len(result.data, result.len);
    pfree(result.data);
 
    PG_RETURN_TEXT_P(rval);
}

References appendStringInfoChar(), appendStringInfoString(), ARR_DIMS, ARR_NDIM, cstring_to_text_with_len(), CStringGetTextDatum, StringInfoData::data, DatumGetPointer(), deconstruct_array_builtin(), ereport, errcode(), errmsg(), ERROR, escape_json_text(), i, initStringInfo(), StringInfoData::len, pfree(), PG_GETARG_ARRAYTYPE_P, PG_RETURN_DATUM, and PG_RETURN_TEXT_P.

json_object_agg_finalfn()

Datum json_object_agg_finalfn

(

PG_FUNCTION_ARGS

)

Definition at line 1188 of file json.c.

{
    JsonAggState *state;
 
    /* cannot be called directly because of internal-type argument */
    Assert(AggCheckCallContext(fcinfo, NULL));
 
    state = PG_ARGISNULL(0) ? NULL : (JsonAggState *) PG_GETARG_POINTER(0);
 
    /* NULL result for no rows in, as is standard with aggregates */
    if (state == NULL)
        PG_RETURN_NULL();
 
    /* Else return state with appropriate object terminator added */
    PG_RETURN_TEXT_P(catenate_stringinfo_string(state->str, " }"));
}

References AggCheckCallContext(), Assert(), catenate_stringinfo_string(), PG_ARGISNULL, PG_GETARG_POINTER, PG_RETURN_NULL, and PG_RETURN_TEXT_P.

json_object_agg_strict_transfn()

Datum json_object_agg_strict_transfn

(

PG_FUNCTION_ARGS

)

Definition at line 1161 of file json.c.

{
    return json_object_agg_transfn_worker(fcinfo, true, false);
}

References json_object_agg_transfn_worker().

json_object_agg_transfn()

Datum json_object_agg_transfn

(

PG_FUNCTION_ARGS

)

Definition at line 1152 of file json.c.

{
    return json_object_agg_transfn_worker(fcinfo, false, false);
}

References json_object_agg_transfn_worker().

json_object_agg_transfn_worker()

static Datum json_object_agg_transfn_worker ( FunctionCallInfo  fcinfo, bool  absent_on_null, bool  unique_keys  )

static

Definition at line 1004 of file json.c.

{
    MemoryContext aggcontext,
                oldcontext;
    JsonAggState *state;
    StringInfo  out;
    Datum       arg;
    bool        skip;
    int         key_offset;
 
    if (!AggCheckCallContext(fcinfo, &aggcontext))
    {
        /* cannot be called directly because of internal-type argument */
        elog(ERROR, "json_object_agg_transfn called in non-aggregate context");
    }
 
    if (PG_ARGISNULL(0))
    {
        Oid         arg_type;
 
        /*
         * Make the StringInfo in a context where it will persist for the
         * duration of the aggregate call. Switching context is only needed
         * for this initial step, as the StringInfo and dynahash routines make
         * sure they use the right context to enlarge the object if necessary.
         */
        oldcontext = MemoryContextSwitchTo(aggcontext);
        state = (JsonAggState *) palloc(sizeof(JsonAggState));
        state->str = makeStringInfo();
        if (unique_keys)
            json_unique_builder_init(&state->unique_check);
        else
            memset(&state->unique_check, 0, sizeof(state->unique_check));
        MemoryContextSwitchTo(oldcontext);
 
        arg_type = get_fn_expr_argtype(fcinfo->flinfo, 1);
 
        if (arg_type == InvalidOid)
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                     errmsg("could not determine data type for argument %d", 1)));
 
        json_categorize_type(arg_type, false, &state->key_category,
                             &state->key_output_func);
 
        arg_type = get_fn_expr_argtype(fcinfo->flinfo, 2);
 
        if (arg_type == InvalidOid)
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                     errmsg("could not determine data type for argument %d", 2)));
 
        json_categorize_type(arg_type, false, &state->val_category,
                             &state->val_output_func);
 
        appendStringInfoString(state->str, "{ ");
    }
    else
    {
        state = (JsonAggState *) PG_GETARG_POINTER(0);
    }
 
    /*
     * Note: since json_object_agg() is declared as taking type "any", the
     * parser will not do any type conversion on unknown-type literals (that
     * is, undecorated strings or NULLs).  Such values will arrive here as
     * type UNKNOWN, which fortunately does not matter to us, since
     * unknownout() works fine.
     */
 
    if (PG_ARGISNULL(1))
        ereport(ERROR,
                (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
                 errmsg("null value not allowed for object key")));
 
    /* Skip null values if absent_on_null */
    skip = absent_on_null && PG_ARGISNULL(2);
 
    if (skip)
    {
        /*
         * We got a NULL value and we're not storing those; if we're not
         * testing key uniqueness, we're done.  If we are, use the throwaway
         * buffer to store the key name so that we can check it.
         */
        if (!unique_keys)
            PG_RETURN_POINTER(state);
 
        out = json_unique_builder_get_throwawaybuf(&state->unique_check);
    }
    else
    {
        out = state->str;
 
        /*
         * Append comma delimiter only if we have already output some fields
         * after the initial string "{ ".
         */
        if (out->len > 2)
            appendStringInfoString(out, ", ");
    }
 
    arg = PG_GETARG_DATUM(1);
 
    key_offset = out->len;
 
    datum_to_json_internal(arg, false, out, state->key_category,
                           state->key_output_func, true);
 
    if (unique_keys)
    {
        /*
         * Copy the key first, instead of pointing into the buffer. It will be
         * added to the hash table, but the buffer may get reallocated as
         * we're appending more data to it. That would invalidate pointers to
         * keys in the current buffer.
         */
        const char *key = MemoryContextStrdup(aggcontext,
                                              &out->data[key_offset]);
 
        if (!json_unique_check_key(&state->unique_check.check, key, 0))
            ereport(ERROR,
                    errcode(ERRCODE_DUPLICATE_JSON_OBJECT_KEY_VALUE),
                    errmsg("duplicate JSON object key value: %s", key));
 
        if (skip)
            PG_RETURN_POINTER(state);
    }
 
    appendStringInfoString(state->str, " : ");
 
    if (PG_ARGISNULL(2))
        arg = (Datum) 0;
    else
        arg = PG_GETARG_DATUM(2);
 
    datum_to_json_internal(arg, PG_ARGISNULL(2), state->str,
                           state->val_category,
                           state->val_output_func, false);
 
    PG_RETURN_POINTER(state);
}

References AggCheckCallContext(), appendStringInfoString(), arg, StringInfoData::data, datum_to_json_internal(), elog, ereport, errcode(), errmsg(), ERROR, FunctionCallInfoBaseData::flinfo, get_fn_expr_argtype(), InvalidOid, json_categorize_type(), json_unique_builder_get_throwawaybuf(), json_unique_builder_init(), json_unique_check_key(), sort-test::key, StringInfoData::len, makeStringInfo(), MemoryContextStrdup(), MemoryContextSwitchTo(), palloc(), PG_ARGISNULL, PG_GETARG_DATUM, PG_GETARG_POINTER, PG_RETURN_POINTER, and skip.

Referenced by json_object_agg_strict_transfn(), json_object_agg_transfn(), json_object_agg_unique_strict_transfn(), and json_object_agg_unique_transfn().

json_object_agg_unique_strict_transfn()

Datum json_object_agg_unique_strict_transfn

(

PG_FUNCTION_ARGS

)

Definition at line 1179 of file json.c.

{
    return json_object_agg_transfn_worker(fcinfo, true, true);
}

References json_object_agg_transfn_worker().

json_object_agg_unique_transfn()

Datum json_object_agg_unique_transfn

(

PG_FUNCTION_ARGS

)

Definition at line 1170 of file json.c.

{
    return json_object_agg_transfn_worker(fcinfo, false, true);
}

References json_object_agg_transfn_worker().

json_object_two_arg()

Datum json_object_two_arg

(

PG_FUNCTION_ARGS

)

Definition at line 1492 of file json.c.

{
    ArrayType  *key_array = PG_GETARG_ARRAYTYPE_P(0);
    ArrayType  *val_array = PG_GETARG_ARRAYTYPE_P(1);
    int         nkdims = ARR_NDIM(key_array);
    int         nvdims = ARR_NDIM(val_array);
    StringInfoData result;
    Datum      *key_datums,
               *val_datums;
    bool       *key_nulls,
               *val_nulls;
    int         key_count,
                val_count,
                i;
    text       *rval;
 
    if (nkdims > 1 || nkdims != nvdims)
        ereport(ERROR,
                (errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
                 errmsg("wrong number of array subscripts")));
 
    if (nkdims == 0)
        PG_RETURN_DATUM(CStringGetTextDatum("{}"));
 
    deconstruct_array_builtin(key_array, TEXTOID, &key_datums, &key_nulls, &key_count);
    deconstruct_array_builtin(val_array, TEXTOID, &val_datums, &val_nulls, &val_count);
 
    if (key_count != val_count)
        ereport(ERROR,
                (errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
                 errmsg("mismatched array dimensions")));
 
    initStringInfo(&result);
 
    appendStringInfoChar(&result, '{');
 
    for (i = 0; i < key_count; ++i)
    {
        if (key_nulls[i])
            ereport(ERROR,
                    (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
                     errmsg("null value not allowed for object key")));
 
        if (i > 0)
            appendStringInfoString(&result, ", ");
        escape_json_text(&result, (text *) DatumGetPointer(key_datums[i]));
        appendStringInfoString(&result, " : ");
        if (val_nulls[i])
            appendStringInfoString(&result, "null");
        else
            escape_json_text(&result,
                             (text *) DatumGetPointer(val_datums[i]));
    }
 
    appendStringInfoChar(&result, '}');
 
    pfree(key_datums);
    pfree(key_nulls);
    pfree(val_datums);
    pfree(val_nulls);
 
    rval = cstring_to_text_with_len(result.data, result.len);
    pfree(result.data);
 
    PG_RETURN_TEXT_P(rval);
}

References appendStringInfoChar(), appendStringInfoString(), ARR_NDIM, cstring_to_text_with_len(), CStringGetTextDatum, StringInfoData::data, DatumGetPointer(), deconstruct_array_builtin(), ereport, errcode(), errmsg(), ERROR, escape_json_text(), i, initStringInfo(), StringInfoData::len, pfree(), PG_GETARG_ARRAYTYPE_P, PG_RETURN_DATUM, and PG_RETURN_TEXT_P.

json_out()

Datum json_out

(

PG_FUNCTION_ARGS

)

Definition at line 127 of file json.c.

{
    /* we needn't detoast because text_to_cstring will handle that */
    Datum       txt = PG_GETARG_DATUM(0);
 
    PG_RETURN_CSTRING(TextDatumGetCString(txt));
}

References PG_GETARG_DATUM, PG_RETURN_CSTRING, and TextDatumGetCString.

json_recv()

Datum json_recv

(

PG_FUNCTION_ARGS

)

Definition at line 153 of file json.c.

{
    StringInfo  buf = (StringInfo) PG_GETARG_POINTER(0);
    char       *str;
    int         nbytes;
    JsonLexContext lex;
 
    str = pq_getmsgtext(buf, buf->len - buf->cursor, &nbytes);
 
    /* Validate it. */
    makeJsonLexContextCstringLen(&lex, str, nbytes, GetDatabaseEncoding(),
                                 false);
    pg_parse_json_or_ereport(&lex, &nullSemAction);
 
    PG_RETURN_TEXT_P(cstring_to_text_with_len(str, nbytes));
}

References buf, cstring_to_text_with_len(), GetDatabaseEncoding(), makeJsonLexContextCstringLen(), nullSemAction, PG_GETARG_POINTER, pg_parse_json_or_ereport, PG_RETURN_TEXT_P, pq_getmsgtext(), and str.

json_send()

Datum json_send

(

PG_FUNCTION_ARGS

)

Definition at line 139 of file json.c.

{
    text       *t = PG_GETARG_TEXT_PP(0);
    StringInfoData buf;
 
    pq_begintypsend(&buf);
    pq_sendtext(&buf, VARDATA_ANY(t), VARSIZE_ANY_EXHDR(t));
    PG_RETURN_BYTEA_P(pq_endtypsend(&buf));
}

References buf, PG_GETARG_TEXT_PP, PG_RETURN_BYTEA_P, pq_begintypsend(), pq_endtypsend(), pq_sendtext(), VARDATA_ANY(), and VARSIZE_ANY_EXHDR().

json_typeof()

Datum json_typeof

(

PG_FUNCTION_ARGS

)

Definition at line 1876 of file json.c.

{
    text       *json = PG_GETARG_TEXT_PP(0);
    JsonLexContext lex;
    char       *type;
    JsonParseErrorType result;
 
    /* Lex exactly one token from the input and check its type. */
    makeJsonLexContext(&lex, json, false);
    result = json_lex(&lex);
    if (result != JSON_SUCCESS)
        json_errsave_error(result, &lex, NULL);
 
    switch (lex.token_type)
    {
        case JSON_TOKEN_OBJECT_START:
            type = "object";
            break;
        case JSON_TOKEN_ARRAY_START:
            type = "array";
            break;
        case JSON_TOKEN_STRING:
            type = "string";
            break;
        case JSON_TOKEN_NUMBER:
            type = "number";
            break;
        case JSON_TOKEN_TRUE:
        case JSON_TOKEN_FALSE:
            type = "boolean";
            break;
        case JSON_TOKEN_NULL:
            type = "null";
            break;
        default:
            elog(ERROR, "unexpected json token: %d", lex.token_type);
    }
 
    PG_RETURN_TEXT_P(cstring_to_text(type));
}

References cstring_to_text(), elog, ERROR, json_errsave_error(), json_lex(), JSON_SUCCESS, JSON_TOKEN_ARRAY_START, JSON_TOKEN_FALSE, JSON_TOKEN_NULL, JSON_TOKEN_NUMBER, JSON_TOKEN_OBJECT_START, JSON_TOKEN_STRING, JSON_TOKEN_TRUE, makeJsonLexContext(), PG_GETARG_TEXT_PP, PG_RETURN_TEXT_P, JsonLexContext::token_type, and type.

json_unique_builder_get_throwawaybuf()

static StringInfo json_unique_builder_get_throwawaybuf ( JsonUniqueBuilderState *  cxt )

static

Definition at line 980 of file json.c.

{
    StringInfo  out = &cxt->skipped_keys;
 
    if (!out->data)
    {
        MemoryContext oldcxt = MemoryContextSwitchTo(cxt->mcxt);
 
        initStringInfo(out);
        MemoryContextSwitchTo(oldcxt);
    }
    else
        /* Just reset the string to empty */
        out->len = 0;
 
    return out;
}

References StringInfoData::data, initStringInfo(), StringInfoData::len, JsonUniqueBuilderState::mcxt, MemoryContextSwitchTo(), and JsonUniqueBuilderState::skipped_keys.

Referenced by json_build_object_worker(), and json_object_agg_transfn_worker().

json_unique_builder_init()

static void json_unique_builder_init ( JsonUniqueBuilderState *  cxt )

static

Definition at line 952 of file json.c.

{
    json_unique_check_init(&cxt->check);
    cxt->mcxt = CurrentMemoryContext;
    cxt->skipped_keys.data = NULL;
}

References JsonUniqueBuilderState::check, CurrentMemoryContext, StringInfoData::data, json_unique_check_init(), JsonUniqueBuilderState::mcxt, and JsonUniqueBuilderState::skipped_keys.

Referenced by json_build_object_worker(), and json_object_agg_transfn_worker().

json_unique_check_init()

static void json_unique_check_init ( JsonUniqueCheckState *  cxt )

static

Definition at line 934 of file json.c.

{
    HASHCTL     ctl;
 
    memset(&ctl, 0, sizeof(ctl));
    ctl.keysize = sizeof(JsonUniqueHashEntry);
    ctl.entrysize = sizeof(JsonUniqueHashEntry);
    ctl.hcxt = CurrentMemoryContext;
    ctl.hash = json_unique_hash;
    ctl.match = json_unique_hash_match;
 
    *cxt = hash_create("json object hashtable",
                       32,
                       &ctl,
                       HASH_ELEM | HASH_CONTEXT | HASH_FUNCTION | HASH_COMPARE);
}

References ctl, CurrentMemoryContext, HASH_COMPARE, HASH_CONTEXT, hash_create(), HASH_ELEM, HASH_FUNCTION, json_unique_hash(), and json_unique_hash_match().

Referenced by json_unique_builder_init(), and json_validate().

json_unique_check_key()

static bool json_unique_check_key ( JsonUniqueCheckState *  cxt, const char *  key, int  object_id  )

static

Definition at line 960 of file json.c.

{
    JsonUniqueHashEntry entry;
    bool        found;
 
    entry.key = key;
    entry.key_len = strlen(key);
    entry.object_id = object_id;
 
    (void) hash_search(*cxt, &entry, HASH_ENTER, &found);
 
    return !found;
}

References HASH_ENTER, hash_search(), JsonUniqueHashEntry::key, sort-test::key, JsonUniqueHashEntry::key_len, and JsonUniqueHashEntry::object_id.

Referenced by json_build_object_worker(), json_object_agg_transfn_worker(), and json_unique_object_field_start().

json_unique_hash()

static uint32 json_unique_hash ( const void *  key, Size  keysize  )

static

Definition at line 902 of file json.c.

{
    const JsonUniqueHashEntry *entry = (JsonUniqueHashEntry *) key;
    uint32      hash = hash_bytes_uint32(entry->object_id);
 
    hash ^= hash_bytes((const unsigned char *) entry->key, entry->key_len);
 
    return hash;
}

References hash(), hash_bytes(), hash_bytes_uint32(), JsonUniqueHashEntry::key, sort-test::key, JsonUniqueHashEntry::key_len, and JsonUniqueHashEntry::object_id.

Referenced by json_unique_check_init().

json_unique_hash_match()

static int json_unique_hash_match ( const void *  key1, const void *  key2, Size  keysize  )

static

Definition at line 913 of file json.c.

{
    const JsonUniqueHashEntry *entry1 = (const JsonUniqueHashEntry *) key1;
    const JsonUniqueHashEntry *entry2 = (const JsonUniqueHashEntry *) key2;
 
    if (entry1->object_id != entry2->object_id)
        return entry1->object_id > entry2->object_id ? 1 : -1;
 
    if (entry1->key_len != entry2->key_len)
        return entry1->key_len > entry2->key_len ? 1 : -1;
 
    return strncmp(entry1->key, entry2->key, entry1->key_len);
}

References JsonUniqueHashEntry::key, JsonUniqueHashEntry::key_len, and JsonUniqueHashEntry::object_id.

Referenced by json_unique_check_init().

json_unique_object_end()

static JsonParseErrorType json_unique_object_end ( void *  _state )

static

Definition at line 1774 of file json.c.

{
    JsonUniqueParsingState *state = _state;
    JsonUniqueStackEntry *entry;
 
    if (!state->unique)
        return JSON_SUCCESS;
 
    entry = state->stack;
    state->stack = entry->parent;   /* pop object from stack */
    pfree(entry);
    return JSON_SUCCESS;
}

References JSON_SUCCESS, JsonUniqueStackEntry::parent, and pfree().

Referenced by json_validate().

json_unique_object_field_start()

static JsonParseErrorType json_unique_object_field_start ( void *  _state, char *  field, bool  isnull  )

static

Definition at line 1789 of file json.c.

{
    JsonUniqueParsingState *state = _state;
    JsonUniqueStackEntry *entry;
 
    if (!state->unique)
        return JSON_SUCCESS;
 
    /* find key collision in the current object */
    if (json_unique_check_key(&state->check, field, state->stack->object_id))
        return JSON_SUCCESS;
 
    state->unique = false;
 
    /* pop all objects entries */
    while ((entry = state->stack))
    {
        state->stack = entry->parent;
        pfree(entry);
    }
    return JSON_SUCCESS;
}

References JSON_SUCCESS, json_unique_check_key(), JsonUniqueStackEntry::parent, and pfree().

Referenced by json_validate().

json_unique_object_start()

static JsonParseErrorType json_unique_object_start ( void *  _state )

static

Definition at line 1756 of file json.c.

{
    JsonUniqueParsingState *state = _state;
    JsonUniqueStackEntry *entry;
 
    if (!state->unique)
        return JSON_SUCCESS;
 
    /* push object entry to stack */
    entry = palloc(sizeof(*entry));
    entry->object_id = state->id_counter++;
    entry->parent = state->stack;
    state->stack = entry;
 
    return JSON_SUCCESS;
}

References JSON_SUCCESS, JsonUniqueStackEntry::object_id, palloc(), and JsonUniqueStackEntry::parent.

Referenced by json_validate().

json_validate()

bool json_validate	(	text *	json,
		bool	check_unique_keys,
		bool	throw_error
	)

Definition at line 1814 of file json.c.

{
    JsonLexContext lex;
    JsonSemAction uniqueSemAction = {0};
    JsonUniqueParsingState state;
    JsonParseErrorType result;
 
    makeJsonLexContext(&lex, json, check_unique_keys);
 
    if (check_unique_keys)
    {
        state.lex = &lex;
        state.stack = NULL;
        state.id_counter = 0;
        state.unique = true;
        json_unique_check_init(&state.check);
 
        uniqueSemAction.semstate = &state;
        uniqueSemAction.object_start = json_unique_object_start;
        uniqueSemAction.object_field_start = json_unique_object_field_start;
        uniqueSemAction.object_end = json_unique_object_end;
    }
 
    result = pg_parse_json(&lex, check_unique_keys ? &uniqueSemAction : &nullSemAction);
 
    if (result != JSON_SUCCESS)
    {
        if (throw_error)
            json_errsave_error(result, &lex, NULL);
 
        return false;           /* invalid json */
    }
 
    if (check_unique_keys && !state.unique)
    {
        if (throw_error)
            ereport(ERROR,
                    (errcode(ERRCODE_DUPLICATE_JSON_OBJECT_KEY_VALUE),
                     errmsg("duplicate JSON object key value")));
 
        return false;           /* not unique keys */
    }
 
    if (check_unique_keys)
        freeJsonLexContext(&lex);
 
    return true;                /* ok */
}

References ereport, errcode(), errmsg(), ERROR, freeJsonLexContext(), json_errsave_error(), JSON_SUCCESS, json_unique_check_init(), json_unique_object_end(), json_unique_object_field_start(), json_unique_object_start(), makeJsonLexContext(), nullSemAction, JsonSemAction::object_end, JsonSemAction::object_field_start, JsonSemAction::object_start, pg_parse_json(), and JsonSemAction::semstate.

Referenced by ExecEvalJsonConstructor(), and ExecEvalJsonIsPredicate().

JsonEncodeDateTime()

char * JsonEncodeDateTime	(	char *	buf,
		Datum	value,
		Oid	typid,
		const int *	tzp
	)

Definition at line 311 of file json.c.

{
    if (!buf)
        buf = palloc(MAXDATELEN + 1);
 
    switch (typid)
    {
        case DATEOID:
            {
                DateADT     date;
                struct pg_tm tm;
 
                date = DatumGetDateADT(value);
 
                /* Same as date_out(), but forcing DateStyle */
                if (DATE_NOT_FINITE(date))
                    EncodeSpecialDate(date, buf);
                else
                {
                    j2date(date + POSTGRES_EPOCH_JDATE,
                           &(tm.tm_year), &(tm.tm_mon), &(tm.tm_mday));
                    EncodeDateOnly(&tm, USE_XSD_DATES, buf);
                }
            }
            break;
        case TIMEOID:
            {
                TimeADT     time = DatumGetTimeADT(value);
                struct pg_tm tt,
                           *tm = &tt;
                fsec_t      fsec;
 
                /* Same as time_out(), but forcing DateStyle */
                time2tm(time, tm, &fsec);
                EncodeTimeOnly(tm, fsec, false, 0, USE_XSD_DATES, buf);
            }
            break;
        case TIMETZOID:
            {
                TimeTzADT  *time = DatumGetTimeTzADTP(value);
                struct pg_tm tt,
                           *tm = &tt;
                fsec_t      fsec;
                int         tz;
 
                /* Same as timetz_out(), but forcing DateStyle */
                timetz2tm(time, tm, &fsec, &tz);
                EncodeTimeOnly(tm, fsec, true, tz, USE_XSD_DATES, buf);
            }
            break;
        case TIMESTAMPOID:
            {
                Timestamp   timestamp;
                struct pg_tm tm;
                fsec_t      fsec;
 
                timestamp = DatumGetTimestamp(value);
                /* Same as timestamp_out(), but forcing DateStyle */
                if (TIMESTAMP_NOT_FINITE(timestamp))
                    EncodeSpecialTimestamp(timestamp, buf);
                else if (timestamp2tm(timestamp, NULL, &tm, &fsec, NULL, NULL) == 0)
                    EncodeDateTime(&tm, fsec, false, 0, NULL, USE_XSD_DATES, buf);
                else
                    ereport(ERROR,
                            (errcode(ERRCODE_DATETIME_VALUE_OUT_OF_RANGE),
                             errmsg("timestamp out of range")));
            }
            break;
        case TIMESTAMPTZOID:
            {
                TimestampTz timestamp;
                struct pg_tm tm;
                int         tz;
                fsec_t      fsec;
                const char *tzn = NULL;
 
                timestamp = DatumGetTimestampTz(value);
 
                /*
                 * If a time zone is specified, we apply the time-zone shift,
                 * convert timestamptz to pg_tm as if it were without a time
                 * zone, and then use the specified time zone for converting
                 * the timestamp into a string.
                 */
                if (tzp)
                {
                    tz = *tzp;
                    timestamp -= (TimestampTz) tz * USECS_PER_SEC;
                }
 
                /* Same as timestamptz_out(), but forcing DateStyle */
                if (TIMESTAMP_NOT_FINITE(timestamp))
                    EncodeSpecialTimestamp(timestamp, buf);
                else if (timestamp2tm(timestamp, tzp ? NULL : &tz, &tm, &fsec,
                                      tzp ? NULL : &tzn, NULL) == 0)
                {
                    if (tzp)
                        tm.tm_isdst = 1;    /* set time-zone presence flag */
 
                    EncodeDateTime(&tm, fsec, true, tz, tzn, USE_XSD_DATES, buf);
                }
                else
                    ereport(ERROR,
                            (errcode(ERRCODE_DATETIME_VALUE_OUT_OF_RANGE),
                             errmsg("timestamp out of range")));
            }
            break;
        default:
            elog(ERROR, "unknown jsonb value datetime type oid %u", typid);
            return NULL;
    }
 
    return buf;
}

References buf, DATE_NOT_FINITE, DatumGetDateADT(), DatumGetTimeADT(), DatumGetTimestamp(), DatumGetTimestampTz(), DatumGetTimeTzADTP(), elog, EncodeDateOnly(), EncodeDateTime(), EncodeSpecialDate(), EncodeSpecialTimestamp(), EncodeTimeOnly(), ereport, errcode(), errmsg(), ERROR, j2date(), MAXDATELEN, palloc(), POSTGRES_EPOCH_JDATE, time2tm(), timestamp2tm(), TIMESTAMP_NOT_FINITE, timetz2tm(), tm, pg_tm::tm_isdst, pg_tm::tm_mday, pg_tm::tm_mon, pg_tm::tm_year, USE_XSD_DATES, USECS_PER_SEC, and value.

Referenced by convertJsonbScalar(), datum_to_json_internal(), datum_to_jsonb_internal(), and executeItemOptUnwrapTarget().

row_to_json()

Datum row_to_json

(

PG_FUNCTION_ARGS

)

Definition at line 664 of file json.c.

{
    Datum       array = PG_GETARG_DATUM(0);
    StringInfoData result;
 
    initStringInfo(&result);
 
    composite_to_json(array, &result, false);
 
    PG_RETURN_TEXT_P(cstring_to_text_with_len(result.data, result.len));
}

References composite_to_json(), cstring_to_text_with_len(), StringInfoData::data, initStringInfo(), StringInfoData::len, PG_GETARG_DATUM, and PG_RETURN_TEXT_P.

row_to_json_pretty()

Datum row_to_json_pretty

(

PG_FUNCTION_ARGS

)

Definition at line 680 of file json.c.

{
    Datum       array = PG_GETARG_DATUM(0);
    bool        use_line_feeds = PG_GETARG_BOOL(1);
    StringInfoData result;
 
    initStringInfo(&result);
 
    composite_to_json(array, &result, use_line_feeds);
 
    PG_RETURN_TEXT_P(cstring_to_text_with_len(result.data, result.len));
}

References composite_to_json(), cstring_to_text_with_len(), StringInfoData::data, initStringInfo(), StringInfoData::len, PG_GETARG_BOOL, PG_GETARG_DATUM, and PG_RETURN_TEXT_P.

to_json()

Datum to_json

(

PG_FUNCTION_ARGS

)

Definition at line 740 of file json.c.

{
    Datum       val = PG_GETARG_DATUM(0);
    Oid         val_type = get_fn_expr_argtype(fcinfo->flinfo, 0);
    JsonTypeCategory tcategory;
    Oid         outfuncoid;
 
    if (val_type == InvalidOid)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                 errmsg("could not determine input data type")));
 
    json_categorize_type(val_type, false,
                         &tcategory, &outfuncoid);
 
    PG_RETURN_DATUM(datum_to_json(val, tcategory, outfuncoid));
}

References datum_to_json(), ereport, errcode(), errmsg(), ERROR, get_fn_expr_argtype(), InvalidOid, json_categorize_type(), PG_GETARG_DATUM, PG_RETURN_DATUM, and val.

to_json_is_immutable()

bool to_json_is_immutable

(

Oid

typoid

)

Definition at line 701 of file json.c.

{
    JsonTypeCategory tcategory;
    Oid         outfuncoid;
 
    json_categorize_type(typoid, false, &tcategory, &outfuncoid);
 
    switch (tcategory)
    {
        case JSONTYPE_BOOL:
        case JSONTYPE_JSON:
        case JSONTYPE_JSONB:
        case JSONTYPE_NULL:
            return true;
 
        case JSONTYPE_DATE:
        case JSONTYPE_TIMESTAMP:
        case JSONTYPE_TIMESTAMPTZ:
            return false;
 
        case JSONTYPE_ARRAY:
            return false;       /* TODO recurse into elements */
 
        case JSONTYPE_COMPOSITE:
            return false;       /* TODO recurse into fields */
 
        case JSONTYPE_NUMERIC:
        case JSONTYPE_CAST:
        case JSONTYPE_OTHER:
            return func_volatile(outfuncoid) == PROVOLATILE_IMMUTABLE;
    }
 
    return false;               /* not reached */
}

References func_volatile(), json_categorize_type(), JSONTYPE_ARRAY, JSONTYPE_BOOL, JSONTYPE_CAST, JSONTYPE_COMPOSITE, JSONTYPE_DATE, JSONTYPE_JSON, JSONTYPE_JSONB, JSONTYPE_NULL, JSONTYPE_NUMERIC, JSONTYPE_OTHER, JSONTYPE_TIMESTAMP, and JSONTYPE_TIMESTAMPTZ.

Referenced by contain_mutable_functions_walker().