PostgreSQL で集合を返すユーザー定義関数の書き方

CREATE TABLE T1 (C1 INT);

INSERT INTO T1 (C1) VALUES (1), (2), (3);

SELECT C1, generate_series(1, C1) AS G1, generate_series(1, C1 * 2) AS G2 FROM T1;

 c1 | g1 | g2
----+----+----
  1 |  1 |  1
  1 |  1 |  2
  2 |  1 |  1
  2 |  2 |  2
  2 |  1 |  3
  2 |  2 |  4
  3 |  1 |  1
  3 |  2 |  2
  3 |  3 |  3
  3 |  1 |  4
  3 |  2 |  5
  3 |  3 |  6
(12 rows)

   8      16 sdb 58721 30112 1645690 68805 700 10529 89832 126220 0 58921 195021
   8      17 sdb1 58665 30108 1645210 68696 700 10529 89832 126220 0 58812 194912
   8       0 sda 74696 37688 2461892 203698 1462510 2120160 28664504 5535643 0 1210183 5739814
   8       1 sda1 578 17 4754 569 7 1 64 4 0 572 572
   8       2 sda2 73612 37639 2452834 201727 1462503 2120159 28664440 5535639 0 1209602 5737840
   8       3 sda3 354 32 3088 1335 0 0 0 0 0 1335 1335

CREATE FUNCTION public.show_diskstats1()
RETURNS TABLE(major int, minor int, diskname text,
              read_completed bigint, read_merged bigint, read_sectors bigint, read_time int,
              write_completed bigint, write_merged bigint, write_sectors bigint, write_time int,
              io int, io_time int, weighted_io_time int)
AS 'MODULE_PATHNAME'
LANGUAGE C VOLATILE STRICT;

CREATE FUNCTION public.show_diskstats1(
              OUT major int, OUT minor int, OUT diskname text,
              OUT read_completed bigint, OUT read_merged bigint, OUT read_sectors bigint, OUT read_time int,
              OUT write_completed bigint, OUT write_merged bigint, OUT write_sectors bigint, OUT write_time int,
              OUT io int, OUT io_time int, OUT weighted_io_time int)
RETURNS SETOF record
AS 'MODULE_PATHNAME'
LANGUAGE C VOLATILE STRICT;

PG_FUNCTION_INFO_V1(show_diskstats1);
Datum
show_diskstats1(PG_FUNCTION_ARGS)
{
    FuncCallContext  *funcctx;
    Datum             result;
    struct
    {
        FILE   *file;
    } *diskstat_ctx;

    if (SRF_IS_FIRSTCALL())
    {
        /* 最初の 1 回はここを通る */
        MemoryContext oldcontext;
	TupleDesc     tupleDesc;

        oldcontext = MemoryContextSwitchTo(funcctx->multi_call_memory_ctx);

        /* このユーザー定義関数の戻り値となる複合型を tupleDesc として取り出す */
        if (get_call_result_type(fcinfo, NULL, &tupleDesc) != TYPEFUNC_COMPOSITE)
            elog(ERROR, "...");

        /* /proc/diskstats をオープンし、そのディスクリプタの情報を funcctx->user_fctx にリンクする */
        diskstat_ctx = palloc0(sizeof(*diskstat_ctx));
        diskstat_ctx->file = fopen("/proc/diskstats", "r");

        if (diskstat_ctx->file == NULL)
            elog(ERROR, "...");

        funcctx->user_fctx = diskstat_ctx;

        /* この関数の戻り値の型と tuple table slot を作成し保存する */
        funcctx->tuple_desc = tupleDesc;
        funcctx->slot = MakeSingleTupleTableSlot(tupleDesc);

        MemoryContextSwitchTo(oldcontext);
    }

    funcctx = SRF_PERCALL_SETUP();
    diskstat_ctx = funcctx->user_fctx;

    /* 全ての行を返し終わったかどうかを判定する。
     * この判定用に funcctx->call_cntr と funcctx->max_calls を使ってもよいが、
     * /proc/diskstats の場合は読み込んで EOF が出た時点が終了になる。
     */
    if (condition)
    {
        /* 返す行がある場合はこちら
         * この時点で funcctx->slot に結果内容を格納しておく。
         */
        HeapTuple tuple;

        /* 最終的な結果は Heap Tuple 形式なので virtual tuple から変換 */
        tuple = ExecCopySlotTuple(funcctx->slot);
        result = HeapTupleGetDatum(tuple);

        SRF_RETURN_NEXT(funcctx, result);
    }
    else
    {
        /* 全ての行を返し終わったのでクリーンナップを行う。
         * この回は結果行を返さない
         */
        fclose(diskstat_ctx->file);

        SRF_RETURN_DONE(funcctx);
    }
}

PG_FUNCTION_INFO_V1(show_diskstats2);
Datum
show_diskstats2(PG_FUNCTION_ARGS)
{
    ReturnSetInfo  *rsi;
    TupleDesc       tupdesc;
    TupleTableSlot *tts;
    MemoryContext   oldcontext;

    rsi = (ReturnSetInfo *) fcinfo->resultinfo;

    /* 本当は rsi が有効かチェックした方がよい。やり方は github コードには載せている */  

    /* fcinfo->resultinfo->returnMode に SFRM_Materialize を指定すると tuplestore を使うことを宣言できる */
    rsi->returnMode = SFRM_Materialize;

    /* このユーザー定義関数の戻り値となる複合型を tupleDesc として取り出す */
    if (get_call_result_type(fcinfo, NULL, &tupleDesc) != TYPEFUNC_COMPOSITE)
        elog(ERROR, "...");

    oldcontext = MemoryContextSwitchTo(funcctx->multi_call_memory_ctx);

    /* この関数の戻り値の型と tuple table slot を作成し保存する */    
    rsi->setDesc = CreateTupleDescCopy(tupdesc);
    BlessTupleDesc(rsi->setDesc);
    tts = MakeSingleTupleTableSlot(rsi->setDesc);

    /* 結果行を格納する tuplestore を作成する */
    rsi->setResult = tuplestore_begin_heap(rsi->allowedModes &SFRM_Materialize_Random, false, work_mem);

    /* /proc/diskstats を一気に読み込む */
    {
        FILE *file;

        file = fopen("/proc/diskstats", "r");
        if (file == NULL)
            ereport(ERROR,
                    (errcode_for_file_access(),
                     errmsg("could not open file \"/proc/diskstats\" for writing: %m")));

        /* /proc/diskstats を EOF になるまで読み込む。
         * 読み込んだ結果は tts に格納しておく。
         */
        while (condition)
            tuplestore_puttupleslot(rsi->setResult, tts);

        fclose(file);
    }

    MemoryContextSwitchTo(oldcontext);

    /* ダミーの戻り値として NULL を返す。*/
    PG_RETURN_NULL();
}