20645520862

Committed 01 Jan 2026 09:04PM UTC coverage: 92.448% (-0.3%) from 92.749%

Build # 20645520862

Build Type

push

github

Committed by

estebanzimanyi

Commit Message

Next

Run Details

36 of 196 new or added lines in 18 files covered. (18.37%)

3 existing lines in 2 files now uncovered.

32280 of 34917 relevant lines covered (92.45%)

1150391.32 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

10.93

/mobilitydb/src/json/tjsonb_jsonfuncs.c

/*****************************************************************************
 *
 * This MobilityDB code is provided under The PostgreSQL License.
 * Copyright (c) 2016-2025, Université libre de Bruxelles and MobilityDB
 * contributors
 *
 * MobilityDB includes portions of PostGIS version 3 source code released
 * under the GNU General Public License (GPLv2 or later).
 * Copyright (c) 2001-2025, PostGIS contributors
 *
 * Permission to use, copy, modify, and distribute this software and its
 * documentation for any purpose, without fee, and without a written
 * agreement is hereby granted, provided that the above copyright notice and
 * this paragraph and the following two paragraphs appear in all copies.
 *
 * IN NO EVENT SHALL UNIVERSITE LIBRE DE BRUXELLES BE LIABLE TO ANY PARTY FOR
 * DIRECT, INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES, INCLUDING
 * LOST PROFITS, ARISING OUT OF THE USE OF THIS SOFTWARE AND ITS DOCUMENTATION,
 * EVEN IF UNIVERSITE LIBRE DE BRUXELLES HAS BEEN ADVISED OF THE POSSIBILITY
 * OF SUCH DAMAGE.
 *
 * UNIVERSITE LIBRE DE BRUXELLES SPECIFICALLY DISCLAIMS ANY WARRANTIES,
 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
 * AND FITNESS FOR A PARTICULAR PURPOSE. THE SOFTWARE PROVIDED HEREUNDER IS ON
 * AN "AS IS" BASIS, AND UNIVERSITE LIBRE DE BRUXELLES HAS NO OBLIGATIONS TO
 * PROVIDE MAINTENANCE, SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS.
 *
 *****************************************************************************/

/**
 * @file
 * @brief JSON functions for temporal JSONB
 */

/* C */
#include <stdbool.h>
/* PostgreSQL */
#include <postgres.h>
#include "utils/array.h"
#include "utils/jsonb.h"
#include "utils/jsonpath.h"
/* MEOS */
#include <meos.h>
#include <meos_json.h>
#include "temporal/span.h"
#include "json/tjsonb.h"
#include <pgtypes.h>
/* MobilityDB */
#include "pg_temporal/temporal.h"

/*****************************************************************************
 * Temporal JSONB casting to/from temporal text
 *****************************************************************************/

PGDLLEXPORT Datum Jsonb_as_text(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Jsonb_as_text);
/**
 * @ingroup mobilitydb_json_json
 * @brief Transform a JSONB value into a text value
 * @sqlfn text()
 * @sqlop @p ::
 */
Datum
Jsonb_as_text(PG_FUNCTION_ARGS)
{
  Jsonb *jb = PG_GETARG_JSONB_P(0);
  text *result = pg_jsonb_to_text(jb);
  PG_FREE_IF_COPY(jb, 0);
  PG_RETURN_TEXT_P(result);
}

PGDLLEXPORT Datum Tjsonb_as_ttext(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_as_ttext);
/**
 * @ingroup mobilitydb_json_json
 * @brief Transform a temporal JSONB value into a temporal text value
 * @sqlfn ttext()
 * @sqlop @p ::
 */
Datum
Tjsonb_as_ttext(PG_FUNCTION_ARGS)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  Temporal *result = tjsonb_to_ttext(temp);
  PG_FREE_IF_COPY(temp, 0);
  PG_RETURN_TEMPORAL_P(result);
}

PGDLLEXPORT Datum Ttext_as_tjsonb(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Ttext_as_tjsonb);
/**
 * @ingroup mobilitydb_json_json
 * @brief Transform a temporal text value into a temporal JSONB value
 * @sqlfn ttext()
 * @sqlop @p ::
 */
Datum
Ttext_as_tjsonb(PG_FUNCTION_ARGS)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  Temporal *result = ttext_to_tjsonb(temp);
  PG_FREE_IF_COPY(temp, 0);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

/*****************************************************************************
 * JSON functions
 *****************************************************************************/

PGDLLEXPORT Datum Tjson_array_length(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjson_array_length);
/**
 * @ingroup mobilitydb_json_json
 * @brief Return the array length of a temporal JSON value
 * @sqlfn tjson_array_length()
 */
Datum
Tjson_array_length(PG_FUNCTION_ARGS)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  Temporal *result = tjson_array_length(temp);
  PG_FREE_IF_COPY(temp, 0);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

PGDLLEXPORT Datum Tjsonb_array_length(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_array_length);
/**
 * @ingroup mobilitydb_json_json
 * @brief Return the array length of a temporal JSONB value
 * @sqlfn tjsonb_array_length()
 */
Datum
Tjsonb_array_length(PG_FUNCTION_ARGS)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  Temporal *result = tjsonb_array_length(temp);
  PG_FREE_IF_COPY(temp, 0);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

/*****************************************************************************/

PGDLLEXPORT Datum Tjson_object_field(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjson_object_field);
/**
 * @ingroup mobilitydb_json_json
 * @brief Extract a field from a temporal JSON value
 * @sqlfn tjson_object_field()
 * @sqlop @p ->
 */
Datum
Tjson_object_field(PG_FUNCTION_ARGS)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  text *key = PG_GETARG_TEXT_P(1);
  Temporal *result = tjson_object_field(temp, key, false);
  PG_FREE_IF_COPY(temp, 0);
  PG_FREE_IF_COPY(key, 1);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

PGDLLEXPORT Datum Tjsonb_object_field(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_object_field);
/**
 * @ingroup mobilitydb_json_json
 * @brief Extract a field from a temporal JSONB value
 * @sqlfn tjsonb_object_field()
 * @sqlop @p ->
 */
Datum
Tjsonb_object_field(PG_FUNCTION_ARGS)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  text *key = PG_GETARG_TEXT_P(1);
  Temporal *result = tjsonb_object_field(temp, key, false);
  PG_FREE_IF_COPY(temp, 0);
  PG_FREE_IF_COPY(key, 1);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

PGDLLEXPORT Datum Tjson_object_field_text(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjson_object_field_text);
/**
 * @ingroup mobilitydb_json_json
 * @brief Extract a field from a temporal JSON value as text
 * @sqlfn tjson_object_field_text()
 * @sqlop @p ->>
 */
Datum
Tjson_object_field_text(PG_FUNCTION_ARGS)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  text *key = PG_GETARG_TEXT_P(1);
  Temporal *result = tjson_object_field(temp, key, true);
  PG_FREE_IF_COPY(temp, 0);
  PG_FREE_IF_COPY(key, 1);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

PGDLLEXPORT Datum Tjsonb_object_field_text(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_object_field_text);
/**
 * @ingroup mobilitydb_json_json
 * @brief Extract a field from a temporal JSONB value as text
 * @sqlfn tjsonb_object_field_text()
 * @sqlop @p ->>
 */
Datum
Tjsonb_object_field_text(PG_FUNCTION_ARGS)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  text *key = PG_GETARG_TEXT_P(1);
  Temporal *result = tjsonb_object_field(temp, key, true);
  PG_FREE_IF_COPY(temp, 0);
  PG_FREE_IF_COPY(key, 1);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

/*****************************************************************************/

PGDLLEXPORT Datum Concat_jsonb_tjsonb(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Concat_jsonb_tjsonb);
/**
 * @ingroup mobilitydb_json_json
 * @brief Concat a JSONB value with a temporal JSONB
 * @sqlfn tjsonb_concat()
 * @sqlop @p ||
 */
Datum
Concat_jsonb_tjsonb(PG_FUNCTION_ARGS)
{
  Jsonb *jb = PG_GETARG_JSONB_P(0);
  Temporal *temp = PG_GETARG_TEMPORAL_P(1);
  Temporal *result = concat_tjsonb_jsonb(temp, jb, INVERT);
  PG_FREE_IF_COPY(jb, 0);
  PG_FREE_IF_COPY(temp, 1);
  PG_RETURN_TEMPORAL_P(result);
}

PGDLLEXPORT Datum Concat_tjsonb_jsonb(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Concat_tjsonb_jsonb);
/**
 * @ingroup mobilitydb_json_json
 * @brief Concat a temporal JSONB with a JSONB value
 * @sqlfn jsonb_concat()
 * @sqlop @p ||
 */
Datum
Concat_tjsonb_jsonb(PG_FUNCTION_ARGS)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  Jsonb *jb = PG_GETARG_JSONB_P(1);
  Temporal *result = concat_tjsonb_jsonb(temp, jb, INVERT_NO);
  PG_FREE_IF_COPY(temp, 0);
  PG_FREE_IF_COPY(jb, 1);
  PG_RETURN_TEMPORAL_P(result);
}

PGDLLEXPORT Datum Concat_tjsonb_tjsonb(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Concat_tjsonb_tjsonb);
/**
 * @ingroup mobilitydb_json_json
 * @brief Concat two temporal JSONB values
 * @sqlfn jsonb_concat()
 * @sqlop @p ||
 */
Datum
Concat_tjsonb_tjsonb(PG_FUNCTION_ARGS)
{
  Temporal *temp1 = PG_GETARG_TEMPORAL_P(0);
  Temporal *temp2 = PG_GETARG_TEMPORAL_P(1);
  Temporal *result = concat_tjsonb_tjsonb(temp1, temp2);
  PG_FREE_IF_COPY(temp1, 0);
  PG_FREE_IF_COPY(temp2, 1);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

/*****************************************************************************/

PGDLLEXPORT Datum Tjsonb_delete_key(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_delete_key);
/**
 * @ingroup mobilitydb_json_json
 * @brief Delete a key from a temporal JSONB value
 * @sqlfn jsonb_delete()
 * @sqlop @p -
 */
Datum
Tjsonb_delete_key(PG_FUNCTION_ARGS)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  text *key = PG_GETARG_TEXT_P(1);
  Temporal *result = tjsonb_delete_key(temp, key);
  PG_FREE_IF_COPY(temp, 0);
  PG_FREE_IF_COPY(key, 1);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

PGDLLEXPORT Datum Tjsonb_delete_key_array(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_delete_key_array);
/**
 * @ingroup mobilitydb_json_json
 * @brief Delete an array of keys from a temporal JSONB value
 * @sqlfn jsonb_delete_array()
 * @sqlop @p -
 */
Datum
Tjsonb_delete_key_array(PG_FUNCTION_ARGS)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  ArrayType *keys = PG_GETARG_ARRAYTYPE_P(1);
  if (ARR_NDIM(keys) > 1)
    ereport(ERROR, (errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
       errmsg("wrong number of array subscripts")));

  /* Extract the keys from the array */
  int keys_len;
  Datum *keys_elems;
  bool *keys_nulls;
  deconstruct_array(keys, TEXTOID, -1, false, 'd', &keys_elems, &keys_nulls,
    &keys_len);
  if (keys_len == 0)
    PG_RETURN_TEMPORAL_P(temp);

  /* Compute the result */
  Temporal *result = tjsonb_delete_key_array(temp, (text **) keys_elems,
    keys_len);
  pfree(keys_elems); pfree(keys_nulls);
  PG_FREE_IF_COPY(temp, 0);
  PG_FREE_IF_COPY(keys, 1);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

PGDLLEXPORT Datum Tjsonb_delete_idx(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_delete_idx);
/**
 * @ingroup mobilitydb_json_json
 * @brief Delete a key specified by an index from a temporal JSONB value
 * @sqlfn jsonb_delete()
 * @sqlop @p -
 */
Datum
Tjsonb_delete_idx(PG_FUNCTION_ARGS)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  int idx = PG_GETARG_INT32(1);
  Temporal *result = tjsonb_delete_idx(temp, idx);
  PG_FREE_IF_COPY(temp, 0);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

/*****************************************************************************/

PGDLLEXPORT Datum Tjson_array_element(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjson_array_element);
/**
 * @ingroup mobilitydb_json_json
 * @brief Extract an array element from a temporal JSON value
 * @sqlfn tjson_array_element()
 */
Datum
Tjson_array_element(PG_FUNCTION_ARGS)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  int element = PG_GETARG_INT32(1);
  Temporal *result = tjson_array_element(temp, element, false);
  PG_FREE_IF_COPY(temp, 0);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

PGDLLEXPORT Datum Tjsonb_array_element(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_array_element);
/**
 * @ingroup mobilitydb_json_json
 * @brief Extract an array element from a temporal JSONB value
 * @sqlfn tjsonb_array_element()
 */
Datum
Tjsonb_array_element(PG_FUNCTION_ARGS)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  int element = PG_GETARG_INT32(1);
  Temporal *result = tjsonb_array_element(temp, element, false);
  PG_FREE_IF_COPY(temp, 0);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

PGDLLEXPORT Datum Tjson_array_element_text(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjson_array_element_text);
/**
 * @ingroup mobilitydb_json_json
 * @brief Extract an array element from a temporal JSON value as text
 * @sqlfn tjson_array_element_text()
 */
Datum
Tjson_array_element_text(PG_FUNCTION_ARGS)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  int element = PG_GETARG_INT32(1);
  Temporal *result = tjson_array_element(temp, element, true);
  PG_FREE_IF_COPY(temp, 0);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

PGDLLEXPORT Datum Tjsonb_array_element_text(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_array_element_text);
/**
 * @ingroup mobilitydb_json_json
 * @brief Extract an array element from a temporal JSONB value as text
 * @sqlfn tjsonb_array_element_text()
 */
Datum
Tjsonb_array_element_text(PG_FUNCTION_ARGS)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  int element = PG_GETARG_INT32(1);
  Temporal *result = tjsonb_array_element(temp, element, true);
  PG_FREE_IF_COPY(temp, 0);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

/*****************************************************************************/

/**
 * @brief Extract a path from a temporal JSONB value
 */
Datum
Tjson_extract_path_common(FunctionCallInfo fcinfo, bool isjsonb, bool astext)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  ArrayType *path = PG_GETARG_ARRAYTYPE_P(1);
  if (ARR_NDIM(path) > 1)
    ereport(ERROR, (errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
       errmsg("wrong number of array subscripts")));

  /* Extract the path from the array */
  int path_len;
  Datum *path_elems;
  bool *path_nulls;
  deconstruct_array(path, TEXTOID, -1, false, 'i', &path_elems, &path_nulls,
    &path_len);
  if (path_len == 0)
    PG_RETURN_TEMPORAL_P(temp);
  /*
   * If the array contains any null elements, return NULL, on the grounds
   * that you'd have gotten NULL if any RHS value were NULL in a nested
   * series of applications of the -> operator.  (Note: because we also
   * return NULL for error cases such as no-such-field, this is true
   * regardless of the contents of the rest of the array.)
   */
  if (array_contains_nulls(path))
    PG_RETURN_NULL();
  
  /* Compute the result */
  Temporal *result = isjsonb ?
    tjsonb_extract_path(temp, (text **) path_elems, path_len, astext) :
    tjson_extract_path(temp, (text **) path_elems, path_len, astext);

  pfree(path_elems); pfree(path_nulls);
  PG_FREE_IF_COPY(temp, 0);
  PG_FREE_IF_COPY(path, 1);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

PGDLLEXPORT Datum Tjson_extract_path(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjson_extract_path);
/**
 * @ingroup mobilitydb_json_json
 * @brief Extract an item specified by a path from a temporal JSON value
 * @sqlfn tjson_extract_path()
 */
Datum
Tjson_extract_path(PG_FUNCTION_ARGS)
{
  return Tjson_extract_path_common(fcinfo, false, false);
}

PGDLLEXPORT Datum Tjson_extract_path_text(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjson_extract_path_text);
/**
 * @ingroup mobilitydb_json_json
 * @brief Extract an item specified by a path from a temporal JSON value as text
 * @sqlfn tjson_extract_path_text()
 */
Datum
Tjson_extract_path_text(PG_FUNCTION_ARGS)
{
  return Tjson_extract_path_common(fcinfo, false, true);
}

PGDLLEXPORT Datum Tjsonb_extract_path(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_extract_path);
/**
 * @ingroup mobilitydb_json_json
 * @brief Extract a path from a temporal JSONB value
 * @sqlfn tjsonb_extract_path()
 */
Datum
Tjsonb_extract_path(PG_FUNCTION_ARGS)
{
  return Tjson_extract_path_common(fcinfo, true, false);
}

PGDLLEXPORT Datum Tjsonb_extract_path_text(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_extract_path_text);
/**
 * @ingroup mobilitydb_json_json
 * @brief Extract a path from a temporal JSONB value as text
 * @sqlfn tjsonb_extract_path_text()
 */
Datum
Tjsonb_extract_path_text(PG_FUNCTION_ARGS)
{
  return Tjson_extract_path_common(fcinfo, true, true);
}

/*****************************************************************************/

/**
 * @brief Replace a value specified by a path with a new value in a temporal
 * JSONB value 
 * @sqlfn tjsonb_set(), tjsonb_set_lax()
 */
Datum
Tjsonb_set_common(FunctionCallInfo fcinfo, bool lax)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  ArrayType *path = PG_GETARG_ARRAYTYPE_P(1);
  Jsonb *newjsonb = PG_GETARG_JSONB_P(2);
  bool create = PG_GETARG_BOOL(3);
  text *handle_null = NULL;
  if (lax)
    handle_null = PG_GETARG_TEXT_P(4);
 
 if (ARR_NDIM(path) > 1)
    ereport(ERROR, (errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
      errmsg("wrong number of array subscripts")));

  Datum *path_elems;
  bool *path_nulls;
  int path_len;
  deconstruct_array(path, TEXTOID, -1, false, 'd', &path_elems, &path_nulls,
    &path_len);
  if (path_len == 0)
    PG_RETURN_TEMPORAL_P(temp);

  /* Compute the result */
  Temporal *result = tjsonb_set(temp, (text **) path_elems, path_len,
    newjsonb, create, handle_null, lax);
  pfree(path_elems); pfree(path_nulls);
  PG_FREE_IF_COPY(temp, 0);
  PG_FREE_IF_COPY(path, 1);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

PGDLLEXPORT Datum Tjsonb_set(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_set);
/**
 * @ingroup mobilitydb_json_json
 * @brief Replace a value specified by a path with a new value in a temporal
 * JSONB value
 * @sqlfn tjsonb_set()
 */
Datum
Tjsonb_set(PG_FUNCTION_ARGS)
{
  return Tjsonb_set_common(fcinfo, false);
}

PGDLLEXPORT Datum Tjsonb_set_lax(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_set_lax);
/**
 * @ingroup mobilitydb_json_json
 * @brief Replace a value specified by a path with a new value in a temporal
 * JSONB value using the lax mode
 * @sqlfn tjsonb_set_lax()
 */
Datum
Tjsonb_set_lax(PG_FUNCTION_ARGS)
{
  return Tjsonb_set_common(fcinfo, true);
}

/*****************************************************************************/

PGDLLEXPORT Datum Tjsonb_delete_path(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_delete_path);
/**
 * @ingroup mobilitydb_json_json
 * @brief Delete a path from a temporal JSONB value
 * @sqlfn Tjsonb_delete_path()
 */
Datum
Tjsonb_delete_path(PG_FUNCTION_ARGS)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  ArrayType *path = PG_GETARG_ARRAYTYPE_P(1);
  if (ARR_NDIM(path) > 1)
    ereport(ERROR, (errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
       errmsg("wrong number of array subscripts")));

  /* Extract the path from the array */
  int path_len;
  Datum *path_elems;
  bool *path_nulls;
  deconstruct_array(path, TEXTOID, -1, false, 'd', &path_elems, &path_nulls,
    &path_len);
  if (path_len == 0)
    PG_RETURN_TEMPORAL_P(temp);

  /* Compute the result */
  Temporal *result = tjsonb_delete_path(temp, (text **) path_elems, path_len);

  pfree(path_elems); pfree(path_nulls);
  PG_FREE_IF_COPY(temp, 0);
  PG_FREE_IF_COPY(path, 1);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

/*****************************************************************************/

PGDLLEXPORT Datum Tjsonb_insert(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_insert);
/**
 * @ingroup mobilitydb_json_json
 * @brief Insert a path into a temporal JSONB value
 * @sqlfn tjsonb_insert()
 */
Datum
Tjsonb_insert(PG_FUNCTION_ARGS)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  ArrayType *path = PG_GETARG_ARRAYTYPE_P(1);
  Jsonb *newjsonb = PG_GETARG_JSONB_P(2);
  bool after = PG_GETARG_BOOL(3);
  if (ARR_NDIM(path) > 1)
    ereport(ERROR, (errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
      errmsg("wrong number of array subscripts")));

  /* Extract the path from the array */
  int path_len;
  Datum *path_elems;
  bool *path_nulls;
  deconstruct_array(path, TEXTOID, -1, false, 'd', &path_elems, &path_nulls,
    &path_len);
  if (path_len == 0)
    PG_RETURN_TEMPORAL_P(temp);

  /* Compute the result */
  Temporal *result = tjsonb_insert(temp, (text **) path_elems, path_len,
    newjsonb, after);
  pfree(path_elems); pfree(path_nulls);
  PG_FREE_IF_COPY(temp, 0);
  PG_FREE_IF_COPY(path, 1);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

/*****************************************************************************/

/**
 * @brief Convert a temporal JSONB into a temporal alphanumeric type
 * @sqlfn tjsonb_to_tint()
 */
Datum
Tjsonb_to_talphanum(FunctionCallInfo fcinfo, meosType temptype)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  text *key_text = PG_GETARG_TEXT_PP(1);
  char *key = pg_text_to_cstring(key_text);
  interpType interp = INTERP_NONE;
  if (PG_NARGS() > 2 && ! PG_ARGISNULL(2))
    interp = input_interp_string(fcinfo, 2);
  Temporal *result = tjsonb_to_talphanum(temp, key, temptype, interp);
  PG_FREE_IF_COPY(temp, 0);
  PG_FREE_IF_COPY(key_text, 1);
  PG_RETURN_POINTER(result);
}

PGDLLEXPORT Datum Tjsonb_to_tbool(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_to_tbool);
/**
 * @ingroup mobilitydb_json_json
 * @brief Convert a temporal JSONB into a temporal boolean
 * @sqlfn tbool()
 */
Datum
Tjsonb_to_tbool(PG_FUNCTION_ARGS)
{
  return Tjsonb_to_talphanum(fcinfo, T_TBOOL);
}

PGDLLEXPORT Datum Tjsonb_to_tint(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_to_tint);
/**
 * @ingroup mobilitydb_json_json
 * @brief Convert a temporal JSONB into a temporal integer
 * @sqlfn tint()
 */
Datum
Tjsonb_to_tint(PG_FUNCTION_ARGS)
{
  return Tjsonb_to_talphanum(fcinfo, T_TINT);
}

PGDLLEXPORT Datum Tjsonb_to_tfloat(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_to_tfloat);
/**
 * @ingroup mobilitydb_json_json
 * @brief Convert a temporal JSONB into a temporal float
 * @sqlfn tfloat()
 */
Datum
Tjsonb_to_tfloat(PG_FUNCTION_ARGS)
{
  return Tjsonb_to_talphanum(fcinfo, T_TFLOAT);
}

PGDLLEXPORT Datum Tjsonb_to_ttext_key(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_to_ttext_key);
/**
 * @ingroup mobilitydb_json_json
 * @brief Convert a temporal JSONB into a temporal text
 * @sqlfn ttext()
 */
Datum
Tjsonb_to_ttext_key(PG_FUNCTION_ARGS)
{
  return Tjsonb_to_talphanum(fcinfo, T_TTEXT);
}

/*****************************************************************************/

PGDLLEXPORT Datum Tjson_strip_nulls(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjson_strip_nulls);
/**
 * @ingroup mobilitydb_json_json
 * @brief Return a temporal JSON value without nulls
 * @sqlfn tjson_strip_nulls()
 */
Datum
Tjson_strip_nulls(PG_FUNCTION_ARGS)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  bool strip_nulls = PG_GETARG_BOOL(1);
  Temporal *result = tjson_strip_nulls(temp, strip_nulls);
  PG_FREE_IF_COPY(temp, 0);
  PG_RETURN_POINTER(result);
}

PGDLLEXPORT Datum Tjsonb_strip_nulls(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_strip_nulls);
/**
 * @ingroup mobilitydb_json_json
 * @brief Return a temporal JSONB value without nulls
 * @sqlfn tjsonb_strip_nulls()
 */
Datum
Tjsonb_strip_nulls(PG_FUNCTION_ARGS)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  bool strip_nulls = PG_GETARG_BOOL(1);
  Temporal *result = tjsonb_strip_nulls(temp, strip_nulls);
  PG_FREE_IF_COPY(temp, 0);
  PG_RETURN_POINTER(result);
}

/*****************************************************************************/

PGDLLEXPORT Datum Tjsonb_pretty(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_pretty);
/**
 * @ingroup mobilitydb_json_json
 * @brief Return a temporal JSONB value without nulls
 * @sqlfn tjsonb_pretty()
 */
Datum
Tjsonb_pretty(PG_FUNCTION_ARGS)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  Temporal *result = tjsonb_pretty(temp);
  PG_FREE_IF_COPY(temp, 0);
  PG_RETURN_POINTER(result);
}

/*****************************************************************************
 * JSON path functions
 *****************************************************************************/

/**
 * @brief Return true if a JSON path returns at least one item for a temporal
 * JSONB value
 * @sqlfn tjsonb_path_exists(), tjsonb_path_exists_tz()
 */
Datum
Tjsonb_path_exists_common(FunctionCallInfo fcinfo, bool tz)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  JsonPath *jp = PG_GETARG_JSONPATH_P(1);
  Jsonb *vars = PG_GETARG_JSONB_P(2);
  bool silent = PG_GETARG_BOOL(3);
  Temporal *result = tjsonb_path_exists(temp, jp, vars, silent, tz);
  PG_FREE_IF_COPY(temp, 0);
  PG_FREE_IF_COPY(jp, 1);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

PGDLLEXPORT Datum Tjsonb_path_exists(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_path_exists);
/**
 * @ingroup mobilitydb_json_json
 * @brief Return true if a JSON path returns at least one item for a temporal
 * JSONB value
 * @sqlfn tjsonb_path_exists()
 */
Datum
Tjsonb_path_exists(PG_FUNCTION_ARGS)
{
 return Tjsonb_path_exists_common(fcinfo, false);
}

PGDLLEXPORT Datum Tjsonb_path_exists_tz(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_path_exists_tz);
/**
 * @ingroup mobilitydb_json_json
 * @brief Return true if a JSON path returns at least one item for a temporal
 * JSONB value
 * @sqlfn tjsonb_path_exists_tz()
 */
Datum
Tjsonb_path_exists_tz(PG_FUNCTION_ARGS)
{
  return Tjsonb_path_exists_common(fcinfo, true);
}

PGDLLEXPORT Datum Tjsonb_path_exists_opr(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_path_exists_opr);
/**
 * @ingroup mobilitydb_json_json
 * @brief Return true if a JSON path returns at least one item for a temporal
 * JSONB value
 * @details Implementation of operator "tjsonb @? jsonpath" (2-argument version
 * of tjsonb_path_exists())
 * @sqlfn tjsonb_path_exists_opr()
 */
Datum
Tjsonb_path_exists_opr(PG_FUNCTION_ARGS)
{
 return Tjsonb_path_exists_common(fcinfo, false);
}

/*****************************************************************************/

/**
 * @brief Extract an item specified by a JSON path predicate from a temporal
 * JSONB value
 * @sqlfn tjsonb_path_match(), tjsonb_path_match_tz()
 */
Datum
Tjsonb_path_match_common(FunctionCallInfo fcinfo, bool tz)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  JsonPath *jp = PG_GETARG_JSONPATH_P(1);
  Jsonb *vars = PG_GETARG_JSONB_P(2);
  bool silent = PG_GETARG_BOOL(3);
  Temporal *result = tjsonb_path_match(temp, jp, vars, silent, tz);
  PG_FREE_IF_COPY(temp, 0);
  PG_FREE_IF_COPY(jp, 1);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

PGDLLEXPORT Datum Tjsonb_path_match(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_path_match);
/**
 * @ingroup mobilitydb_json_json
 * @brief Extract an item specified by a JSON path predicate from a temporal
 * JSONB value
 * @sqlfn tjsonb_path_match()
 */
Datum
Tjsonb_path_match(PG_FUNCTION_ARGS)
{
 return Tjsonb_path_match_common(fcinfo, false);
}

PGDLLEXPORT Datum Tjsonb_path_match_tz(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_path_match_tz);
/**
 * @ingroup mobilitydb_json_json
 * @brief Extract an item specified by a JSON path predicate from a temporal
 * JSONB value
 * @sqlfn tjsonb_path_match_tz()
 */
Datum
Tjsonb_path_match_tz(PG_FUNCTION_ARGS)
{
  return Tjsonb_path_match_common(fcinfo, true);
}

/*****************************************************************************/

/**
 * @brief Extract the items specified by a JSON path predicate from a
 * temporal JSONB value as a JSONB array
 * @sqlfn tjsonb_path_query_array(), tjsonb_path_query_array_tz()
 */
Datum
Tjsonb_path_query_array_common(FunctionCallInfo fcinfo, bool tz)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  JsonPath *jp = PG_GETARG_JSONPATH_P(1);
  Jsonb *vars = PG_GETARG_JSONB_P(2);
  bool silent = PG_GETARG_BOOL(3);
  Temporal *result = tjsonb_path_query_array(temp, jp, vars, silent, tz);
  PG_FREE_IF_COPY(temp, 0);
  PG_FREE_IF_COPY(jp, 1);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

PGDLLEXPORT Datum Tjsonb_path_query_array(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_path_query_array);
/**
 * @ingroup mobilitydb_json_json
 * @brief Extract the items specified by a JSON path predicate from a
 * temporal JSONB value as a JSONB array
 * @sqlfn tjsonb_path_query_array()
 */
Datum
Tjsonb_path_query_array(PG_FUNCTION_ARGS)
{
 return Tjsonb_path_query_array_common(fcinfo, false);
}

PGDLLEXPORT Datum Tjsonb_path_query_array_tz(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_path_query_array_tz);
/**
 * @ingroup mobilitydb_json_json
 * @brief Extract the items specified by a JSON path predicate from a
 * temporal JSONB value as a JSONB array
 * @sqlfn tjsonb_path_query_array_tz()
 */
Datum
Tjsonb_path_query_array_tz(PG_FUNCTION_ARGS)
{
  return Tjsonb_path_query_array_common(fcinfo, true);
}

/*****************************************************************************/

/**
 * @brief Extract the first item specified by a JSON path predicate from a
 * temporal JSONB value. If there are no items, return NULL.
 * @sqlfn tjsonb_query_first(), tjsonb_query_first_tz()
 */
Datum
Tjsonb_path_query_first_common(FunctionCallInfo fcinfo, bool tz)
{
  Temporal *temp = PG_GETARG_TEMPORAL_P(0);
  JsonPath *jp = PG_GETARG_JSONPATH_P(1);
  Jsonb *vars = PG_GETARG_JSONB_P(2);
  bool silent = PG_GETARG_BOOL(3);
  Temporal *result = tjsonb_path_query_first(temp, jp, vars, silent, tz);
  PG_FREE_IF_COPY(temp, 0);
  PG_FREE_IF_COPY(jp, 1);
  if (! result)
    PG_RETURN_NULL();
  PG_RETURN_TEMPORAL_P(result);
}

PGDLLEXPORT Datum Tjsonb_path_query_first(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_path_query_first);
/**
 * @ingroup mobilitydb_json_json
 * @brief Extract the first item specified by a JSON path predicate from a
 * temporal JSONB value. If there are no items, return NULL.
 * @sqlfn tjsonb_path_query_first()
 */
Datum
Tjsonb_path_query_first(PG_FUNCTION_ARGS)
{
 return Tjsonb_path_query_first_common(fcinfo, false);
}

PGDLLEXPORT Datum Tjsonb_path_query_first_tz(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(Tjsonb_path_query_first_tz);
/**
 * @ingroup mobilitydb_json_json
 * @brief Extract the first item specified by a JSON path predicate from a
 * temporal JSONB value. If there are no items, return NULL.
 * @sqlfn tjsonb_path_query_first_tz()
 */
Datum
Tjsonb_path_query_first_tz(PG_FUNCTION_ARGS)
{
  return Tjsonb_path_query_first_common(fcinfo, true);
}

/*****************************************************************************/

estebanzimanyi / MobilityDB / 20645520862

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous