/*
* int2vectorin - converts "num num ..." to internal form
*/
Datum
int2vectorin(PG_FUNCTION_ARGS)
{
char *intString = PG_GETARG_CSTRING(0);
int2vector *result;
int n;
result = (int2vector *) palloc0(Int2VectorSize(FUNC_MAX_ARGS));
for (n = 0; *intString && n < FUNC_MAX_ARGS; n++)
{
if (sscanf(intString, "%hd", &result->values[n]) != 1)
break;
while (*intString && isspace((unsigned char) *intString))
intString++;
while (*intString && !isspace((unsigned char) *intString))
intString++;
}
while (*intString && isspace((unsigned char) *intString))
intString++;
if (*intString)
ereport(ERROR,
(errcode(ERRCODE_INVALID_PARAMETER_VALUE),
errmsg("int2vector has too many elements")));
result->size = Int2VectorSize(n);
result->ndim = 1;
result->dataoffset = 0; /* never any nulls */
result->elemtype = INT2OID;
result->dim1 = n;
result->lbound1 = 0;
PG_RETURN_POINTER(result);
}
/*
* construct int2_vector_s given a raw array of int2s
*
* If int2s is NULL then caller must fill values[] afterward
*/
int2_vector_s *buildint2vector(const int2 * int2s, int n)
{
int2_vector_s *result;
result = (int2_vector_s*) pzalloc(Int2VectorSize(n));
if (n > 0 && int2s)
memcpy(result->values, int2s, n * sizeof(int2));
/*
* Attach standard array header. For historical reasons, we set the index
* lower bound to 0 not 1.
*/
VLA_SET_SZ_STND(result, Int2VectorSize(n));
result->ndim = 1;
result->dataoffset = 0; /* never any nulls */
result->elemtype = INT2OID;
result->dim1 = n;
result->lbound1 = 0;
return result;
}
/*
* int2vectorin - converts "num num ..." to internal form
*/
datum_t int2vectorin(PG_FUNC_ARGS)
{
char *intString = ARG_CSTRING(0);
int2_vector_s *result;
int n;
result = (int2_vector_s*) pzalloc(Int2VectorSize(FUNC_MAX_ARGS));
for (n = 0; *intString && n < FUNC_MAX_ARGS; n++) {
while (*intString && isspace((unsigned char)*intString))
intString++;
if (*intString == '\0')
break;
result->values[n] = pg_atoi(intString, sizeof(int16), ' ');
while (*intString && !isspace((unsigned char)*intString))
intString++;
}
while (*intString && isspace((unsigned char)*intString))
intString++;
if (*intString) {
ereport(ERROR, (
errcode(E_INVALID_PARAMETER_VALUE),
errmsg("int2vector has too many elements")));
}
VLA_SET_SZ_STND(result, Int2VectorSize(n));
result->ndim = 1;
result->dataoffset = 0; /* never any nulls */
result->elemtype = INT2OID;
result->dim1 = n;
result->lbound1 = 0;
RET_POINTER(result);
}
Relation DirectOpen_Open(
DirectOpen *direct,
Oid relationId,
Oid tablespace,
Oid database,
Oid relfilenode,
FormData_pg_class *pgClass,
FormData_pg_attribute *attrArray,
FormData_pg_am *pgAm,
FormData_pg_index *pgIndex,
int2 *indKeyArray,
Oid *indClassArray,
bool relHasOid)
{
int natts;
int i;
Assert(pgClass != NULL);
natts = pgClass->relnatts;
if (relationId == -1)
relationId = pgClass->relfilenode; // Assume it is ok to use the relfilenode as the relationId in our limited usage.
if (relfilenode == -1)
relfilenode = pgClass->relfilenode;
if (!direct->isInit)
{
/*
* Lots of Hard-coded construction of the gp_persistent* RelationS and
* dependent objects like tuple descriptors, etc.
*/
direct->relationData.rd_refcnt = 0;
direct->relationData.rd_isvalid = true;
direct->relationData.rd_id = relationId;
direct->relationData.rd_rel = pgClass;
if (pgIndex != NULL)
{
int pgIndexFixedLen = offsetof(FormData_pg_index, indkey);
int indKeyVectorLen = Int2VectorSize(natts);
int2vector *indKeyVector;
oidvector *indClassVector;
uint16 amstrategies;
uint16 amsupport;
Oid *operator;
RegProcedure *support;
FmgrInfo *supportinfo;
Assert(pgAm != NULL);
Assert(indKeyArray != NULL);
Assert(indClassArray != NULL);
/*
* Allocate Formdata_pg_index with fields through indkey
* where indkey is a variable length int2vector with indKeyArray values.
*/
direct->relationData.rd_index =
(FormData_pg_index*)palloc(
pgIndexFixedLen + indKeyVectorLen);
memcpy(direct->relationData.rd_index, pgIndex, pgIndexFixedLen);
indKeyVector = buildint2vector(
indKeyArray,
natts);
memcpy(
&direct->relationData.rd_index->indkey,
indKeyVector,
indKeyVectorLen);
pfree(indKeyVector);
direct->relationData.rd_am = pgAm;
amstrategies = pgAm->amstrategies;
amsupport = pgAm->amsupport;
direct->relationData.rd_indexcxt = TopMemoryContext;
/*
* Allocate arrays to hold data
*/
direct->relationData.rd_aminfo = (RelationAmInfo *)
MemoryContextAllocZero(TopMemoryContext, sizeof(RelationAmInfo));
direct->relationData.rd_opfamily = (Oid *)
MemoryContextAllocZero(TopMemoryContext, natts * sizeof(Oid));
direct->relationData.rd_opcintype = (Oid *)
MemoryContextAllocZero(TopMemoryContext, natts * sizeof(Oid));
if (amstrategies > 0)
operator = (Oid *)
MemoryContextAllocZero(TopMemoryContext,
natts * amstrategies * sizeof(Oid));
else
operator = NULL;
if (amsupport > 0)
{
int nsupport = natts * amsupport;
support = (RegProcedure *)
MemoryContextAllocZero(TopMemoryContext, nsupport * sizeof(RegProcedure));
supportinfo = (FmgrInfo *)
MemoryContextAllocZero(TopMemoryContext, nsupport * sizeof(FmgrInfo));
}
else
{
support = NULL;
supportinfo = NULL;
}
direct->relationData.rd_operator = operator;
direct->relationData.rd_support = support;
direct->relationData.rd_supportinfo = supportinfo;
direct->relationData.rd_indoption = (int16 *)
MemoryContextAllocZero(TopMemoryContext, natts * sizeof(int16));
/*
* Create oidvector in rd_indclass with values from indClassArray.
*/
indClassVector = buildoidvector(indClassArray, natts);
/*
* Fill the operator and support procedure OID arrays. (aminfo and
* supportinfo are left as zeroes, and are filled on-the-fly when used)
*/
IndexSupportInitialize(indClassVector,
operator, support,
direct->relationData.rd_opfamily,
direct->relationData.rd_opcintype,
amstrategies, amsupport, natts);
/*
* expressions and predicate cache will be filled later.
*/
direct->relationData.rd_indexprs = NIL;
direct->relationData.rd_indpred = NIL;
direct->relationData.rd_amcache = NULL;
}
// Not much in terms of contraints.
direct->constrData.has_not_null = true;
/*
* Setup tuple descriptor for columns.
*/
direct->descData.natts = pgClass->relnatts;
// Make the array of pointers.
direct->descData.attrs =
(Form_pg_attribute*)
MemoryContextAllocZero(
TopMemoryContext,
sizeof(Form_pg_attribute*) * pgClass->relnatts);
for (i = 0; i < pgClass->relnatts; i++)
{
direct->descData.attrs[i] =
(Form_pg_attribute)
MemoryContextAllocZero(
TopMemoryContext,
sizeof(FormData_pg_attribute));
memcpy(direct->descData.attrs[i], &(attrArray[i]), sizeof(FormData_pg_attribute));
// Patch up relation id.
direct->descData.attrs[i]->attrelid = relationId;
}
direct->descData.constr = &direct->constrData;
direct->descData.tdtypeid = pgClass->reltype;
direct->descData.tdtypmod = -1;
direct->descData.tdqdtypmod = -1;
direct->descData.tdhasoid = relHasOid;
direct->descData.tdrefcount = 1;
direct->relationData.rd_att = &direct->descData;
direct->pgStat.t_id = relationId;
direct->pgStat.t_shared = 1;
direct->relationData.pgstat_info = &direct->pgStat;
direct->isInit = true;
}
// UNDONE: Should verify for NON-SHARED relations we don't open relations in different databases / or
// UNDONE: open different relations in same database at same time !!!
direct->relationData.rd_node.spcNode = tablespace;
direct->relationData.rd_node.dbNode = database;
direct->relationData.rd_node.relNode = relfilenode;
direct->relationData.rd_targblock = InvalidBlockNumber;
for (i = 0; i < direct->relationData.rd_rel->relnatts; i++)
{
Assert(direct->descData.attrs[i] != NULL);
// Patch up relation id.
direct->descData.attrs[i]->attrelid = direct->relationData.rd_id;
}
direct->relationData.rd_refcnt++;
RelationOpenSmgr(&direct->relationData);
return &direct->relationData;
}