Age Owner Branch data TLA Line data Source code
1 : : /* -------------------------------------------------------------------------
2 : : *
3 : : * pgstat_subscription.c
4 : : * Implementation of subscription statistics.
5 : : *
6 : : * This file contains the implementation of subscription statistics. It is kept
7 : : * separate from pgstat.c to enforce the line between the statistics access /
8 : : * storage implementation and the details about individual types of
9 : : * statistics.
10 : : *
11 : : * Copyright (c) 2001-2024, PostgreSQL Global Development Group
12 : : *
13 : : * IDENTIFICATION
14 : : * src/backend/utils/activity/pgstat_subscription.c
15 : : * -------------------------------------------------------------------------
16 : : */
17 : :
18 : : #include "postgres.h"
19 : :
20 : : #include "utils/pgstat_internal.h"
21 : :
22 : :
23 : : /*
24 : : * Report a subscription error.
25 : : */
26 : : void
755 andres@anarazel.de 27 :CBC 65 : pgstat_report_subscription_error(Oid subid, bool is_apply_error)
28 : : {
29 : : PgStat_EntryRef *entry_ref;
30 : : PgStat_BackendSubEntry *pending;
31 : :
739 32 : 65 : entry_ref = pgstat_prep_pending_entry(PGSTAT_KIND_SUBSCRIPTION,
33 : : InvalidOid, subid, NULL);
34 : 65 : pending = entry_ref->pending;
35 : :
36 [ + + ]: 65 : if (is_apply_error)
37 : 57 : pending->apply_error_count++;
38 : : else
39 : 8 : pending->sync_error_count++;
755 40 : 65 : }
41 : :
42 : : /*
43 : : * Report creating the subscription.
44 : : */
45 : : void
739 46 : 138 : pgstat_create_subscription(Oid subid)
47 : : {
48 : : /* Ensures that stats are dropped if transaction rolls back */
49 : 138 : pgstat_create_transactional(PGSTAT_KIND_SUBSCRIPTION,
50 : : InvalidOid, subid);
51 : :
52 : : /* Create and initialize the subscription stats entry */
556 53 : 138 : pgstat_get_entry_ref(PGSTAT_KIND_SUBSCRIPTION, InvalidOid, subid,
54 : : true, NULL);
55 : 138 : pgstat_reset_entry(PGSTAT_KIND_SUBSCRIPTION, InvalidOid, subid, 0);
739 56 : 138 : }
57 : :
58 : : /*
59 : : * Report dropping the subscription.
60 : : *
61 : : * Ensures that stats are dropped if transaction commits.
62 : : */
63 : : void
64 : 87 : pgstat_drop_subscription(Oid subid)
65 : : {
66 : 87 : pgstat_drop_transactional(PGSTAT_KIND_SUBSCRIPTION,
67 : : InvalidOid, subid);
755 68 : 87 : }
69 : :
70 : : /*
71 : : * Support function for the SQL-callable pgstat* functions. Returns
72 : : * the collected statistics for one subscription or NULL.
73 : : */
74 : : PgStat_StatSubEntry *
739 75 : 29 : pgstat_fetch_stat_subscription(Oid subid)
76 : : {
77 : 29 : return (PgStat_StatSubEntry *)
78 : 29 : pgstat_fetch_entry(PGSTAT_KIND_SUBSCRIPTION, InvalidOid, subid);
79 : : }
80 : :
81 : : /*
82 : : * Flush out pending stats for the entry
83 : : *
84 : : * If nowait is true, this function returns false if lock could not
85 : : * immediately acquired, otherwise true is returned.
86 : : */
87 : : bool
88 : 65 : pgstat_subscription_flush_cb(PgStat_EntryRef *entry_ref, bool nowait)
89 : : {
90 : : PgStat_BackendSubEntry *localent;
91 : : PgStatShared_Subscription *shsubent;
92 : :
93 : 65 : localent = (PgStat_BackendSubEntry *) entry_ref->pending;
94 : 65 : shsubent = (PgStatShared_Subscription *) entry_ref->shared_stats;
95 : :
96 : : /* localent always has non-zero content */
97 : :
98 [ - + ]: 65 : if (!pgstat_lock_entry(entry_ref, nowait))
739 andres@anarazel.de 99 :UBC 0 : return false;
100 : :
101 : : #define SUB_ACC(fld) shsubent->stats.fld += localent->fld
739 andres@anarazel.de 102 :CBC 65 : SUB_ACC(apply_error_count);
103 : 65 : SUB_ACC(sync_error_count);
104 : : #undef SUB_ACC
105 : :
106 : 65 : pgstat_unlock_entry(entry_ref);
107 : 65 : return true;
108 : : }
109 : :
110 : : void
111 : 150 : pgstat_subscription_reset_timestamp_cb(PgStatShared_Common *header, TimestampTz ts)
112 : : {
113 : 150 : ((PgStatShared_Subscription *) header)->stats.stat_reset_timestamp = ts;
114 : 150 : }
|