Line data    Source code

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