Line data    Source code

       1             : /*-------------------------------------------------------------------------
       2             :  *
       3             :  * printsimple.c
       4             :  *    Routines to print out tuples containing only a limited range of
       5             :  *    builtin types without catalog access.  This is intended for
       6             :  *    backends that don't have catalog access because they are not bound
       7             :  *    to a specific database, such as some walsender processes.  It
       8             :  *    doesn't handle standalone backends or protocol versions other than
       9             :  *    3.0, because we don't need such handling for current applications.
      10             :  *
      11             :  * Portions Copyright (c) 1996-2022, PostgreSQL Global Development Group
      12             :  * Portions Copyright (c) 1994, Regents of the University of California
      13             :  *
      14             :  * IDENTIFICATION
      15             :  *    src/backend/access/common/printsimple.c
      16             :  *
      17             :  *-------------------------------------------------------------------------
      18             :  */
      19             : #include "postgres.h"
      20             : 
      21             : #include "access/printsimple.h"
      22             : #include "catalog/pg_type.h"
      23             : #include "libpq/pqformat.h"
      24             : #include "utils/builtins.h"
      25             : 
      26             : /*
      27             :  * At startup time, send a RowDescription message.
      28             :  */
      29             : void
      30        2934 : printsimple_startup(DestReceiver *self, int operation, TupleDesc tupdesc)
      31             : {
      32             :     StringInfoData buf;
      33             :     int         i;
      34             : 
      35        2934 :     pq_beginmessage(&buf, 'T'); /* RowDescription */
      36        2934 :     pq_sendint16(&buf, tupdesc->natts);
      37             : 
      38       11152 :     for (i = 0; i < tupdesc->natts; ++i)
      39             :     {
      40        8218 :         Form_pg_attribute attr = TupleDescAttr(tupdesc, i);
      41             : 
      42        8218 :         pq_sendstring(&buf, NameStr(attr->attname));
      43        8218 :         pq_sendint32(&buf, 0);  /* table oid */
      44        8218 :         pq_sendint16(&buf, 0);  /* attnum */
      45        8218 :         pq_sendint32(&buf, (int) attr->atttypid);
      46        8218 :         pq_sendint16(&buf, attr->attlen);
      47        8218 :         pq_sendint32(&buf, attr->atttypmod);
      48        8218 :         pq_sendint16(&buf, 0);  /* format code */
      49             :     }
      50             : 
      51        2934 :     pq_endmessage(&buf);
      52        2934 : }
      53             : 
      54             : /*
      55             :  * For each tuple, send a DataRow message.
      56             :  */
      57             : bool
      58        4356 : printsimple(TupleTableSlot *slot, DestReceiver *self)
      59             : {
      60        4356 :     TupleDesc   tupdesc = slot->tts_tupleDescriptor;
      61             :     StringInfoData buf;
      62             :     int         i;
      63             : 
      64             :     /* Make sure the tuple is fully deconstructed */
      65        4356 :     slot_getallattrs(slot);
      66             : 
      67             :     /* Prepare and send message */
      68        4356 :     pq_beginmessage(&buf, 'D');
      69        4356 :     pq_sendint16(&buf, tupdesc->natts);
      70             : 
      71       16858 :     for (i = 0; i < tupdesc->natts; ++i)
      72             :     {
      73       12502 :         Form_pg_attribute attr = TupleDescAttr(tupdesc, i);
      74             :         Datum       value;
      75             : 
      76       12502 :         if (slot->tts_isnull[i])
      77             :         {
      78        1886 :             pq_sendint32(&buf, -1);
      79        1886 :             continue;
      80             :         }
      81             : 
      82       10616 :         value = slot->tts_values[i];
      83             : 
      84             :         /*
      85             :          * We can't call the regular type output functions here because we
      86             :          * might not have catalog access.  Instead, we must hard-wire
      87             :          * knowledge of the required types.
      88             :          */
      89       10616 :         switch (attr->atttypid)
      90             :         {
      91        8990 :             case TEXTOID:
      92             :                 {
      93        8990 :                     text       *t = DatumGetTextPP(value);
      94             : 
      95       17980 :                     pq_sendcountedtext(&buf,
      96        8990 :                                        VARDATA_ANY(t),
      97        8990 :                                        VARSIZE_ANY_EXHDR(t),
      98             :                                        false);
      99             :                 }
     100        8990 :                 break;
     101             : 
     102           0 :             case INT4OID:
     103             :                 {
     104           0 :                     int32       num = DatumGetInt32(value);
     105             :                     char        str[12];    /* sign, 10 digits and '\0' */
     106             :                     int         len;
     107             : 
     108           0 :                     len = pg_ltoa(num, str);
     109           0 :                     pq_sendcountedtext(&buf, str, len, false);
     110             :                 }
     111           0 :                 break;
     112             : 
     113        1594 :             case INT8OID:
     114             :                 {
     115        1594 :                     int64       num = DatumGetInt64(value);
     116             :                     char        str[MAXINT8LEN + 1];
     117             :                     int         len;
     118             : 
     119        1594 :                     len = pg_lltoa(num, str);
     120        1594 :                     pq_sendcountedtext(&buf, str, len, false);
     121             :                 }
     122        1594 :                 break;
     123             : 
     124          32 :             case OIDOID:
     125             :                 {
     126          32 :                     Oid         num = ObjectIdGetDatum(value);
     127             :                     char        str[10];    /* 10 digits */
     128             :                     int         len;
     129             : 
     130          32 :                     len = pg_ultoa_n(num, str);
     131          32 :                     pq_sendcountedtext(&buf, str, len, false);
     132             :                 }
     133          32 :                 break;
     134             : 
     135           0 :             default:
     136           0 :                 elog(ERROR, "unsupported type OID: %u", attr->atttypid);
     137             :         }
     138             :     }
     139             : 
     140        4356 :     pq_endmessage(&buf);
     141             : 
     142        4356 :     return true;
     143             : }