PostgreSQL Source Code  git master
hstore_plpython.c
Go to the documentation of this file.
1 #include "postgres.h"
2 
3 #include "fmgr.h"
4 #include "hstore/hstore.h"
5 #include "plpy_typeio.h"
6 #include "plpython.h"
7 
9 
10 /* Linkage to functions in plpython module */
11 typedef char *(*PLyObject_AsString_t) (PyObject *plrv);
13 typedef PyObject *(*PLyUnicode_FromStringAndSize_t) (const char *s, Py_ssize_t size);
15 
16 /* Linkage to functions in hstore module */
17 typedef HStore *(*hstoreUpgrade_t) (Datum orig);
19 typedef int (*hstoreUniquePairs_t) (Pairs *a, int32 l, int32 *buflen);
21 typedef HStore *(*hstorePairs_t) (Pairs *pairs, int32 pcount, int32 buflen);
23 typedef size_t (*hstoreCheckKeyLen_t) (size_t len);
25 typedef size_t (*hstoreCheckValLen_t) (size_t len);
27 
28 
29 /*
30  * Module initialize function: fetch function pointers for cross-module calls.
31  */
32 void
33 _PG_init(void)
34 {
35  /* Asserts verify that typedefs above match original declarations */
38  load_external_function("$libdir/" PLPYTHON_LIBNAME, "PLyObject_AsString",
39  true, NULL);
42  load_external_function("$libdir/" PLPYTHON_LIBNAME, "PLyUnicode_FromStringAndSize",
43  true, NULL);
46  load_external_function("$libdir/hstore", "hstoreUpgrade",
47  true, NULL);
50  load_external_function("$libdir/hstore", "hstoreUniquePairs",
51  true, NULL);
54  load_external_function("$libdir/hstore", "hstorePairs",
55  true, NULL);
58  load_external_function("$libdir/hstore", "hstoreCheckKeyLen",
59  true, NULL);
62  load_external_function("$libdir/hstore", "hstoreCheckValLen",
63  true, NULL);
64 }
65 
66 
67 /* These defines must be after the module init function */
68 #define PLyObject_AsString PLyObject_AsString_p
69 #define PLyUnicode_FromStringAndSize PLyUnicode_FromStringAndSize_p
70 #define hstoreUpgrade hstoreUpgrade_p
71 #define hstoreUniquePairs hstoreUniquePairs_p
72 #define hstorePairs hstorePairs_p
73 #define hstoreCheckKeyLen hstoreCheckKeyLen_p
74 #define hstoreCheckValLen hstoreCheckValLen_p
75 
76 
78 
79 Datum
81 {
82  HStore *in = PG_GETARG_HSTORE_P(0);
83  int i;
84  int count = HS_COUNT(in);
85  char *base = STRPTR(in);
86  HEntry *entries = ARRPTR(in);
87  PyObject *dict;
88 
89  dict = PyDict_New();
90  if (!dict)
91  ereport(ERROR,
92  (errcode(ERRCODE_OUT_OF_MEMORY),
93  errmsg("out of memory")));
94 
95  for (i = 0; i < count; i++)
96  {
97  PyObject *key;
98 
99  key = PLyUnicode_FromStringAndSize(HSTORE_KEY(entries, base, i),
100  HSTORE_KEYLEN(entries, i));
101  if (HSTORE_VALISNULL(entries, i))
102  PyDict_SetItem(dict, key, Py_None);
103  else
104  {
105  PyObject *value;
106 
108  HSTORE_VALLEN(entries, i));
109  PyDict_SetItem(dict, key, value);
110  Py_XDECREF(value);
111  }
112  Py_XDECREF(key);
113  }
114 
115  return PointerGetDatum(dict);
116 }
117 
118 
120 
121 Datum
123 {
124  PyObject *dict;
125  PyObject *volatile items;
126  Py_ssize_t pcount;
127  HStore *volatile out;
128 
129  dict = (PyObject *) PG_GETARG_POINTER(0);
130  if (!PyMapping_Check(dict))
131  ereport(ERROR,
132  (errcode(ERRCODE_WRONG_OBJECT_TYPE),
133  errmsg("not a Python mapping")));
134 
135  pcount = PyMapping_Size(dict);
136  items = PyMapping_Items(dict);
137 
138  PG_TRY();
139  {
140  int32 buflen;
141  Py_ssize_t i;
142  Pairs *pairs;
143 
144  pairs = palloc(pcount * sizeof(*pairs));
145 
146  for (i = 0; i < pcount; i++)
147  {
148  PyObject *tuple;
149  PyObject *key;
150  PyObject *value;
151 
152  tuple = PyList_GetItem(items, i);
153  key = PyTuple_GetItem(tuple, 0);
154  value = PyTuple_GetItem(tuple, 1);
155 
156  pairs[i].key = PLyObject_AsString(key);
157  pairs[i].keylen = hstoreCheckKeyLen(strlen(pairs[i].key));
158  pairs[i].needfree = true;
159 
160  if (value == Py_None)
161  {
162  pairs[i].val = NULL;
163  pairs[i].vallen = 0;
164  pairs[i].isnull = true;
165  }
166  else
167  {
168  pairs[i].val = PLyObject_AsString(value);
169  pairs[i].vallen = hstoreCheckValLen(strlen(pairs[i].val));
170  pairs[i].isnull = false;
171  }
172  }
173 
174  pcount = hstoreUniquePairs(pairs, pcount, &buflen);
175  out = hstorePairs(pairs, pcount, buflen);
176  }
177  PG_FINALLY();
178  {
179  Py_DECREF(items);
180  }
181  PG_END_TRY();
182 
183  PG_RETURN_POINTER(out);
184 }
signed int int32
Definition: c.h:430
#define AssertVariableIsOfType(varname, typename)
Definition: c.h:914
#define ARRPTR(x)
Definition: cube.c:25
void * load_external_function(const char *filename, const char *funcname, bool signalNotFound, void **filehandle)
Definition: dfmgr.c:105
int errcode(int sqlerrcode)
Definition: elog.c:735
int errmsg(const char *fmt,...)
Definition: elog.c:946
#define PG_TRY(...)
Definition: elog.h:309
#define PG_END_TRY(...)
Definition: elog.h:334
#define ERROR
Definition: elog.h:35
#define PG_FINALLY(...)
Definition: elog.h:326
#define ereport(elevel,...)
Definition: elog.h:145
#define PG_GETARG_POINTER(n)
Definition: fmgr.h:276
#define PG_RETURN_POINTER(x)
Definition: fmgr.h:361
#define PG_FUNCTION_ARGS
Definition: fmgr.h:193
#define HS_COUNT(hsp_)
Definition: hstore.h:61
#define HSTORE_KEY(arr_, str_, i_)
Definition: hstore.h:79
#define PG_GETARG_HSTORE_P(x)
Definition: hstore.h:154
#define HSTORE_VALISNULL(arr_, i_)
Definition: hstore.h:83
#define HSTORE_VALLEN(arr_, i_)
Definition: hstore.h:82
#define HSTORE_KEYLEN(arr_, i_)
Definition: hstore.h:81
#define HSTORE_VAL(arr_, str_, i_)
Definition: hstore.h:80
#define STRPTR(x)
Definition: hstore.h:76
#define hstoreUpgrade
HStore *(* hstoreUpgrade_t)(Datum orig)
Datum plpython_to_hstore(PG_FUNCTION_ARGS)
#define PLyUnicode_FromStringAndSize
int(* hstoreUniquePairs_t)(Pairs *a, int32 l, int32 *buflen)
static hstoreUpgrade_t hstoreUpgrade_p
void _PG_init(void)
char *(* PLyObject_AsString_t)(PyObject *plrv)
static hstoreCheckKeyLen_t hstoreCheckKeyLen_p
#define hstoreCheckKeyLen
PyObject *(* PLyUnicode_FromStringAndSize_t)(const char *s, Py_ssize_t size)
#define hstoreUniquePairs
PG_MODULE_MAGIC
HStore *(* hstorePairs_t)(Pairs *pairs, int32 pcount, int32 buflen)
static hstorePairs_t hstorePairs_p
static PLyObject_AsString_t PLyObject_AsString_p
#define hstorePairs
Datum hstore_to_plpython(PG_FUNCTION_ARGS)
static hstoreCheckValLen_t hstoreCheckValLen_p
static hstoreUniquePairs_t hstoreUniquePairs_p
size_t(* hstoreCheckValLen_t)(size_t len)
static PLyUnicode_FromStringAndSize_t PLyUnicode_FromStringAndSize_p
#define hstoreCheckValLen
PG_FUNCTION_INFO_V1(hstore_to_plpython)
size_t(* hstoreCheckKeyLen_t)(size_t len)
#define PLyObject_AsString
static struct @143 value
long val
Definition: informix.c:664
int a
Definition: isn.c:69
int i
Definition: isn.c:73
void * palloc(Size size)
Definition: mcxt.c:1199
const void size_t len
static Datum PointerGetDatum(const void *X)
Definition: postgres.h:670
uintptr_t Datum
Definition: postgres.h:412
Definition: hstore.h:19
Definition: hstore.h:45
Definition: hstore.h:162
char * val
Definition: hstore.h:164
bool isnull
Definition: hstore.h:167
size_t keylen
Definition: hstore.h:165
char * key
Definition: hstore.h:163
bool needfree
Definition: hstore.h:168
size_t vallen
Definition: hstore.h:166