Postgresql c函数返回一行有2个字段

时间:2013-11-09 10:41:54

标签: postgresql

我是postgresql c函数的新手,我开始关注示例。

我想编写一个简单的函数,它包含在SQL内部并接收参数evaluete,并分别返回2个字段作为2列的总和(现在更简单)。

以下功能通过检查时出现问题

(get_call_result_type(fcinfo, &resultTypeId, &resultTupleDesc) != TYPEFUNC_COMPOSITE)

如果删除此行,我会从此查询中获得1个整数

select * from pdc_imuanno(2012);

和来自

的错误
select (a).* from pdc_imuanno(2012) a;

因为它不是复合类型。

问题是我如何为元组准备模板如果它不正确

resultTupleDesc = CreateTemplateTupleDesc(2, false);
TupleDescInitEntry(resultTupleDesc, (AttrNumber) 1, "abp1", FLOAT4OID, -1, 0);
TupleDescInitEntry(resultTupleDesc, (AttrNumber) 2, "abp2", FLOAT4OID, -1, 0);

更多

get_call_result_type(fcinfo, &resultTypeId, &resultTupleDesc)

fcinfo是什么以及来自哪里?

源表:

CREATE TABLE imu.calcolo (
  codfis character varying(16) NOT NULL,
  anno integer NOT NULL,
  abp1 numeric,
  abp2 numeric,
  CONSTRAINT imucalcolo_pkey PRIMARY KEY (codfis, anno)
)
WITH ( OIDS=FALSE );
-------------------------------------------------------
#include "postgres.h"
#include "fmgr.h"

#include "catalog/pg_type.h"
#include "funcapi.h"
#include "executor/spi.h"
#include "lib/stringinfo.h"
#include "miscadmin.h"

#include <math.h>

#include "utils/builtins.h"
#include "utils/guc.h"
#include "utils/lsyscache.h"
#include "utils/memutils.h"
#include "utils/numeric.h"

#include "access/htup_details.h"

#ifdef PG_MODULE_MAGIC
PG_MODULE_MAGIC;
#endif

PG_FUNCTION_INFO_V1(test_query);
Datum test_query(PG_FUNCTION_ARGS);

Datum
test_query(PG_FUNCTION_ARGS)
{
  TupleDesc resultTupleDesc, tupledesc;
  bool bisnull, cisnull;
  Oid resultTypeId;
  Datum retvals[2];
  bool retnulls[2];
  HeapTuple rettuple;

  sprintf(query,"SELECT anno, abp1::real, abp2::real "
                "FROM imu.calcolo WHERE anno = %d;",PG_GETARG_INT32(0));

  int ret;
  int proc;
  float abp1 = 0;
  float abp2 = 0;
  SPI_connect();
  ret = SPI_exec(query,0);
  proc = SPI_processed;

  if (ret > 0 && SPI_tuptable != NULL)
  {
    HeapTuple tuple;
    tupledesc = SPI_tuptable->tupdesc;
    SPITupleTable *tuptable = SPI_tuptable;
    for (j = 0; j < proc; j++)
    {
      tuple = tuptable->vals[j];
      abp1 += DatumGetFloat4(SPI_getbinval(tuple, tupledesc, 2, &bisnull));
      abp2 += DatumGetFloat4(SPI_getbinval(tuple, tupledesc, 3, &cisnull));
    }
  }
  resultTupleDesc = CreateTemplateTupleDesc(2, false);
  TupleDescInitEntry(resultTupleDesc, (AttrNumber) 1, "abp1", FLOAT4OID, -1, 0);
  TupleDescInitEntry(resultTupleDesc, (AttrNumber) 2, "abp2", FLOAT4OID, -1, 0);

  if (get_call_result_type(fcinfo, &resultTypeId, &resultTupleDesc) != TYPEFUNC_COMPOSITE) {
    ereport(ERROR, (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                    errmsg("function returning record called in context that cannot accept type record")));
  }
  resultTupleDesc = BlessTupleDesc(resultTupleDesc);
  SPI_finish();
  retvals[0] = Float4GetDatum(abp1);
  retvals[1] = Float4GetDatum(abp2);
  retnulls[1] = bisnull;
  retnulls[2] = cisnull;
  rettuple = heap_form_tuple( resultTupleDesc, retvals, retnulls);
  PG_RETURN_DATUM( HeapTupleGetDatum( rettuple ) );

}

创建功能:

CREATE FUNCTION pdc_imuanno(integer)
RETURNS float
AS 'pdc','test_query'
LANGUAGE C STABLE STRICT;
ALTER FUNCTION pdc_imuanno(integer) OWNER TO www;

查询:

select * from pdc_imuanno(2012);

好的,我发现了简单而愚蠢的错误 我自己创建了函数retunrning 1字段,并且我将其视为返回2个字段。

使用这个创建sql它可以工作

CREATE FUNCTION pdc_imuanno(integer)
RETURNS TABLE(abp1 real, abp2 real)
AS 'pdc','test_query'
LANGUAGE C STABLE STRICT;

无论如何,它使用有限数量的行来求和,如果我扩展行数,它就会崩溃 在这一点上

rettuple = heap_form_tuple( resultTupleDesc, retvals, retnulls);

我想值的类型有一些错误

所以我在表字段上查询为numeric :: real,我将它们作为float4输出,然后将它们作为基准输出。

我的错误在哪里?

非常感谢您的帮助。

这是我在StackOverflow中的第一篇文章。

卢卡

1 个答案:

答案 0 :(得分:2)

  

在:

     

get_call_result_type(fcinfo, &resultTypeId, &resultTupleDesc)

     

fcinfo - 来自何处以及来自哪里?

fcinfo是v1调用约定中PG_FUNCTION_ARGS的一部分。它是函数调用的上下文,包含各种细节,如参数等。大部分内容由GETARG宏等在幕后处理,但你需要将它传递给辅助函数。

由于拼写错误等问题,其余的问题很难理解。我是猜测您想在C中编写一个函数来将两个值一起添加并返回结果?如果是这样,那么返回复合类型中的两个字段没有多大意义。请显示相关的CREATE TYPE语句,用于定义函数的CREATE OR REPLACE FUNCTION声明,并说明您要执行的操作以及原因。

(如果你编辑你的问题,发表评论以回复此答案,否则我将不知道你编辑过。)