Age Owner Branch data TLA Line data Source code
1 : : #include "postgres.h"
2 : :
3 : : #include "fmgr.h"
4 : : #include "hstore/hstore.h"
5 : : #include "plpy_typeio.h"
6 : : #include "plpython.h"
7 : :
3276 peter_e@gmx.net 8 :CBC 1 : PG_MODULE_MAGIC;
9 : :
10 : : /* Linkage to functions in plpython module */
11 : : typedef char *(*PLyObject_AsString_t) (PyObject *plrv);
12 : : static PLyObject_AsString_t PLyObject_AsString_p;
13 : : typedef PyObject *(*PLyUnicode_FromStringAndSize_t) (const char *s, Py_ssize_t size);
14 : : static PLyUnicode_FromStringAndSize_t PLyUnicode_FromStringAndSize_p;
15 : :
16 : : /* Linkage to functions in hstore module */
17 : : typedef HStore *(*hstoreUpgrade_t) (Datum orig);
18 : : static hstoreUpgrade_t hstoreUpgrade_p;
19 : : typedef int (*hstoreUniquePairs_t) (Pairs *a, int32 l, int32 *buflen);
20 : : static hstoreUniquePairs_t hstoreUniquePairs_p;
21 : : typedef HStore *(*hstorePairs_t) (Pairs *pairs, int32 pcount, int32 buflen);
22 : : static hstorePairs_t hstorePairs_p;
23 : : typedef size_t (*hstoreCheckKeyLen_t) (size_t len);
24 : : static hstoreCheckKeyLen_t hstoreCheckKeyLen_p;
25 : : typedef size_t (*hstoreCheckValLen_t) (size_t len);
26 : : static hstoreCheckValLen_t hstoreCheckValLen_p;
27 : :
28 : :
29 : : /*
30 : : * Module initialize function: fetch function pointers for cross-module calls.
31 : : */
32 : : void
2750 tgl@sss.pgh.pa.us 33 : 1 : _PG_init(void)
34 : : {
35 : : /* Asserts verify that typedefs above match original declarations */
36 : : AssertVariableIsOfType(&PLyObject_AsString, PLyObject_AsString_t);
37 : 1 : PLyObject_AsString_p = (PLyObject_AsString_t)
38 : 1 : load_external_function("$libdir/" PLPYTHON_LIBNAME, "PLyObject_AsString",
39 : : true, NULL);
40 : : AssertVariableIsOfType(&PLyUnicode_FromStringAndSize, PLyUnicode_FromStringAndSize_t);
2749 41 : 1 : PLyUnicode_FromStringAndSize_p = (PLyUnicode_FromStringAndSize_t)
42 : 1 : load_external_function("$libdir/" PLPYTHON_LIBNAME, "PLyUnicode_FromStringAndSize",
43 : : true, NULL);
44 : : AssertVariableIsOfType(&hstoreUpgrade, hstoreUpgrade_t);
2750 45 : 1 : hstoreUpgrade_p = (hstoreUpgrade_t)
46 : 1 : load_external_function("$libdir/hstore", "hstoreUpgrade",
47 : : true, NULL);
48 : : AssertVariableIsOfType(&hstoreUniquePairs, hstoreUniquePairs_t);
49 : 1 : hstoreUniquePairs_p = (hstoreUniquePairs_t)
50 : 1 : load_external_function("$libdir/hstore", "hstoreUniquePairs",
51 : : true, NULL);
52 : : AssertVariableIsOfType(&hstorePairs, hstorePairs_t);
53 : 1 : hstorePairs_p = (hstorePairs_t)
54 : 1 : load_external_function("$libdir/hstore", "hstorePairs",
55 : : true, NULL);
56 : : AssertVariableIsOfType(&hstoreCheckKeyLen, hstoreCheckKeyLen_t);
57 : 1 : hstoreCheckKeyLen_p = (hstoreCheckKeyLen_t)
58 : 1 : load_external_function("$libdir/hstore", "hstoreCheckKeyLen",
59 : : true, NULL);
60 : : AssertVariableIsOfType(&hstoreCheckValLen, hstoreCheckValLen_t);
61 : 1 : hstoreCheckValLen_p = (hstoreCheckValLen_t)
62 : 1 : load_external_function("$libdir/hstore", "hstoreCheckValLen",
63 : : true, NULL);
64 : 1 : }
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 : :
3276 peter_e@gmx.net 77 : 2 : PG_FUNCTION_INFO_V1(hstore_to_plpython);
78 : :
79 : : Datum
80 : 7 : hstore_to_plpython(PG_FUNCTION_ARGS)
81 : : {
2400 tgl@sss.pgh.pa.us 82 : 7 : HStore *in = PG_GETARG_HSTORE_P(0);
83 : : int i;
3276 peter_e@gmx.net 84 : 7 : int count = HS_COUNT(in);
85 : 7 : char *base = STRPTR(in);
86 : 7 : HEntry *entries = ARRPTR(in);
87 : : PyObject *dict;
88 : :
89 : 7 : dict = PyDict_New();
2357 90 [ - + ]: 7 : if (!dict)
2357 peter_e@gmx.net 91 [ # # ]:UBC 0 : ereport(ERROR,
92 : : (errcode(ERRCODE_OUT_OF_MEMORY),
93 : : errmsg("out of memory")));
94 : :
3276 peter_e@gmx.net 95 [ + + ]:CBC 20 : for (i = 0; i < count; i++)
96 : : {
97 : : PyObject *key;
98 : :
769 andres@anarazel.de 99 [ + + ]: 13 : key = PLyUnicode_FromStringAndSize(HSTORE_KEY(entries, base, i),
703 tgl@sss.pgh.pa.us 100 [ + + ]: 13 : HSTORE_KEYLEN(entries, i));
3069 101 [ + + ]: 13 : if (HSTORE_VALISNULL(entries, i))
3276 peter_e@gmx.net 102 : 6 : PyDict_SetItem(dict, key, Py_None);
103 : : else
104 : : {
105 : : PyObject *value;
106 : :
769 andres@anarazel.de 107 [ + - ]: 7 : value = PLyUnicode_FromStringAndSize(HSTORE_VAL(entries, base, i),
703 tgl@sss.pgh.pa.us 108 [ - + ]: 7 : HSTORE_VALLEN(entries, i));
3276 peter_e@gmx.net 109 : 7 : PyDict_SetItem(dict, key, value);
110 : 7 : Py_XDECREF(value);
111 : : }
112 : 13 : Py_XDECREF(key);
113 : : }
114 : :
115 : 7 : return PointerGetDatum(dict);
116 : : }
117 : :
118 : :
119 : 2 : PG_FUNCTION_INFO_V1(plpython_to_hstore);
120 : :
121 : : Datum
122 : 8 : plpython_to_hstore(PG_FUNCTION_ARGS)
123 : : {
124 : : PyObject *dict;
125 : : PyObject *volatile items;
126 : : Py_ssize_t pcount;
127 : : HStore *volatile out;
128 : :
129 : 8 : dict = (PyObject *) PG_GETARG_POINTER(0);
130 : :
131 : : /*
132 : : * As of Python 3, PyMapping_Check() is unreliable unless one first checks
133 : : * that the object isn't a sequence. (Cleaner solutions exist, but not
134 : : * before Python 3.10, which we're not prepared to require yet.)
135 : : */
353 tgl@sss.pgh.pa.us 136 [ + + - + ]: 8 : if (PySequence_Check(dict) || !PyMapping_Check(dict))
3276 peter_e@gmx.net 137 [ + - ]: 1 : ereport(ERROR,
138 : : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
139 : : errmsg("not a Python mapping")));
140 : :
141 : 7 : pcount = PyMapping_Size(dict);
1858 peter@eisentraut.org 142 : 7 : items = PyMapping_Items(dict);
143 : :
3276 peter_e@gmx.net 144 [ + - ]: 7 : PG_TRY();
145 : : {
146 : : int32 buflen;
147 : : Py_ssize_t i;
148 : : Pairs *pairs;
149 : :
150 : 7 : pairs = palloc(pcount * sizeof(*pairs));
151 : :
152 [ + + ]: 26 : for (i = 0; i < pcount; i++)
153 : : {
154 : : PyObject *tuple;
155 : : PyObject *key;
156 : : PyObject *value;
157 : :
158 : 19 : tuple = PyList_GetItem(items, i);
159 : 19 : key = PyTuple_GetItem(tuple, 0);
160 : 19 : value = PyTuple_GetItem(tuple, 1);
161 : :
162 : 19 : pairs[i].key = PLyObject_AsString(key);
163 : 19 : pairs[i].keylen = hstoreCheckKeyLen(strlen(pairs[i].key));
164 : 19 : pairs[i].needfree = true;
165 : :
166 [ + + ]: 19 : if (value == Py_None)
167 : : {
168 : 6 : pairs[i].val = NULL;
169 : 6 : pairs[i].vallen = 0;
170 : 6 : pairs[i].isnull = true;
171 : : }
172 : : else
173 : : {
174 : 13 : pairs[i].val = PLyObject_AsString(value);
175 : 13 : pairs[i].vallen = hstoreCheckValLen(strlen(pairs[i].val));
176 : 13 : pairs[i].isnull = false;
177 : : }
178 : : }
179 : :
180 : 7 : pcount = hstoreUniquePairs(pairs, pcount, &buflen);
181 : 7 : out = hstorePairs(pairs, pcount, buflen);
182 : : }
1626 peter@eisentraut.org 183 :UBC 0 : PG_FINALLY();
184 : : {
1858 peter@eisentraut.org 185 :CBC 7 : Py_DECREF(items);
186 : : }
3276 peter_e@gmx.net 187 [ - + ]: 7 : PG_END_TRY();
188 : :
189 : 7 : PG_RETURN_POINTER(out);
190 : : }
|
