PostgreSQL: C भाषा में एक्सटेंशन (फ़ंक्शन) का विकास

सरल सी फ़ंक्शन

#include "postgres.h"
#include "fmgr.h"
PG_MODULE_MAGIC;
PG_FUNCTION_INFO_V1(add_ab);
Datum add_ab(PG_FUNCTION_ARGS)
{
    int32 arg_a = PG_GETARG_INT32(0);
    int32 arg_b = PG_GETARG_INT32(1);
    PG_RETURN_INT32(arg_a + arg_b);
}

• #include "postgresql.h": आपको हमेशा इस हेडर फ़ाइल को शामिल करना होगा, इसमें विभिन्न बुनियादी प्रकार और कार्य शामिल हैं।
• #include "fmgr.h": हेडर फ़ाइल में विभिन्न PG_ * मैक्रोज़ होते हैं।
• PG_MODULE_MAGIC: एक मैक्रो जो निर्धारित करता है कि हम संस्करण 8.2 से ऊपर PostgreSQL के लिए एक मॉड्यूल विकसित कर रहे हैं।
• PG_FUNC_INFO_V1: PostgreSQL के भीतर एक मैक्रो डिफाइनिंग फ़ंक्शन कॉलिंग कन्वेंशन। यदि आप इसके बिना एक फ़ंक्शन घोषित करते हैं, तो संस्करण 0, अन्यथा संस्करण 1 पर कॉल करने पर एक समझौता होगा।
• Datum: . , PostgreSQL. - VARIANT Microsoft. «» - . , .
• add_ab(PG_FUNCTION_ARGS): . - . , .
• int32 arg_a = PG_GETARG_INT32(0): ( ).
• PG_RETURN_INT32(arg_a + arg_b): .

MODULES = add_func
PG_CONFIG = pg_config
PGXS := $(shell $(PG_CONFIG) --pgxs)
include $(PGXS)

हम SQL फ़ंक्शन बनाते हैं

CREATE FUNCTION add(int, int)
  RETURNS int
AS '/usr/lib/postgresql/9.4/lib/add_func', 'add_ab'
LANGUAGE C STRICT;

SELECT add(1, 2);

स्थापना को स्वचालित करें

CREATE FUNCTION add(int, int)
  RETURNS int
AS 'MODULE_PATHNAME', 'add_ab'
LANGUAGE C STRICT;

MODULES = add_func
DATA_built = add_funcs.sql
PG_CONFIG = pg_config
PGXS := $(shell $(PG_CONFIG) --pgxs)
include $(PGXS)

थोड़ा सुधार करो

तर्क की जाँच

• सुनिश्चित करें कि फ़ंक्शन कार्य करता है, भले ही कोई तर्क NULL हो
• यदि दोनों तर्क NULL हैं , तो NULL वापस करें

PG_FUNCTION_INFO_V1(add_ab_null);
Datum add_ab_null(PG_FUNCTION_ARGS)
{
    int32 not_null = 0;
    int32 sum = 0;
    if (!PG_ARGISNULL(0))
    {
        sum += PG_GETARG_INT32(0);
        not_null = 1;
    }
    if (!PG_ARGISNULL(1))
    {
        sum += PG_GETARG_INT32(1);
        not_null = 1;
    }
    if (not_null)
    {
        PG_RETURN_INT32(sum);
    }
    PG_RETURN_NULL();
}

CREATE FUNCTION add(int, int)
 RETURNS int
AS '$libdir/add_func', 'add_ab_null'
LANGUAGE C;

SELECT add(NULL, NULL) AS must_be_null, add(NULL, 1) AS must_be_one;
-[ RECORD 1 ]+--
must_be_null |
must_be_one  | 1

SELECT (CASE WHEN (a IS null) AND (b IS null)
(THEN null
ELSE coalesce(a, 0) + coalesce(b,0)
END)
FROM (SELECT 1::int AS a, null::int AS b) s;
-[ RECORD 1 ]
 case | 1

किसी फ़ंक्शन में कोई भी तर्क

if (!PG_ARGISNULL(i))
{
    sum += PG_GETARG_INT32(i);
    not_null = 1;
}

#include "utils/array.h"     //       
#include "catalog/pg_type.h" // INT4OID
PG_MODULE_MAGIC;
Datum add_int32_array(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(add_int32_array);
Datum add_int32_array(PG_FUNCTION_ARGS)
{
    //     .
    //          ,       int.
    ArrayType *input_array;
    int32 sum = 0;
    bool not_null = false;

    Datum *datums;
    bool *nulls;
    int count;
    int i;
    input_array = PG_GETARG_ARRAYTYPE_P(0); //     .  *_P  
                                            //        ,   INT32

    //          INT32 (INT4)
    Assert(ARR_ELEMTYPE(input_array) == INT4OID);

    //      
    if (ARR_NDIM(input_array) > 1)
        ereport(ERROR, (errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR), errmsg("1-dimensional array needed")));

    deconstruct_array(input_array, //  
                      INT4OID,     //  
                      4,           //    
                      true,        // int4   
                      'i',         //  'i'
                      &datums, &nulls, &count); //   

    for(i = 0; i < count; i++)
    {
        //     
        if (nulls[i])
            continue;

        // ,       
        sum += DatumGetInt32(datums[i]);
        not_null = true;
    }
    if (not_null)
        PG_RETURN_INT32(sum);

    PG_RETURN_NULL();
}

• पूर्णांक सरणियों के लिए कोई विशेष प्रकार नहीं है, इसलिए हम सामान्य प्रकार ArrayType का उपयोग करते हैं , जो किसी भी प्रकार के सरणी के लिए काम करता है
• पहले तर्क के साथ सरणी को आरम्भ करने के लिए, हमने विशेष मैक्रो PG_GETARG_ARRAYTYPE_P का उपयोग किया
• यह देखने के लिए भी जांच की जाती है कि क्या सरणी वास्तव में एक आयामी ARR_NDIM है
• Int4 (= 23) के लिए OID प्रकार को INT4OID के रूप में परिभाषित किया गया है। अन्य प्रकार की परिभाषाएँ देखने के लिए, आप SQL का उपयोग कर सकते हैं,
  
  

  select oid, typlen, typbyval, typalign from pg_type
  where typname = 'int4';
  -[ RECORD 1 ]
  oid | 23
  typlen | 4
  typbyval | t
  typalign | i
  

  
  

CREATE OR REPLACE FUNCTION add_arr(int[]) RETURNS int
AS '$libdir/add_func', 'add_int32_array'
LANGUAGE C STRICT;

SELECT add_arr('{1,2,3,4,5,6,7,8,9}');
-[ RECORD 1 ]
add_arr | 45
SELECT add_arr(ARRAY[1,2,NULL]);
-[ RECORD 1 ]
add_arr | 3
SELECT add_arr(ARRAY[NULL::int]);
-[ RECORD 1 ]
add_arr |

अनुशंसाएँ

स्मृति के साथ काम करें

पैलोक () और pfree () का उपयोग करना

संरचनाओं का प्रारंभ

कनेक्टेड फ़ाइलें

उपयोगी कड़ियाँ