使用C编写的动态链接库为PostgreSQL数据库添加自定义函数


不得不说，在这点上，PostgreSQL功能非常强大，不亚于任何家大型商业数据库。要添加一些自定义函数，不是那么复杂。

下边的示例会添加如下函数：test(), concat_text(string, string)  还有另外两个，不作举例，您可以自行尝试

1. 代码如下：

  1. /* $PostgreSQL: pgsql/src/tutorial/funcs_new.c,v 1.13 2007/02/27 23:48:10 tgl Exp $ */ 
  2.  
  3. /****************************************************************************** 
  4.   These are user-defined functions that can be bound to a Postgres backend 
  5.   and called by Postgres to execute SQL functions of the same name. 
  6.  
  7.   The calling format for these functions is defined by the CREATE FUNCTION 
  8.   SQL statement that binds them to the backend. 
  9.  
  10.   NOTE: this file shows examples of "new style" function call conventions. 
  11.   See funcs.c for examples of "old style". 
  12. *****************************************************************************/ 
  13.  
  14. #include "postgres.h"           /* general Postgres declarations */  
  15.  
  16. #include "executor/executor.h"  /* for GetAttributeByName() */  
  17. #include "utils/geo_decls.h"    /* for point type */  
  18.  
  19. #include "executor/spi.h"  
  20.  
  21. PG_MODULE_MAGIC; 
  22.  
  23. /* These prototypes just prevent possible warnings from gcc. */ 
  24. //  
  25. //Datum     add_one(PG_FUNCTION_ARGS);  
  26. //Datum     add_one_float8(PG_FUNCTION_ARGS);  
  27. //Datum     makepoint(PG_FUNCTION_ARGS);  
  28. //Datum     copytext(PG_FUNCTION_ARGS);  
  29. //Datum     concat_text(PG_FUNCTION_ARGS);  
  30. //Datum     c_overpaid(PG_FUNCTION_ARGS);  
  31.  
  32.  
  33. /* By Value */ 
  34.  
  35. PG_FUNCTION_INFO_V1(add_one); 
  36.  
  37. __declspec (dllexport) Datum 
  38. add_one(PG_FUNCTION_ARGS) 
  39. { 
  40.     int32       arg = PG_GETARG_INT32(0); 
  41.  
  42.     PG_RETURN_INT32(arg + 1); 
  43. } 
  44.  
  45. /* By Reference, Fixed Length */ 
  46.  
  47. PG_FUNCTION_INFO_V1(add_one_float8); 
  48.  
  49. __declspec (dllexport) Datum 
  50. add_one_float8(PG_FUNCTION_ARGS) 
  51. { 
  52.     /* The macros for FLOAT8 hide its pass-by-reference nature */ 
  53.     float8      arg = PG_GETARG_FLOAT8(0); 
  54.  
  55.     PG_RETURN_FLOAT8(arg + 1.0); 
  56. } 
  57.  
  58. PG_FUNCTION_INFO_V1(makepoint); 
  59.  
  60. __declspec (dllexport) Datum 
  61. makepoint(PG_FUNCTION_ARGS) 
  62. { 
  63.     Point      *pointx = PG_GETARG_POINT_P(0); 
  64.     Point      *pointy = PG_GETARG_POINT_P(1); 
  65.     Point      *new_point = (Point *) palloc(sizeof(Point)); 
  66.  
  67.     new_point->x = pointx->x; 
  68.     new_point->y = pointy->y; 
  69.  
  70.     PG_RETURN_POINT_P(new_point); 
  71. } 
  72.  
  73. /* By Reference, Variable Length */ 
  74.  
  75. PG_FUNCTION_INFO_V1(copytext); 
  76.  
  77. __declspec (dllexport) Datum 
  78. copytext(PG_FUNCTION_ARGS) 
  79. { 
  80.     text       *t = PG_GETARG_TEXT_P(0); 
  81.  
  82.     /* 
  83.      * VARSIZE is the total size of the struct in bytes. 
  84.      */ 
  85.     text       *new_t = (text *) palloc(VARSIZE(t)); 
  86.  
  87.     SET_VARSIZE(new_t, VARSIZE(t)); 
  88.  
  89.     /* 
  90.      * VARDATA is a pointer to the data region of the struct. 
  91.      */ 
  92.     memcpy((void *) VARDATA(new_t),     /* destination */ 
  93.            (void *) VARDATA(t), /* source */ 
  94.            VARSIZE(t) - VARHDRSZ);      /* how many bytes */ 
  95.     PG_RETURN_TEXT_P(new_t); 
  96. } 
  97.  
  98. PG_FUNCTION_INFO_V1(concat_text); 
  99.  
  100. __declspec (dllexport) Datum 
  101. concat_text(PG_FUNCTION_ARGS) 
  102. { 
  103.     text       *arg1 = PG_GETARG_TEXT_P(0); 
  104.     text       *arg2 = PG_GETARG_TEXT_P(1); 
  105.     int32       arg1_size = VARSIZE(arg1) - VARHDRSZ; 
  106.     int32       arg2_size = VARSIZE(arg2) - VARHDRSZ; 
  107.     int32       new_text_size = arg1_size + arg2_size + VARHDRSZ; 
  108.     text       *new_text = (text *) palloc(new_text_size); 
  109.  
  110.     SET_VARSIZE(new_text, new_text_size); 
  111.     memcpy(VARDATA(new_text), VARDATA(arg1), arg1_size); 
  112.     memcpy(VARDATA(new_text) + arg1_size, VARDATA(arg2), arg2_size); 
  113.     PG_RETURN_TEXT_P(new_text); 
  114. } 
  115.  
  116. /* Composite types */ 
  117.  
  118. PG_FUNCTION_INFO_V1(c_overpaid); 
  119.  
  120. __declspec (dllexport) Datum 
  121. c_overpaid(PG_FUNCTION_ARGS) 
  122. { 
  123.     HeapTupleHeader t = PG_GETARG_HEAPTUPLEHEADER(0); 
  124.     int32       limit = PG_GETARG_INT32(1); 
  125.     bool        isnull; 
  126.     int32       salary; 
  127.  
  128.     salary = DatumGetInt32(GetAttributeByName(t, "salary", &isnull)); 
  129.     if (isnull) 
  130.         PG_RETURN_BOOL(false); 
  131.  
  132.     /* 
  133.      * Alternatively, we might prefer to do PG_RETURN_NULL() for null salary 
  134.      */ 
  135.  
  136.     PG_RETURN_BOOL(salary > limit); 
  137. } 
  138.  
  139. PG_FUNCTION_INFO_V1(test); 
  140. __declspec (dllexport)  Datum test(PG_FUNCTION_ARGS) 
  141. { 
  142.        char *command; 
  143.        int ret; 
  144.        int proc; 
  145.  
  146.        command = "SELECT 1"; 
  147.  
  148.        elog(INFO,"1"); 
  149.  
  150.        SPI_connect(); 
  151.  
  152.        elog(INFO,"2"); 
  153.  
  154.        ret = SPI_exec(command, 1); 
  155.  
  156.        elog(INFO,"3"); 
  157.  
  158.        proc = SPI_processed; 
  159.  
  160.        elog(INFO,"4"); 
  161.        if (ret > 0 && SPI_tuptable != NULL) 
  162.        { 
  163.            TupleDesc tupdesc = SPI_tuptable->tupdesc; 
  164.            SPITupleTable *tuptable = SPI_tuptable; 
  165.            char buf[8192]; 
  166.            int i, j; 
  167.  
  168.            elog(INFO,"5"); 
  169.            for (j = 0; j < proc; j++) 
  170.            { 
  171.                HeapTuple tuple = tuptable->vals[j]; 
  172.  
  173.                elog(INFO,"6"); 
  174.                for (i = 1, buf[0] = 0; i <= tupdesc->natts; i++) 
  175.                { 
  176.                        snprintf(buf + strlen (buf), sizeof(buf) - strlen(buf), " %s%s", 
  177.                                 SPI_getvalue(tuple, tupdesc, i), 
  178.                                 (i == tupdesc->natts) ? " " : " |"); 
  179.  
  180.                        elog(INFO,"7"); 
  181.                } 
  182.                elog(INFO, "EXECQ: %s", buf); 
  183.            } 
  184.        } 
  185.  
  186.        SPI_finish(); 
  187.        // pfree(command);  
  188.  
  189. } 
  • 1
  • 2
  • 下一页

相关内容