Branch data     Line data    Source code

       1                 :             : /*-------------------------------------------------------------------------
       2                 :             :  *
       3                 :             :  * Facilities for frontend code to query a databases.
       4                 :             :  *
       5                 :             :  * Portions Copyright (c) 1996-2026, PostgreSQL Global Development Group
       6                 :             :  * Portions Copyright (c) 1994, Regents of the University of California
       7                 :             :  *
       8                 :             :  * src/fe_utils/query_utils.c
       9                 :             :  *
      10                 :             :  *-------------------------------------------------------------------------
      11                 :             :  */
      12                 :             : #include "postgres_fe.h"
      13                 :             : 
      14                 :             : #include "common/logging.h"
      15                 :             : #include "fe_utils/cancel.h"
      16                 :             : #include "fe_utils/query_utils.h"
      17                 :             : 
      18                 :             : /*
      19                 :             :  * Run a query, return the results, exit program on failure.
      20                 :             :  */
      21                 :             : PGresult *
      22                 :         873 : executeQuery(PGconn *conn, const char *query, bool echo)
      23                 :             : {
      24                 :             :     PGresult   *res;
      25                 :             : 
      26         [ +  + ]:         873 :     if (echo)
      27                 :          42 :         printf("%s\n", query);
      28                 :             : 
      29                 :         873 :     res = PQexec(conn, query);
      30   [ +  -  +  + ]:        1746 :     if (!res ||
      31                 :         873 :         PQresultStatus(res) != PGRES_TUPLES_OK)
      32                 :             :     {
      33                 :           2 :         pg_log_error("query failed: %s", PQerrorMessage(conn));
      34                 :           2 :         pg_log_error_detail("Query was: %s", query);
      35                 :           2 :         PQfinish(conn);
      36                 :           2 :         exit(1);
      37                 :             :     }
      38                 :             : 
      39                 :         871 :     return res;
      40                 :             : }
      41                 :             : 
      42                 :             : 
      43                 :             : /*
      44                 :             :  * As above for a SQL command (which returns nothing).
      45                 :             :  */
      46                 :             : void
      47                 :         173 : executeCommand(PGconn *conn, const char *query, bool echo)
      48                 :             : {
      49                 :             :     PGresult   *res;
      50                 :             : 
      51         [ +  + ]:         173 :     if (echo)
      52                 :           7 :         printf("%s\n", query);
      53                 :             : 
      54                 :         173 :     res = PQexec(conn, query);
      55   [ +  -  -  + ]:         346 :     if (!res ||
      56                 :         173 :         PQresultStatus(res) != PGRES_COMMAND_OK)
      57                 :             :     {
      58                 :           0 :         pg_log_error("query failed: %s", PQerrorMessage(conn));
      59                 :           0 :         pg_log_error_detail("Query was: %s", query);
      60                 :           0 :         PQfinish(conn);
      61                 :           0 :         exit(1);
      62                 :             :     }
      63                 :             : 
      64                 :         173 :     PQclear(res);
      65                 :         173 : }
      66                 :             : 
      67                 :             : 
      68                 :             : /*
      69                 :             :  * As above for a SQL maintenance command (returns command success).
      70                 :             :  * Command is executed with a cancel handler set, so Ctrl-C can
      71                 :             :  * interrupt it.
      72                 :             :  */
      73                 :             : bool
      74                 :          15 : executeMaintenanceCommand(PGconn *conn, const char *query, bool echo)
      75                 :             : {
      76                 :             :     PGresult   *res;
      77                 :             :     bool        r;
      78                 :             : 
      79         [ +  + ]:          15 :     if (echo)
      80                 :           7 :         printf("%s\n", query);
      81                 :             : 
      82                 :          15 :     SetCancelConn(conn);
      83                 :          15 :     res = PQexec(conn, query);
      84                 :          15 :     ResetCancelConn();
      85                 :             : 
      86   [ +  -  +  - ]:          15 :     r = (res && PQresultStatus(res) == PGRES_COMMAND_OK);
      87                 :             : 
      88                 :          15 :     PQclear(res);
      89                 :             : 
      90                 :          15 :     return r;
      91                 :             : }