Age Owner Branch data TLA Line data Source code
1 : : /*-------------------------------------------------------------------------
2 : : *
3 : : * Facilities for frontend code to query a databases.
4 : : *
5 : : * Portions Copyright (c) 1996-2024, 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 *
1164 rhaas@postgresql.org 22 :CBC 684 : executeQuery(PGconn *conn, const char *query, bool echo)
23 : : {
24 : : PGresult *res;
25 : :
26 [ + + ]: 684 : if (echo)
27 : 47 : printf("%s\n", query);
28 : :
29 : 684 : res = PQexec(conn, query);
30 [ + - + + ]: 1368 : if (!res ||
31 : 684 : PQresultStatus(res) != PGRES_TUPLES_OK)
32 : : {
33 : 2 : pg_log_error("query failed: %s", PQerrorMessage(conn));
737 tgl@sss.pgh.pa.us 34 : 2 : pg_log_error_detail("Query was: %s", query);
1164 rhaas@postgresql.org 35 : 2 : PQfinish(conn);
36 : 2 : exit(1);
37 : : }
38 : :
39 : 682 : return res;
40 : : }
41 : :
42 : :
43 : : /*
44 : : * As above for a SQL command (which returns nothing).
45 : : */
46 : : void
47 : 113 : executeCommand(PGconn *conn, const char *query, bool echo)
48 : : {
49 : : PGresult *res;
50 : :
51 [ + + ]: 113 : if (echo)
52 : 8 : printf("%s\n", query);
53 : :
54 : 113 : res = PQexec(conn, query);
55 [ + - - + ]: 226 : if (!res ||
56 : 113 : PQresultStatus(res) != PGRES_COMMAND_OK)
57 : : {
1164 rhaas@postgresql.org 58 :UBC 0 : pg_log_error("query failed: %s", PQerrorMessage(conn));
737 tgl@sss.pgh.pa.us 59 : 0 : pg_log_error_detail("Query was: %s", query);
1164 rhaas@postgresql.org 60 : 0 : PQfinish(conn);
61 : 0 : exit(1);
62 : : }
63 : :
1164 rhaas@postgresql.org 64 :CBC 113 : PQclear(res);
65 : 113 : }
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 : 16 : executeMaintenanceCommand(PGconn *conn, const char *query, bool echo)
75 : : {
76 : : PGresult *res;
77 : : bool r;
78 : :
79 [ + + ]: 16 : if (echo)
80 : 8 : printf("%s\n", query);
81 : :
82 : 16 : SetCancelConn(conn);
83 : 16 : res = PQexec(conn, query);
84 : 16 : ResetCancelConn();
85 : :
86 [ + - + - ]: 16 : r = (res && PQresultStatus(res) == PGRES_COMMAND_OK);
87 : :
651 peter@eisentraut.org 88 : 16 : PQclear(res);
89 : :
1164 rhaas@postgresql.org 90 : 16 : return r;
91 : : }
|
