Line data Source code
1 : #include "postgres.h"
2 :
3 : #include "fmgr.h"
4 : #include "hstore/hstore.h"
5 : #include "plperl.h"
6 :
7 0 : PG_MODULE_MAGIC_EXT(
8 : .name = "hstore_plperl",
9 : .version = PG_VERSION
10 : );
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
29 0 : _PG_init(void)
30 : {
31 : /* Asserts verify that typedefs above match original declarations */
32 0 : AssertVariableIsOfType(&hstoreUpgrade, hstoreUpgrade_t);
33 0 : hstoreUpgrade_p = (hstoreUpgrade_t)
34 0 : load_external_function("$libdir/hstore", "hstoreUpgrade",
35 : true, NULL);
36 0 : AssertVariableIsOfType(&hstoreUniquePairs, hstoreUniquePairs_t);
37 0 : hstoreUniquePairs_p = (hstoreUniquePairs_t)
38 0 : load_external_function("$libdir/hstore", "hstoreUniquePairs",
39 : true, NULL);
40 0 : AssertVariableIsOfType(&hstorePairs, hstorePairs_t);
41 0 : hstorePairs_p = (hstorePairs_t)
42 0 : load_external_function("$libdir/hstore", "hstorePairs",
43 : true, NULL);
44 0 : AssertVariableIsOfType(&hstoreCheckKeyLen, hstoreCheckKeyLen_t);
45 0 : hstoreCheckKeyLen_p = (hstoreCheckKeyLen_t)
46 0 : load_external_function("$libdir/hstore", "hstoreCheckKeyLen",
47 : true, NULL);
48 0 : AssertVariableIsOfType(&hstoreCheckValLen, hstoreCheckValLen_t);
49 0 : hstoreCheckValLen_p = (hstoreCheckValLen_t)
50 0 : load_external_function("$libdir/hstore", "hstoreCheckValLen",
51 : true, NULL);
52 0 : }
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 0 : PG_FUNCTION_INFO_V1(hstore_to_plperl);
64 :
65 : Datum
66 0 : hstore_to_plperl(PG_FUNCTION_ARGS)
67 : {
68 0 : dTHX;
69 0 : HStore *in = PG_GETARG_HSTORE_P(0);
70 0 : int i;
71 0 : int count = HS_COUNT(in);
72 0 : char *base = STRPTR(in);
73 0 : HEntry *entries = ARRPTR(in);
74 0 : HV *hv;
75 :
76 0 : hv = newHV();
77 :
78 0 : for (i = 0; i < count; i++)
79 : {
80 0 : const char *key;
81 0 : SV *value;
82 :
83 0 : key = pnstrdup(HSTORE_KEY(entries, base, i),
84 0 : HSTORE_KEYLEN(entries, i));
85 0 : value = HSTORE_VALISNULL(entries, i) ? newSV(0) :
86 0 : cstr2sv(pnstrdup(HSTORE_VAL(entries, base, i),
87 0 : HSTORE_VALLEN(entries, i)));
88 :
89 0 : (void) hv_store(hv, key, strlen(key), value, 0);
90 0 : }
91 :
92 0 : return PointerGetDatum(newRV((SV *) hv));
93 0 : }
94 :
95 :
96 0 : PG_FUNCTION_INFO_V1(plperl_to_hstore);
97 :
98 : Datum
99 0 : plperl_to_hstore(PG_FUNCTION_ARGS)
100 : {
101 0 : dTHX;
102 0 : SV *in = (SV *) PG_GETARG_POINTER(0);
103 0 : HV *hv;
104 0 : HE *he;
105 0 : int32 buflen;
106 0 : int32 i;
107 0 : int32 pcount;
108 0 : HStore *out;
109 0 : Pairs *pairs;
110 :
111 : /* Dereference references recursively. */
112 0 : while (SvROK(in))
113 0 : in = SvRV(in);
114 :
115 : /* Now we must have a hash. */
116 0 : if (SvTYPE(in) != SVt_PVHV)
117 0 : ereport(ERROR,
118 : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
119 : errmsg("cannot transform non-hash Perl value to hstore")));
120 0 : hv = (HV *) in;
121 :
122 0 : pcount = hv_iterinit(hv);
123 :
124 0 : pairs = palloc(pcount * sizeof(Pairs));
125 :
126 0 : i = 0;
127 0 : while ((he = hv_iternext(hv)))
128 : {
129 0 : char *key = sv2cstr(HeSVKEY_force(he));
130 0 : SV *value = HeVAL(he);
131 :
132 0 : pairs[i].key = pstrdup(key);
133 0 : pairs[i].keylen = hstoreCheckKeyLen(strlen(pairs[i].key));
134 0 : pairs[i].needfree = true;
135 :
136 0 : if (!SvOK(value))
137 : {
138 0 : pairs[i].val = NULL;
139 0 : pairs[i].vallen = 0;
140 0 : pairs[i].isnull = true;
141 0 : }
142 : else
143 : {
144 0 : pairs[i].val = pstrdup(sv2cstr(value));
145 0 : pairs[i].vallen = hstoreCheckValLen(strlen(pairs[i].val));
146 0 : pairs[i].isnull = false;
147 : }
148 :
149 0 : i++;
150 0 : }
151 :
152 0 : pcount = hstoreUniquePairs(pairs, pcount, &buflen);
153 0 : out = hstorePairs(pairs, pcount, buflen);
154 0 : PG_RETURN_POINTER(out);
155 0 : }
|
