Line data    Source code

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