Age Owner Branch data TLA Line data Source code
1 : : /* ----------
2 : : * backend_progress.c
3 : : *
4 : : * Command progress reporting infrastructure.
5 : : *
6 : : * Copyright (c) 2001-2025, PostgreSQL Global Development Group
7 : : *
8 : : * src/backend/utils/activity/backend_progress.c
9 : : * ----------
10 : : */
11 : : #include "postgres.h"
12 : :
13 : : #include "access/parallel.h"
14 : : #include "libpq/pqformat.h"
15 : : #include "utils/backend_progress.h"
16 : : #include "utils/backend_status.h"
17 : :
18 : :
19 : : /*-----------
20 : : * pgstat_progress_start_command() -
21 : : *
22 : : * Set st_progress_command (and st_progress_command_target) in own backend
23 : : * entry. Also, zero-initialize st_progress_param array.
24 : : *-----------
25 : : */
26 : : void
1617 andres@anarazel.de 27 :CBC 43629 : pgstat_progress_start_command(ProgressCommandType cmdtype, Oid relid)
28 : : {
29 : 43629 : volatile PgBackendStatus *beentry = MyBEEntry;
30 : :
31 [ + - - + ]: 43629 : if (!beentry || !pgstat_track_activities)
1617 andres@anarazel.de 32 :UBC 0 : return;
33 : :
1617 andres@anarazel.de 34 :CBC 43629 : PGSTAT_BEGIN_WRITE_ACTIVITY(beentry);
35 : 43629 : beentry->st_progress_command = cmdtype;
36 : 43629 : beentry->st_progress_command_target = relid;
37 [ + - + - : 916209 : MemSet(&beentry->st_progress_param, 0, sizeof(beentry->st_progress_param));
+ - + - +
+ ]
38 [ - + - + ]: 43629 : PGSTAT_END_WRITE_ACTIVITY(beentry);
39 : : }
40 : :
41 : : /*-----------
42 : : * pgstat_progress_update_param() -
43 : : *
44 : : * Update index'th member in st_progress_param[] of own backend entry.
45 : : *-----------
46 : : */
47 : : void
48 : 11428865 : pgstat_progress_update_param(int index, int64 val)
49 : : {
50 : 11428865 : volatile PgBackendStatus *beentry = MyBEEntry;
51 : :
52 [ + - - + ]: 11428865 : Assert(index >= 0 && index < PGSTAT_NUM_PROGRESS_PARAM);
53 : :
54 [ + - - + ]: 11428865 : if (!beentry || !pgstat_track_activities)
1617 andres@anarazel.de 55 :UBC 0 : return;
56 : :
1617 andres@anarazel.de 57 :CBC 11428865 : PGSTAT_BEGIN_WRITE_ACTIVITY(beentry);
58 : 11428865 : beentry->st_progress_param[index] = val;
59 [ - + - + ]: 11428865 : PGSTAT_END_WRITE_ACTIVITY(beentry);
60 : : }
61 : :
62 : : /*-----------
63 : : * pgstat_progress_incr_param() -
64 : : *
65 : : * Increment index'th member in st_progress_param[] of own backend entry.
66 : : *-----------
67 : : */
68 : : void
896 tgl@sss.pgh.pa.us 69 : 1348 : pgstat_progress_incr_param(int index, int64 incr)
70 : : {
71 : 1348 : volatile PgBackendStatus *beentry = MyBEEntry;
72 : :
73 [ + - - + ]: 1348 : Assert(index >= 0 && index < PGSTAT_NUM_PROGRESS_PARAM);
74 : :
75 [ + - - + ]: 1348 : if (!beentry || !pgstat_track_activities)
896 tgl@sss.pgh.pa.us 76 :UBC 0 : return;
77 : :
896 tgl@sss.pgh.pa.us 78 :CBC 1348 : PGSTAT_BEGIN_WRITE_ACTIVITY(beentry);
79 : 1348 : beentry->st_progress_param[index] += incr;
80 [ - + - + ]: 1348 : PGSTAT_END_WRITE_ACTIVITY(beentry);
81 : : }
82 : :
83 : : /*-----------
84 : : * pgstat_progress_parallel_incr_param() -
85 : : *
86 : : * A variant of pgstat_progress_incr_param to allow a worker to poke at
87 : : * a leader to do an incremental progress update.
88 : : *-----------
89 : : */
90 : : void
788 msawada@postgresql.o 91 : 94 : pgstat_progress_parallel_incr_param(int index, int64 incr)
92 : : {
93 : : /*
94 : : * Parallel workers notify a leader through a PqMsg_Progress message to
95 : : * update progress, passing the progress index and incremented value.
96 : : * Leaders can just call pgstat_progress_incr_param directly.
97 : : */
98 [ - + ]: 94 : if (IsParallelWorker())
99 : : {
100 : : static StringInfoData progress_message;
101 : :
788 msawada@postgresql.o 102 :UBC 0 : initStringInfo(&progress_message);
103 : :
416 nathan@postgresql.or 104 : 0 : pq_beginmessage(&progress_message, PqMsg_Progress);
788 msawada@postgresql.o 105 : 0 : pq_sendint32(&progress_message, index);
106 : 0 : pq_sendint64(&progress_message, incr);
107 : 0 : pq_endmessage(&progress_message);
108 : : }
109 : : else
788 msawada@postgresql.o 110 :CBC 94 : pgstat_progress_incr_param(index, incr);
111 : 94 : }
112 : :
113 : : /*-----------
114 : : * pgstat_progress_update_multi_param() -
115 : : *
116 : : * Update multiple members in st_progress_param[] of own backend entry.
117 : : * This is atomic; readers won't see intermediate states.
118 : : *-----------
119 : : */
120 : : void
1617 andres@anarazel.de 121 : 579977 : pgstat_progress_update_multi_param(int nparam, const int *index,
122 : : const int64 *val)
123 : : {
124 : 579977 : volatile PgBackendStatus *beentry = MyBEEntry;
125 : : int i;
126 : :
127 [ + - + - : 579977 : if (!beentry || !pgstat_track_activities || nparam == 0)
- + ]
1617 andres@anarazel.de 128 :UBC 0 : return;
129 : :
1617 andres@anarazel.de 130 :CBC 579977 : PGSTAT_BEGIN_WRITE_ACTIVITY(beentry);
131 : :
132 [ + + ]: 1525983 : for (i = 0; i < nparam; ++i)
133 : : {
134 [ + - - + ]: 946006 : Assert(index[i] >= 0 && index[i] < PGSTAT_NUM_PROGRESS_PARAM);
135 : :
136 : 946006 : beentry->st_progress_param[index[i]] = val[i];
137 : : }
138 : :
139 [ - + - + ]: 579977 : PGSTAT_END_WRITE_ACTIVITY(beentry);
140 : : }
141 : :
142 : : /*-----------
143 : : * pgstat_progress_end_command() -
144 : : *
145 : : * Reset st_progress_command (and st_progress_command_target) in own backend
146 : : * entry. This signals the end of the command.
147 : : *-----------
148 : : */
149 : : void
150 : 71940 : pgstat_progress_end_command(void)
151 : : {
152 : 71940 : volatile PgBackendStatus *beentry = MyBEEntry;
153 : :
154 [ + - - + ]: 71940 : if (!beentry || !pgstat_track_activities)
1617 andres@anarazel.de 155 :UBC 0 : return;
156 : :
1617 andres@anarazel.de 157 [ + + ]:CBC 71940 : if (beentry->st_progress_command == PROGRESS_COMMAND_INVALID)
158 : 29057 : return;
159 : :
160 : 42883 : PGSTAT_BEGIN_WRITE_ACTIVITY(beentry);
161 : 42883 : beentry->st_progress_command = PROGRESS_COMMAND_INVALID;
162 : 42883 : beentry->st_progress_command_target = InvalidOid;
163 [ - + - + ]: 42883 : PGSTAT_END_WRITE_ACTIVITY(beentry);
164 : : }
|