Как вернуть массив массивов из C-фунции для Postgres?
Форум — Development
Возвращать массив (1) я научился, а как положить его внутрь другого (2) не получается.
1: CREATE TYPE libc_collect_item AS (goods_id bigint, priority integer);
2: CREATE TYPE libc_collect AS (collect_number integer, collect_items libc_collect_item[]);
Реализация функции возвращающей первый тип:
//CREATE OR REPLACE FUNCTION f_get_collects() RETURNS SETOF libc_collect_item[] AS 'lib', 'f_get_collects' LANGUAGE C STRICT STABLE;
PG_FUNCTION_INFO_V1(f_get_collects);
Datum
f_get_collects(PG_FUNCTION_ARGS)
{
//....
TupleDesc tupleDesc;
Oid outputTypeId;
get_call_result_type(fcinfo, &outputTypeId, &tupleDesc);
Oid outputBaseTypeId = get_base_element_type(outputTypeId);
Oid outputBaseTypRelId = get_typ_typrelid(outputBaseTypeId);
int16 typlen;
bool typbyval;
char typalign;
get_typlenbyvalalign(outputBaseTypeId, &typlen, &typbyval, &typalign);
const char* attributeNameOne = get_relid_attribute_name(outputBaseTypRelId, 1);
const char* attributeNameTwo = get_relid_attribute_name(outputBaseTypRelId, 2);
Oid attributeOneOid = get_atttype(outputBaseTypRelId, 1);
Oid attributeTwoOid = get_atttype(outputBaseTypRelId, 2);
int32 attributeOneTypMod = get_atttypmod(outputBaseTypRelId, 1);
int32 attributeTwoTypMod = get_atttypmod(outputBaseTypRelId, 2);
TupleDesc my_tupleDesc = CreateTemplateTupleDesc(2, false);
TupleDescInitEntry(my_tupleDesc, 1, attributeNameOne, attributeOneOid, attributeOneTypMod, 0);
TupleDescInitEntry(my_tupleDesc, 2, attributeNameTwo, attributeTwoOid, attributeTwoTypMod, 0);
my_tupleDesc = BlessTupleDesc(my_tupleDesc);
vector<OutCollection> collectionList;
OutCollection item(100002, 1);
collectionList.push_back(item);
OutCollection item2(100003, 2);
collectionList.push_back(item2);
Datum* pResultDatums = (Datum*)palloc(sizeof(Datum) * (int)collectionList.size());
for(int i =0; i < (int)collectionList.size(); i++){
Datum* valueItem = (Datum*) palloc(sizeof(Datum) *2);
valueItem[0] = collectionList[i].orderItemId;
valueItem[1] = collectionList[i].priority;
bool* isNullItem = (bool*) palloc(sizeof(bool) * 2);
isNullItem[0] = false;
isNullItem[1] = false;
HeapTuple resultHeapTupleItem = heap_form_tuple(my_tupleDesc, valueItem, isNullItem);
my_tupleDesc = BlessTupleDesc(my_tupleDesc);
pResultDatums[i] = HeapTupleGetDatum(resultHeapTupleItem);
}
ArrayType* pResult = construct_array(pResultDatums, collectionList.size(), outputBaseTypeId, typlen, typbyval, typalign);
PG_RETURN_ARRAYTYPE_P(pResult);
}
f_sys_collects_generate_single_simple
---------------------------------------
{ "(1,1)","(10,2)" }
(1 строка)
Теперь попытаемся затолкать массив в массив, для чего сначала сделаем вкладываемый:
ArrayType* createArray(/*FunctionCallInfo fcinfo, */vector collectionList)
{
Oid outputTypeId;
//Типы пока захардкожены
Oid outputBaseTypeId = 106553;
Oid outputBaseTypRelId = 106551;
int16 typlen;
bool typbyval;
char typalign;
get_typlenbyvalalign(outputBaseTypeId, &typlen, &typbyval, &typalign);
const char* attributeNameOne = "goods_id"; get_relid_attribute_name(outputBaseTypRelId, 1);
const char* attributeNameTwo = "priority"; get_relid_attribute_name(outputBaseTypRelId, 2);
ereport(INFO, (errmsg("Attribute names: %s %s ", attributeNameOne, attributeNameTwo)));
Oid attributeOneOid = 20; get_atttype(outputBaseTypRelId, 1);
Oid attributeTwoOid = 23; get_atttype(outputBaseTypRelId, 2);
int32 attributeOneTypMod = get_atttypmod(outputBaseTypRelId, 1);
int32 attributeTwoTypMod = get_atttypmod(outputBaseTypRelId, 2);
TupleDesc my_tupleDesc = CreateTemplateTupleDesc(2, false);
TupleDescInitEntry(my_tupleDesc, 1, attributeNameOne, attributeOneOid, attributeOneTypMod, 0);
TupleDescInitEntry(my_tupleDesc, 2, attributeNameTwo, attributeTwoOid, attributeTwoTypMod, 0);
my_tupleDesc = BlessTupleDesc(my_tupleDesc);
Datum* pResultDatums = (Datum*)palloc(sizeof(Datum) * (int)collectionList.size());
return pResult;
}
Теперь птаюсь воткнуть массив в элемент массива, предварительно его разобрав, чтобы получить Datum
//CREATE OR REPLACE FUNCTION f_sys_collects_generate_single_simple2( goods_item libc_goods_item[], point_item libc_point_item[] ) RETURNS SETOF libc_collect[] AS 'lib', 'f_sys_collects_generate_single_simple2' LANGUAGE C STRICT STABLE;
PG_FUNCTION_INFO_V1(f_sys_collects_generate_single_simple2);
Datum
f_sys_collects_generate_single_simple2(PG_FUNCTION_ARGS)
{
vector<OutCollection> collectionList;
OutCollection item(100002, 1);
collectionList.push_back(item);
OutCollection item2(100003, 2);
collectionList.push_back(item2);
TupleDesc tupleDesc;
Oid outputTypeId;
get_call_result_type(fcinfo, &outputTypeId, &tupleDesc);
Oid outputBaseTypeId = get_base_element_type(outputTypeId);
Oid outputBaseTypRelId = get_typ_typrelid(outputBaseTypeId);
int16 typlen;
bool typbyval;
char typalign;
get_typlenbyvalalign(outputBaseTypeId, &typlen, &typbyval, &typalign);
const char* attributeNameOne = get_relid_attribute_name(outputBaseTypRelId, 1);
const char* attributeNameTwo = get_relid_attribute_name(outputBaseTypRelId, 2);
Oid attributeOneOid = get_atttype(outputBaseTypRelId, 1);
Oid attributeTwoOid = get_atttype(outputBaseTypRelId, 2);
int32 attributeOneTypMod = get_atttypmod(outputBaseTypRelId, 1);
int32 attributeTwoTypMod = get_atttypmod(outputBaseTypRelId, 2);
ereport(INFO, (errmsg("Attribute type mods: %i %i ", attributeOneTypMod, attributeTwoTypMod)));
TupleDesc my_tupleDesc = CreateTemplateTupleDesc(2, false);
TupleDescInitEntry(my_tupleDesc, 1, attributeNameOne, attributeOneOid, attributeOneTypMod, 0);
TupleDescInitEntry(my_tupleDesc, 2, attributeNameTwo, attributeTwoOid, attributeTwoTypMod, 0);
my_tupleDesc = BlessTupleDesc(my_tupleDesc);
Datum* pResultDatums = (Datum*)palloc(sizeof(Datum) * (int)collectionList.size());
for(int i =0; i < (int)collectionList.size(); i++){
Datum* valueItem = (Datum*) palloc(sizeof(Datum) *2);
valueItem[0] = i;
ArrayType* pResult2 = createArray(/*fcinfo, */collectionList);
Datum* datums;
Oid elemType = 106553;
int16 elemWidth;
bool elemTypeByVal;
char elemAlignmentCode;
bool* nulls;
int count;
get_typlenbyvalalign(elemType, &elemWidth, &elemTypeByVal, &elemAlignmentCode);
ereport(INFO, (errmsg("elemType: %i", elemType)));
ereport(INFO, (errmsg("elemWidth: %i", elemWidth)));
ereport(INFO, (errmsg("elemTypeByVal: %i", elemTypeByVal)));
deconstruct_array(pResult2, elemType, elemWidth, elemTypeByVal, elemAlignmentCode, &datums, &nulls, &count);
ereport(INFO, (errmsg("count: %i", count)));
valueItem[1] = *datums;
/*
// На всякий случай проверяем, там все в порядке.
for (int i = 0; i < count; i++)
{
if (!nulls[i])
{
HeapTupleHeader lt = DatumGetHeapTupleHeader(datums[i]);
//field = DatumGetInt16(GetAttributeByNum(lt, 1, &isNull));
int64 field1 = DatumGetInt64(GetAttributeByNum(lt, 1, &isNull));
if (!isNull)
ereport(INFO, (errmsg("elemTypeByVal: %llu", field1)));
int field2 = DatumGetInt16(GetAttributeByNum(lt, 2, &isNull));
if (!isNull)
ereport(INFO, (errmsg("elemTypeByVal: %i", field2)));
}
}
*/
bool* isNullItem = (bool*) palloc(sizeof(bool) * 2);
isNullItem[0] = false;
isNullItem[1] = false;
HeapTuple resultHeapTupleItem = heap_form_tuple(my_tupleDesc, valueItem, isNullItem);
my_tupleDesc = BlessTupleDesc(my_tupleDesc);
pResultDatums[i] = HeapTupleGetDatum(resultHeapTupleItem);
ereport(INFO, (errmsg("outputBaseTypeId: %d", outputBaseTypeId)));
}
ArrayType* pResult = construct_array(pResultDatums, collectionList.size(), outputBaseTypeId, typlen, typbyval, typalign);
PG_RETURN_ARRAYTYPE_P(pResult);
}
Получаю:
......
ИНФОРМАЦИЯ: elemType: 106553
ИНФОРМАЦИЯ: elemWidth: -1
ИНФОРМАЦИЯ: elemTypeByVal: 0
ИНФОРМАЦИЯ: count: 2
ИНФОРМАЦИЯ: outputBaseTypeId: 106556
ИНФОРМАЦИЯ: Object type of the return type: 0
ИНФОРМАЦИЯ: Object base type of the return type: 106553
ИНФОРМАЦИЯ: Object base type rel id of the return type: 106551
ИНФОРМАЦИЯ: Attribute names: goods_id priority
ИНФОРМАЦИЯ: Attribute oids: 20 23
ИНФОРМАЦИЯ: Attribute type mods: -1 -1
ИНФОРМАЦИЯ: elemType: 106553
ИНФОРМАЦИЯ: elemWidth: -1
ИНФОРМАЦИЯ: elemTypeByVal: 0
ИНФОРМАЦИЯ: count: 2
ИНФОРМАЦИЯ: outputBaseTypeId: 106556
ОШИБКА: cache lookup failed for type 4294967295
Понятное дело, что этого типа в базе нет и быть не может.
Если в f_sys_collects_generate_single_simple2 поменять на:
isNullItem[1] = true;
...
ИНФОРМАЦИЯ: elemTypeByVal: 0
ИНФОРМАЦИЯ: count: 2
ИНФОРМАЦИЯ: outputBaseTypeId: 106556
f_sys_collects_generate_single_simple2
----------------------------------------
{ "(0,)","(1,)" }
Судя по типу 4294967295 что-то где-то сползает.