LINUX.ORG.RU

Сообщения Palmman

 

Как вернуть массив массивов из 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 что-то где-то сползает.

 , ,

Palmman
()

RSS подписка на новые темы