TLA  Line data    Source code

       1                 : /*--------------------------------------------------------------------------
       2                 :  *
       3                 :  * test_copy_callbacks.c
       4                 :  *      Code for testing COPY callbacks.
       5                 :  *
       6                 :  * Portions Copyright (c) 1996-2023, PostgreSQL Global Development Group
       7                 :  * Portions Copyright (c) 1994, Regents of the University of California
       8                 :  *
       9                 :  * IDENTIFICATION
      10                 :  *      src/test/modules/test_copy_callbacks/test_copy_callbacks.c
      11                 :  *
      12                 :  * -------------------------------------------------------------------------
      13                 :  */
      14                 : 
      15                 : #include "postgres.h"
      16                 : 
      17                 : #include "access/table.h"
      18                 : #include "commands/copy.h"
      19                 : #include "fmgr.h"
      20                 : #include "utils/rel.h"
      21                 : 
      22 GNC           1 : PG_MODULE_MAGIC;
      23                 : 
      24                 : static void
      25               3 : to_cb(void *data, int len)
      26                 : {
      27               3 :     ereport(NOTICE,
      28                 :             (errmsg("COPY TO callback called with data \"%s\" and length %d",
      29                 :                     (char *) data, len)));
      30               3 : }
      31                 : 
      32               2 : PG_FUNCTION_INFO_V1(test_copy_to_callback);
      33                 : Datum
      34               1 : test_copy_to_callback(PG_FUNCTION_ARGS)
      35                 : {
      36               1 :     Relation    rel = table_open(PG_GETARG_OID(0), AccessShareLock);
      37                 :     CopyToState cstate;
      38                 :     int64       processed;
      39                 : 
      40               1 :     cstate = BeginCopyTo(NULL, rel, NULL, RelationGetRelid(rel), NULL, false,
      41                 :                          to_cb, NIL, NIL);
      42               1 :     processed = DoCopyTo(cstate);
      43               1 :     EndCopyTo(cstate);
      44                 : 
      45               1 :     ereport(NOTICE, (errmsg("COPY TO callback has processed %lld rows",
      46                 :                             (long long) processed)));
      47                 : 
      48               1 :     table_close(rel, NoLock);
      49                 : 
      50               1 :     PG_RETURN_VOID();
      51                 : }