Line data    Source code

       1              : /*--------------------------------------------------------------------------
       2              :  *
       3              :  * test_copy_callbacks.c
       4              :  *      Code for testing COPY callbacks.
       5              :  *
       6              :  * Portions Copyright (c) 1996-2026, 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            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 %" PRId64 " rows",
      46              :                             processed)));
      47              : 
      48            1 :     table_close(rel, NoLock);
      49              : 
      50            1 :     PG_RETURN_VOID();
      51              : }