1 #include "postgres.h"
2
3 #include "fmgr.h"
4 #include "hstore/hstore.h"
5 #include "plperl.h"
6 #include "plperl_helpers.h"
7
8 PG_MODULE_MAGIC;
9
10 extern void _PG_init(void);
11
12 /* Linkage to functions in hstore module */
13 typedef HStore *(*hstoreUpgrade_t) (Datum orig);
14 static hstoreUpgrade_t hstoreUpgrade_p;
15 typedef int (*hstoreUniquePairs_t) (Pairs *a, int32 l, int32 *buflen);
16 static hstoreUniquePairs_t hstoreUniquePairs_p;
17 typedef HStore *(*hstorePairs_t) (Pairs *pairs, int32 pcount, int32 buflen);
18 static hstorePairs_t hstorePairs_p;
19 typedef size_t (*hstoreCheckKeyLen_t) (size_t len);
20 static hstoreCheckKeyLen_t hstoreCheckKeyLen_p;
21 typedef size_t (*hstoreCheckValLen_t) (size_t len);
22 static hstoreCheckValLen_t hstoreCheckValLen_p;
23
24
25 /*
26 * Module initialize function: fetch function pointers for cross-module calls.
27 */
28 void
_PG_init(void)29 _PG_init(void)
30 {
31 /* Asserts verify that typedefs above match original declarations */
32 AssertVariableIsOfType(&hstoreUpgrade, hstoreUpgrade_t);
33 hstoreUpgrade_p = (hstoreUpgrade_t)
34 load_external_function("$libdir/hstore", "hstoreUpgrade",
35 true, NULL);
36 AssertVariableIsOfType(&hstoreUniquePairs, hstoreUniquePairs_t);
37 hstoreUniquePairs_p = (hstoreUniquePairs_t)
38 load_external_function("$libdir/hstore", "hstoreUniquePairs",
39 true, NULL);
40 AssertVariableIsOfType(&hstorePairs, hstorePairs_t);
41 hstorePairs_p = (hstorePairs_t)
42 load_external_function("$libdir/hstore", "hstorePairs",
43 true, NULL);
44 AssertVariableIsOfType(&hstoreCheckKeyLen, hstoreCheckKeyLen_t);
45 hstoreCheckKeyLen_p = (hstoreCheckKeyLen_t)
46 load_external_function("$libdir/hstore", "hstoreCheckKeyLen",
47 true, NULL);
48 AssertVariableIsOfType(&hstoreCheckValLen, hstoreCheckValLen_t);
49 hstoreCheckValLen_p = (hstoreCheckValLen_t)
50 load_external_function("$libdir/hstore", "hstoreCheckValLen",
51 true, NULL);
52 }
53
54
55 /* These defines must be after the module init function */
56 #define hstoreUpgrade hstoreUpgrade_p
57 #define hstoreUniquePairs hstoreUniquePairs_p
58 #define hstorePairs hstorePairs_p
59 #define hstoreCheckKeyLen hstoreCheckKeyLen_p
60 #define hstoreCheckValLen hstoreCheckValLen_p
61
62
63 PG_FUNCTION_INFO_V1(hstore_to_plperl);
64
65 Datum
hstore_to_plperl(PG_FUNCTION_ARGS)66 hstore_to_plperl(PG_FUNCTION_ARGS)
67 {
68 dTHX;
69 HStore *in = PG_GETARG_HSTORE_P(0);
70 int i;
71 int count = HS_COUNT(in);
72 char *base = STRPTR(in);
73 HEntry *entries = ARRPTR(in);
74 HV *hv;
75
76 hv = newHV();
77
78 for (i = 0; i < count; i++)
79 {
80 const char *key;
81 SV *value;
82
83 key = pnstrdup(HSTORE_KEY(entries, base, i),
84 HSTORE_KEYLEN(entries, i));
85 value = HSTORE_VALISNULL(entries, i) ? newSV(0) :
86 cstr2sv(pnstrdup(HSTORE_VAL(entries, base, i),
87 HSTORE_VALLEN(entries, i)));
88
89 (void) hv_store(hv, key, strlen(key), value, 0);
90 }
91
92 return PointerGetDatum(newRV((SV *) hv));
93 }
94
95
96 PG_FUNCTION_INFO_V1(plperl_to_hstore);
97
98 Datum
plperl_to_hstore(PG_FUNCTION_ARGS)99 plperl_to_hstore(PG_FUNCTION_ARGS)
100 {
101 dTHX;
102 SV *in = (SV *) PG_GETARG_POINTER(0);
103 HV *hv;
104 HE *he;
105 int32 buflen;
106 int32 i;
107 int32 pcount;
108 HStore *out;
109 Pairs *pairs;
110
111 /* Dereference references recursively. */
112 while (SvROK(in))
113 in = SvRV(in);
114
115 /* Now we must have a hash. */
116 if (SvTYPE(in) != SVt_PVHV)
117 ereport(ERROR,
118 (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
119 errmsg("cannot transform non-hash Perl value to hstore")));
120 hv = (HV *) in;
121
122 pcount = hv_iterinit(hv);
123
124 pairs = palloc(pcount * sizeof(Pairs));
125
126 i = 0;
127 while ((he = hv_iternext(hv)))
128 {
129 char *key = sv2cstr(HeSVKEY_force(he));
130 SV *value = HeVAL(he);
131
132 pairs[i].key = pstrdup(key);
133 pairs[i].keylen = hstoreCheckKeyLen(strlen(pairs[i].key));
134 pairs[i].needfree = true;
135
136 if (!SvOK(value))
137 {
138 pairs[i].val = NULL;
139 pairs[i].vallen = 0;
140 pairs[i].isnull = true;
141 }
142 else
143 {
144 pairs[i].val = pstrdup(sv2cstr(value));
145 pairs[i].vallen = hstoreCheckValLen(strlen(pairs[i].val));
146 pairs[i].isnull = false;
147 }
148
149 i++;
150 }
151
152 pcount = hstoreUniquePairs(pairs, pcount, &buflen);
153 out = hstorePairs(pairs, pcount, buflen);
154 PG_RETURN_POINTER(out);
155 }
156