2008-05-12 08:00:54 +08:00
|
|
|
/*
|
2010-09-21 04:08:53 +08:00
|
|
|
* contrib/hstore/hstore_io.c
|
2008-05-12 08:00:54 +08:00
|
|
|
*/
|
|
|
|
#include "postgres.h"
|
|
|
|
|
2006-09-06 02:00:58 +08:00
|
|
|
#include <ctype.h>
|
|
|
|
|
2009-10-01 03:50:22 +08:00
|
|
|
#include "catalog/pg_type.h"
|
|
|
|
#include "funcapi.h"
|
|
|
|
#include "libpq/pqformat.h"
|
|
|
|
#include "utils/lsyscache.h"
|
|
|
|
#include "utils/typcache.h"
|
|
|
|
|
2008-05-12 08:00:54 +08:00
|
|
|
#include "hstore.h"
|
|
|
|
|
2006-09-06 02:00:58 +08:00
|
|
|
PG_MODULE_MAGIC;
|
|
|
|
|
2009-10-01 03:50:22 +08:00
|
|
|
/* old names for C functions */
|
2010-02-26 10:01:40 +08:00
|
|
|
HSTORE_POLLUTE(hstore_from_text, tconvert);
|
2009-10-01 03:50:22 +08:00
|
|
|
|
|
|
|
|
2006-10-04 08:30:14 +08:00
|
|
|
typedef struct
|
|
|
|
{
|
|
|
|
char *begin;
|
|
|
|
char *ptr;
|
|
|
|
char *cur;
|
|
|
|
char *word;
|
|
|
|
int wordlen;
|
|
|
|
|
|
|
|
Pairs *pairs;
|
|
|
|
int pcur;
|
|
|
|
int plen;
|
2009-06-11 22:49:15 +08:00
|
|
|
} HSParser;
|
2006-09-06 02:00:58 +08:00
|
|
|
|
|
|
|
#define RESIZEPRSBUF \
|
|
|
|
do { \
|
2006-10-04 08:30:14 +08:00
|
|
|
if ( state->cur - state->word + 1 >= state->wordlen ) \
|
|
|
|
{ \
|
|
|
|
int4 clen = state->cur - state->word; \
|
|
|
|
state->wordlen *= 2; \
|
|
|
|
state->word = (char*)repalloc( (void*)state->word, state->wordlen ); \
|
|
|
|
state->cur = state->word + clen; \
|
|
|
|
} \
|
2006-09-06 02:00:58 +08:00
|
|
|
} while (0)
|
|
|
|
|
|
|
|
|
2006-10-04 08:30:14 +08:00
|
|
|
#define GV_WAITVAL 0
|
|
|
|
#define GV_INVAL 1
|
|
|
|
#define GV_INESCVAL 2
|
|
|
|
#define GV_WAITESCIN 3
|
|
|
|
#define GV_WAITESCESCIN 4
|
2006-09-06 02:00:58 +08:00
|
|
|
|
|
|
|
static bool
|
2009-06-11 22:49:15 +08:00
|
|
|
get_val(HSParser *state, bool ignoreeq, bool *escaped)
|
2006-10-04 08:30:14 +08:00
|
|
|
{
|
|
|
|
int st = GV_WAITVAL;
|
|
|
|
|
|
|
|
state->wordlen = 32;
|
|
|
|
state->cur = state->word = palloc(state->wordlen);
|
|
|
|
*escaped = false;
|
|
|
|
|
|
|
|
while (1)
|
|
|
|
{
|
|
|
|
if (st == GV_WAITVAL)
|
|
|
|
{
|
|
|
|
if (*(state->ptr) == '"')
|
|
|
|
{
|
|
|
|
*escaped = true;
|
2006-09-06 02:00:58 +08:00
|
|
|
st = GV_INESCVAL;
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
else if (*(state->ptr) == '\0')
|
|
|
|
{
|
2006-09-06 02:00:58 +08:00
|
|
|
return false;
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
else if (*(state->ptr) == '=' && !ignoreeq)
|
|
|
|
{
|
|
|
|
elog(ERROR, "Syntax error near '%c' at postion %d", *(state->ptr), (int4) (state->ptr - state->begin));
|
|
|
|
}
|
|
|
|
else if (*(state->ptr) == '\\')
|
|
|
|
{
|
2006-09-06 02:00:58 +08:00
|
|
|
st = GV_WAITESCIN;
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
else if (!isspace((unsigned char) *(state->ptr)))
|
|
|
|
{
|
2006-09-06 02:00:58 +08:00
|
|
|
*(state->cur) = *(state->ptr);
|
|
|
|
state->cur++;
|
|
|
|
st = GV_INVAL;
|
|
|
|
}
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
else if (st == GV_INVAL)
|
|
|
|
{
|
|
|
|
if (*(state->ptr) == '\\')
|
|
|
|
{
|
2006-09-06 02:00:58 +08:00
|
|
|
st = GV_WAITESCIN;
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
else if (*(state->ptr) == '=' && !ignoreeq)
|
|
|
|
{
|
2006-09-06 02:00:58 +08:00
|
|
|
state->ptr--;
|
|
|
|
return true;
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
else if (*(state->ptr) == ',' && ignoreeq)
|
|
|
|
{
|
2006-09-06 02:00:58 +08:00
|
|
|
state->ptr--;
|
|
|
|
return true;
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
else if (isspace((unsigned char) *(state->ptr)))
|
|
|
|
{
|
2006-09-06 02:00:58 +08:00
|
|
|
return true;
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
else if (*(state->ptr) == '\0')
|
|
|
|
{
|
2006-09-06 02:00:58 +08:00
|
|
|
state->ptr--;
|
|
|
|
return true;
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2006-09-06 02:00:58 +08:00
|
|
|
RESIZEPRSBUF;
|
|
|
|
*(state->cur) = *(state->ptr);
|
|
|
|
state->cur++;
|
|
|
|
}
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
else if (st == GV_INESCVAL)
|
|
|
|
{
|
|
|
|
if (*(state->ptr) == '\\')
|
|
|
|
{
|
2006-09-06 02:00:58 +08:00
|
|
|
st = GV_WAITESCESCIN;
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
else if (*(state->ptr) == '"')
|
|
|
|
{
|
2006-09-06 02:00:58 +08:00
|
|
|
return true;
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
else if (*(state->ptr) == '\0')
|
|
|
|
{
|
|
|
|
elog(ERROR, "Unexpected end of string");
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2006-09-06 02:00:58 +08:00
|
|
|
RESIZEPRSBUF;
|
|
|
|
*(state->cur) = *(state->ptr);
|
|
|
|
state->cur++;
|
|
|
|
}
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
else if (st == GV_WAITESCIN)
|
|
|
|
{
|
|
|
|
if (*(state->ptr) == '\0')
|
|
|
|
elog(ERROR, "Unexpected end of string");
|
2006-09-06 02:00:58 +08:00
|
|
|
RESIZEPRSBUF;
|
|
|
|
*(state->cur) = *(state->ptr);
|
|
|
|
state->cur++;
|
2006-10-04 08:30:14 +08:00
|
|
|
st = GV_INVAL;
|
|
|
|
}
|
|
|
|
else if (st == GV_WAITESCESCIN)
|
|
|
|
{
|
|
|
|
if (*(state->ptr) == '\0')
|
|
|
|
elog(ERROR, "Unexpected end of string");
|
2006-09-06 02:00:58 +08:00
|
|
|
RESIZEPRSBUF;
|
|
|
|
*(state->cur) = *(state->ptr);
|
|
|
|
state->cur++;
|
|
|
|
st = GV_INESCVAL;
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
else
|
2006-12-06 01:36:00 +08:00
|
|
|
elog(ERROR, "Unknown state %d at position line %d in file '%s'", st, __LINE__, __FILE__);
|
2006-09-06 02:00:58 +08:00
|
|
|
|
|
|
|
state->ptr++;
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
2006-09-06 02:00:58 +08:00
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
#define WKEY 0
|
|
|
|
#define WVAL 1
|
2006-10-04 08:30:14 +08:00
|
|
|
#define WEQ 2
|
|
|
|
#define WGT 3
|
2006-09-06 02:00:58 +08:00
|
|
|
#define WDEL 4
|
|
|
|
|
|
|
|
|
|
|
|
static void
|
2009-06-11 22:49:15 +08:00
|
|
|
parse_hstore(HSParser *state)
|
2006-10-04 08:30:14 +08:00
|
|
|
{
|
|
|
|
int st = WKEY;
|
|
|
|
bool escaped = false;
|
|
|
|
|
|
|
|
state->plen = 16;
|
|
|
|
state->pairs = (Pairs *) palloc(sizeof(Pairs) * state->plen);
|
|
|
|
state->pcur = 0;
|
2006-09-06 02:00:58 +08:00
|
|
|
state->ptr = state->begin;
|
2006-10-04 08:30:14 +08:00
|
|
|
state->word = NULL;
|
2006-09-06 02:00:58 +08:00
|
|
|
|
2006-10-04 08:30:14 +08:00
|
|
|
while (1)
|
|
|
|
{
|
|
|
|
if (st == WKEY)
|
|
|
|
{
|
|
|
|
if (!get_val(state, false, &escaped))
|
2006-09-06 02:00:58 +08:00
|
|
|
return;
|
2006-10-04 08:30:14 +08:00
|
|
|
if (state->pcur >= state->plen)
|
|
|
|
{
|
2006-09-06 02:00:58 +08:00
|
|
|
state->plen *= 2;
|
2006-10-04 08:30:14 +08:00
|
|
|
state->pairs = (Pairs *) repalloc(state->pairs, sizeof(Pairs) * state->plen);
|
2006-09-06 02:00:58 +08:00
|
|
|
}
|
2006-10-04 08:30:14 +08:00
|
|
|
state->pairs[state->pcur].key = state->word;
|
2009-03-16 06:05:17 +08:00
|
|
|
state->pairs[state->pcur].keylen = hstoreCheckKeyLen(state->cur - state->word);
|
2006-10-04 08:30:14 +08:00
|
|
|
state->pairs[state->pcur].val = NULL;
|
|
|
|
state->word = NULL;
|
2006-09-06 02:00:58 +08:00
|
|
|
st = WEQ;
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
else if (st == WEQ)
|
|
|
|
{
|
|
|
|
if (*(state->ptr) == '=')
|
|
|
|
{
|
2006-09-06 02:00:58 +08:00
|
|
|
st = WGT;
|
|
|
|
}
|
2006-10-04 08:30:14 +08:00
|
|
|
else if (*(state->ptr) == '\0')
|
|
|
|
{
|
2006-12-06 01:36:00 +08:00
|
|
|
elog(ERROR, "Unexpected end of string");
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
else if (!isspace((unsigned char) *(state->ptr)))
|
|
|
|
{
|
2006-12-06 01:36:00 +08:00
|
|
|
elog(ERROR, "Syntax error near '%c' at position %d", *(state->ptr), (int4) (state->ptr - state->begin));
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
else if (st == WGT)
|
|
|
|
{
|
|
|
|
if (*(state->ptr) == '>')
|
|
|
|
{
|
2006-09-06 02:00:58 +08:00
|
|
|
st = WVAL;
|
|
|
|
}
|
2006-10-04 08:30:14 +08:00
|
|
|
else if (*(state->ptr) == '\0')
|
|
|
|
{
|
2006-12-06 01:36:00 +08:00
|
|
|
elog(ERROR, "Unexpected end of string");
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2006-12-06 01:36:00 +08:00
|
|
|
elog(ERROR, "Syntax error near '%c' at position %d", *(state->ptr), (int4) (state->ptr - state->begin));
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
else if (st == WVAL)
|
|
|
|
{
|
|
|
|
if (!get_val(state, true, &escaped))
|
|
|
|
elog(ERROR, "Unexpected end of string");
|
|
|
|
state->pairs[state->pcur].val = state->word;
|
2009-03-16 06:05:17 +08:00
|
|
|
state->pairs[state->pcur].vallen = hstoreCheckValLen(state->cur - state->word);
|
2006-10-04 08:30:14 +08:00
|
|
|
state->pairs[state->pcur].isnull = false;
|
|
|
|
state->pairs[state->pcur].needfree = true;
|
|
|
|
if (state->cur - state->word == 4 && !escaped)
|
|
|
|
{
|
2006-09-06 02:00:58 +08:00
|
|
|
state->word[4] = '\0';
|
2006-10-04 08:30:14 +08:00
|
|
|
if (0 == pg_strcasecmp(state->word, "null"))
|
|
|
|
state->pairs[state->pcur].isnull = true;
|
|
|
|
}
|
|
|
|
state->word = NULL;
|
2006-09-06 02:00:58 +08:00
|
|
|
state->pcur++;
|
|
|
|
st = WDEL;
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
else if (st == WDEL)
|
|
|
|
{
|
|
|
|
if (*(state->ptr) == ',')
|
|
|
|
{
|
2006-09-06 02:00:58 +08:00
|
|
|
st = WKEY;
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
else if (*(state->ptr) == '\0')
|
|
|
|
{
|
2006-09-06 02:00:58 +08:00
|
|
|
return;
|
|
|
|
}
|
2006-10-04 08:30:14 +08:00
|
|
|
else if (!isspace((unsigned char) *(state->ptr)))
|
|
|
|
{
|
2006-12-06 01:36:00 +08:00
|
|
|
elog(ERROR, "Syntax error near '%c' at position %d", *(state->ptr), (int4) (state->ptr - state->begin));
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
elog(ERROR, "Unknown state %d at line %d in file '%s'", st, __LINE__, __FILE__);
|
2006-09-06 02:00:58 +08:00
|
|
|
|
|
|
|
state->ptr++;
|
|
|
|
}
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
2006-09-06 02:00:58 +08:00
|
|
|
|
2009-10-01 03:50:22 +08:00
|
|
|
static int
|
2006-10-04 08:30:14 +08:00
|
|
|
comparePairs(const void *a, const void *b)
|
|
|
|
{
|
2011-09-12 02:54:32 +08:00
|
|
|
const Pairs *pa = a;
|
|
|
|
const Pairs *pb = b;
|
|
|
|
|
|
|
|
if (pa->keylen == pb->keylen)
|
2006-10-04 08:30:14 +08:00
|
|
|
{
|
2011-09-12 02:54:32 +08:00
|
|
|
int res = memcmp(pa->key, pb->key, pa->keylen);
|
2006-10-04 08:30:14 +08:00
|
|
|
|
|
|
|
if (res)
|
2006-09-06 02:00:58 +08:00
|
|
|
return res;
|
|
|
|
|
2006-12-06 01:36:00 +08:00
|
|
|
/* guarantee that needfree will be later */
|
2011-09-12 02:54:32 +08:00
|
|
|
if (pb->needfree == pa->needfree)
|
2006-09-06 02:00:58 +08:00
|
|
|
return 0;
|
2011-09-12 02:54:32 +08:00
|
|
|
else if (pa->needfree)
|
2006-09-06 02:00:58 +08:00
|
|
|
return 1;
|
|
|
|
else
|
2006-10-04 08:30:14 +08:00
|
|
|
return -1;
|
2006-09-06 02:00:58 +08:00
|
|
|
}
|
2011-09-12 02:54:32 +08:00
|
|
|
return (pa->keylen > pb->keylen) ? 1 : -1;
|
2006-09-06 02:00:58 +08:00
|
|
|
}
|
|
|
|
|
2009-10-01 03:50:22 +08:00
|
|
|
/*
|
|
|
|
* this code still respects pairs.needfree, even though in general
|
|
|
|
* it should never be called in a context where anything needs freeing.
|
|
|
|
* we keep it because (a) those calls are in a rare code path anyway,
|
|
|
|
* and (b) who knows whether they might be needed by some caller.
|
|
|
|
*/
|
2006-09-06 02:00:58 +08:00
|
|
|
int
|
2009-10-01 03:50:22 +08:00
|
|
|
hstoreUniquePairs(Pairs *a, int4 l, int4 *buflen)
|
2006-10-04 08:30:14 +08:00
|
|
|
{
|
|
|
|
Pairs *ptr,
|
|
|
|
*res;
|
|
|
|
|
|
|
|
*buflen = 0;
|
|
|
|
if (l < 2)
|
|
|
|
{
|
|
|
|
if (l == 1)
|
|
|
|
*buflen = a->keylen + ((a->isnull) ? 0 : a->vallen);
|
2006-09-06 02:00:58 +08:00
|
|
|
return l;
|
|
|
|
}
|
|
|
|
|
|
|
|
qsort((void *) a, l, sizeof(Pairs), comparePairs);
|
2006-10-04 08:30:14 +08:00
|
|
|
ptr = a + 1;
|
|
|
|
res = a;
|
|
|
|
while (ptr - a < l)
|
|
|
|
{
|
2009-10-01 03:50:22 +08:00
|
|
|
if (ptr->keylen == res->keylen &&
|
2010-12-22 11:11:40 +08:00
|
|
|
memcmp(ptr->key, res->key, res->keylen) == 0)
|
2006-10-04 08:30:14 +08:00
|
|
|
{
|
|
|
|
if (ptr->needfree)
|
|
|
|
{
|
2006-09-06 02:00:58 +08:00
|
|
|
pfree(ptr->key);
|
|
|
|
pfree(ptr->val);
|
|
|
|
}
|
2006-10-04 08:30:14 +08:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
*buflen += res->keylen + ((res->isnull) ? 0 : res->vallen);
|
2006-09-06 02:00:58 +08:00
|
|
|
res++;
|
2006-10-04 08:30:14 +08:00
|
|
|
memcpy(res, ptr, sizeof(Pairs));
|
2006-09-06 02:00:58 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
ptr++;
|
|
|
|
}
|
|
|
|
|
2006-10-04 08:30:14 +08:00
|
|
|
*buflen += res->keylen + ((res->isnull) ? 0 : res->vallen);
|
2006-09-06 02:00:58 +08:00
|
|
|
return res + 1 - a;
|
|
|
|
}
|
|
|
|
|
2009-03-16 06:05:17 +08:00
|
|
|
size_t
|
|
|
|
hstoreCheckKeyLen(size_t len)
|
|
|
|
{
|
|
|
|
if (len > HSTORE_MAX_KEY_LEN)
|
|
|
|
ereport(ERROR,
|
|
|
|
(errcode(ERRCODE_STRING_DATA_RIGHT_TRUNCATION),
|
|
|
|
errmsg("string too long for hstore key")));
|
|
|
|
return len;
|
|
|
|
}
|
|
|
|
|
|
|
|
size_t
|
|
|
|
hstoreCheckValLen(size_t len)
|
|
|
|
{
|
|
|
|
if (len > HSTORE_MAX_VALUE_LEN)
|
|
|
|
ereport(ERROR,
|
|
|
|
(errcode(ERRCODE_STRING_DATA_RIGHT_TRUNCATION),
|
|
|
|
errmsg("string too long for hstore value")));
|
|
|
|
return len;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2009-10-01 03:50:22 +08:00
|
|
|
HStore *
|
|
|
|
hstorePairs(Pairs *pairs, int4 pcount, int4 buflen)
|
|
|
|
{
|
2010-02-26 10:01:40 +08:00
|
|
|
HStore *out;
|
2009-10-01 03:50:22 +08:00
|
|
|
HEntry *entry;
|
|
|
|
char *ptr;
|
|
|
|
char *buf;
|
2010-02-26 10:01:40 +08:00
|
|
|
int4 len;
|
|
|
|
int4 i;
|
2009-10-01 03:50:22 +08:00
|
|
|
|
|
|
|
len = CALCDATASIZE(pcount, buflen);
|
|
|
|
out = palloc(len);
|
|
|
|
SET_VARSIZE(out, len);
|
|
|
|
HS_SETCOUNT(out, pcount);
|
|
|
|
|
|
|
|
if (pcount == 0)
|
|
|
|
return out;
|
|
|
|
|
|
|
|
entry = ARRPTR(out);
|
|
|
|
buf = ptr = STRPTR(out);
|
|
|
|
|
|
|
|
for (i = 0; i < pcount; i++)
|
2010-02-26 10:01:40 +08:00
|
|
|
HS_ADDITEM(entry, buf, ptr, pairs[i]);
|
2009-10-01 03:50:22 +08:00
|
|
|
|
2010-02-26 10:01:40 +08:00
|
|
|
HS_FINALIZE(out, pcount, buf, ptr);
|
2009-10-01 03:50:22 +08:00
|
|
|
|
|
|
|
return out;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2006-09-06 02:00:58 +08:00
|
|
|
PG_FUNCTION_INFO_V1(hstore_in);
|
2006-10-04 08:30:14 +08:00
|
|
|
Datum hstore_in(PG_FUNCTION_ARGS);
|
2006-09-06 02:00:58 +08:00
|
|
|
Datum
|
2006-10-04 08:30:14 +08:00
|
|
|
hstore_in(PG_FUNCTION_ARGS)
|
|
|
|
{
|
|
|
|
HSParser state;
|
2009-10-01 03:50:22 +08:00
|
|
|
int4 buflen;
|
2006-10-04 08:30:14 +08:00
|
|
|
HStore *out;
|
|
|
|
|
|
|
|
state.begin = PG_GETARG_CSTRING(0);
|
2006-09-06 02:00:58 +08:00
|
|
|
|
|
|
|
parse_hstore(&state);
|
|
|
|
|
2009-10-01 03:50:22 +08:00
|
|
|
state.pcur = hstoreUniquePairs(state.pairs, state.pcur, &buflen);
|
|
|
|
|
|
|
|
out = hstorePairs(state.pairs, state.pcur, buflen);
|
|
|
|
|
|
|
|
PG_RETURN_POINTER(out);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
PG_FUNCTION_INFO_V1(hstore_recv);
|
|
|
|
Datum hstore_recv(PG_FUNCTION_ARGS);
|
|
|
|
Datum
|
|
|
|
hstore_recv(PG_FUNCTION_ARGS)
|
|
|
|
{
|
|
|
|
int4 buflen;
|
|
|
|
HStore *out;
|
|
|
|
Pairs *pairs;
|
2010-02-26 10:01:40 +08:00
|
|
|
int4 i;
|
|
|
|
int4 pcount;
|
|
|
|
StringInfo buf = (StringInfo) PG_GETARG_POINTER(0);
|
2009-10-01 03:50:22 +08:00
|
|
|
|
|
|
|
pcount = pq_getmsgint(buf, 4);
|
|
|
|
|
|
|
|
if (pcount == 0)
|
2006-10-04 08:30:14 +08:00
|
|
|
{
|
2009-10-01 03:50:22 +08:00
|
|
|
out = hstorePairs(NULL, 0, 0);
|
2006-09-06 02:00:58 +08:00
|
|
|
PG_RETURN_POINTER(out);
|
|
|
|
}
|
|
|
|
|
2009-10-01 03:50:22 +08:00
|
|
|
pairs = palloc(pcount * sizeof(Pairs));
|
2006-09-06 02:00:58 +08:00
|
|
|
|
2009-10-01 03:50:22 +08:00
|
|
|
for (i = 0; i < pcount; ++i)
|
|
|
|
{
|
2010-02-26 10:01:40 +08:00
|
|
|
int rawlen = pq_getmsgint(buf, 4);
|
|
|
|
int len;
|
2009-10-01 03:50:22 +08:00
|
|
|
|
|
|
|
if (rawlen < 0)
|
|
|
|
ereport(ERROR,
|
|
|
|
(errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
|
|
|
|
errmsg("null value not allowed for hstore key")));
|
|
|
|
|
|
|
|
pairs[i].key = pq_getmsgtext(buf, rawlen, &len);
|
|
|
|
pairs[i].keylen = hstoreCheckKeyLen(len);
|
|
|
|
pairs[i].needfree = true;
|
|
|
|
|
|
|
|
rawlen = pq_getmsgint(buf, 4);
|
|
|
|
if (rawlen < 0)
|
|
|
|
{
|
|
|
|
pairs[i].val = NULL;
|
|
|
|
pairs[i].vallen = 0;
|
|
|
|
pairs[i].isnull = true;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
pairs[i].val = pq_getmsgtext(buf, rawlen, &len);
|
|
|
|
pairs[i].vallen = hstoreCheckValLen(len);
|
|
|
|
pairs[i].isnull = false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pcount = hstoreUniquePairs(pairs, pcount, &buflen);
|
|
|
|
|
|
|
|
out = hstorePairs(pairs, pcount, buflen);
|
|
|
|
|
|
|
|
PG_RETURN_POINTER(out);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
PG_FUNCTION_INFO_V1(hstore_from_text);
|
|
|
|
Datum hstore_from_text(PG_FUNCTION_ARGS);
|
|
|
|
Datum
|
|
|
|
hstore_from_text(PG_FUNCTION_ARGS)
|
|
|
|
{
|
2010-02-26 10:01:40 +08:00
|
|
|
text *key;
|
|
|
|
text *val = NULL;
|
|
|
|
Pairs p;
|
2009-10-01 03:50:22 +08:00
|
|
|
HStore *out;
|
|
|
|
|
|
|
|
if (PG_ARGISNULL(0))
|
|
|
|
PG_RETURN_NULL();
|
|
|
|
|
|
|
|
p.needfree = false;
|
|
|
|
key = PG_GETARG_TEXT_PP(0);
|
|
|
|
p.key = VARDATA_ANY(key);
|
|
|
|
p.keylen = hstoreCheckKeyLen(VARSIZE_ANY_EXHDR(key));
|
|
|
|
|
|
|
|
if (PG_ARGISNULL(1))
|
|
|
|
{
|
|
|
|
p.vallen = 0;
|
|
|
|
p.isnull = true;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
val = PG_GETARG_TEXT_PP(1);
|
|
|
|
p.val = VARDATA_ANY(val);
|
|
|
|
p.vallen = hstoreCheckValLen(VARSIZE_ANY_EXHDR(val));
|
|
|
|
p.isnull = false;
|
|
|
|
}
|
|
|
|
|
|
|
|
out = hstorePairs(&p, 1, p.keylen + p.vallen);
|
|
|
|
|
|
|
|
PG_RETURN_POINTER(out);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
PG_FUNCTION_INFO_V1(hstore_from_arrays);
|
|
|
|
Datum hstore_from_arrays(PG_FUNCTION_ARGS);
|
|
|
|
Datum
|
|
|
|
hstore_from_arrays(PG_FUNCTION_ARGS)
|
|
|
|
{
|
|
|
|
int4 buflen;
|
|
|
|
HStore *out;
|
|
|
|
Pairs *pairs;
|
|
|
|
Datum *key_datums;
|
|
|
|
bool *key_nulls;
|
2010-02-26 10:01:40 +08:00
|
|
|
int key_count;
|
2009-10-01 03:50:22 +08:00
|
|
|
Datum *value_datums;
|
|
|
|
bool *value_nulls;
|
2010-02-26 10:01:40 +08:00
|
|
|
int value_count;
|
2009-10-01 03:50:22 +08:00
|
|
|
ArrayType *key_array;
|
|
|
|
ArrayType *value_array;
|
2010-02-26 10:01:40 +08:00
|
|
|
int i;
|
2009-10-01 03:50:22 +08:00
|
|
|
|
|
|
|
if (PG_ARGISNULL(0))
|
|
|
|
PG_RETURN_NULL();
|
|
|
|
|
|
|
|
key_array = PG_GETARG_ARRAYTYPE_P(0);
|
|
|
|
|
|
|
|
Assert(ARR_ELEMTYPE(key_array) == TEXTOID);
|
|
|
|
|
|
|
|
/*
|
2010-02-26 10:01:40 +08:00
|
|
|
* must check >1 rather than != 1 because empty arrays have 0 dimensions,
|
|
|
|
* not 1
|
2009-10-01 03:50:22 +08:00
|
|
|
*/
|
2006-09-06 02:00:58 +08:00
|
|
|
|
2009-10-01 03:50:22 +08:00
|
|
|
if (ARR_NDIM(key_array) > 1)
|
|
|
|
ereport(ERROR,
|
|
|
|
(errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
|
|
|
|
errmsg("wrong number of array subscripts")));
|
|
|
|
|
|
|
|
deconstruct_array(key_array,
|
|
|
|
TEXTOID, -1, false, 'i',
|
|
|
|
&key_datums, &key_nulls, &key_count);
|
|
|
|
|
|
|
|
/* value_array might be NULL */
|
2006-09-06 02:00:58 +08:00
|
|
|
|
2009-10-01 03:50:22 +08:00
|
|
|
if (PG_ARGISNULL(1))
|
|
|
|
{
|
|
|
|
value_array = NULL;
|
|
|
|
value_count = key_count;
|
|
|
|
value_datums = NULL;
|
|
|
|
value_nulls = NULL;
|
|
|
|
}
|
|
|
|
else
|
2006-10-04 08:30:14 +08:00
|
|
|
{
|
2009-10-01 03:50:22 +08:00
|
|
|
value_array = PG_GETARG_ARRAYTYPE_P(1);
|
|
|
|
|
|
|
|
Assert(ARR_ELEMTYPE(value_array) == TEXTOID);
|
|
|
|
|
|
|
|
if (ARR_NDIM(value_array) > 1)
|
|
|
|
ereport(ERROR,
|
|
|
|
(errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
|
|
|
|
errmsg("wrong number of array subscripts")));
|
|
|
|
|
|
|
|
if ((ARR_NDIM(key_array) > 0 || ARR_NDIM(value_array) > 0) &&
|
|
|
|
(ARR_NDIM(key_array) != ARR_NDIM(value_array) ||
|
|
|
|
ARR_DIMS(key_array)[0] != ARR_DIMS(value_array)[0] ||
|
|
|
|
ARR_LBOUND(key_array)[0] != ARR_LBOUND(value_array)[0]))
|
|
|
|
ereport(ERROR,
|
|
|
|
(errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
|
|
|
|
errmsg("arrays must have same bounds")));
|
|
|
|
|
|
|
|
deconstruct_array(value_array,
|
|
|
|
TEXTOID, -1, false, 'i',
|
|
|
|
&value_datums, &value_nulls, &value_count);
|
2006-09-06 02:00:58 +08:00
|
|
|
|
2009-10-01 03:50:22 +08:00
|
|
|
Assert(key_count == value_count);
|
|
|
|
}
|
|
|
|
|
|
|
|
pairs = palloc(key_count * sizeof(Pairs));
|
|
|
|
|
|
|
|
for (i = 0; i < key_count; ++i)
|
|
|
|
{
|
|
|
|
if (key_nulls[i])
|
|
|
|
ereport(ERROR,
|
|
|
|
(errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
|
|
|
|
errmsg("null value not allowed for hstore key")));
|
|
|
|
|
|
|
|
if (!value_nulls || value_nulls[i])
|
|
|
|
{
|
|
|
|
pairs[i].key = VARDATA_ANY(key_datums[i]);
|
|
|
|
pairs[i].val = NULL;
|
|
|
|
pairs[i].keylen = hstoreCheckKeyLen(VARSIZE_ANY_EXHDR(key_datums[i]));
|
|
|
|
pairs[i].vallen = 4;
|
|
|
|
pairs[i].isnull = true;
|
|
|
|
pairs[i].needfree = false;
|
|
|
|
}
|
2006-10-04 08:30:14 +08:00
|
|
|
else
|
|
|
|
{
|
2009-10-01 03:50:22 +08:00
|
|
|
pairs[i].key = VARDATA_ANY(key_datums[i]);
|
|
|
|
pairs[i].val = VARDATA_ANY(value_datums[i]);
|
|
|
|
pairs[i].keylen = hstoreCheckKeyLen(VARSIZE_ANY_EXHDR(key_datums[i]));
|
|
|
|
pairs[i].vallen = hstoreCheckValLen(VARSIZE_ANY_EXHDR(value_datums[i]));
|
|
|
|
pairs[i].isnull = false;
|
|
|
|
pairs[i].needfree = false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
key_count = hstoreUniquePairs(pairs, key_count, &buflen);
|
|
|
|
|
|
|
|
out = hstorePairs(pairs, key_count, buflen);
|
|
|
|
|
|
|
|
PG_RETURN_POINTER(out);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
PG_FUNCTION_INFO_V1(hstore_from_array);
|
|
|
|
Datum hstore_from_array(PG_FUNCTION_ARGS);
|
|
|
|
Datum
|
|
|
|
hstore_from_array(PG_FUNCTION_ARGS)
|
|
|
|
{
|
|
|
|
ArrayType *in_array = PG_GETARG_ARRAYTYPE_P(0);
|
2010-02-26 10:01:40 +08:00
|
|
|
int ndims = ARR_NDIM(in_array);
|
|
|
|
int count;
|
2009-10-01 03:50:22 +08:00
|
|
|
int4 buflen;
|
|
|
|
HStore *out;
|
|
|
|
Pairs *pairs;
|
|
|
|
Datum *in_datums;
|
|
|
|
bool *in_nulls;
|
2010-02-26 10:01:40 +08:00
|
|
|
int in_count;
|
|
|
|
int i;
|
2009-10-01 03:50:22 +08:00
|
|
|
|
|
|
|
Assert(ARR_ELEMTYPE(in_array) == TEXTOID);
|
|
|
|
|
|
|
|
switch (ndims)
|
|
|
|
{
|
|
|
|
case 0:
|
|
|
|
out = hstorePairs(NULL, 0, 0);
|
|
|
|
PG_RETURN_POINTER(out);
|
|
|
|
|
|
|
|
case 1:
|
|
|
|
if ((ARR_DIMS(in_array)[0]) % 2)
|
|
|
|
ereport(ERROR,
|
|
|
|
(errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
|
|
|
|
errmsg("array must have even number of elements")));
|
|
|
|
break;
|
|
|
|
|
|
|
|
case 2:
|
|
|
|
if ((ARR_DIMS(in_array)[1]) != 2)
|
|
|
|
ereport(ERROR,
|
|
|
|
(errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
|
|
|
|
errmsg("array must have two columns")));
|
|
|
|
break;
|
|
|
|
|
|
|
|
default:
|
|
|
|
ereport(ERROR,
|
|
|
|
(errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
|
|
|
|
errmsg("wrong number of array subscripts")));
|
2010-02-26 10:01:40 +08:00
|
|
|
}
|
2009-10-01 03:50:22 +08:00
|
|
|
|
|
|
|
deconstruct_array(in_array,
|
|
|
|
TEXTOID, -1, false, 'i',
|
|
|
|
&in_datums, &in_nulls, &in_count);
|
|
|
|
|
|
|
|
count = in_count / 2;
|
|
|
|
|
|
|
|
pairs = palloc(count * sizeof(Pairs));
|
|
|
|
|
|
|
|
for (i = 0; i < count; ++i)
|
|
|
|
{
|
2010-02-26 10:01:40 +08:00
|
|
|
if (in_nulls[i * 2])
|
2009-10-01 03:50:22 +08:00
|
|
|
ereport(ERROR,
|
|
|
|
(errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
|
|
|
|
errmsg("null value not allowed for hstore key")));
|
|
|
|
|
2010-02-26 10:01:40 +08:00
|
|
|
if (in_nulls[i * 2 + 1])
|
2009-10-01 03:50:22 +08:00
|
|
|
{
|
2010-02-26 10:01:40 +08:00
|
|
|
pairs[i].key = VARDATA_ANY(in_datums[i * 2]);
|
2009-10-01 03:50:22 +08:00
|
|
|
pairs[i].val = NULL;
|
2010-02-26 10:01:40 +08:00
|
|
|
pairs[i].keylen = hstoreCheckKeyLen(VARSIZE_ANY_EXHDR(in_datums[i * 2]));
|
2009-10-01 03:50:22 +08:00
|
|
|
pairs[i].vallen = 4;
|
|
|
|
pairs[i].isnull = true;
|
|
|
|
pairs[i].needfree = false;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2010-02-26 10:01:40 +08:00
|
|
|
pairs[i].key = VARDATA_ANY(in_datums[i * 2]);
|
|
|
|
pairs[i].val = VARDATA_ANY(in_datums[i * 2 + 1]);
|
|
|
|
pairs[i].keylen = hstoreCheckKeyLen(VARSIZE_ANY_EXHDR(in_datums[i * 2]));
|
|
|
|
pairs[i].vallen = hstoreCheckValLen(VARSIZE_ANY_EXHDR(in_datums[i * 2 + 1]));
|
2009-10-01 03:50:22 +08:00
|
|
|
pairs[i].isnull = false;
|
|
|
|
pairs[i].needfree = false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
count = hstoreUniquePairs(pairs, count, &buflen);
|
|
|
|
|
|
|
|
out = hstorePairs(pairs, count, buflen);
|
|
|
|
|
|
|
|
PG_RETURN_POINTER(out);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* most of hstore_from_record is shamelessly swiped from record_out */
|
|
|
|
|
|
|
|
/*
|
|
|
|
* structure to cache metadata needed for record I/O
|
|
|
|
*/
|
|
|
|
typedef struct ColumnIOData
|
|
|
|
{
|
|
|
|
Oid column_type;
|
|
|
|
Oid typiofunc;
|
|
|
|
Oid typioparam;
|
|
|
|
FmgrInfo proc;
|
|
|
|
} ColumnIOData;
|
|
|
|
|
|
|
|
typedef struct RecordIOData
|
|
|
|
{
|
|
|
|
Oid record_type;
|
|
|
|
int32 record_typmod;
|
|
|
|
int ncolumns;
|
|
|
|
ColumnIOData columns[1]; /* VARIABLE LENGTH ARRAY */
|
|
|
|
} RecordIOData;
|
|
|
|
|
|
|
|
PG_FUNCTION_INFO_V1(hstore_from_record);
|
|
|
|
Datum hstore_from_record(PG_FUNCTION_ARGS);
|
|
|
|
Datum
|
|
|
|
hstore_from_record(PG_FUNCTION_ARGS)
|
|
|
|
{
|
|
|
|
HeapTupleHeader rec;
|
|
|
|
int4 buflen;
|
|
|
|
HStore *out;
|
2010-02-26 10:01:40 +08:00
|
|
|
Pairs *pairs;
|
2009-10-01 03:50:22 +08:00
|
|
|
Oid tupType;
|
|
|
|
int32 tupTypmod;
|
|
|
|
TupleDesc tupdesc;
|
|
|
|
HeapTupleData tuple;
|
|
|
|
RecordIOData *my_extra;
|
|
|
|
int ncolumns;
|
2010-02-26 10:01:40 +08:00
|
|
|
int i,
|
|
|
|
j;
|
2009-10-01 03:50:22 +08:00
|
|
|
Datum *values;
|
|
|
|
bool *nulls;
|
|
|
|
|
|
|
|
if (PG_ARGISNULL(0))
|
|
|
|
{
|
2010-02-26 10:01:40 +08:00
|
|
|
Oid argtype = get_fn_expr_argtype(fcinfo->flinfo, 0);
|
2009-10-01 03:50:22 +08:00
|
|
|
|
|
|
|
/*
|
2010-02-26 10:01:40 +08:00
|
|
|
* have no tuple to look at, so the only source of type info is the
|
|
|
|
* argtype. The lookup_rowtype_tupdesc call below will error out if we
|
|
|
|
* don't have a known composite type oid here.
|
2009-10-01 03:50:22 +08:00
|
|
|
*/
|
|
|
|
tupType = argtype;
|
|
|
|
tupTypmod = -1;
|
|
|
|
|
|
|
|
rec = NULL;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
rec = PG_GETARG_HEAPTUPLEHEADER(0);
|
|
|
|
|
|
|
|
/* Extract type info from the tuple itself */
|
|
|
|
tupType = HeapTupleHeaderGetTypeId(rec);
|
|
|
|
tupTypmod = HeapTupleHeaderGetTypMod(rec);
|
|
|
|
}
|
|
|
|
|
|
|
|
tupdesc = lookup_rowtype_tupdesc(tupType, tupTypmod);
|
|
|
|
ncolumns = tupdesc->natts;
|
|
|
|
|
|
|
|
/*
|
|
|
|
* We arrange to look up the needed I/O info just once per series of
|
|
|
|
* calls, assuming the record type doesn't change underneath us.
|
|
|
|
*/
|
|
|
|
my_extra = (RecordIOData *) fcinfo->flinfo->fn_extra;
|
|
|
|
if (my_extra == NULL ||
|
|
|
|
my_extra->ncolumns != ncolumns)
|
|
|
|
{
|
|
|
|
fcinfo->flinfo->fn_extra =
|
|
|
|
MemoryContextAlloc(fcinfo->flinfo->fn_mcxt,
|
|
|
|
sizeof(RecordIOData) - sizeof(ColumnIOData)
|
|
|
|
+ ncolumns * sizeof(ColumnIOData));
|
|
|
|
my_extra = (RecordIOData *) fcinfo->flinfo->fn_extra;
|
|
|
|
my_extra->record_type = InvalidOid;
|
|
|
|
my_extra->record_typmod = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (my_extra->record_type != tupType ||
|
|
|
|
my_extra->record_typmod != tupTypmod)
|
|
|
|
{
|
|
|
|
MemSet(my_extra, 0,
|
|
|
|
sizeof(RecordIOData) - sizeof(ColumnIOData)
|
|
|
|
+ ncolumns * sizeof(ColumnIOData));
|
|
|
|
my_extra->record_type = tupType;
|
|
|
|
my_extra->record_typmod = tupTypmod;
|
|
|
|
my_extra->ncolumns = ncolumns;
|
|
|
|
}
|
|
|
|
|
|
|
|
pairs = palloc(ncolumns * sizeof(Pairs));
|
|
|
|
|
|
|
|
if (rec)
|
|
|
|
{
|
|
|
|
/* Build a temporary HeapTuple control structure */
|
|
|
|
tuple.t_len = HeapTupleHeaderGetDatumLength(rec);
|
|
|
|
ItemPointerSetInvalid(&(tuple.t_self));
|
|
|
|
tuple.t_tableOid = InvalidOid;
|
|
|
|
tuple.t_data = rec;
|
|
|
|
|
|
|
|
values = (Datum *) palloc(ncolumns * sizeof(Datum));
|
|
|
|
nulls = (bool *) palloc(ncolumns * sizeof(bool));
|
|
|
|
|
|
|
|
/* Break down the tuple into fields */
|
|
|
|
heap_deform_tuple(&tuple, tupdesc, values, nulls);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
values = NULL;
|
|
|
|
nulls = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
for (i = 0, j = 0; i < ncolumns; ++i)
|
|
|
|
{
|
|
|
|
ColumnIOData *column_info = &my_extra->columns[i];
|
|
|
|
Oid column_type = tupdesc->attrs[i]->atttypid;
|
|
|
|
char *value;
|
|
|
|
|
|
|
|
/* Ignore dropped columns in datatype */
|
|
|
|
if (tupdesc->attrs[i]->attisdropped)
|
|
|
|
continue;
|
|
|
|
|
|
|
|
pairs[j].key = NameStr(tupdesc->attrs[i]->attname);
|
|
|
|
pairs[j].keylen = hstoreCheckKeyLen(strlen(NameStr(tupdesc->attrs[i]->attname)));
|
|
|
|
|
|
|
|
if (!nulls || nulls[i])
|
|
|
|
{
|
|
|
|
pairs[j].val = NULL;
|
|
|
|
pairs[j].vallen = 4;
|
|
|
|
pairs[j].isnull = true;
|
|
|
|
pairs[j].needfree = false;
|
|
|
|
++j;
|
|
|
|
continue;
|
2006-09-06 02:00:58 +08:00
|
|
|
}
|
2009-10-01 03:50:22 +08:00
|
|
|
|
|
|
|
/*
|
|
|
|
* Convert the column value to text
|
|
|
|
*/
|
|
|
|
if (column_info->column_type != column_type)
|
|
|
|
{
|
2010-02-26 10:01:40 +08:00
|
|
|
bool typIsVarlena;
|
2009-10-01 03:50:22 +08:00
|
|
|
|
|
|
|
getTypeOutputInfo(column_type,
|
|
|
|
&column_info->typiofunc,
|
|
|
|
&typIsVarlena);
|
|
|
|
fmgr_info_cxt(column_info->typiofunc, &column_info->proc,
|
|
|
|
fcinfo->flinfo->fn_mcxt);
|
|
|
|
column_info->column_type = column_type;
|
|
|
|
}
|
|
|
|
|
|
|
|
value = OutputFunctionCall(&column_info->proc, values[i]);
|
|
|
|
|
|
|
|
pairs[j].val = value;
|
|
|
|
pairs[j].vallen = hstoreCheckValLen(strlen(value));
|
|
|
|
pairs[j].isnull = false;
|
|
|
|
pairs[j].needfree = false;
|
|
|
|
++j;
|
2006-09-06 02:00:58 +08:00
|
|
|
}
|
|
|
|
|
2009-10-01 03:50:22 +08:00
|
|
|
ncolumns = hstoreUniquePairs(pairs, j, &buflen);
|
|
|
|
|
|
|
|
out = hstorePairs(pairs, ncolumns, buflen);
|
|
|
|
|
|
|
|
ReleaseTupleDesc(tupdesc);
|
|
|
|
|
2006-09-06 02:00:58 +08:00
|
|
|
PG_RETURN_POINTER(out);
|
|
|
|
}
|
|
|
|
|
2009-10-01 03:50:22 +08:00
|
|
|
|
|
|
|
PG_FUNCTION_INFO_V1(hstore_populate_record);
|
|
|
|
Datum hstore_populate_record(PG_FUNCTION_ARGS);
|
|
|
|
Datum
|
|
|
|
hstore_populate_record(PG_FUNCTION_ARGS)
|
|
|
|
{
|
2010-02-26 10:01:40 +08:00
|
|
|
Oid argtype = get_fn_expr_argtype(fcinfo->flinfo, 0);
|
|
|
|
HStore *hs;
|
|
|
|
HEntry *entries;
|
|
|
|
char *ptr;
|
2009-10-01 03:50:22 +08:00
|
|
|
HeapTupleHeader rec;
|
|
|
|
Oid tupType;
|
|
|
|
int32 tupTypmod;
|
|
|
|
TupleDesc tupdesc;
|
|
|
|
HeapTupleData tuple;
|
2010-02-26 10:01:40 +08:00
|
|
|
HeapTuple rettuple;
|
2009-10-01 03:50:22 +08:00
|
|
|
RecordIOData *my_extra;
|
2010-02-26 10:01:40 +08:00
|
|
|
int ncolumns;
|
2009-10-01 03:50:22 +08:00
|
|
|
int i;
|
|
|
|
Datum *values;
|
|
|
|
bool *nulls;
|
|
|
|
|
|
|
|
if (!type_is_rowtype(argtype))
|
|
|
|
ereport(ERROR,
|
|
|
|
(errcode(ERRCODE_DATATYPE_MISMATCH),
|
|
|
|
errmsg("first argument must be a rowtype")));
|
|
|
|
|
|
|
|
if (PG_ARGISNULL(0))
|
|
|
|
{
|
|
|
|
if (PG_ARGISNULL(1))
|
|
|
|
PG_RETURN_NULL();
|
|
|
|
|
|
|
|
rec = NULL;
|
|
|
|
|
|
|
|
/*
|
2010-02-26 10:01:40 +08:00
|
|
|
* have no tuple to look at, so the only source of type info is the
|
|
|
|
* argtype. The lookup_rowtype_tupdesc call below will error out if we
|
|
|
|
* don't have a known composite type oid here.
|
2009-10-01 03:50:22 +08:00
|
|
|
*/
|
|
|
|
tupType = argtype;
|
|
|
|
tupTypmod = -1;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
rec = PG_GETARG_HEAPTUPLEHEADER(0);
|
|
|
|
|
|
|
|
if (PG_ARGISNULL(1))
|
|
|
|
PG_RETURN_POINTER(rec);
|
|
|
|
|
|
|
|
/* Extract type info from the tuple itself */
|
|
|
|
tupType = HeapTupleHeaderGetTypeId(rec);
|
|
|
|
tupTypmod = HeapTupleHeaderGetTypMod(rec);
|
|
|
|
}
|
|
|
|
|
|
|
|
hs = PG_GETARG_HS(1);
|
|
|
|
entries = ARRPTR(hs);
|
|
|
|
ptr = STRPTR(hs);
|
|
|
|
|
|
|
|
/*
|
2010-02-26 10:01:40 +08:00
|
|
|
* if the input hstore is empty, we can only skip the rest if we were
|
|
|
|
* passed in a non-null record, since otherwise there may be issues with
|
|
|
|
* domain nulls.
|
2009-10-01 03:50:22 +08:00
|
|
|
*/
|
|
|
|
|
|
|
|
if (HS_COUNT(hs) == 0 && rec)
|
|
|
|
PG_RETURN_POINTER(rec);
|
|
|
|
|
|
|
|
tupdesc = lookup_rowtype_tupdesc(tupType, tupTypmod);
|
|
|
|
ncolumns = tupdesc->natts;
|
|
|
|
|
|
|
|
if (rec)
|
|
|
|
{
|
|
|
|
/* Build a temporary HeapTuple control structure */
|
|
|
|
tuple.t_len = HeapTupleHeaderGetDatumLength(rec);
|
|
|
|
ItemPointerSetInvalid(&(tuple.t_self));
|
|
|
|
tuple.t_tableOid = InvalidOid;
|
|
|
|
tuple.t_data = rec;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* We arrange to look up the needed I/O info just once per series of
|
|
|
|
* calls, assuming the record type doesn't change underneath us.
|
|
|
|
*/
|
|
|
|
my_extra = (RecordIOData *) fcinfo->flinfo->fn_extra;
|
|
|
|
if (my_extra == NULL ||
|
|
|
|
my_extra->ncolumns != ncolumns)
|
|
|
|
{
|
|
|
|
fcinfo->flinfo->fn_extra =
|
|
|
|
MemoryContextAlloc(fcinfo->flinfo->fn_mcxt,
|
|
|
|
sizeof(RecordIOData) - sizeof(ColumnIOData)
|
|
|
|
+ ncolumns * sizeof(ColumnIOData));
|
|
|
|
my_extra = (RecordIOData *) fcinfo->flinfo->fn_extra;
|
|
|
|
my_extra->record_type = InvalidOid;
|
|
|
|
my_extra->record_typmod = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (my_extra->record_type != tupType ||
|
|
|
|
my_extra->record_typmod != tupTypmod)
|
|
|
|
{
|
|
|
|
MemSet(my_extra, 0,
|
|
|
|
sizeof(RecordIOData) - sizeof(ColumnIOData)
|
|
|
|
+ ncolumns * sizeof(ColumnIOData));
|
|
|
|
my_extra->record_type = tupType;
|
|
|
|
my_extra->record_typmod = tupTypmod;
|
|
|
|
my_extra->ncolumns = ncolumns;
|
|
|
|
}
|
|
|
|
|
|
|
|
values = (Datum *) palloc(ncolumns * sizeof(Datum));
|
|
|
|
nulls = (bool *) palloc(ncolumns * sizeof(bool));
|
|
|
|
|
|
|
|
if (rec)
|
|
|
|
{
|
|
|
|
/* Break down the tuple into fields */
|
|
|
|
heap_deform_tuple(&tuple, tupdesc, values, nulls);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
for (i = 0; i < ncolumns; ++i)
|
|
|
|
{
|
|
|
|
values[i] = (Datum) 0;
|
|
|
|
nulls[i] = true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
for (i = 0; i < ncolumns; ++i)
|
|
|
|
{
|
|
|
|
ColumnIOData *column_info = &my_extra->columns[i];
|
|
|
|
Oid column_type = tupdesc->attrs[i]->atttypid;
|
|
|
|
char *value;
|
2010-02-26 10:01:40 +08:00
|
|
|
int idx;
|
|
|
|
int vallen;
|
2009-10-01 03:50:22 +08:00
|
|
|
|
|
|
|
/* Ignore dropped columns in datatype */
|
|
|
|
if (tupdesc->attrs[i]->attisdropped)
|
|
|
|
{
|
|
|
|
nulls[i] = true;
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
idx = hstoreFindKey(hs, 0,
|
|
|
|
NameStr(tupdesc->attrs[i]->attname),
|
|
|
|
strlen(NameStr(tupdesc->attrs[i]->attname)));
|
2010-02-26 10:01:40 +08:00
|
|
|
|
2009-10-01 03:50:22 +08:00
|
|
|
/*
|
2010-02-26 10:01:40 +08:00
|
|
|
* we can't just skip here if the key wasn't found since we might have
|
|
|
|
* a domain to deal with. If we were passed in a non-null record
|
|
|
|
* datum, we assume that the existing values are valid (if they're
|
|
|
|
* not, then it's not our fault), but if we were passed in a null,
|
|
|
|
* then every field which we don't populate needs to be run through
|
|
|
|
* the input function just in case it's a domain type.
|
2009-10-01 03:50:22 +08:00
|
|
|
*/
|
|
|
|
if (idx < 0 && rec)
|
|
|
|
continue;
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Prepare to convert the column value from text
|
|
|
|
*/
|
|
|
|
if (column_info->column_type != column_type)
|
|
|
|
{
|
|
|
|
getTypeInputInfo(column_type,
|
|
|
|
&column_info->typiofunc,
|
|
|
|
&column_info->typioparam);
|
|
|
|
fmgr_info_cxt(column_info->typiofunc, &column_info->proc,
|
|
|
|
fcinfo->flinfo->fn_mcxt);
|
|
|
|
column_info->column_type = column_type;
|
|
|
|
}
|
|
|
|
|
2010-02-26 10:01:40 +08:00
|
|
|
if (idx < 0 || HS_VALISNULL(entries, idx))
|
2009-10-01 03:50:22 +08:00
|
|
|
{
|
|
|
|
/*
|
2010-02-26 10:01:40 +08:00
|
|
|
* need InputFunctionCall to happen even for nulls, so that domain
|
|
|
|
* checks are done
|
2009-10-01 03:50:22 +08:00
|
|
|
*/
|
|
|
|
values[i] = InputFunctionCall(&column_info->proc, NULL,
|
|
|
|
column_info->typioparam,
|
|
|
|
tupdesc->attrs[i]->atttypmod);
|
|
|
|
nulls[i] = true;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2010-02-26 10:01:40 +08:00
|
|
|
vallen = HS_VALLEN(entries, idx);
|
2009-10-01 03:50:22 +08:00
|
|
|
value = palloc(1 + vallen);
|
2010-02-26 10:01:40 +08:00
|
|
|
memcpy(value, HS_VAL(entries, ptr, idx), vallen);
|
2009-10-01 03:50:22 +08:00
|
|
|
value[vallen] = 0;
|
|
|
|
|
|
|
|
values[i] = InputFunctionCall(&column_info->proc, value,
|
|
|
|
column_info->typioparam,
|
|
|
|
tupdesc->attrs[i]->atttypmod);
|
|
|
|
nulls[i] = false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
rettuple = heap_form_tuple(tupdesc, values, nulls);
|
|
|
|
|
|
|
|
ReleaseTupleDesc(tupdesc);
|
|
|
|
|
|
|
|
PG_RETURN_DATUM(HeapTupleGetDatum(rettuple));
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2006-10-04 08:30:14 +08:00
|
|
|
static char *
|
|
|
|
cpw(char *dst, char *src, int len)
|
|
|
|
{
|
|
|
|
char *ptr = src;
|
2006-09-06 02:00:58 +08:00
|
|
|
|
2006-10-04 08:30:14 +08:00
|
|
|
while (ptr - src < len)
|
|
|
|
{
|
|
|
|
if (*ptr == '"' || *ptr == '\\')
|
|
|
|
*dst++ = '\\';
|
2006-09-06 02:00:58 +08:00
|
|
|
*dst++ = *ptr++;
|
|
|
|
}
|
|
|
|
return dst;
|
|
|
|
}
|
|
|
|
|
|
|
|
PG_FUNCTION_INFO_V1(hstore_out);
|
2006-10-04 08:30:14 +08:00
|
|
|
Datum hstore_out(PG_FUNCTION_ARGS);
|
2006-09-06 02:00:58 +08:00
|
|
|
Datum
|
2006-10-04 08:30:14 +08:00
|
|
|
hstore_out(PG_FUNCTION_ARGS)
|
|
|
|
{
|
|
|
|
HStore *in = PG_GETARG_HS(0);
|
|
|
|
int buflen,
|
2009-10-01 03:50:22 +08:00
|
|
|
i;
|
2010-02-26 10:01:40 +08:00
|
|
|
int count = HS_COUNT(in);
|
2006-10-04 08:30:14 +08:00
|
|
|
char *out,
|
|
|
|
*ptr;
|
|
|
|
char *base = STRPTR(in);
|
|
|
|
HEntry *entries = ARRPTR(in);
|
|
|
|
|
2009-10-01 03:50:22 +08:00
|
|
|
if (count == 0)
|
2006-10-04 08:30:14 +08:00
|
|
|
{
|
|
|
|
out = palloc(1);
|
|
|
|
*out = '\0';
|
2006-09-06 02:00:58 +08:00
|
|
|
PG_RETURN_CSTRING(out);
|
|
|
|
}
|
|
|
|
|
2009-10-01 03:50:22 +08:00
|
|
|
buflen = 0;
|
|
|
|
|
|
|
|
/*
|
2010-02-26 10:01:40 +08:00
|
|
|
* this loop overestimates due to pessimistic assumptions about escaping,
|
|
|
|
* so very large hstore values can't be output. this could be fixed, but
|
|
|
|
* many other data types probably have the same issue. This replaced code
|
|
|
|
* that used the original varlena size for calculations, which was wrong
|
|
|
|
* in some subtle ways.
|
2009-10-01 03:50:22 +08:00
|
|
|
*/
|
2009-04-03 01:57:05 +08:00
|
|
|
|
2009-10-01 03:50:22 +08:00
|
|
|
for (i = 0; i < count; i++)
|
|
|
|
{
|
|
|
|
/* include "" and => and comma-space */
|
2010-02-26 10:01:40 +08:00
|
|
|
buflen += 6 + 2 * HS_KEYLEN(entries, i);
|
2009-10-01 03:50:22 +08:00
|
|
|
/* include "" only if nonnull */
|
2010-02-26 10:01:40 +08:00
|
|
|
buflen += 2 + (HS_VALISNULL(entries, i)
|
2009-10-01 03:50:22 +08:00
|
|
|
? 2
|
2010-02-26 10:01:40 +08:00
|
|
|
: 2 * HS_VALLEN(entries, i));
|
2009-10-01 03:50:22 +08:00
|
|
|
}
|
2006-10-04 08:30:14 +08:00
|
|
|
|
|
|
|
out = ptr = palloc(buflen);
|
2009-10-01 03:50:22 +08:00
|
|
|
|
|
|
|
for (i = 0; i < count; i++)
|
2006-10-04 08:30:14 +08:00
|
|
|
{
|
|
|
|
*ptr++ = '"';
|
2010-02-26 10:01:40 +08:00
|
|
|
ptr = cpw(ptr, HS_KEY(entries, base, i), HS_KEYLEN(entries, i));
|
2006-10-04 08:30:14 +08:00
|
|
|
*ptr++ = '"';
|
|
|
|
*ptr++ = '=';
|
|
|
|
*ptr++ = '>';
|
2010-02-26 10:01:40 +08:00
|
|
|
if (HS_VALISNULL(entries, i))
|
2006-10-04 08:30:14 +08:00
|
|
|
{
|
|
|
|
*ptr++ = 'N';
|
|
|
|
*ptr++ = 'U';
|
|
|
|
*ptr++ = 'L';
|
|
|
|
*ptr++ = 'L';
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
*ptr++ = '"';
|
2010-02-26 10:01:40 +08:00
|
|
|
ptr = cpw(ptr, HS_VAL(entries, base, i), HS_VALLEN(entries, i));
|
2006-10-04 08:30:14 +08:00
|
|
|
*ptr++ = '"';
|
2006-09-06 02:00:58 +08:00
|
|
|
}
|
|
|
|
|
2009-10-01 03:50:22 +08:00
|
|
|
if (i + 1 != count)
|
2006-10-04 08:30:14 +08:00
|
|
|
{
|
|
|
|
*ptr++ = ',';
|
|
|
|
*ptr++ = ' ';
|
2006-09-06 02:00:58 +08:00
|
|
|
}
|
|
|
|
}
|
2006-10-04 08:30:14 +08:00
|
|
|
*ptr = '\0';
|
2006-09-06 02:00:58 +08:00
|
|
|
|
|
|
|
PG_RETURN_CSTRING(out);
|
|
|
|
}
|
2009-10-01 03:50:22 +08:00
|
|
|
|
|
|
|
|
|
|
|
PG_FUNCTION_INFO_V1(hstore_send);
|
|
|
|
Datum hstore_send(PG_FUNCTION_ARGS);
|
|
|
|
Datum
|
|
|
|
hstore_send(PG_FUNCTION_ARGS)
|
|
|
|
{
|
|
|
|
HStore *in = PG_GETARG_HS(0);
|
2010-02-26 10:01:40 +08:00
|
|
|
int i;
|
|
|
|
int count = HS_COUNT(in);
|
2009-10-01 03:50:22 +08:00
|
|
|
char *base = STRPTR(in);
|
|
|
|
HEntry *entries = ARRPTR(in);
|
|
|
|
StringInfoData buf;
|
|
|
|
|
|
|
|
pq_begintypsend(&buf);
|
|
|
|
|
|
|
|
pq_sendint(&buf, count, 4);
|
|
|
|
|
|
|
|
for (i = 0; i < count; i++)
|
|
|
|
{
|
2010-02-26 10:01:40 +08:00
|
|
|
int32 keylen = HS_KEYLEN(entries, i);
|
|
|
|
|
2009-10-01 03:50:22 +08:00
|
|
|
pq_sendint(&buf, keylen, 4);
|
2010-02-26 10:01:40 +08:00
|
|
|
pq_sendtext(&buf, HS_KEY(entries, base, i), keylen);
|
|
|
|
if (HS_VALISNULL(entries, i))
|
2009-10-01 03:50:22 +08:00
|
|
|
{
|
|
|
|
pq_sendint(&buf, -1, 4);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2010-02-26 10:01:40 +08:00
|
|
|
int32 vallen = HS_VALLEN(entries, i);
|
|
|
|
|
2009-10-01 03:50:22 +08:00
|
|
|
pq_sendint(&buf, vallen, 4);
|
2010-02-26 10:01:40 +08:00
|
|
|
pq_sendtext(&buf, HS_VAL(entries, base, i), vallen);
|
2009-10-01 03:50:22 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
PG_RETURN_BYTEA_P(pq_endtypsend(&buf));
|
|
|
|
}
|