PostgreSQL Source Code  git master
hstore_plperl.c
Go to the documentation of this file.
1 #include "postgres.h"
2 
3 #include "fmgr.h"
4 #include "hstore/hstore.h"
5 #include "plperl.h"
6 #include "plperl_helpers.h"
7 
9 
10 extern void _PG_init(void);
11 
12 /* Linkage to functions in hstore module */
13 typedef HStore *(*hstoreUpgrade_t) (Datum orig);
15 typedef int (*hstoreUniquePairs_t) (Pairs *a, int32 l, int32 *buflen);
17 typedef HStore *(*hstorePairs_t) (Pairs *pairs, int32 pcount, int32 buflen);
19 typedef size_t (*hstoreCheckKeyLen_t) (size_t len);
21 typedef size_t (*hstoreCheckValLen_t) (size_t len);
23 
24 
25 /*
26  * Module initialize function: fetch function pointers for cross-module calls.
27  */
28 void
29 _PG_init(void)
30 {
31  /* Asserts verify that typedefs above match original declarations */
34  load_external_function("$libdir/hstore", "hstoreUpgrade",
35  true, NULL);
38  load_external_function("$libdir/hstore", "hstoreUniquePairs",
39  true, NULL);
42  load_external_function("$libdir/hstore", "hstorePairs",
43  true, NULL);
46  load_external_function("$libdir/hstore", "hstoreCheckKeyLen",
47  true, NULL);
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 
64 
65 Datum
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 
97 
98 Datum
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 }
size_t(* hstoreCheckKeyLen_t)(size_t len)
Definition: hstore_plperl.c:19
#define hstorePairs
Definition: hstore_plperl.c:58
#define PG_RETURN_POINTER(x)
Definition: fmgr.h:351
Definition: hstore.h:44
PG_MODULE_MAGIC
Definition: hstore_plperl.c:8
char * pnstrdup(const char *in, Size len)
Definition: mcxt.c:1172
#define PG_GETARG_HSTORE_P(x)
Definition: hstore.h:154
#define dTHX
Definition: ppport.h:3208
static hstorePairs_t hstorePairs_p
Definition: hstore_plperl.c:18
#define PointerGetDatum(X)
Definition: postgres.h:556
char * pstrdup(const char *in)
Definition: mcxt.c:1161
#define HSTORE_KEYLEN(arr_, i_)
Definition: hstore.h:81
static struct @144 value
int errcode(int sqlerrcode)
Definition: elog.c:570
#define HSTORE_VALLEN(arr_, i_)
Definition: hstore.h:82
#define PG_GETARG_POINTER(n)
Definition: fmgr.h:271
size_t vallen
Definition: hstore.h:166
#define hstoreCheckValLen
Definition: hstore_plperl.c:60
bool needfree
Definition: hstore.h:168
static hstoreCheckValLen_t hstoreCheckValLen_p
Definition: hstore_plperl.c:22
signed int int32
Definition: c.h:346
char * val
Definition: hstore.h:164
char * key
Definition: hstore.h:163
#define HSTORE_VALISNULL(arr_, i_)
Definition: hstore.h:83
#define ERROR
Definition: elog.h:43
size_t(* hstoreCheckValLen_t)(size_t len)
Definition: hstore_plperl.c:21
#define HS_COUNT(hsp_)
Definition: hstore.h:61
static hstoreCheckKeyLen_t hstoreCheckKeyLen_p
Definition: hstore_plperl.c:20
HStore *(* hstoreUpgrade_t)(Datum orig)
Definition: hstore_plperl.c:13
#define HSTORE_KEY(arr_, str_, i_)
Definition: hstore.h:79
PGFunction load_external_function(const char *filename, const char *funcname, bool signalNotFound, void **filehandle)
Definition: dfmgr.c:107
#define ereport(elevel, rest)
Definition: elog.h:141
Datum hstore_to_plperl(PG_FUNCTION_ARGS)
Definition: hstore_plperl.c:66
#define hstoreUniquePairs
Definition: hstore_plperl.c:57
PG_FUNCTION_INFO_V1(hstore_to_plperl)
uintptr_t Datum
Definition: postgres.h:367
HStore *(* hstorePairs_t)(Pairs *pairs, int32 pcount, int32 buflen)
Definition: hstore_plperl.c:17
#define hstoreUpgrade
Definition: hstore_plperl.c:56
static SV * cstr2sv(const char *str)
size_t keylen
Definition: hstore.h:165
bool isnull
Definition: hstore.h:167
#define hstoreCheckKeyLen
Definition: hstore_plperl.c:59
Definition: hstore.h:18
void _PG_init(void)
Definition: hstore_plperl.c:29
#define HSTORE_VAL(arr_, str_, i_)
Definition: hstore.h:80
Datum plperl_to_hstore(PG_FUNCTION_ARGS)
Definition: hstore_plperl.c:99
void * palloc(Size size)
Definition: mcxt.c:924
int errmsg(const char *fmt,...)
Definition: elog.c:784
#define STRPTR(x)
Definition: hstore.h:76
int i
int(* hstoreUniquePairs_t)(Pairs *a, int32 l, int32 *buflen)
Definition: hstore_plperl.c:15
static hstoreUniquePairs_t hstoreUniquePairs_p
Definition: hstore_plperl.c:16
#define PG_FUNCTION_ARGS
Definition: fmgr.h:188
#define ARRPTR(x)
Definition: cube.c:25
static hstoreUpgrade_t hstoreUpgrade_p
Definition: hstore_plperl.c:14
Definition: hstore.h:161
long val
Definition: informix.c:684
static char * sv2cstr(SV *sv)
#define AssertVariableIsOfType(varname, typename)
Definition: c.h:881