Line data    Source code

       1              : #include "postgres.h"
       2              : 
       3              : #include "fmgr.h"
       4              : #include "ltree/ltree.h"
       5              : #include "plpy_util.h"
       6              : 
       7            1 : PG_MODULE_MAGIC_EXT(
       8              :                     .name = "ltree_plpython",
       9              :                     .version = PG_VERSION
      10              : );
      11              : 
      12              : /* Linkage to functions in plpython module */
      13              : typedef PyObject *(*PLyUnicode_FromStringAndSize_t) (const char *s, Py_ssize_t size);
      14              : static PLyUnicode_FromStringAndSize_t PLyUnicode_FromStringAndSize_p;
      15              : 
      16              : /* Static asserts verify that typedefs above match original declarations */
      17              : StaticAssertVariableIsOfType(&PLyUnicode_FromStringAndSize, PLyUnicode_FromStringAndSize_t);
      18              : 
      19              : 
      20              : /*
      21              :  * Module initialize function: fetch function pointers for cross-module calls.
      22              :  */
      23              : void
      24            1 : _PG_init(void)
      25              : {
      26            1 :     PLyUnicode_FromStringAndSize_p = (PLyUnicode_FromStringAndSize_t)
      27            1 :         load_external_function("$libdir/" PLPYTHON_LIBNAME, "PLyUnicode_FromStringAndSize",
      28              :                                true, NULL);
      29            1 : }
      30              : 
      31              : 
      32              : /* These defines must be after the module init function */
      33              : #define PLyUnicode_FromStringAndSize PLyUnicode_FromStringAndSize_p
      34              : 
      35              : 
      36            2 : PG_FUNCTION_INFO_V1(ltree_to_plpython);
      37              : 
      38              : Datum
      39            2 : ltree_to_plpython(PG_FUNCTION_ARGS)
      40              : {
      41            2 :     ltree      *in = PG_GETARG_LTREE_P(0);
      42              :     int         i;
      43              :     PyObject   *list;
      44              :     ltree_level *curlevel;
      45              : 
      46            2 :     list = PyList_New(in->numlevel);
      47            2 :     if (!list)
      48            0 :         ereport(ERROR,
      49              :                 (errcode(ERRCODE_OUT_OF_MEMORY),
      50              :                  errmsg("out of memory")));
      51              : 
      52            2 :     curlevel = LTREE_FIRST(in);
      53            8 :     for (i = 0; i < in->numlevel; i++)
      54              :     {
      55            6 :         PyList_SetItem(list, i, PLyUnicode_FromStringAndSize(curlevel->name, curlevel->len));
      56            6 :         curlevel = LEVEL_NEXT(curlevel);
      57              :     }
      58              : 
      59            2 :     PG_FREE_IF_COPY(in, 0);
      60              : 
      61            2 :     return PointerGetDatum(list);
      62              : }