LCOV - differential code coverage report
Current view: top level - src/bin/psql - crosstabview.c (source / functions) Coverage Total Hit UBC CBC
Current: Differential Code Coverage HEAD vs 15 Lines: 95.4 % 239 228 11 228
Current Date: 2023-04-08 17:13:01 Functions: 100.0 % 15 15 15
Baseline: 15 Line coverage date bins:
Baseline Date: 2023-04-08 15:09:40 (240..) days: 95.4 % 239 228 11 228
Legend: Lines: hit not hit Function coverage date bins:
(240..) days: 100.0 % 15 15 15

 Age         Owner                  TLA  Line data    Source code
                                  1                 : /*
                                  2                 :  * psql - the PostgreSQL interactive terminal
                                  3                 :  *
                                  4                 :  * Copyright (c) 2000-2023, PostgreSQL Global Development Group
                                  5                 :  *
                                  6                 :  * src/bin/psql/crosstabview.c
                                  7                 :  */
                                  8                 : #include "postgres_fe.h"
                                  9                 : 
                                 10                 : #include "common.h"
                                 11                 : #include "common/logging.h"
                                 12                 : #include "crosstabview.h"
                                 13                 : #include "pqexpbuffer.h"
                                 14                 : #include "psqlscanslash.h"
                                 15                 : #include "settings.h"
                                 16                 : 
                                 17                 : /*
                                 18                 :  * Value/position from the resultset that goes into the horizontal or vertical
                                 19                 :  * crosstabview header.
                                 20                 :  */
                                 21                 : typedef struct _pivot_field
                                 22                 : {
                                 23                 :     /*
                                 24                 :      * Pointer obtained from PQgetvalue() for colV or colH. Each distinct
                                 25                 :      * value becomes an entry in the vertical header (colV), or horizontal
                                 26                 :      * header (colH). A Null value is represented by a NULL pointer.
                                 27                 :      */
                                 28                 :     char       *name;
                                 29                 : 
                                 30                 :     /*
                                 31                 :      * When a sort is requested on an alternative column, this holds
                                 32                 :      * PQgetvalue() for the sort column corresponding to <name>. If <name>
                                 33                 :      * appear multiple times, it's the first value in the order of the results
                                 34                 :      * that is kept. A Null value is represented by a NULL pointer.
                                 35                 :      */
                                 36                 :     char       *sort_value;
                                 37                 : 
                                 38                 :     /*
                                 39                 :      * Rank of this value, starting at 0. Initially, it's the relative
                                 40                 :      * position of the first appearance of <name> in the resultset. For
                                 41                 :      * example, if successive rows contain B,A,C,A,D then it's B:0,A:1,C:2,D:3
                                 42                 :      * When a sort column is specified, ranks get updated in a final pass to
                                 43                 :      * reflect the desired order.
                                 44                 :      */
                                 45                 :     int         rank;
                                 46                 : } pivot_field;
                                 47                 : 
                                 48                 : /* Node in avl_tree */
                                 49                 : typedef struct _avl_node
                                 50                 : {
                                 51                 :     /* Node contents */
                                 52                 :     pivot_field field;
                                 53                 : 
                                 54                 :     /*
                                 55                 :      * Height of this node in the tree (number of nodes on the longest path to
                                 56                 :      * a leaf).
                                 57                 :      */
                                 58                 :     int         height;
                                 59                 : 
                                 60                 :     /*
                                 61                 :      * Child nodes. [0] points to left subtree, [1] to right subtree. Never
                                 62                 :      * NULL, points to the empty node avl_tree.end when no left or right
                                 63                 :      * value.
                                 64                 :      */
                                 65                 :     struct _avl_node *children[2];
                                 66                 : } avl_node;
                                 67                 : 
                                 68                 : /*
                                 69                 :  * Control structure for the AVL tree (binary search tree kept
                                 70                 :  * balanced with the AVL algorithm)
                                 71                 :  */
                                 72                 : typedef struct _avl_tree
                                 73                 : {
                                 74                 :     int         count;          /* Total number of nodes */
                                 75                 :     avl_node   *root;           /* root of the tree */
                                 76                 :     avl_node   *end;            /* Immutable dereferenceable empty tree */
                                 77                 : } avl_tree;
                                 78                 : 
                                 79                 : 
                                 80                 : static bool printCrosstab(const PGresult *result,
                                 81                 :                           int num_columns, pivot_field *piv_columns, int field_for_columns,
                                 82                 :                           int num_rows, pivot_field *piv_rows, int field_for_rows,
                                 83                 :                           int field_for_data);
                                 84                 : static void avlInit(avl_tree *tree);
                                 85                 : static void avlMergeValue(avl_tree *tree, char *name, char *sort_value);
                                 86                 : static int  avlCollectFields(avl_tree *tree, avl_node *node,
                                 87                 :                              pivot_field *fields, int idx);
                                 88                 : static void avlFree(avl_tree *tree, avl_node *node);
                                 89                 : static void rankSort(int num_columns, pivot_field *piv_columns);
                                 90                 : static int  indexOfColumn(char *arg, const PGresult *res);
                                 91                 : static int  pivotFieldCompare(const void *a, const void *b);
                                 92                 : static int  rankCompare(const void *a, const void *b);
                                 93                 : 
                                 94                 : 
                                 95                 : /*
                                 96                 :  * Main entry point to this module.
                                 97                 :  *
                                 98                 :  * Process the data from *res according to the options in pset (global),
                                 99                 :  * to generate the horizontal and vertical headers contents,
                                100                 :  * then call printCrosstab() for the actual output.
                                101                 :  */
                                102                 : bool
  423 peter                     103 CBC          66 : PrintResultInCrosstab(const PGresult *res)
                                104                 : {
 2551 tgl                       105              66 :     bool        retval = false;
                                106                 :     avl_tree    piv_columns;
                                107                 :     avl_tree    piv_rows;
 2557 alvherre                  108              66 :     pivot_field *array_columns = NULL;
                                109              66 :     pivot_field *array_rows = NULL;
                                110              66 :     int         num_columns = 0;
                                111              66 :     int         num_rows = 0;
                                112                 :     int         field_for_rows;
                                113                 :     int         field_for_columns;
                                114                 :     int         field_for_data;
                                115                 :     int         sort_field_for_columns;
                                116                 :     int         rn;
                                117                 : 
                                118              66 :     avlInit(&piv_rows);
                                119              66 :     avlInit(&piv_columns);
                                120                 : 
                                121              66 :     if (PQresultStatus(res) != PGRES_TUPLES_OK)
                                122                 :     {
 1469 peter                     123 UBC           0 :         pg_log_error("\\crosstabview: statement did not return a result set");
 2557 alvherre                  124               0 :         goto error_return;
                                125                 :     }
                                126                 : 
 2551 tgl                       127 CBC          66 :     if (PQnfields(res) < 3)
                                128                 :     {
 1469 peter                     129               6 :         pg_log_error("\\crosstabview: query must return at least three columns");
 2557 alvherre                  130               6 :         goto error_return;
                                131                 :     }
                                132                 : 
                                133                 :     /* Process first optional arg (vertical header column) */
 2551 tgl                       134              60 :     if (pset.ctv_args[0] == NULL)
 2557 alvherre                  135              15 :         field_for_rows = 0;
                                136                 :     else
                                137                 :     {
 2551 tgl                       138              45 :         field_for_rows = indexOfColumn(pset.ctv_args[0], res);
                                139              45 :         if (field_for_rows < 0)
 2557 alvherre                  140 UBC           0 :             goto error_return;
                                141                 :     }
                                142                 : 
                                143                 :     /* Process second optional arg (horizontal header column) */
 2551 tgl                       144 CBC          60 :     if (pset.ctv_args[1] == NULL)
 2557 alvherre                  145              15 :         field_for_columns = 1;
                                146                 :     else
                                147                 :     {
 2551 tgl                       148              45 :         field_for_columns = indexOfColumn(pset.ctv_args[1], res);
 2557 alvherre                  149              45 :         if (field_for_columns < 0)
                                150               3 :             goto error_return;
                                151                 :     }
                                152                 : 
                                153                 :     /* Insist that header columns be distinct */
                                154              57 :     if (field_for_columns == field_for_rows)
                                155                 :     {
 1469 peter                     156               3 :         pg_log_error("\\crosstabview: vertical and horizontal headers must be different columns");
 2557 alvherre                  157               3 :         goto error_return;
                                158                 :     }
                                159                 : 
                                160                 :     /* Process third optional arg (data column) */
 2551 tgl                       161              54 :     if (pset.ctv_args[2] == NULL)
                                162                 :     {
                                163                 :         int         i;
                                164                 : 
                                165                 :         /*
                                166                 :          * If the data column was not specified, we search for the one not
                                167                 :          * used as either vertical or horizontal headers.  Must be exactly
                                168                 :          * three columns, or this won't be unique.
                                169                 :          */
                                170              15 :         if (PQnfields(res) != 3)
                                171                 :         {
 1469 peter                     172 UBC           0 :             pg_log_error("\\crosstabview: data column must be specified when query returns more than three columns");
 2557 alvherre                  173               0 :             goto error_return;
                                174                 :         }
                                175                 : 
 2551 tgl                       176 CBC          15 :         field_for_data = -1;
 2557 alvherre                  177              45 :         for (i = 0; i < PQnfields(res); i++)
                                178                 :         {
                                179              45 :             if (i != field_for_rows && i != field_for_columns)
                                180                 :             {
                                181              15 :                 field_for_data = i;
                                182              15 :                 break;
                                183                 :             }
                                184                 :         }
                                185              15 :         Assert(field_for_data >= 0);
                                186                 :     }
                                187                 :     else
                                188                 :     {
 2551 tgl                       189              39 :         field_for_data = indexOfColumn(pset.ctv_args[2], res);
                                190              39 :         if (field_for_data < 0)
                                191               9 :             goto error_return;
                                192                 :     }
                                193                 : 
                                194                 :     /* Process fourth optional arg (horizontal header sort column) */
                                195              45 :     if (pset.ctv_args[3] == NULL)
                                196              30 :         sort_field_for_columns = -1;    /* no sort column */
                                197                 :     else
                                198                 :     {
                                199              15 :         sort_field_for_columns = indexOfColumn(pset.ctv_args[3], res);
                                200              15 :         if (sort_field_for_columns < 0)
 2557 alvherre                  201 UBC           0 :             goto error_return;
                                202                 :     }
                                203                 : 
                                204                 :     /*
                                205                 :      * First part: accumulate the names that go into the vertical and
                                206                 :      * horizontal headers, each into an AVL binary tree to build the set of
                                207                 :      * DISTINCT values.
                                208                 :      */
                                209                 : 
 2557 alvherre                  210 CBC        5061 :     for (rn = 0; rn < PQntuples(res); rn++)
                                211                 :     {
                                212                 :         char       *val;
                                213                 :         char       *val1;
                                214                 : 
                                215                 :         /* horizontal */
                                216            5019 :         val = PQgetisnull(res, rn, field_for_columns) ? NULL :
                                217            4998 :             PQgetvalue(res, rn, field_for_columns);
                                218            5019 :         val1 = NULL;
                                219                 : 
                                220            5094 :         if (sort_field_for_columns >= 0 &&
                                221              75 :             !PQgetisnull(res, rn, sort_field_for_columns))
                                222              75 :             val1 = PQgetvalue(res, rn, sort_field_for_columns);
                                223                 : 
                                224            5019 :         avlMergeValue(&piv_columns, val, val1);
                                225                 : 
                                226            5019 :         if (piv_columns.count > CROSSTABVIEW_MAX_COLUMNS)
                                227                 :         {
 1469 peter                     228               3 :             pg_log_error("\\crosstabview: maximum number of columns (%d) exceeded",
                                229                 :                          CROSSTABVIEW_MAX_COLUMNS);
 2557 alvherre                  230               3 :             goto error_return;
                                231                 :         }
                                232                 : 
                                233                 :         /* vertical */
                                234            5016 :         val = PQgetisnull(res, rn, field_for_rows) ? NULL :
                                235            5010 :             PQgetvalue(res, rn, field_for_rows);
                                236                 : 
                                237            5016 :         avlMergeValue(&piv_rows, val, NULL);
                                238                 :     }
                                239                 : 
                                240                 :     /*
                                241                 :      * Second part: Generate sorted arrays from the AVL trees.
                                242                 :      */
                                243                 : 
                                244              42 :     num_columns = piv_columns.count;
                                245              42 :     num_rows = piv_rows.count;
                                246                 : 
                                247                 :     array_columns = (pivot_field *)
                                248              42 :         pg_malloc(sizeof(pivot_field) * num_columns);
                                249                 : 
                                250                 :     array_rows = (pivot_field *)
                                251              42 :         pg_malloc(sizeof(pivot_field) * num_rows);
                                252                 : 
                                253              42 :     avlCollectFields(&piv_columns, piv_columns.root, array_columns, 0);
                                254              42 :     avlCollectFields(&piv_rows, piv_rows.root, array_rows, 0);
                                255                 : 
                                256                 :     /*
                                257                 :      * Third part: optionally, process the ranking data for the horizontal
                                258                 :      * header
                                259                 :      */
                                260              42 :     if (sort_field_for_columns >= 0)
                                261              15 :         rankSort(num_columns, array_columns);
                                262                 : 
                                263                 :     /*
                                264                 :      * Fourth part: print the crosstab'ed result.
                                265                 :      */
                                266              42 :     retval = printCrosstab(res,
                                267                 :                            num_columns, array_columns, field_for_columns,
                                268                 :                            num_rows, array_rows, field_for_rows,
                                269                 :                            field_for_data);
                                270                 : 
                                271              66 : error_return:
                                272              66 :     avlFree(&piv_columns, piv_columns.root);
                                273              66 :     avlFree(&piv_rows, piv_rows.root);
                                274              66 :     pg_free(array_columns);
                                275              66 :     pg_free(array_rows);
                                276                 : 
                                277              66 :     return retval;
                                278                 : }
                                279                 : 
                                280                 : /*
                                281                 :  * Output the pivoted resultset with the printTable* functions.  Return true
                                282                 :  * if successful, false otherwise.
                                283                 :  */
                                284                 : static bool
  423 peter                     285              42 : printCrosstab(const PGresult *result,
                                286                 :               int num_columns, pivot_field *piv_columns, int field_for_columns,
                                287                 :               int num_rows, pivot_field *piv_rows, int field_for_rows,
                                288                 :               int field_for_data)
                                289                 : {
 2557 alvherre                  290              42 :     printQueryOpt popt = pset.popt;
                                291                 :     printTableContent cont;
                                292                 :     int         i,
                                293                 :                 rn;
                                294                 :     char        col_align;
                                295                 :     int        *horiz_map;
                                296              42 :     bool        retval = false;
                                297                 : 
                                298              42 :     printTableInit(&cont, &popt.topt, popt.title, num_columns + 1, num_rows);
                                299                 : 
                                300                 :     /* Step 1: set target column names (horizontal header) */
                                301                 : 
                                302                 :     /* The name of the first column is kept unchanged by the pivoting */
                                303              42 :     printTableAddHeader(&cont,
                                304                 :                         PQfname(result, field_for_rows),
                                305                 :                         false,
  423 peter                     306              42 :                         column_type_alignment(PQftype(result,
                                307                 :                                                       field_for_rows)));
                                308                 : 
                                309                 :     /*
                                310                 :      * To iterate over piv_columns[] by piv_columns[].rank, create a reverse
                                311                 :      * map associating each piv_columns[].rank to its index in piv_columns.
                                312                 :      * This avoids an O(N^2) loop later.
                                313                 :      */
 2557 alvherre                  314              42 :     horiz_map = (int *) pg_malloc(sizeof(int) * num_columns);
                                315             231 :     for (i = 0; i < num_columns; i++)
                                316             189 :         horiz_map[piv_columns[i].rank] = i;
                                317                 : 
                                318                 :     /*
                                319                 :      * The display alignment depends on its PQftype().
                                320                 :      */
  423 peter                     321              42 :     col_align = column_type_alignment(PQftype(result, field_for_data));
                                322                 : 
 2557 alvherre                  323             231 :     for (i = 0; i < num_columns; i++)
                                324                 :     {
                                325                 :         char       *colname;
                                326                 : 
                                327             378 :         colname = piv_columns[horiz_map[i]].name ?
                                328             210 :             piv_columns[horiz_map[i]].name :
                                329              21 :             (popt.nullPrint ? popt.nullPrint : "");
                                330                 : 
                                331             189 :         printTableAddHeader(&cont, colname, false, col_align);
                                332                 :     }
                                333              42 :     pg_free(horiz_map);
                                334                 : 
                                335                 :     /* Step 2: set row names in the first output column (vertical header) */
                                336             153 :     for (i = 0; i < num_rows; i++)
                                337                 :     {
                                338             111 :         int         k = piv_rows[i].rank;
                                339                 : 
                                340             111 :         cont.cells[k * (num_columns + 1)] = piv_rows[i].name ?
                                341             117 :             piv_rows[i].name :
                                342               6 :             (popt.nullPrint ? popt.nullPrint : "");
                                343                 :     }
                                344              42 :     cont.cellsadded = num_rows * (num_columns + 1);
                                345                 : 
                                346                 :     /*
                                347                 :      * Step 3: fill in the content cells.
                                348                 :      */
  423 peter                     349             255 :     for (rn = 0; rn < PQntuples(result); rn++)
                                350                 :     {
                                351                 :         int         row_number;
                                352                 :         int         col_number;
                                353                 :         pivot_field *rp,
                                354                 :                    *cp;
                                355                 :         pivot_field elt;
                                356                 : 
                                357                 :         /* Find target row */
                                358             216 :         if (!PQgetisnull(result, rn, field_for_rows))
                                359             210 :             elt.name = PQgetvalue(result, rn, field_for_rows);
                                360                 :         else
 2557 alvherre                  361               6 :             elt.name = NULL;
 2296 tgl                       362             216 :         rp = (pivot_field *) bsearch(&elt,
                                363                 :                                      piv_rows,
                                364                 :                                      num_rows,
                                365                 :                                      sizeof(pivot_field),
                                366                 :                                      pivotFieldCompare);
                                367             216 :         Assert(rp != NULL);
                                368             216 :         row_number = rp->rank;
                                369                 : 
                                370                 :         /* Find target column */
  423 peter                     371             216 :         if (!PQgetisnull(result, rn, field_for_columns))
                                372             195 :             elt.name = PQgetvalue(result, rn, field_for_columns);
                                373                 :         else
 2557 alvherre                  374              21 :             elt.name = NULL;
                                375                 : 
 2296 tgl                       376             216 :         cp = (pivot_field *) bsearch(&elt,
                                377                 :                                      piv_columns,
                                378                 :                                      num_columns,
                                379                 :                                      sizeof(pivot_field),
                                380                 :                                      pivotFieldCompare);
                                381             216 :         Assert(cp != NULL);
                                382             216 :         col_number = cp->rank;
                                383                 : 
                                384                 :         /* Place value into cell */
 2557 alvherre                  385             216 :         if (col_number >= 0 && row_number >= 0)
                                386                 :         {
                                387                 :             int         idx;
                                388                 : 
                                389                 :             /* index into the cont.cells array */
                                390             216 :             idx = 1 + col_number + row_number * (num_columns + 1);
                                391                 : 
                                392                 :             /*
                                393                 :              * If the cell already contains a value, raise an error.
                                394                 :              */
                                395             216 :             if (cont.cells[idx] != NULL)
                                396                 :             {
 1469 peter                     397               3 :                 pg_log_error("\\crosstabview: query result contains multiple data values for row \"%s\", column \"%s\"",
                                398                 :                              rp->name ? rp->name :
                                399                 :                              (popt.nullPrint ? popt.nullPrint : "(null)"),
                                400                 :                              cp->name ? cp->name :
                                401                 :                              (popt.nullPrint ? popt.nullPrint : "(null)"));
 2557 alvherre                  402               3 :                 goto error;
                                403                 :             }
                                404                 : 
  423 peter                     405             426 :             cont.cells[idx] = !PQgetisnull(result, rn, field_for_data) ?
                                406             219 :                 PQgetvalue(result, rn, field_for_data) :
 2557 alvherre                  407               6 :                 (popt.nullPrint ? popt.nullPrint : "");
                                408                 :         }
                                409                 :     }
                                410                 : 
                                411                 :     /*
                                412                 :      * The non-initialized cells must be set to an empty string for the print
                                413                 :      * functions
                                414                 :      */
                                415             576 :     for (i = 0; i < cont.cellsadded; i++)
                                416                 :     {
                                417             537 :         if (cont.cells[i] == NULL)
                                418             246 :             cont.cells[i] = "";
                                419                 :     }
                                420                 : 
                                421              39 :     printTable(&cont, pset.queryFout, false, pset.logfile);
                                422              39 :     retval = true;
                                423                 : 
                                424              42 : error:
                                425              42 :     printTableCleanup(&cont);
                                426                 : 
                                427              42 :     return retval;
                                428                 : }
                                429                 : 
                                430                 : /*
                                431                 :  * The avl* functions below provide a minimalistic implementation of AVL binary
                                432                 :  * trees, to efficiently collect the distinct values that will form the horizontal
                                433                 :  * and vertical headers. It only supports adding new values, no removal or even
                                434                 :  * search.
                                435                 :  */
                                436                 : static void
                                437             132 : avlInit(avl_tree *tree)
                                438                 : {
                                439             132 :     tree->end = (avl_node *) pg_malloc0(sizeof(avl_node));
                                440             132 :     tree->end->children[0] = tree->end->children[1] = tree->end;
                                441             132 :     tree->count = 0;
                                442             132 :     tree->root = tree->end;
                                443             132 : }
                                444                 : 
                                445                 : /* Deallocate recursively an AVL tree, starting from node */
                                446                 : static void
                                447            9945 : avlFree(avl_tree *tree, avl_node *node)
                                448                 : {
                                449            9945 :     if (node->children[0] != tree->end)
                                450                 :     {
                                451            5028 :         avlFree(tree, node->children[0]);
                                452            5028 :         pg_free(node->children[0]);
                                453                 :     }
                                454            9945 :     if (node->children[1] != tree->end)
                                455                 :     {
                                456            4785 :         avlFree(tree, node->children[1]);
                                457            4785 :         pg_free(node->children[1]);
                                458                 :     }
                                459            9945 :     if (node == tree->root)
                                460                 :     {
                                461                 :         /* free the root separately as it's not child of anything */
                                462             132 :         if (node != tree->end)
                                463              90 :             pg_free(node);
                                464                 :         /* free the tree->end struct only once and when all else is freed */
                                465             132 :         pg_free(tree->end);
                                466                 :     }
                                467            9945 : }
                                468                 : 
                                469                 : /* Set the height to 1 plus the greatest of left and right heights */
                                470                 : static void
                                471          112743 : avlUpdateHeight(avl_node *n)
                                472                 : {
                                473          112743 :     n->height = 1 + (n->children[0]->height > n->children[1]->height ?
                                474          112743 :                      n->children[0]->height :
                                475                 :                      n->children[1]->height);
                                476          112743 : }
                                477                 : 
                                478                 : /* Rotate a subtree left (dir=0) or right (dir=1). Not recursive */
                                479                 : static avl_node *
                                480           12105 : avlRotate(avl_node **current, int dir)
                                481                 : {
                                482           12105 :     avl_node   *before = *current;
                                483           12105 :     avl_node   *after = (*current)->children[dir];
                                484                 : 
                                485           12105 :     *current = after;
                                486           12105 :     before->children[dir] = after->children[!dir];
                                487           12105 :     avlUpdateHeight(before);
                                488           12105 :     after->children[!dir] = before;
                                489                 : 
                                490           12105 :     return after;
                                491                 : }
                                492                 : 
                                493                 : static int
                                494          109551 : avlBalance(avl_node *n)
                                495                 : {
                                496          109551 :     return n->children[0]->height - n->children[1]->height;
                                497                 : }
                                498                 : 
                                499                 : /*
                                500                 :  * After an insertion, possibly rebalance the tree so that the left and right
                                501                 :  * node heights don't differ by more than 1.
                                502                 :  * May update *node.
                                503                 :  */
                                504                 : static void
                                505          100638 : avlAdjustBalance(avl_tree *tree, avl_node **node)
                                506                 : {
                                507          100638 :     avl_node   *current = *node;
                                508          100638 :     int         b = avlBalance(current) / 2;
                                509                 : 
                                510          100638 :     if (b != 0)
                                511                 :     {
                                512            8913 :         int         dir = (1 - b) / 2;
                                513                 : 
                                514            8913 :         if (avlBalance(current->children[dir]) == -b)
                                515            3192 :             avlRotate(&current->children[dir], !dir);
                                516            8913 :         current = avlRotate(node, dir);
                                517                 :     }
                                518          100638 :     if (current != tree->end)
                                519          100638 :         avlUpdateHeight(current);
                                520          100638 : }
                                521                 : 
                                522                 : /*
                                523                 :  * Insert a new value/field, starting from *node, reaching the correct position
                                524                 :  * in the tree by recursion.  Possibly rebalance the tree and possibly update
                                525                 :  * *node.  Do nothing if the value is already present in the tree.
                                526                 :  */
                                527                 : static void
                                528          110673 : avlInsertNode(avl_tree *tree, avl_node **node, pivot_field field)
                                529                 : {
                                530          110673 :     avl_node   *current = *node;
                                531                 : 
                                532          110673 :     if (current == tree->end)
                                533                 :     {
                                534                 :         avl_node   *new_node = (avl_node *)
                                535            9903 :         pg_malloc(sizeof(avl_node));
                                536                 : 
                                537            9903 :         new_node->height = 1;
                                538            9903 :         new_node->field = field;
                                539            9903 :         new_node->children[0] = new_node->children[1] = tree->end;
                                540            9903 :         tree->count++;
                                541            9903 :         *node = new_node;
                                542                 :     }
                                543                 :     else
                                544                 :     {
                                545          100770 :         int         cmp = pivotFieldCompare(&field, &current->field);
                                546                 : 
                                547          100770 :         if (cmp != 0)
                                548                 :         {
                                549          100638 :             avlInsertNode(tree,
                                550                 :                           cmp > 0 ? &current->children[1] : &current->children[0],
                                551                 :                           field);
                                552          100638 :             avlAdjustBalance(tree, node);
                                553                 :         }
                                554                 :     }
                                555          110673 : }
                                556                 : 
                                557                 : /* Insert the value into the AVL tree, if it does not preexist */
                                558                 : static void
                                559           10035 : avlMergeValue(avl_tree *tree, char *name, char *sort_value)
                                560                 : {
                                561                 :     pivot_field field;
                                562                 : 
                                563           10035 :     field.name = name;
                                564           10035 :     field.rank = tree->count;
                                565           10035 :     field.sort_value = sort_value;
                                566           10035 :     avlInsertNode(tree, &tree->root, field);
                                567           10035 : }
                                568                 : 
                                569                 : /*
                                570                 :  * Recursively extract node values into the names array, in sorted order with a
                                571                 :  * left-to-right tree traversal.
                                572                 :  * Return the next candidate offset to write into the names array.
                                573                 :  * fields[] must be preallocated to hold tree->count entries
                                574                 :  */
                                575                 : static int
                                576             684 : avlCollectFields(avl_tree *tree, avl_node *node, pivot_field *fields, int idx)
                                577                 : {
                                578             684 :     if (node == tree->end)
                                579             384 :         return idx;
                                580                 : 
                                581             300 :     idx = avlCollectFields(tree, node->children[0], fields, idx);
                                582             300 :     fields[idx] = node->field;
                                583             300 :     return avlCollectFields(tree, node->children[1], fields, idx + 1);
                                584                 : }
                                585                 : 
                                586                 : static void
                                587              15 : rankSort(int num_columns, pivot_field *piv_columns)
                                588                 : {
                                589                 :     int        *hmap;           /* [[offset in piv_columns, rank], ...for
                                590                 :                                  * every header entry] */
                                591                 :     int         i;
                                592                 : 
                                593              15 :     hmap = (int *) pg_malloc(sizeof(int) * num_columns * 2);
                                594              78 :     for (i = 0; i < num_columns; i++)
                                595                 :     {
                                596              63 :         char       *val = piv_columns[i].sort_value;
                                597                 : 
                                598                 :         /* ranking information is valid if non null and matches /^-?\d+$/ */
                                599              63 :         if (val &&
                                600              63 :             ((*val == '-' &&
 2557 alvherre                  601 UBC           0 :               strspn(val + 1, "0123456789") == strlen(val + 1)) ||
 2557 alvherre                  602 CBC          63 :              strspn(val, "0123456789") == strlen(val)))
                                603                 :         {
                                604              63 :             hmap[i * 2] = atoi(val);
                                605              63 :             hmap[i * 2 + 1] = i;
                                606                 :         }
                                607                 :         else
                                608                 :         {
                                609                 :             /* invalid rank information ignored (equivalent to rank 0) */
 2557 alvherre                  610 UBC           0 :             hmap[i * 2] = 0;
                                611               0 :             hmap[i * 2 + 1] = i;
                                612                 :         }
                                613                 :     }
                                614                 : 
 2557 alvherre                  615 CBC          15 :     qsort(hmap, num_columns, sizeof(int) * 2, rankCompare);
                                616                 : 
                                617              78 :     for (i = 0; i < num_columns; i++)
                                618                 :     {
                                619              63 :         piv_columns[hmap[i * 2 + 1]].rank = i;
                                620                 :     }
                                621                 : 
                                622              15 :     pg_free(hmap);
                                623              15 : }
                                624                 : 
                                625                 : /*
                                626                 :  * Look up a column reference, which can be either:
                                627                 :  * - a number from 1 to PQnfields(res)
                                628                 :  * - a column name matching one of PQfname(res,...)
                                629                 :  *
                                630                 :  * Returns zero-based column number, or -1 if not found or ambiguous.
                                631                 :  *
                                632                 :  * Note: may modify contents of "arg" string.
                                633                 :  */
                                634                 : static int
 2551 tgl                       635             144 : indexOfColumn(char *arg, const PGresult *res)
                                636                 : {
                                637                 :     int         idx;
                                638                 : 
                                639             144 :     if (arg[0] && strspn(arg, "0123456789") == strlen(arg))
                                640                 :     {
                                641                 :         /* if arg contains only digits, it's a column number */
 2557 alvherre                  642              48 :         idx = atoi(arg) - 1;
                                643              48 :         if (idx < 0 || idx >= PQnfields(res))
                                644                 :         {
 1469 peter                     645               3 :             pg_log_error("\\crosstabview: column number %d is out of range 1..%d",
                                646                 :                          idx + 1, PQnfields(res));
 2557 alvherre                  647               3 :             return -1;
                                648                 :         }
                                649                 :     }
                                650                 :     else
                                651                 :     {
                                652                 :         int         i;
                                653                 : 
                                654                 :         /*
                                655                 :          * Dequote and downcase the column name.  By checking for all-digits
                                656                 :          * before doing this, we can ensure that a quoted name is treated as a
                                657                 :          * name even if it's all digits.
                                658                 :          */
 2548 tgl                       659              96 :         dequote_downcase_identifier(arg, true, pset.encoding);
                                660                 : 
                                661                 :         /* Now look for match(es) among res' column names */
 2557 alvherre                  662              96 :         idx = -1;
                                663             456 :         for (i = 0; i < PQnfields(res); i++)
                                664                 :         {
 2551 tgl                       665             360 :             if (strcmp(arg, PQfname(res, i)) == 0)
                                666                 :             {
 2557 alvherre                  667              87 :                 if (idx >= 0)
                                668                 :                 {
                                669                 :                     /* another idx was already found for the same name */
 1469 peter                     670 UBC           0 :                     pg_log_error("\\crosstabview: ambiguous column name: \"%s\"", arg);
 2557 alvherre                  671               0 :                     return -1;
                                672                 :                 }
 2557 alvherre                  673 CBC          87 :                 idx = i;
                                674                 :             }
                                675                 :         }
                                676              96 :         if (idx == -1)
                                677                 :         {
 1469 peter                     678               9 :             pg_log_error("\\crosstabview: column name not found: \"%s\"", arg);
 2557 alvherre                  679               9 :             return -1;
                                680                 :         }
                                681                 :     }
                                682                 : 
                                683             132 :     return idx;
                                684                 : }
                                685                 : 
                                686                 : /*
                                687                 :  * Value comparator for vertical and horizontal headers
                                688                 :  * used for deduplication only.
                                689                 :  * - null values are considered equal
                                690                 :  * - non-null < null
                                691                 :  * - non-null values are compared with strcmp()
                                692                 :  */
                                693                 : static int
                                694          101568 : pivotFieldCompare(const void *a, const void *b)
                                695                 : {
 2296 tgl                       696          101568 :     const pivot_field *pa = (const pivot_field *) a;
                                697          101568 :     const pivot_field *pb = (const pivot_field *) b;
                                698                 : 
                                699                 :     /* test null values */
 2557 alvherre                  700          101568 :     if (!pb->name)
                                701              66 :         return pa->name ? -1 : 0;
                                702          101502 :     else if (!pa->name)
                                703              51 :         return 1;
                                704                 : 
                                705                 :     /* non-null values */
 2296 tgl                       706          101451 :     return strcmp(pa->name, pb->name);
                                707                 : }
                                708                 : 
                                709                 : static int
 2557 alvherre                  710              72 : rankCompare(const void *a, const void *b)
                                711                 : {
 2296 tgl                       712              72 :     return *((const int *) a) - *((const int *) b);
                                713                 : }
        

Generated by: LCOV version v1.16-55-g56c0a2a