为PostgreSQL实现C扩展函数 - 我该怎么做？(在C/PostgreSQL之间传递数据)

Question

我正在为PostgreSQL编写一个C扩展(v 8.4).我目前仍然坚持如何将PostgreSQL中的列数据传递给我的C函数.我也有一个内存所有权的问题,因为PostgreSQL似乎在场景中做了很多内存管理.

如果有人能帮助我"加入点",我将不胜感激,以获得一个基本的骨架代码,我可以从中构建库.

这是我到目前为止:

/*******************************************************/
/*                  C header file                      */
/*******************************************************/
typedef struct _myarray
{
    double *data;
    size_t len;
} MyArray;


MyArray * NEW_MyArray(const size_t len);
void Destroy_MyArray(MyArray * arr);
size_t NumElements_MyArray(MyArray * arr);      /* trivial function returns number of elements */
MyArray * NotTrivial_MyArray(MyArray * arr);    /* non trivial function returns MyArray (a float8[] in PG) */
double HeapFunc_MyArray(MyArray * arr);         /* allocs from heap */


/*******************************************************/
/*                   C Source file                     */
/*******************************************************/

/* utility conversion funcs */
/* How do I convert from the structure returned by array_agg to float8[] (or int4[] ?) */


MyArray * NEW_MyArray(const size_t len){
    /* Do I use palloc0() or calloc() here ? */
}

void Destroy_MyArray(MyArray * arr){
    /* Do I use pfree() or free() here ? */
}

size_t NumElements_MyArray(MyArray * arr){
    assert(arr != 0);
    return arr->len;
}

MyArray * NotTrivial_MyArray(MyArray * arr){
    assert(arr != 0);
    MyArray * ptr = NEW_MyArray(arr->len);
    return ptr;
}

double HeapFunc_MyArray(MyArray * arr){
    /* Create temporary variables on heap (use palloc0() or calloc()?) */
    /* Cleanup temp variables (use pfree() or free() ? */
    return 42/1.0;
}



/*******************************************************/
/* PostgreSQL wrapper funcs implementation source file */
/*******************************************************/

/* Prototypes */
PG_FUNCTION_INFO_V1(test_num_elements);
PG_FUNCTION_INFO_V1(test_not_trivial);
PG_FUNCTION_INFO_V1(test_heapfunc);

Datum test_num_elements(PG_FUNCTION_ARGS);
Datum test_not_trivial(PG_FUNCTION_ARGS);
Datum test_heapfunc(PG_FUNCTION_ARGS);


Datum
test_num_elements(PG_FUNCTION_ARGS)
{
    /* Convert data returned by array_agg() into MyArray * (how?) */
    /* invoke NumElements_MyArray() */
    /* Do I free temporary MyArray * ptr or will PG clean up 
       - if I have to clean up (like I suspect), do I use pfree() or free() ?*/
    PG_RETURN_INT32(result);
}

Datum
test_not_trivial(PG_FUNCTION_ARGS)
{
    /* Ditto, as above */
    PG_RETURN_POINTER(/* utility function to convert MyArray* to float8[] equiv for PG (how) */); 
}

Datum
test_heapfunc(PG_FUNCTION_ARGS)
{
    /* Ditto, as above */
    PG_RETURN_FLOAT8(result);
}


-- SQL FUNCTIONS

CREATE OR REPLACE FUNCTION test_num_elements(float8[])  RETURNS int4
AS '$libdir/pg_testlib.so' LANGUAGE 'c';

CREATE OR REPLACE FUNCTION test_not_trivial(float8[])  RETURNS float8[]
AS '$libdir/pg_testlib.so' LANGUAGE 'c';

CREATE OR REPLACE FUNCTION test_heapfunc(float8[])  RETURNS float8
AS '$libdir/pg_testlib.so' LANGUAGE 'c';


-- SQL TEST
SELECT test_num_elements(array_agg(salary)) FROM employees;
SELECT test_not_trivial(array_agg(salary)) FROM employees;
SELECT test_heapfunc(array_agg(salary)) FROM employees;

总之,我的问题是:

1. 如何将array_agg()中的列数据转换为双精度C数组(或者根据具体情况而定)
2. 如何将一个C数组的int(或双精度数)转换回int4 []或float8 []以便在PostgreSQL中使用？
3. 内存分配原则 - 我使用PostgreSQL内存管理函数palloc()/ pfree()还是可以使用calloc/free？.另外,当使用PG mem funcs时,我是否负责释放我分配的内存？

Answer 1

首先，您应该始终使用 palloc/pfree 来管理内存。PostgreSQL 不会为你管理内存，但每个连接都由一个独立的进程处理，所以如果你泄漏内存，它只会持续一个连接的生命周期（当然，这可能会很长）。

通常，我们将数据分配为：

void *data = palloc(size_of_your_data + VARHDRSZ);
SET_VARSIZE(data, size_of_your_data + VARHDRSZ);

然后使用以下方式访问您的数据：

MyArray *myarr = (MyArray*) VARDATA(data);

完成后，您可以：

PG_RETURN_POINTER(data);

如果您返回数据，则不会释放它。如果您希望 palloc 临时存储，则必须 pfree 它。

现在，MyArray 并不是您想要的float8[]。您需要使用ArrayType，如 34.9.11 中所述 。多态参数和返回类型。