granicus.if.org Git - postgresql/blob - src/backend/postmaster/pgstat.c

   1 /* ----------
   2  * pgstat.c
   3  *
   4  *      All the statistics collector stuff hacked up in one big, ugly file.
   5  *
   6  *      TODO:   - Separate collector, postmaster and backend stuff
   7  *                        into different files.
   8  *
   9  *                      - Add some automatic call for pgstat vacuuming.
  10  *
  11  *                      - Add a pgstat config column to pg_database, so this
  12  *                        entire thing can be enabled/disabled on a per db basis.
  13  *
  14  *      Copyright (c) 2001-2017, PostgreSQL Global Development Group
  15  *
  16  *      src/backend/postmaster/pgstat.c
  17  * ----------
  18  */
  19 #include "postgres.h"
  20
  21 #include <unistd.h>
  22 #include <fcntl.h>
  23 #include <sys/param.h>
  24 #include <sys/time.h>
  25 #include <sys/socket.h>
  26 #include <netdb.h>
  27 #include <netinet/in.h>
  28 #include <arpa/inet.h>
  29 #include <signal.h>
  30 #include <time.h>
  31 #ifdef HAVE_SYS_SELECT_H
  32 #include <sys/select.h>
  33 #endif
  34
  35 #include "pgstat.h"
  36
  37 #include "access/heapam.h"
  38 #include "access/htup_details.h"
  39 #include "access/transam.h"
  40 #include "access/twophase_rmgr.h"
  41 #include "access/xact.h"
  42 #include "catalog/pg_database.h"
  43 #include "catalog/pg_proc.h"
  44 #include "common/ip.h"
  45 #include "libpq/libpq.h"
  46 #include "libpq/pqsignal.h"
  47 #include "mb/pg_wchar.h"
  48 #include "miscadmin.h"
  49 #include "pg_trace.h"
  50 #include "postmaster/autovacuum.h"
  51 #include "postmaster/fork_process.h"
  52 #include "postmaster/postmaster.h"
  53 #include "replication/walsender.h"
  54 #include "storage/backendid.h"
  55 #include "storage/dsm.h"
  56 #include "storage/fd.h"
  57 #include "storage/ipc.h"
  58 #include "storage/latch.h"
  59 #include "storage/lmgr.h"
  60 #include "storage/pg_shmem.h"
  61 #include "storage/procsignal.h"
  62 #include "storage/sinvaladt.h"
  63 #include "utils/ascii.h"
  64 #include "utils/guc.h"
  65 #include "utils/memutils.h"
  66 #include "utils/ps_status.h"
  67 #include "utils/rel.h"
  68 #include "utils/snapmgr.h"
  69 #include "utils/timestamp.h"
  70 #include "utils/tqual.h"
  71
  72
  73 /* ----------
  74  * Timer definitions.
  75  * ----------
  76  */
  77 #define PGSTAT_STAT_INTERVAL    500 /* Minimum time between stats file
  78                                                                          * updates; in milliseconds. */
  79
  80 #define PGSTAT_RETRY_DELAY              10      /* How long to wait between checks for a
  81                                                                          * new file; in milliseconds. */
  82
  83 #define PGSTAT_MAX_WAIT_TIME    10000   /* Maximum time to wait for a stats
  84                                                                                  * file update; in milliseconds. */
  85
  86 #define PGSTAT_INQ_INTERVAL             640 /* How often to ping the collector for a
  87                                                                          * new file; in milliseconds. */
  88
  89 #define PGSTAT_RESTART_INTERVAL 60      /* How often to attempt to restart a
  90                                                                          * failed statistics collector; in
  91                                                                          * seconds. */
  92
  93 #define PGSTAT_POLL_LOOP_COUNT  (PGSTAT_MAX_WAIT_TIME / PGSTAT_RETRY_DELAY)
  94 #define PGSTAT_INQ_LOOP_COUNT   (PGSTAT_INQ_INTERVAL / PGSTAT_RETRY_DELAY)
  95
  96 /* Minimum receive buffer size for the collector's socket. */
  97 #define PGSTAT_MIN_RCVBUF               (100 * 1024)
  98
  99
 100 /* ----------
 101  * The initial size hints for the hash tables used in the collector.
 102  * ----------
 103  */
 104 #define PGSTAT_DB_HASH_SIZE             16
 105 #define PGSTAT_TAB_HASH_SIZE    512
 106 #define PGSTAT_FUNCTION_HASH_SIZE       512
 107
 108
 109 /* ----------
 110  * Total number of backends including auxiliary
 111  *
 112  * We reserve a slot for each possible BackendId, plus one for each
 113  * possible auxiliary process type.  (This scheme assumes there is not
 114  * more than one of any auxiliary process type at a time.) MaxBackends
 115  * includes autovacuum workers and background workers as well.
 116  * ----------
 117  */
 118 #define NumBackendStatSlots (MaxBackends + NUM_AUXPROCTYPES)
 119
 120
 121 /* ----------
 122  * GUC parameters
 123  * ----------
 124  */
 125 bool            pgstat_track_activities = false;
 126 bool            pgstat_track_counts = false;
 127 int                     pgstat_track_functions = TRACK_FUNC_OFF;
 128 int                     pgstat_track_activity_query_size = 1024;
 129
 130 /* ----------
 131  * Built from GUC parameter
 132  * ----------
 133  */
 134 char       *pgstat_stat_directory = NULL;
 135 char       *pgstat_stat_filename = NULL;
 136 char       *pgstat_stat_tmpname = NULL;
 137
 138 /*
 139  * BgWriter global statistics counters (unused in other processes).
 140  * Stored directly in a stats message structure so it can be sent
 141  * without needing to copy things around.  We assume this inits to zeroes.
 142  */
 143 PgStat_MsgBgWriter BgWriterStats;
 144
 145 /* ----------
 146  * Local data
 147  * ----------
 148  */
 149 NON_EXEC_STATIC pgsocket pgStatSock = PGINVALID_SOCKET;
 150
 151 static struct sockaddr_storage pgStatAddr;
 152
 153 static time_t last_pgstat_start_time;
 154
 155 static bool pgStatRunningInCollector = false;
 156
 157 /*
 158  * Structures in which backends store per-table info that's waiting to be
 159  * sent to the collector.
 160  *
 161  * NOTE: once allocated, TabStatusArray structures are never moved or deleted
 162  * for the life of the backend.  Also, we zero out the t_id fields of the
 163  * contained PgStat_TableStatus structs whenever they are not actively in use.
 164  * This allows relcache pgstat_info pointers to be treated as long-lived data,
 165  * avoiding repeated searches in pgstat_initstats() when a relation is
 166  * repeatedly opened during a transaction.
 167  */
 168 #define TABSTAT_QUANTUM         100 /* we alloc this many at a time */
 169
 170 typedef struct TabStatusArray
 171 {
 172         struct TabStatusArray *tsa_next;        /* link to next array, if any */
 173         int                     tsa_used;               /* # entries currently used */
 174         PgStat_TableStatus tsa_entries[TABSTAT_QUANTUM];        /* per-table data */
 175 } TabStatusArray;
 176
 177 static TabStatusArray *pgStatTabList = NULL;
 178
 179 /*
 180  * pgStatTabHash entry: map from relation OID to PgStat_TableStatus pointer
 181  */
 182 typedef struct TabStatHashEntry
 183 {
 184         Oid                     t_id;
 185         PgStat_TableStatus *tsa_entry;
 186 } TabStatHashEntry;
 187
 188 /*
 189  * Hash table for O(1) t_id -> tsa_entry lookup
 190  */
 191 static HTAB *pgStatTabHash = NULL;
 192
 193 /*
 194  * Backends store per-function info that's waiting to be sent to the collector
 195  * in this hash table (indexed by function OID).
 196  */
 197 static HTAB *pgStatFunctions = NULL;
 198
 199 /*
 200  * Indicates if backend has some function stats that it hasn't yet
 201  * sent to the collector.
 202  */
 203 static bool have_function_stats = false;
 204
 205 /*
 206  * Tuple insertion/deletion counts for an open transaction can't be propagated
 207  * into PgStat_TableStatus counters until we know if it is going to commit
 208  * or abort.  Hence, we keep these counts in per-subxact structs that live
 209  * in TopTransactionContext.  This data structure is designed on the assumption
 210  * that subxacts won't usually modify very many tables.
 211  */
 212 typedef struct PgStat_SubXactStatus
 213 {
 214         int                     nest_level;             /* subtransaction nest level */
 215         struct PgStat_SubXactStatus *prev;      /* higher-level subxact if any */
 216         PgStat_TableXactStatus *first;  /* head of list for this subxact */
 217 } PgStat_SubXactStatus;
 218
 219 static PgStat_SubXactStatus *pgStatXactStack = NULL;
 220
 221 static int      pgStatXactCommit = 0;
 222 static int      pgStatXactRollback = 0;
 223 PgStat_Counter pgStatBlockReadTime = 0;
 224 PgStat_Counter pgStatBlockWriteTime = 0;
 225
 226 /* Record that's written to 2PC state file when pgstat state is persisted */
 227 typedef struct TwoPhasePgStatRecord
 228 {
 229         PgStat_Counter tuples_inserted; /* tuples inserted in xact */
 230         PgStat_Counter tuples_updated;  /* tuples updated in xact */
 231         PgStat_Counter tuples_deleted;  /* tuples deleted in xact */
 232         PgStat_Counter inserted_pre_trunc;      /* tuples inserted prior to truncate */
 233         PgStat_Counter updated_pre_trunc;       /* tuples updated prior to truncate */
 234         PgStat_Counter deleted_pre_trunc;       /* tuples deleted prior to truncate */
 235         Oid                     t_id;                   /* table's OID */
 236         bool            t_shared;               /* is it a shared catalog? */
 237         bool            t_truncated;    /* was the relation truncated? */
 238 } TwoPhasePgStatRecord;
 239
 240 /*
 241  * Info about current "snapshot" of stats file
 242  */
 243 static MemoryContext pgStatLocalContext = NULL;
 244 static HTAB *pgStatDBHash = NULL;
 245
 246 /* Status for backends including auxiliary */
 247 static LocalPgBackendStatus *localBackendStatusTable = NULL;
 248
 249 /* Total number of backends including auxiliary */
 250 static int      localNumBackends = 0;
 251
 252 /*
 253  * Cluster wide statistics, kept in the stats collector.
 254  * Contains statistics that are not collected per database
 255  * or per table.
 256  */
 257 static PgStat_ArchiverStats archiverStats;
 258 static PgStat_GlobalStats globalStats;
 259
 260 /*
 261  * List of OIDs of databases we need to write out.  If an entry is InvalidOid,
 262  * it means to write only the shared-catalog stats ("DB 0"); otherwise, we
 263  * will write both that DB's data and the shared stats.
 264  */
 265 static List *pending_write_requests = NIL;
 266
 267 /* Signal handler flags */
 268 static volatile bool need_exit = false;
 269 static volatile bool got_SIGHUP = false;
 270
 271 /*
 272  * Total time charged to functions so far in the current backend.
 273  * We use this to help separate "self" and "other" time charges.
 274  * (We assume this initializes to zero.)
 275  */
 276 static instr_time total_func_time;
 277
 278
 279 /* ----------
 280  * Local function forward declarations
 281  * ----------
 282  */
 283 #ifdef EXEC_BACKEND
 284 static pid_t pgstat_forkexec(void);
 285 #endif
 286
 287 NON_EXEC_STATIC void PgstatCollectorMain(int argc, char *argv[]) pg_attribute_noreturn();
 288 static void pgstat_exit(SIGNAL_ARGS);
 289 static void pgstat_beshutdown_hook(int code, Datum arg);
 290 static void pgstat_sighup_handler(SIGNAL_ARGS);
 291
 292 static PgStat_StatDBEntry *pgstat_get_db_entry(Oid databaseid, bool create);
 293 static PgStat_StatTabEntry *pgstat_get_tab_entry(PgStat_StatDBEntry *dbentry,
 294                                          Oid tableoid, bool create);
 295 static void pgstat_write_statsfiles(bool permanent, bool allDbs);
 296 static void pgstat_write_db_statsfile(PgStat_StatDBEntry *dbentry, bool permanent);
 297 static HTAB *pgstat_read_statsfiles(Oid onlydb, bool permanent, bool deep);
 298 static void pgstat_read_db_statsfile(Oid databaseid, HTAB *tabhash, HTAB *funchash, bool permanent);
 299 static void backend_read_statsfile(void);
 300 static void pgstat_read_current_status(void);
 301
 302 static bool pgstat_write_statsfile_needed(void);
 303 static bool pgstat_db_requested(Oid databaseid);
 304
 305 static void pgstat_send_tabstat(PgStat_MsgTabstat *tsmsg);
 306 static void pgstat_send_funcstats(void);
 307 static HTAB *pgstat_collect_oids(Oid catalogid);
 308
 309 static PgStat_TableStatus *get_tabstat_entry(Oid rel_id, bool isshared);
 310
 311 static void pgstat_setup_memcxt(void);
 312
 313 static const char *pgstat_get_wait_activity(WaitEventActivity w);
 314 static const char *pgstat_get_wait_client(WaitEventClient w);
 315 static const char *pgstat_get_wait_ipc(WaitEventIPC w);
 316 static const char *pgstat_get_wait_timeout(WaitEventTimeout w);
 317 static const char *pgstat_get_wait_io(WaitEventIO w);
 318
 319 static void pgstat_setheader(PgStat_MsgHdr *hdr, StatMsgType mtype);
 320 static void pgstat_send(void *msg, int len);
 321
 322 static void pgstat_recv_inquiry(PgStat_MsgInquiry *msg, int len);
 323 static void pgstat_recv_tabstat(PgStat_MsgTabstat *msg, int len);
 324 static void pgstat_recv_tabpurge(PgStat_MsgTabpurge *msg, int len);
 325 static void pgstat_recv_dropdb(PgStat_MsgDropdb *msg, int len);
 326 static void pgstat_recv_resetcounter(PgStat_MsgResetcounter *msg, int len);
 327 static void pgstat_recv_resetsharedcounter(PgStat_MsgResetsharedcounter *msg, int len);
 328 static void pgstat_recv_resetsinglecounter(PgStat_MsgResetsinglecounter *msg, int len);
 329 static void pgstat_recv_autovac(PgStat_MsgAutovacStart *msg, int len);
 330 static void pgstat_recv_vacuum(PgStat_MsgVacuum *msg, int len);
 331 static void pgstat_recv_analyze(PgStat_MsgAnalyze *msg, int len);
 332 static void pgstat_recv_archiver(PgStat_MsgArchiver *msg, int len);
 333 static void pgstat_recv_bgwriter(PgStat_MsgBgWriter *msg, int len);
 334 static void pgstat_recv_funcstat(PgStat_MsgFuncstat *msg, int len);
 335 static void pgstat_recv_funcpurge(PgStat_MsgFuncpurge *msg, int len);
 336 static void pgstat_recv_recoveryconflict(PgStat_MsgRecoveryConflict *msg, int len);
 337 static void pgstat_recv_deadlock(PgStat_MsgDeadlock *msg, int len);
 338 static void pgstat_recv_tempfile(PgStat_MsgTempFile *msg, int len);
 339
 340 /* ------------------------------------------------------------
 341  * Public functions called from postmaster follow
 342  * ------------------------------------------------------------
 343  */
 344
 345 /* ----------
 346  * pgstat_init() -
 347  *
 348  *      Called from postmaster at startup. Create the resources required
 349  *      by the statistics collector process.  If unable to do so, do not
 350  *      fail --- better to let the postmaster start with stats collection
 351  *      disabled.
 352  * ----------
 353  */
 354 void
 355 pgstat_init(void)
 356 {
 357         ACCEPT_TYPE_ARG3 alen;
 358         struct addrinfo *addrs = NULL,
 359                            *addr,
 360                                 hints;
 361         int                     ret;
 362         fd_set          rset;
 363         struct timeval tv;
 364         char            test_byte;
 365         int                     sel_res;
 366         int                     tries = 0;
 367
 368 #define TESTBYTEVAL ((char) 199)
 369
 370         /*
 371          * This static assertion verifies that we didn't mess up the calculations
 372          * involved in selecting maximum payload sizes for our UDP messages.
 373          * Because the only consequence of overrunning PGSTAT_MAX_MSG_SIZE would
 374          * be silent performance loss from fragmentation, it seems worth having a
 375          * compile-time cross-check that we didn't.
 376          */
 377         StaticAssertStmt(sizeof(PgStat_Msg) <= PGSTAT_MAX_MSG_SIZE,
 378                                          "maximum stats message size exceeds PGSTAT_MAX_MSG_SIZE");
 379
 380         /*
 381          * Create the UDP socket for sending and receiving statistic messages
 382          */
 383         hints.ai_flags = AI_PASSIVE;
 384         hints.ai_family = AF_UNSPEC;
 385         hints.ai_socktype = SOCK_DGRAM;
 386         hints.ai_protocol = 0;
 387         hints.ai_addrlen = 0;
 388         hints.ai_addr = NULL;
 389         hints.ai_canonname = NULL;
 390         hints.ai_next = NULL;
 391         ret = pg_getaddrinfo_all("localhost", NULL, &hints, &addrs);
 392         if (ret || !addrs)
 393         {
 394                 ereport(LOG,
 395                                 (errmsg("could not resolve \"localhost\": %s",
 396                                                 gai_strerror(ret))));
 397                 goto startup_failed;
 398         }
 399
 400         /*
 401          * On some platforms, pg_getaddrinfo_all() may return multiple addresses
 402          * only one of which will actually work (eg, both IPv6 and IPv4 addresses
 403          * when kernel will reject IPv6).  Worse, the failure may occur at the
 404          * bind() or perhaps even connect() stage.  So we must loop through the
 405          * results till we find a working combination. We will generate LOG
 406          * messages, but no error, for bogus combinations.
 407          */
 408         for (addr = addrs; addr; addr = addr->ai_next)
 409         {
 410 #ifdef HAVE_UNIX_SOCKETS
 411                 /* Ignore AF_UNIX sockets, if any are returned. */
 412                 if (addr->ai_family == AF_UNIX)
 413                         continue;
 414 #endif
 415
 416                 if (++tries > 1)
 417                         ereport(LOG,
 418                                         (errmsg("trying another address for the statistics collector")));
 419
 420                 /*
 421                  * Create the socket.
 422                  */
 423                 if ((pgStatSock = socket(addr->ai_family, SOCK_DGRAM, 0)) == PGINVALID_SOCKET)
 424                 {
 425                         ereport(LOG,
 426                                         (errcode_for_socket_access(),
 427                                          errmsg("could not create socket for statistics collector: %m")));
 428                         continue;
 429                 }
 430
 431                 /*
 432                  * Bind it to a kernel assigned port on localhost and get the assigned
 433                  * port via getsockname().
 434                  */
 435                 if (bind(pgStatSock, addr->ai_addr, addr->ai_addrlen) < 0)
 436                 {
 437                         ereport(LOG,
 438                                         (errcode_for_socket_access(),
 439                                          errmsg("could not bind socket for statistics collector: %m")));
 440                         closesocket(pgStatSock);
 441                         pgStatSock = PGINVALID_SOCKET;
 442                         continue;
 443                 }
 444
 445                 alen = sizeof(pgStatAddr);
 446                 if (getsockname(pgStatSock, (struct sockaddr *) &pgStatAddr, &alen) < 0)
 447                 {
 448                         ereport(LOG,
 449                                         (errcode_for_socket_access(),
 450                                          errmsg("could not get address of socket for statistics collector: %m")));
 451                         closesocket(pgStatSock);
 452                         pgStatSock = PGINVALID_SOCKET;
 453                         continue;
 454                 }
 455
 456                 /*
 457                  * Connect the socket to its own address.  This saves a few cycles by
 458                  * not having to respecify the target address on every send. This also
 459                  * provides a kernel-level check that only packets from this same
 460                  * address will be received.
 461                  */
 462                 if (connect(pgStatSock, (struct sockaddr *) &pgStatAddr, alen) < 0)
 463                 {
 464                         ereport(LOG,
 465                                         (errcode_for_socket_access(),
 466                                          errmsg("could not connect socket for statistics collector: %m")));
 467                         closesocket(pgStatSock);
 468                         pgStatSock = PGINVALID_SOCKET;
 469                         continue;
 470                 }
 471
 472                 /*
 473                  * Try to send and receive a one-byte test message on the socket. This
 474                  * is to catch situations where the socket can be created but will not
 475                  * actually pass data (for instance, because kernel packet filtering
 476                  * rules prevent it).
 477                  */
 478                 test_byte = TESTBYTEVAL;
 479
 480 retry1:
 481                 if (send(pgStatSock, &test_byte, 1, 0) != 1)
 482                 {
 483                         if (errno == EINTR)
 484                                 goto retry1;    /* if interrupted, just retry */
 485                         ereport(LOG,
 486                                         (errcode_for_socket_access(),
 487                                          errmsg("could not send test message on socket for statistics collector: %m")));
 488                         closesocket(pgStatSock);
 489                         pgStatSock = PGINVALID_SOCKET;
 490                         continue;
 491                 }
 492
 493                 /*
 494                  * There could possibly be a little delay before the message can be
 495                  * received.  We arbitrarily allow up to half a second before deciding
 496                  * it's broken.
 497                  */
 498                 for (;;)                                /* need a loop to handle EINTR */
 499                 {
 500                         FD_ZERO(&rset);
 501                         FD_SET(pgStatSock, &rset);
 502
 503                         tv.tv_sec = 0;
 504                         tv.tv_usec = 500000;
 505                         sel_res = select(pgStatSock + 1, &rset, NULL, NULL, &tv);
 506                         if (sel_res >= 0 || errno != EINTR)
 507                                 break;
 508                 }
 509                 if (sel_res < 0)
 510                 {
 511                         ereport(LOG,
 512                                         (errcode_for_socket_access(),
 513                                          errmsg("select() failed in statistics collector: %m")));
 514                         closesocket(pgStatSock);
 515                         pgStatSock = PGINVALID_SOCKET;
 516                         continue;
 517                 }
 518                 if (sel_res == 0 || !FD_ISSET(pgStatSock, &rset))
 519                 {
 520                         /*
 521                          * This is the case we actually think is likely, so take pains to
 522                          * give a specific message for it.
 523                          *
 524                          * errno will not be set meaningfully here, so don't use it.
 525                          */
 526                         ereport(LOG,
 527                                         (errcode(ERRCODE_CONNECTION_FAILURE),
 528                                          errmsg("test message did not get through on socket for statistics collector")));
 529                         closesocket(pgStatSock);
 530                         pgStatSock = PGINVALID_SOCKET;
 531                         continue;
 532                 }
 533
 534                 test_byte++;                    /* just make sure variable is changed */
 535
 536 retry2:
 537                 if (recv(pgStatSock, &test_byte, 1, 0) != 1)
 538                 {
 539                         if (errno == EINTR)
 540                                 goto retry2;    /* if interrupted, just retry */
 541                         ereport(LOG,
 542                                         (errcode_for_socket_access(),
 543                                          errmsg("could not receive test message on socket for statistics collector: %m")));
 544                         closesocket(pgStatSock);
 545                         pgStatSock = PGINVALID_SOCKET;
 546                         continue;
 547                 }
 548
 549                 if (test_byte != TESTBYTEVAL)   /* strictly paranoia ... */
 550                 {
 551                         ereport(LOG,
 552                                         (errcode(ERRCODE_INTERNAL_ERROR),
 553                                          errmsg("incorrect test message transmission on socket for statistics collector")));
 554                         closesocket(pgStatSock);
 555                         pgStatSock = PGINVALID_SOCKET;
 556                         continue;
 557                 }
 558
 559                 /* If we get here, we have a working socket */
 560                 break;
 561         }
 562
 563         /* Did we find a working address? */
 564         if (!addr || pgStatSock == PGINVALID_SOCKET)
 565                 goto startup_failed;
 566
 567         /*
 568          * Set the socket to non-blocking IO.  This ensures that if the collector
 569          * falls behind, statistics messages will be discarded; backends won't
 570          * block waiting to send messages to the collector.
 571          */
 572         if (!pg_set_noblock(pgStatSock))
 573         {
 574                 ereport(LOG,
 575                                 (errcode_for_socket_access(),
 576                                  errmsg("could not set statistics collector socket to nonblocking mode: %m")));
 577                 goto startup_failed;
 578         }
 579
 580         /*
 581          * Try to ensure that the socket's receive buffer is at least
 582          * PGSTAT_MIN_RCVBUF bytes, so that it won't easily overflow and lose
 583          * data.  Use of UDP protocol means that we are willing to lose data under
 584          * heavy load, but we don't want it to happen just because of ridiculously
 585          * small default buffer sizes (such as 8KB on older Windows versions).
 586          */
 587         {
 588                 int                     old_rcvbuf;
 589                 int                     new_rcvbuf;
 590                 ACCEPT_TYPE_ARG3 rcvbufsize = sizeof(old_rcvbuf);
 591
 592                 if (getsockopt(pgStatSock, SOL_SOCKET, SO_RCVBUF,
 593                                            (char *) &old_rcvbuf, &rcvbufsize) < 0)
 594                 {
 595                         elog(LOG, "getsockopt(SO_RCVBUF) failed: %m");
 596                         /* if we can't get existing size, always try to set it */
 597                         old_rcvbuf = 0;
 598                 }
 599
 600                 new_rcvbuf = PGSTAT_MIN_RCVBUF;
 601                 if (old_rcvbuf < new_rcvbuf)
 602                 {
 603                         if (setsockopt(pgStatSock, SOL_SOCKET, SO_RCVBUF,
 604                                                    (char *) &new_rcvbuf, sizeof(new_rcvbuf)) < 0)
 605                                 elog(LOG, "setsockopt(SO_RCVBUF) failed: %m");
 606                 }
 607         }
 608
 609         pg_freeaddrinfo_all(hints.ai_family, addrs);
 610
 611         return;
 612
 613 startup_failed:
 614         ereport(LOG,
 615                         (errmsg("disabling statistics collector for lack of working socket")));
 616
 617         if (addrs)
 618                 pg_freeaddrinfo_all(hints.ai_family, addrs);
 619
 620         if (pgStatSock != PGINVALID_SOCKET)
 621                 closesocket(pgStatSock);
 622         pgStatSock = PGINVALID_SOCKET;
 623
 624         /*
 625          * Adjust GUC variables to suppress useless activity, and for debugging
 626          * purposes (seeing track_counts off is a clue that we failed here). We
 627          * use PGC_S_OVERRIDE because there is no point in trying to turn it back
 628          * on from postgresql.conf without a restart.
 629          */
 630         SetConfigOption("track_counts", "off", PGC_INTERNAL, PGC_S_OVERRIDE);
 631 }
 632
 633 /*
 634  * subroutine for pgstat_reset_all
 635  */
 636 static void
 637 pgstat_reset_remove_files(const char *directory)
 638 {
 639         DIR                *dir;
 640         struct dirent *entry;
 641         char            fname[MAXPGPATH * 2];
 642
 643         dir = AllocateDir(directory);
 644         while ((entry = ReadDir(dir, directory)) != NULL)
 645         {
 646                 int                     nchars;
 647                 Oid                     tmp_oid;
 648
 649                 /*
 650                  * Skip directory entries that don't match the file names we write.
 651                  * See get_dbstat_filename for the database-specific pattern.
 652                  */
 653                 if (strncmp(entry->d_name, "global.", 7) == 0)
 654                         nchars = 7;
 655                 else
 656                 {
 657                         nchars = 0;
 658                         (void) sscanf(entry->d_name, "db_%u.%n",
 659                                                   &tmp_oid, &nchars);
 660                         if (nchars <= 0)
 661                                 continue;
 662                         /* %u allows leading whitespace, so reject that */
 663                         if (strchr("0123456789", entry->d_name[3]) == NULL)
 664                                 continue;
 665                 }
 666
 667                 if (strcmp(entry->d_name + nchars, "tmp") != 0 &&
 668                         strcmp(entry->d_name + nchars, "stat") != 0)
 669                         continue;
 670
 671                 snprintf(fname, sizeof(fname), "%s/%s", directory,
 672                                  entry->d_name);
 673                 unlink(fname);
 674         }
 675         FreeDir(dir);
 676 }
 677
 678 /*
 679  * pgstat_reset_all() -
 680  *
 681  * Remove the stats files.  This is currently used only if WAL
 682  * recovery is needed after a crash.
 683  */
 684 void
 685 pgstat_reset_all(void)
 686 {
 687         pgstat_reset_remove_files(pgstat_stat_directory);
 688         pgstat_reset_remove_files(PGSTAT_STAT_PERMANENT_DIRECTORY);
 689 }
 690
 691 #ifdef EXEC_BACKEND
 692
 693 /*
 694  * pgstat_forkexec() -
 695  *
 696  * Format up the arglist for, then fork and exec, statistics collector process
 697  */
 698 static pid_t
 699 pgstat_forkexec(void)
 700 {
 701         char       *av[10];
 702         int                     ac = 0;
 703
 704         av[ac++] = "postgres";
 705         av[ac++] = "--forkcol";
 706         av[ac++] = NULL;                        /* filled in by postmaster_forkexec */
 707
 708         av[ac] = NULL;
 709         Assert(ac < lengthof(av));
 710
 711         return postmaster_forkexec(ac, av);
 712 }
 713 #endif                                                  /* EXEC_BACKEND */
 714
 715
 716 /*
 717  * pgstat_start() -
 718  *
 719  *      Called from postmaster at startup or after an existing collector
 720  *      died.  Attempt to fire up a fresh statistics collector.
 721  *
 722  *      Returns PID of child process, or 0 if fail.
 723  *
 724  *      Note: if fail, we will be called again from the postmaster main loop.
 725  */
 726 int
 727 pgstat_start(void)
 728 {
 729         time_t          curtime;
 730         pid_t           pgStatPid;
 731
 732         /*
 733          * Check that the socket is there, else pgstat_init failed and we can do
 734          * nothing useful.
 735          */
 736         if (pgStatSock == PGINVALID_SOCKET)
 737                 return 0;
 738
 739         /*
 740          * Do nothing if too soon since last collector start.  This is a safety
 741          * valve to protect against continuous respawn attempts if the collector
 742          * is dying immediately at launch.  Note that since we will be re-called
 743          * from the postmaster main loop, we will get another chance later.
 744          */
 745         curtime = time(NULL);
 746         if ((unsigned int) (curtime - last_pgstat_start_time) <
 747                 (unsigned int) PGSTAT_RESTART_INTERVAL)
 748                 return 0;
 749         last_pgstat_start_time = curtime;
 750
 751         /*
 752          * Okay, fork off the collector.
 753          */
 754 #ifdef EXEC_BACKEND
 755         switch ((pgStatPid = pgstat_forkexec()))
 756 #else
 757         switch ((pgStatPid = fork_process()))
 758 #endif
 759         {
 760                 case -1:
 761                         ereport(LOG,
 762                                         (errmsg("could not fork statistics collector: %m")));
 763                         return 0;
 764
 765 #ifndef EXEC_BACKEND
 766                 case 0:
 767                         /* in postmaster child ... */
 768                         InitPostmasterChild();
 769
 770                         /* Close the postmaster's sockets */
 771                         ClosePostmasterPorts(false);
 772
 773                         /* Drop our connection to postmaster's shared memory, as well */
 774                         dsm_detach_all();
 775                         PGSharedMemoryDetach();
 776
 777                         PgstatCollectorMain(0, NULL);
 778                         break;
 779 #endif
 780
 781                 default:
 782                         return (int) pgStatPid;
 783         }
 784
 785         /* shouldn't get here */
 786         return 0;
 787 }
 788
 789 void
 790 allow_immediate_pgstat_restart(void)
 791 {
 792         last_pgstat_start_time = 0;
 793 }
 794
 795 /* ------------------------------------------------------------
 796  * Public functions used by backends follow
 797  *------------------------------------------------------------
 798  */
 799
 800
 801 /* ----------
 802  * pgstat_report_stat() -
 803  *
 804  *      Must be called by processes that performs DML: tcop/postgres.c, logical
 805  *      receiver processes, SPI worker, etc. to send the so far collected
 806  *      per-table and function usage statistics to the collector.  Note that this
 807  *      is called only when not within a transaction, so it is fair to use
 808  *      transaction stop time as an approximation of current time.
 809  * ----------
 810  */
 811 void
 812 pgstat_report_stat(bool force)
 813 {
 814         /* we assume this inits to all zeroes: */
 815         static const PgStat_TableCounts all_zeroes;
 816         static TimestampTz last_report = 0;
 817
 818         TimestampTz now;
 819         PgStat_MsgTabstat regular_msg;
 820         PgStat_MsgTabstat shared_msg;
 821         TabStatusArray *tsa;
 822         int                     i;
 823
 824         /* Don't expend a clock check if nothing to do */
 825         if ((pgStatTabList == NULL || pgStatTabList->tsa_used == 0) &&
 826                 pgStatXactCommit == 0 && pgStatXactRollback == 0 &&
 827                 !have_function_stats)
 828                 return;
 829
 830         /*
 831          * Don't send a message unless it's been at least PGSTAT_STAT_INTERVAL
 832          * msec since we last sent one, or the caller wants to force stats out.
 833          */
 834         now = GetCurrentTransactionStopTimestamp();
 835         if (!force &&
 836                 !TimestampDifferenceExceeds(last_report, now, PGSTAT_STAT_INTERVAL))
 837                 return;
 838         last_report = now;
 839
 840         /*
 841          * Destroy pgStatTabHash before we start invalidating PgStat_TableEntry
 842          * entries it points to.  (Should we fail partway through the loop below,
 843          * it's okay to have removed the hashtable already --- the only
 844          * consequence is we'd get multiple entries for the same table in the
 845          * pgStatTabList, and that's safe.)
 846          */
 847         if (pgStatTabHash)
 848                 hash_destroy(pgStatTabHash);
 849         pgStatTabHash = NULL;
 850
 851         /*
 852          * Scan through the TabStatusArray struct(s) to find tables that actually
 853          * have counts, and build messages to send.  We have to separate shared
 854          * relations from regular ones because the databaseid field in the message
 855          * header has to depend on that.
 856          */
 857         regular_msg.m_databaseid = MyDatabaseId;
 858         shared_msg.m_databaseid = InvalidOid;
 859         regular_msg.m_nentries = 0;
 860         shared_msg.m_nentries = 0;
 861
 862         for (tsa = pgStatTabList; tsa != NULL; tsa = tsa->tsa_next)
 863         {
 864                 for (i = 0; i < tsa->tsa_used; i++)
 865                 {
 866                         PgStat_TableStatus *entry = &tsa->tsa_entries[i];
 867                         PgStat_MsgTabstat *this_msg;
 868                         PgStat_TableEntry *this_ent;
 869
 870                         /* Shouldn't have any pending transaction-dependent counts */
 871                         Assert(entry->trans == NULL);
 872
 873                         /*
 874                          * Ignore entries that didn't accumulate any actual counts, such
 875                          * as indexes that were opened by the planner but not used.
 876                          */
 877                         if (memcmp(&entry->t_counts, &all_zeroes,
 878                                            sizeof(PgStat_TableCounts)) == 0)
 879                                 continue;
 880
 881                         /*
 882                          * OK, insert data into the appropriate message, and send if full.
 883                          */
 884                         this_msg = entry->t_shared ? &shared_msg : &regular_msg;
 885                         this_ent = &this_msg->m_entry[this_msg->m_nentries];
 886                         this_ent->t_id = entry->t_id;
 887                         memcpy(&this_ent->t_counts, &entry->t_counts,
 888                                    sizeof(PgStat_TableCounts));
 889                         if (++this_msg->m_nentries >= PGSTAT_NUM_TABENTRIES)
 890                         {
 891                                 pgstat_send_tabstat(this_msg);
 892                                 this_msg->m_nentries = 0;
 893                         }
 894                 }
 895                 /* zero out TableStatus structs after use */
 896                 MemSet(tsa->tsa_entries, 0,
 897                            tsa->tsa_used * sizeof(PgStat_TableStatus));
 898                 tsa->tsa_used = 0;
 899         }
 900
 901         /*
 902          * Send partial messages.  Make sure that any pending xact commit/abort
 903          * gets counted, even if there are no table stats to send.
 904          */
 905         if (regular_msg.m_nentries > 0 ||
 906                 pgStatXactCommit > 0 || pgStatXactRollback > 0)
 907                 pgstat_send_tabstat(&regular_msg);
 908         if (shared_msg.m_nentries > 0)
 909                 pgstat_send_tabstat(&shared_msg);
 910
 911         /* Now, send function statistics */
 912         pgstat_send_funcstats();
 913 }
 914
 915 /*
 916  * Subroutine for pgstat_report_stat: finish and send a tabstat message
 917  */
 918 static void
 919 pgstat_send_tabstat(PgStat_MsgTabstat *tsmsg)
 920 {
 921         int                     n;
 922         int                     len;
 923
 924         /* It's unlikely we'd get here with no socket, but maybe not impossible */
 925         if (pgStatSock == PGINVALID_SOCKET)
 926                 return;
 927
 928         /*
 929          * Report and reset accumulated xact commit/rollback and I/O timings
 930          * whenever we send a normal tabstat message
 931          */
 932         if (OidIsValid(tsmsg->m_databaseid))
 933         {
 934                 tsmsg->m_xact_commit = pgStatXactCommit;
 935                 tsmsg->m_xact_rollback = pgStatXactRollback;
 936                 tsmsg->m_block_read_time = pgStatBlockReadTime;
 937                 tsmsg->m_block_write_time = pgStatBlockWriteTime;
 938                 pgStatXactCommit = 0;
 939                 pgStatXactRollback = 0;
 940                 pgStatBlockReadTime = 0;
 941                 pgStatBlockWriteTime = 0;
 942         }
 943         else
 944         {
 945                 tsmsg->m_xact_commit = 0;
 946                 tsmsg->m_xact_rollback = 0;
 947                 tsmsg->m_block_read_time = 0;
 948                 tsmsg->m_block_write_time = 0;
 949         }
 950
 951         n = tsmsg->m_nentries;
 952         len = offsetof(PgStat_MsgTabstat, m_entry[0]) +
 953                 n * sizeof(PgStat_TableEntry);
 954
 955         pgstat_setheader(&tsmsg->m_hdr, PGSTAT_MTYPE_TABSTAT);
 956         pgstat_send(tsmsg, len);
 957 }
 958
 959 /*
 960  * Subroutine for pgstat_report_stat: populate and send a function stat message
 961  */
 962 static void
 963 pgstat_send_funcstats(void)
 964 {
 965         /* we assume this inits to all zeroes: */
 966         static const PgStat_FunctionCounts all_zeroes;
 967
 968         PgStat_MsgFuncstat msg;
 969         PgStat_BackendFunctionEntry *entry;
 970         HASH_SEQ_STATUS fstat;
 971
 972         if (pgStatFunctions == NULL)
 973                 return;
 974
 975         pgstat_setheader(&msg.m_hdr, PGSTAT_MTYPE_FUNCSTAT);
 976         msg.m_databaseid = MyDatabaseId;
 977         msg.m_nentries = 0;
 978
 979         hash_seq_init(&fstat, pgStatFunctions);
 980         while ((entry = (PgStat_BackendFunctionEntry *) hash_seq_search(&fstat)) != NULL)
 981         {
 982                 PgStat_FunctionEntry *m_ent;
 983
 984                 /* Skip it if no counts accumulated since last time */
 985                 if (memcmp(&entry->f_counts, &all_zeroes,
 986                                    sizeof(PgStat_FunctionCounts)) == 0)
 987                         continue;
 988
 989                 /* need to convert format of time accumulators */
 990                 m_ent = &msg.m_entry[msg.m_nentries];
 991                 m_ent->f_id = entry->f_id;
 992                 m_ent->f_numcalls = entry->f_counts.f_numcalls;
 993                 m_ent->f_total_time = INSTR_TIME_GET_MICROSEC(entry->f_counts.f_total_time);
 994                 m_ent->f_self_time = INSTR_TIME_GET_MICROSEC(entry->f_counts.f_self_time);
 995
 996                 if (++msg.m_nentries >= PGSTAT_NUM_FUNCENTRIES)
 997                 {
 998                         pgstat_send(&msg, offsetof(PgStat_MsgFuncstat, m_entry[0]) +
 999                                                 msg.m_nentries * sizeof(PgStat_FunctionEntry));
1000                         msg.m_nentries = 0;
1001                 }
1002
1003                 /* reset the entry's counts */
1004                 MemSet(&entry->f_counts, 0, sizeof(PgStat_FunctionCounts));
1005         }
1006
1007         if (msg.m_nentries > 0)
1008                 pgstat_send(&msg, offsetof(PgStat_MsgFuncstat, m_entry[0]) +
1009                                         msg.m_nentries * sizeof(PgStat_FunctionEntry));
1010
1011         have_function_stats = false;
1012 }
1013
1014
1015 /* ----------
1016  * pgstat_vacuum_stat() -
1017  *
1018  *      Will tell the collector about objects he can get rid of.
1019  * ----------
1020  */
1021 void
1022 pgstat_vacuum_stat(void)
1023 {
1024         HTAB       *htab;
1025         PgStat_MsgTabpurge msg;
1026         PgStat_MsgFuncpurge f_msg;
1027         HASH_SEQ_STATUS hstat;
1028         PgStat_StatDBEntry *dbentry;
1029         PgStat_StatTabEntry *tabentry;
1030         PgStat_StatFuncEntry *funcentry;
1031         int                     len;
1032
1033         if (pgStatSock == PGINVALID_SOCKET)
1034                 return;
1035
1036         /*
1037          * If not done for this transaction, read the statistics collector stats
1038          * file into some hash tables.
1039          */
1040         backend_read_statsfile();
1041
1042         /*
1043          * Read pg_database and make a list of OIDs of all existing databases
1044          */
1045         htab = pgstat_collect_oids(DatabaseRelationId);
1046
1047         /*
1048          * Search the database hash table for dead databases and tell the
1049          * collector to drop them.
1050          */
1051         hash_seq_init(&hstat, pgStatDBHash);
1052         while ((dbentry = (PgStat_StatDBEntry *) hash_seq_search(&hstat)) != NULL)
1053         {
1054                 Oid                     dbid = dbentry->databaseid;
1055
1056                 CHECK_FOR_INTERRUPTS();
1057
1058                 /* the DB entry for shared tables (with InvalidOid) is never dropped */
1059                 if (OidIsValid(dbid) &&
1060                         hash_search(htab, (void *) &dbid, HASH_FIND, NULL) == NULL)
1061                         pgstat_drop_database(dbid);
1062         }
1063
1064         /* Clean up */
1065         hash_destroy(htab);
1066
1067         /*
1068          * Lookup our own database entry; if not found, nothing more to do.
1069          */
1070         dbentry = (PgStat_StatDBEntry *) hash_search(pgStatDBHash,
1071                                                                                                  (void *) &MyDatabaseId,
1072                                                                                                  HASH_FIND, NULL);
1073         if (dbentry == NULL || dbentry->tables == NULL)
1074                 return;
1075
1076         /*
1077          * Similarly to above, make a list of all known relations in this DB.
1078          */
1079         htab = pgstat_collect_oids(RelationRelationId);
1080
1081         /*
1082          * Initialize our messages table counter to zero
1083          */
1084         msg.m_nentries = 0;
1085
1086         /*
1087          * Check for all tables listed in stats hashtable if they still exist.
1088          */
1089         hash_seq_init(&hstat, dbentry->tables);
1090         while ((tabentry = (PgStat_StatTabEntry *) hash_seq_search(&hstat)) != NULL)
1091         {
1092                 Oid                     tabid = tabentry->tableid;
1093
1094                 CHECK_FOR_INTERRUPTS();
1095
1096                 if (hash_search(htab, (void *) &tabid, HASH_FIND, NULL) != NULL)
1097                         continue;
1098
1099                 /*
1100                  * Not there, so add this table's Oid to the message
1101                  */
1102                 msg.m_tableid[msg.m_nentries++] = tabid;
1103
1104                 /*
1105                  * If the message is full, send it out and reinitialize to empty
1106                  */
1107                 if (msg.m_nentries >= PGSTAT_NUM_TABPURGE)
1108                 {
1109                         len = offsetof(PgStat_MsgTabpurge, m_tableid[0])
1110                                 + msg.m_nentries * sizeof(Oid);
1111
1112                         pgstat_setheader(&msg.m_hdr, PGSTAT_MTYPE_TABPURGE);
1113                         msg.m_databaseid = MyDatabaseId;
1114                         pgstat_send(&msg, len);
1115
1116                         msg.m_nentries = 0;
1117                 }
1118         }
1119
1120         /*
1121          * Send the rest
1122          */
1123         if (msg.m_nentries > 0)
1124         {
1125                 len = offsetof(PgStat_MsgTabpurge, m_tableid[0])
1126                         + msg.m_nentries * sizeof(Oid);
1127
1128                 pgstat_setheader(&msg.m_hdr, PGSTAT_MTYPE_TABPURGE);
1129                 msg.m_databaseid = MyDatabaseId;
1130                 pgstat_send(&msg, len);
1131         }
1132
1133         /* Clean up */
1134         hash_destroy(htab);
1135
1136         /*
1137          * Now repeat the above steps for functions.  However, we needn't bother
1138          * in the common case where no function stats are being collected.
1139          */
1140         if (dbentry->functions != NULL &&
1141                 hash_get_num_entries(dbentry->functions) > 0)
1142         {
1143                 htab = pgstat_collect_oids(ProcedureRelationId);
1144
1145                 pgstat_setheader(&f_msg.m_hdr, PGSTAT_MTYPE_FUNCPURGE);
1146                 f_msg.m_databaseid = MyDatabaseId;
1147                 f_msg.m_nentries = 0;
1148
1149                 hash_seq_init(&hstat, dbentry->functions);
1150                 while ((funcentry = (PgStat_StatFuncEntry *) hash_seq_search(&hstat)) != NULL)
1151                 {
1152                         Oid                     funcid = funcentry->functionid;
1153
1154                         CHECK_FOR_INTERRUPTS();
1155
1156                         if (hash_search(htab, (void *) &funcid, HASH_FIND, NULL) != NULL)
1157                                 continue;
1158
1159                         /*
1160                          * Not there, so add this function's Oid to the message
1161                          */
1162                         f_msg.m_functionid[f_msg.m_nentries++] = funcid;
1163
1164                         /*
1165                          * If the message is full, send it out and reinitialize to empty
1166                          */
1167                         if (f_msg.m_nentries >= PGSTAT_NUM_FUNCPURGE)
1168                         {
1169                                 len = offsetof(PgStat_MsgFuncpurge, m_functionid[0])
1170                                         + f_msg.m_nentries * sizeof(Oid);
1171
1172                                 pgstat_send(&f_msg, len);
1173
1174                                 f_msg.m_nentries = 0;
1175                         }
1176                 }
1177
1178                 /*
1179                  * Send the rest
1180                  */
1181                 if (f_msg.m_nentries > 0)
1182                 {
1183                         len = offsetof(PgStat_MsgFuncpurge, m_functionid[0])
1184                                 + f_msg.m_nentries * sizeof(Oid);
1185
1186                         pgstat_send(&f_msg, len);
1187                 }
1188
1189                 hash_destroy(htab);
1190         }
1191 }
1192
1193
1194 /* ----------
1195  * pgstat_collect_oids() -
1196  *
1197  *      Collect the OIDs of all objects listed in the specified system catalog
1198  *      into a temporary hash table.  Caller should hash_destroy the result
1199  *      when done with it.  (However, we make the table in CurrentMemoryContext
1200  *      so that it will be freed properly in event of an error.)
1201  * ----------
1202  */
1203 static HTAB *
1204 pgstat_collect_oids(Oid catalogid)
1205 {
1206         HTAB       *htab;
1207         HASHCTL         hash_ctl;
1208         Relation        rel;
1209         HeapScanDesc scan;
1210         HeapTuple       tup;
1211         Snapshot        snapshot;
1212
1213         memset(&hash_ctl, 0, sizeof(hash_ctl));
1214         hash_ctl.keysize = sizeof(Oid);
1215         hash_ctl.entrysize = sizeof(Oid);
1216         hash_ctl.hcxt = CurrentMemoryContext;
1217         htab = hash_create("Temporary table of OIDs",
1218                                            PGSTAT_TAB_HASH_SIZE,
1219                                            &hash_ctl,
1220                                            HASH_ELEM | HASH_BLOBS | HASH_CONTEXT);
1221
1222         rel = heap_open(catalogid, AccessShareLock);
1223         snapshot = RegisterSnapshot(GetLatestSnapshot());
1224         scan = heap_beginscan(rel, snapshot, 0, NULL);
1225         while ((tup = heap_getnext(scan, ForwardScanDirection)) != NULL)
1226         {
1227                 Oid                     thisoid = HeapTupleGetOid(tup);
1228
1229                 CHECK_FOR_INTERRUPTS();
1230
1231                 (void) hash_search(htab, (void *) &thisoid, HASH_ENTER, NULL);
1232         }
1233         heap_endscan(scan);
1234         UnregisterSnapshot(snapshot);
1235         heap_close(rel, AccessShareLock);
1236
1237         return htab;
1238 }
1239
1240
1241 /* ----------
1242  * pgstat_drop_database() -
1243  *
1244  *      Tell the collector that we just dropped a database.
1245  *      (If the message gets lost, we will still clean the dead DB eventually
1246  *      via future invocations of pgstat_vacuum_stat().)
1247  * ----------
1248  */
1249 void
1250 pgstat_drop_database(Oid databaseid)
1251 {
1252         PgStat_MsgDropdb msg;
1253
1254         if (pgStatSock == PGINVALID_SOCKET)
1255                 return;
1256
1257         pgstat_setheader(&msg.m_hdr, PGSTAT_MTYPE_DROPDB);
1258         msg.m_databaseid = databaseid;
1259         pgstat_send(&msg, sizeof(msg));
1260 }
1261
1262
1263 /* ----------
1264  * pgstat_drop_relation() -
1265  *
1266  *      Tell the collector that we just dropped a relation.
1267  *      (If the message gets lost, we will still clean the dead entry eventually
1268  *      via future invocations of pgstat_vacuum_stat().)
1269  *
1270  *      Currently not used for lack of any good place to call it; we rely
1271  *      entirely on pgstat_vacuum_stat() to clean out stats for dead rels.
1272  * ----------
1273  */
1274 #ifdef NOT_USED
1275 void
1276 pgstat_drop_relation(Oid relid)
1277 {
1278         PgStat_MsgTabpurge msg;
1279         int                     len;
1280
1281         if (pgStatSock == PGINVALID_SOCKET)
1282                 return;
1283
1284         msg.m_tableid[0] = relid;
1285         msg.m_nentries = 1;
1286
1287         len = offsetof(PgStat_MsgTabpurge, m_tableid[0]) + sizeof(Oid);
1288
1289         pgstat_setheader(&msg.m_hdr, PGSTAT_MTYPE_TABPURGE);
1290         msg.m_databaseid = MyDatabaseId;
1291         pgstat_send(&msg, len);
1292 }
1293 #endif                                                  /* NOT_USED */
1294
1295
1296 /* ----------
1297  * pgstat_reset_counters() -
1298  *
1299  *      Tell the statistics collector to reset counters for our database.
1300  *
1301  *      Permission checking for this function is managed through the normal
1302  *      GRANT system.
1303  * ----------
1304  */
1305 void
1306 pgstat_reset_counters(void)
1307 {
1308         PgStat_MsgResetcounter msg;
1309
1310         if (pgStatSock == PGINVALID_SOCKET)
1311                 return;
1312
1313         pgstat_setheader(&msg.m_hdr, PGSTAT_MTYPE_RESETCOUNTER);
1314         msg.m_databaseid = MyDatabaseId;
1315         pgstat_send(&msg, sizeof(msg));
1316 }
1317
1318 /* ----------
1319  * pgstat_reset_shared_counters() -
1320  *
1321  *      Tell the statistics collector to reset cluster-wide shared counters.
1322  *
1323  *      Permission checking for this function is managed through the normal
1324  *      GRANT system.
1325  * ----------
1326  */
1327 void
1328 pgstat_reset_shared_counters(const char *target)
1329 {
1330         PgStat_MsgResetsharedcounter msg;
1331
1332         if (pgStatSock == PGINVALID_SOCKET)
1333                 return;
1334
1335         if (strcmp(target, "archiver") == 0)
1336                 msg.m_resettarget = RESET_ARCHIVER;
1337         else if (strcmp(target, "bgwriter") == 0)
1338                 msg.m_resettarget = RESET_BGWRITER;
1339         else
1340                 ereport(ERROR,
1341                                 (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
1342                                  errmsg("unrecognized reset target: \"%s\"", target),
1343                                  errhint("Target must be \"archiver\" or \"bgwriter\".")));
1344
1345         pgstat_setheader(&msg.m_hdr, PGSTAT_MTYPE_RESETSHAREDCOUNTER);
1346         pgstat_send(&msg, sizeof(msg));
1347 }
1348
1349 /* ----------
1350  * pgstat_reset_single_counter() -
1351  *
1352  *      Tell the statistics collector to reset a single counter.
1353  *
1354  *      Permission checking for this function is managed through the normal
1355  *      GRANT system.
1356  * ----------
1357  */
1358 void
1359 pgstat_reset_single_counter(Oid objoid, PgStat_Single_Reset_Type type)
1360 {
1361         PgStat_MsgResetsinglecounter msg;
1362
1363         if (pgStatSock == PGINVALID_SOCKET)
1364                 return;
1365
1366         pgstat_setheader(&msg.m_hdr, PGSTAT_MTYPE_RESETSINGLECOUNTER);
1367         msg.m_databaseid = MyDatabaseId;
1368         msg.m_resettype = type;
1369         msg.m_objectid = objoid;
1370
1371         pgstat_send(&msg, sizeof(msg));
1372 }
1373
1374 /* ----------
1375  * pgstat_report_autovac() -
1376  *
1377  *      Called from autovacuum.c to report startup of an autovacuum process.
1378  *      We are called before InitPostgres is done, so can't rely on MyDatabaseId;
1379  *      the db OID must be passed in, instead.
1380  * ----------
1381  */
1382 void
1383 pgstat_report_autovac(Oid dboid)
1384 {
1385         PgStat_MsgAutovacStart msg;
1386
1387         if (pgStatSock == PGINVALID_SOCKET)
1388                 return;
1389
1390         pgstat_setheader(&msg.m_hdr, PGSTAT_MTYPE_AUTOVAC_START);
1391         msg.m_databaseid = dboid;
1392         msg.m_start_time = GetCurrentTimestamp();
1393
1394         pgstat_send(&msg, sizeof(msg));
1395 }
1396
1397
1398 /* ---------
1399  * pgstat_report_vacuum() -
1400  *
1401  *      Tell the collector about the table we just vacuumed.
1402  * ---------
1403  */
1404 void
1405 pgstat_report_vacuum(Oid tableoid, bool shared,
1406                                          PgStat_Counter livetuples, PgStat_Counter deadtuples)
1407 {
1408         PgStat_MsgVacuum msg;
1409
1410         if (pgStatSock == PGINVALID_SOCKET || !pgstat_track_counts)
1411                 return;
1412
1413         pgstat_setheader(&msg.m_hdr, PGSTAT_MTYPE_VACUUM);
1414         msg.m_databaseid = shared ? InvalidOid : MyDatabaseId;
1415         msg.m_tableoid = tableoid;
1416         msg.m_autovacuum = IsAutoVacuumWorkerProcess();
1417         msg.m_vacuumtime = GetCurrentTimestamp();
1418         msg.m_live_tuples = livetuples;
1419         msg.m_dead_tuples = deadtuples;
1420         pgstat_send(&msg, sizeof(msg));
1421 }
1422
1423 /* --------
1424  * pgstat_report_analyze() -
1425  *
1426  *      Tell the collector about the table we just analyzed.
1427  *
1428  * Caller must provide new live- and dead-tuples estimates, as well as a
1429  * flag indicating whether to reset the changes_since_analyze counter.
1430  * --------
1431  */
1432 void
1433 pgstat_report_analyze(Relation rel,
1434                                           PgStat_Counter livetuples, PgStat_Counter deadtuples,
1435                                           bool resetcounter)
1436 {
1437         PgStat_MsgAnalyze msg;
1438
1439         if (pgStatSock == PGINVALID_SOCKET || !pgstat_track_counts)
1440                 return;
1441
1442         /*
1443          * Unlike VACUUM, ANALYZE might be running inside a transaction that has
1444          * already inserted and/or deleted rows in the target table. ANALYZE will
1445          * have counted such rows as live or dead respectively. Because we will
1446          * report our counts of such rows at transaction end, we should subtract
1447          * off these counts from what we send to the collector now, else they'll
1448          * be double-counted after commit.  (This approach also ensures that the
1449          * collector ends up with the right numbers if we abort instead of
1450          * committing.)
1451          */
1452         if (rel->pgstat_info != NULL)
1453         {
1454                 PgStat_TableXactStatus *trans;
1455
1456                 for (trans = rel->pgstat_info->trans; trans; trans = trans->upper)
1457                 {
1458                         livetuples -= trans->tuples_inserted - trans->tuples_deleted;
1459                         deadtuples -= trans->tuples_updated + trans->tuples_deleted;
1460                 }
1461                 /* count stuff inserted by already-aborted subxacts, too */
1462                 deadtuples -= rel->pgstat_info->t_counts.t_delta_dead_tuples;
1463                 /* Since ANALYZE's counts are estimates, we could have underflowed */
1464                 livetuples = Max(livetuples, 0);
1465                 deadtuples = Max(deadtuples, 0);
1466         }
1467
1468         pgstat_setheader(&msg.m_hdr, PGSTAT_MTYPE_ANALYZE);
1469         msg.m_databaseid = rel->rd_rel->relisshared ? InvalidOid : MyDatabaseId;
1470         msg.m_tableoid = RelationGetRelid(rel);
1471         msg.m_autovacuum = IsAutoVacuumWorkerProcess();
1472         msg.m_resetcounter = resetcounter;
1473         msg.m_analyzetime = GetCurrentTimestamp();
1474         msg.m_live_tuples = livetuples;
1475         msg.m_dead_tuples = deadtuples;
1476         pgstat_send(&msg, sizeof(msg));
1477 }
1478
1479 /* --------
1480  * pgstat_report_recovery_conflict() -
1481  *
1482  *      Tell the collector about a Hot Standby recovery conflict.
1483  * --------
1484  */
1485 void
1486 pgstat_report_recovery_conflict(int reason)
1487 {
1488         PgStat_MsgRecoveryConflict msg;
1489
1490         if (pgStatSock == PGINVALID_SOCKET || !pgstat_track_counts)
1491                 return;
1492
1493         pgstat_setheader(&msg.m_hdr, PGSTAT_MTYPE_RECOVERYCONFLICT);
1494         msg.m_databaseid = MyDatabaseId;
1495         msg.m_reason = reason;
1496         pgstat_send(&msg, sizeof(msg));
1497 }
1498
1499 /* --------
1500  * pgstat_report_deadlock() -
1501  *
1502  *      Tell the collector about a deadlock detected.
1503  * --------
1504  */
1505 void
1506 pgstat_report_deadlock(void)
1507 {
1508         PgStat_MsgDeadlock msg;
1509
1510         if (pgStatSock == PGINVALID_SOCKET || !pgstat_track_counts)
1511                 return;
1512
1513         pgstat_setheader(&msg.m_hdr, PGSTAT_MTYPE_DEADLOCK);
1514         msg.m_databaseid = MyDatabaseId;
1515         pgstat_send(&msg, sizeof(msg));
1516 }
1517
1518 /* --------
1519  * pgstat_report_tempfile() -
1520  *
1521  *      Tell the collector about a temporary file.
1522  * --------
1523  */
1524 void
1525 pgstat_report_tempfile(size_t filesize)
1526 {
1527         PgStat_MsgTempFile msg;
1528
1529         if (pgStatSock == PGINVALID_SOCKET || !pgstat_track_counts)
1530                 return;
1531
1532         pgstat_setheader(&msg.m_hdr, PGSTAT_MTYPE_TEMPFILE);
1533         msg.m_databaseid = MyDatabaseId;
1534         msg.m_filesize = filesize;
1535         pgstat_send(&msg, sizeof(msg));
1536 }
1537
1538
1539 /* ----------
1540  * pgstat_ping() -
1541  *
1542  *      Send some junk data to the collector to increase traffic.
1543  * ----------
1544  */
1545 void
1546 pgstat_ping(void)
1547 {
1548         PgStat_MsgDummy msg;
1549
1550         if (pgStatSock == PGINVALID_SOCKET)
1551                 return;
1552
1553         pgstat_setheader(&msg.m_hdr, PGSTAT_MTYPE_DUMMY);
1554         pgstat_send(&msg, sizeof(msg));
1555 }
1556
1557 /* ----------
1558  * pgstat_send_inquiry() -
1559  *
1560  *      Notify collector that we need fresh data.
1561  * ----------
1562  */
1563 static void
1564 pgstat_send_inquiry(TimestampTz clock_time, TimestampTz cutoff_time, Oid databaseid)
1565 {
1566         PgStat_MsgInquiry msg;
1567
1568         pgstat_setheader(&msg.m_hdr, PGSTAT_MTYPE_INQUIRY);
1569         msg.clock_time = clock_time;
1570         msg.cutoff_time = cutoff_time;
1571         msg.databaseid = databaseid;
1572         pgstat_send(&msg, sizeof(msg));
1573 }
1574
1575
1576 /*
1577  * Initialize function call usage data.
1578  * Called by the executor before invoking a function.
1579  */
1580 void
1581 pgstat_init_function_usage(FunctionCallInfoData *fcinfo,
1582                                                    PgStat_FunctionCallUsage *fcu)
1583 {
1584         PgStat_BackendFunctionEntry *htabent;
1585         bool            found;
1586
1587         if (pgstat_track_functions <= fcinfo->flinfo->fn_stats)
1588         {
1589                 /* stats not wanted */
1590                 fcu->fs = NULL;
1591                 return;
1592         }
1593
1594         if (!pgStatFunctions)
1595         {
1596                 /* First time through - initialize function stat table */
1597                 HASHCTL         hash_ctl;
1598
1599                 memset(&hash_ctl, 0, sizeof(hash_ctl));
1600                 hash_ctl.keysize = sizeof(Oid);
1601                 hash_ctl.entrysize = sizeof(PgStat_BackendFunctionEntry);
1602                 pgStatFunctions = hash_create("Function stat entries",
1603                                                                           PGSTAT_FUNCTION_HASH_SIZE,
1604                                                                           &hash_ctl,
1605                                                                           HASH_ELEM | HASH_BLOBS);
1606         }
1607
1608         /* Get the stats entry for this function, create if necessary */
1609         htabent = hash_search(pgStatFunctions, &fcinfo->flinfo->fn_oid,
1610                                                   HASH_ENTER, &found);
1611         if (!found)
1612                 MemSet(&htabent->f_counts, 0, sizeof(PgStat_FunctionCounts));
1613
1614         fcu->fs = &htabent->f_counts;
1615
1616         /* save stats for this function, later used to compensate for recursion */
1617         fcu->save_f_total_time = htabent->f_counts.f_total_time;
1618
1619         /* save current backend-wide total time */
1620         fcu->save_total = total_func_time;
1621
1622         /* get clock time as of function start */
1623         INSTR_TIME_SET_CURRENT(fcu->f_start);
1624 }
1625
1626 /*
1627  * find_funcstat_entry - find any existing PgStat_BackendFunctionEntry entry
1628  *              for specified function
1629  *
1630  * If no entry, return NULL, don't create a new one
1631  */
1632 PgStat_BackendFunctionEntry *
1633 find_funcstat_entry(Oid func_id)
1634 {
1635         if (pgStatFunctions == NULL)
1636                 return NULL;
1637
1638         return (PgStat_BackendFunctionEntry *) hash_search(pgStatFunctions,
1639                                                                                                            (void *) &func_id,
1640                                                                                                            HASH_FIND, NULL);
1641 }
1642
1643 /*
1644  * Calculate function call usage and update stat counters.
1645  * Called by the executor after invoking a function.
1646  *
1647  * In the case of a set-returning function that runs in value-per-call mode,
1648  * we will see multiple pgstat_init_function_usage/pgstat_end_function_usage
1649  * calls for what the user considers a single call of the function.  The
1650  * finalize flag should be TRUE on the last call.
1651  */
1652 void
1653 pgstat_end_function_usage(PgStat_FunctionCallUsage *fcu, bool finalize)
1654 {
1655         PgStat_FunctionCounts *fs = fcu->fs;
1656         instr_time      f_total;
1657         instr_time      f_others;
1658         instr_time      f_self;
1659
1660         /* stats not wanted? */
1661         if (fs == NULL)
1662                 return;
1663
1664         /* total elapsed time in this function call */
1665         INSTR_TIME_SET_CURRENT(f_total);
1666         INSTR_TIME_SUBTRACT(f_total, fcu->f_start);
1667
1668         /* self usage: elapsed minus anything already charged to other calls */
1669         f_others = total_func_time;
1670         INSTR_TIME_SUBTRACT(f_others, fcu->save_total);
1671         f_self = f_total;
1672         INSTR_TIME_SUBTRACT(f_self, f_others);
1673
1674         /* update backend-wide total time */
1675         INSTR_TIME_ADD(total_func_time, f_self);
1676
1677         /*
1678          * Compute the new f_total_time as the total elapsed time added to the
1679          * pre-call value of f_total_time.  This is necessary to avoid
1680          * double-counting any time taken by recursive calls of myself.  (We do
1681          * not need any similar kluge for self time, since that already excludes
1682          * any recursive calls.)
1683          */
1684         INSTR_TIME_ADD(f_total, fcu->save_f_total_time);
1685
1686         /* update counters in function stats table */
1687         if (finalize)
1688                 fs->f_numcalls++;
1689         fs->f_total_time = f_total;
1690         INSTR_TIME_ADD(fs->f_self_time, f_self);
1691
1692         /* indicate that we have something to send */
1693         have_function_stats = true;
1694 }
1695
1696
1697 /* ----------
1698  * pgstat_initstats() -
1699  *
1700  *      Initialize a relcache entry to count access statistics.
1701  *      Called whenever a relation is opened.
1702  *
1703  *      We assume that a relcache entry's pgstat_info field is zeroed by
1704  *      relcache.c when the relcache entry is made; thereafter it is long-lived
1705  *      data.  We can avoid repeated searches of the TabStatus arrays when the
1706  *      same relation is touched repeatedly within a transaction.
1707  * ----------
1708  */
1709 void
1710 pgstat_initstats(Relation rel)
1711 {
1712         Oid                     rel_id = rel->rd_id;
1713         char            relkind = rel->rd_rel->relkind;
1714
1715         /* We only count stats for things that have storage */
1716         if (!(relkind == RELKIND_RELATION ||
1717                   relkind == RELKIND_MATVIEW ||
1718                   relkind == RELKIND_INDEX ||
1719                   relkind == RELKIND_TOASTVALUE ||
1720                   relkind == RELKIND_SEQUENCE))
1721         {
1722                 rel->pgstat_info = NULL;
1723                 return;
1724         }
1725
1726         if (pgStatSock == PGINVALID_SOCKET || !pgstat_track_counts)
1727         {
1728                 /* We're not counting at all */
1729                 rel->pgstat_info = NULL;
1730                 return;
1731         }
1732
1733         /*
1734          * If we already set up this relation in the current transaction, nothing
1735          * to do.
1736          */
1737         if (rel->pgstat_info != NULL &&
1738                 rel->pgstat_info->t_id == rel_id)
1739                 return;
1740
1741         /* Else find or make the PgStat_TableStatus entry, and update link */
1742         rel->pgstat_info = get_tabstat_entry(rel_id, rel->rd_rel->relisshared);
1743 }
1744
1745 /*
1746  * get_tabstat_entry - find or create a PgStat_TableStatus entry for rel
1747  */
1748 static PgStat_TableStatus *
1749 get_tabstat_entry(Oid rel_id, bool isshared)
1750 {
1751         TabStatHashEntry *hash_entry;
1752         PgStat_TableStatus *entry;
1753         TabStatusArray *tsa;
1754         bool            found;
1755
1756         /*
1757          * Create hash table if we don't have it already.
1758          */
1759         if (pgStatTabHash == NULL)
1760         {
1761                 HASHCTL         ctl;
1762
1763                 memset(&ctl, 0, sizeof(ctl));
1764                 ctl.keysize = sizeof(Oid);
1765                 ctl.entrysize = sizeof(TabStatHashEntry);
1766
1767                 pgStatTabHash = hash_create("pgstat TabStatusArray lookup hash table",
1768                                                                         TABSTAT_QUANTUM,
1769                                                                         &ctl,
1770                                                                         HASH_ELEM | HASH_BLOBS);
1771         }
1772
1773         /*
1774          * Find an entry or create a new one.
1775          */
1776         hash_entry = hash_search(pgStatTabHash, &rel_id, HASH_ENTER, &found);
1777         if (!found)
1778         {
1779                 /* initialize new entry with null pointer */
1780                 hash_entry->tsa_entry = NULL;
1781         }
1782
1783         /*
1784          * If entry is already valid, we're done.
1785          */
1786         if (hash_entry->tsa_entry)
1787                 return hash_entry->tsa_entry;
1788
1789         /*
1790          * Locate the first pgStatTabList entry with free space, making a new list
1791          * entry if needed.  Note that we could get an OOM failure here, but if so
1792          * we have left the hashtable and the list in a consistent state.
1793          */
1794         if (pgStatTabList == NULL)
1795         {
1796                 /* Set up first pgStatTabList entry */
1797                 pgStatTabList = (TabStatusArray *)
1798                         MemoryContextAllocZero(TopMemoryContext,
1799                                                                    sizeof(TabStatusArray));
1800         }
1801
1802         tsa = pgStatTabList;
1803         while (tsa->tsa_used >= TABSTAT_QUANTUM)
1804         {
1805                 if (tsa->tsa_next == NULL)
1806                         tsa->tsa_next = (TabStatusArray *)
1807                                 MemoryContextAllocZero(TopMemoryContext,
1808                                                                            sizeof(TabStatusArray));
1809                 tsa = tsa->tsa_next;
1810         }
1811
1812         /*
1813          * Allocate a PgStat_TableStatus entry within this list entry.  We assume
1814          * the entry was already zeroed, either at creation or after last use.
1815          */
1816         entry = &tsa->tsa_entries[tsa->tsa_used++];
1817         entry->t_id = rel_id;
1818         entry->t_shared = isshared;
1819
1820         /*
1821          * Now we can fill the entry in pgStatTabHash.
1822          */
1823         hash_entry->tsa_entry = entry;
1824
1825         return entry;
1826 }
1827
1828 /*
1829  * find_tabstat_entry - find any existing PgStat_TableStatus entry for rel
1830  *
1831  * If no entry, return NULL, don't create a new one
1832  *
1833  * Note: if we got an error in the most recent execution of pgstat_report_stat,
1834  * it's possible that an entry exists but there's no hashtable entry for it.
1835  * That's okay, we'll treat this case as "doesn't exist".
1836  */
1837 PgStat_TableStatus *
1838 find_tabstat_entry(Oid rel_id)
1839 {
1840         TabStatHashEntry *hash_entry;
1841
1842         /* If hashtable doesn't exist, there are no entries at all */
1843         if (!pgStatTabHash)
1844                 return NULL;
1845
1846         hash_entry = hash_search(pgStatTabHash, &rel_id, HASH_FIND, NULL);
1847         if (!hash_entry)
1848                 return NULL;
1849
1850         /* Note that this step could also return NULL, but that's correct */
1851         return hash_entry->tsa_entry;
1852 }
1853
1854 /*
1855  * get_tabstat_stack_level - add a new (sub)transaction stack entry if needed
1856  */
1857 static PgStat_SubXactStatus *
1858 get_tabstat_stack_level(int nest_level)
1859 {
1860         PgStat_SubXactStatus *xact_state;
1861
1862         xact_state = pgStatXactStack;
1863         if (xact_state == NULL || xact_state->nest_level != nest_level)
1864         {
1865                 xact_state = (PgStat_SubXactStatus *)
1866                         MemoryContextAlloc(TopTransactionContext,
1867                                                            sizeof(PgStat_SubXactStatus));
1868                 xact_state->nest_level = nest_level;
1869                 xact_state->prev = pgStatXactStack;
1870                 xact_state->first = NULL;
1871                 pgStatXactStack = xact_state;
1872         }
1873         return xact_state;
1874 }
1875
1876 /*
1877  * add_tabstat_xact_level - add a new (sub)transaction state record
1878  */
1879 static void
1880 add_tabstat_xact_level(PgStat_TableStatus *pgstat_info, int nest_level)
1881 {
1882         PgStat_SubXactStatus *xact_state;
1883         PgStat_TableXactStatus *trans;
1884
1885         /*
1886          * If this is the first rel to be modified at the current nest level, we
1887          * first have to push a transaction stack entry.
1888          */
1889         xact_state = get_tabstat_stack_level(nest_level);
1890
1891         /* Now make a per-table stack entry */
1892         trans = (PgStat_TableXactStatus *)
1893                 MemoryContextAllocZero(TopTransactionContext,
1894                                                            sizeof(PgStat_TableXactStatus));
1895         trans->nest_level = nest_level;
1896         trans->upper = pgstat_info->trans;
1897         trans->parent = pgstat_info;
1898         trans->next = xact_state->first;
1899         xact_state->first = trans;
1900         pgstat_info->trans = trans;
1901 }
1902
1903 /*
1904  * pgstat_count_heap_insert - count a tuple insertion of n tuples
1905  */
1906 void
1907 pgstat_count_heap_insert(Relation rel, PgStat_Counter n)
1908 {
1909         PgStat_TableStatus *pgstat_info = rel->pgstat_info;
1910
1911         if (pgstat_info != NULL)
1912         {
1913                 /* We have to log the effect at the proper transactional level */
1914                 int                     nest_level = GetCurrentTransactionNestLevel();
1915
1916                 if (pgstat_info->trans == NULL ||
1917                         pgstat_info->trans->nest_level != nest_level)
1918                         add_tabstat_xact_level(pgstat_info, nest_level);
1919
1920                 pgstat_info->trans->tuples_inserted += n;
1921         }
1922 }
1923
1924 /*
1925  * pgstat_count_heap_update - count a tuple update
1926  */
1927 void
1928 pgstat_count_heap_update(Relation rel, bool hot)
1929 {
1930         PgStat_TableStatus *pgstat_info = rel->pgstat_info;
1931
1932         if (pgstat_info != NULL)
1933         {
1934                 /* We have to log the effect at the proper transactional level */
1935                 int                     nest_level = GetCurrentTransactionNestLevel();
1936
1937                 if (pgstat_info->trans == NULL ||
1938                         pgstat_info->trans->nest_level != nest_level)
1939                         add_tabstat_xact_level(pgstat_info, nest_level);
1940
1941                 pgstat_info->trans->tuples_updated++;
1942
1943                 /* t_tuples_hot_updated is nontransactional, so just advance it */
1944                 if (hot)
1945                         pgstat_info->t_counts.t_tuples_hot_updated++;
1946         }
1947 }
1948
1949 /*
1950  * pgstat_count_heap_delete - count a tuple deletion
1951  */
1952 void
1953 pgstat_count_heap_delete(Relation rel)
1954 {
1955         PgStat_TableStatus *pgstat_info = rel->pgstat_info;
1956
1957         if (pgstat_info != NULL)
1958         {
1959                 /* We have to log the effect at the proper transactional level */
1960                 int                     nest_level = GetCurrentTransactionNestLevel();
1961
1962                 if (pgstat_info->trans == NULL ||
1963                         pgstat_info->trans->nest_level != nest_level)
1964                         add_tabstat_xact_level(pgstat_info, nest_level);
1965
1966                 pgstat_info->trans->tuples_deleted++;
1967         }
1968 }
1969
1970 /*
1971  * pgstat_truncate_save_counters
1972  *
1973  * Whenever a table is truncated, we save its i/u/d counters so that they can
1974  * be cleared, and if the (sub)xact that executed the truncate later aborts,
1975  * the counters can be restored to the saved (pre-truncate) values.  Note we do
1976  * this on the first truncate in any particular subxact level only.
1977  */
1978 static void
1979 pgstat_truncate_save_counters(PgStat_TableXactStatus *trans)
1980 {
1981         if (!trans->truncated)
1982         {
1983                 trans->inserted_pre_trunc = trans->tuples_inserted;
1984                 trans->updated_pre_trunc = trans->tuples_updated;
1985                 trans->deleted_pre_trunc = trans->tuples_deleted;
1986                 trans->truncated = true;
1987         }
1988 }
1989
1990 /*
1991  * pgstat_truncate_restore_counters - restore counters when a truncate aborts
1992  */
1993 static void
1994 pgstat_truncate_restore_counters(PgStat_TableXactStatus *trans)
1995 {
1996         if (trans->truncated)
1997         {
1998                 trans->tuples_inserted = trans->inserted_pre_trunc;
1999                 trans->tuples_updated = trans->updated_pre_trunc;
2000                 trans->tuples_deleted = trans->deleted_pre_trunc;
2001         }
2002 }
2003
2004 /*
2005  * pgstat_count_truncate - update tuple counters due to truncate
2006  */
2007 void
2008 pgstat_count_truncate(Relation rel)
2009 {
2010         PgStat_TableStatus *pgstat_info = rel->pgstat_info;
2011
2012         if (pgstat_info != NULL)
2013         {
2014                 /* We have to log the effect at the proper transactional level */
2015                 int                     nest_level = GetCurrentTransactionNestLevel();
2016
2017                 if (pgstat_info->trans == NULL ||
2018                         pgstat_info->trans->nest_level != nest_level)
2019                         add_tabstat_xact_level(pgstat_info, nest_level);
2020
2021                 pgstat_truncate_save_counters(pgstat_info->trans);
2022                 pgstat_info->trans->tuples_inserted = 0;
2023                 pgstat_info->trans->tuples_updated = 0;
2024                 pgstat_info->trans->tuples_deleted = 0;
2025         }
2026 }
2027
2028 /*
2029  * pgstat_update_heap_dead_tuples - update dead-tuples count
2030  *
2031  * The semantics of this are that we are reporting the nontransactional
2032  * recovery of "delta" dead tuples; so t_delta_dead_tuples decreases
2033  * rather than increasing, and the change goes straight into the per-table
2034  * counter, not into transactional state.
2035  */
2036 void
2037 pgstat_update_heap_dead_tuples(Relation rel, int delta)
2038 {
2039         PgStat_TableStatus *pgstat_info = rel->pgstat_info;
2040
2041         if (pgstat_info != NULL)
2042                 pgstat_info->t_counts.t_delta_dead_tuples -= delta;
2043 }
2044
2045
2046 /* ----------
2047  * AtEOXact_PgStat
2048  *
2049  *      Called from access/transam/xact.c at top-level transaction commit/abort.
2050  * ----------
2051  */
2052 void
2053 AtEOXact_PgStat(bool isCommit)
2054 {
2055         PgStat_SubXactStatus *xact_state;
2056
2057         /*
2058          * Count transaction commit or abort.  (We use counters, not just bools,
2059          * in case the reporting message isn't sent right away.)
2060          */
2061         if (isCommit)
2062                 pgStatXactCommit++;
2063         else
2064                 pgStatXactRollback++;
2065
2066         /*
2067          * Transfer transactional insert/update counts into the base tabstat
2068          * entries.  We don't bother to free any of the transactional state, since
2069          * it's all in TopTransactionContext and will go away anyway.
2070          */
2071         xact_state = pgStatXactStack;
2072         if (xact_state != NULL)
2073         {
2074                 PgStat_TableXactStatus *trans;
2075
2076                 Assert(xact_state->nest_level == 1);
2077                 Assert(xact_state->prev == NULL);
2078                 for (trans = xact_state->first; trans != NULL; trans = trans->next)
2079                 {
2080                         PgStat_TableStatus *tabstat;
2081
2082                         Assert(trans->nest_level == 1);
2083                         Assert(trans->upper == NULL);
2084                         tabstat = trans->parent;
2085                         Assert(tabstat->trans == trans);
2086                         /* restore pre-truncate stats (if any) in case of aborted xact */
2087                         if (!isCommit)
2088                                 pgstat_truncate_restore_counters(trans);
2089                         /* count attempted actions regardless of commit/abort */
2090                         tabstat->t_counts.t_tuples_inserted += trans->tuples_inserted;
2091                         tabstat->t_counts.t_tuples_updated += trans->tuples_updated;
2092                         tabstat->t_counts.t_tuples_deleted += trans->tuples_deleted;
2093                         if (isCommit)
2094                         {
2095                                 tabstat->t_counts.t_truncated = trans->truncated;
2096                                 if (trans->truncated)
2097                                 {
2098                                         /* forget live/dead stats seen by backend thus far */
2099                                         tabstat->t_counts.t_delta_live_tuples = 0;
2100                                         tabstat->t_counts.t_delta_dead_tuples = 0;
2101                                 }
2102                                 /* insert adds a live tuple, delete removes one */
2103                                 tabstat->t_counts.t_delta_live_tuples +=
2104                                         trans->tuples_inserted - trans->tuples_deleted;
2105                                 /* update and delete each create a dead tuple */
2106                                 tabstat->t_counts.t_delta_dead_tuples +=
2107                                         trans->tuples_updated + trans->tuples_deleted;
2108                                 /* insert, update, delete each count as one change event */
2109                                 tabstat->t_counts.t_changed_tuples +=
2110                                         trans->tuples_inserted + trans->tuples_updated +
2111                                         trans->tuples_deleted;
2112                         }
2113                         else
2114                         {
2115                                 /* inserted tuples are dead, deleted tuples are unaffected */
2116                                 tabstat->t_counts.t_delta_dead_tuples +=
2117                                         trans->tuples_inserted + trans->tuples_updated;
2118                                 /* an aborted xact generates no changed_tuple events */
2119                         }
2120                         tabstat->trans = NULL;
2121                 }
2122         }
2123         pgStatXactStack = NULL;
2124
2125         /* Make sure any stats snapshot is thrown away */
2126         pgstat_clear_snapshot();
2127 }
2128
2129 /* ----------
2130  * AtEOSubXact_PgStat
2131  *
2132  *      Called from access/transam/xact.c at subtransaction commit/abort.
2133  * ----------
2134  */
2135 void
2136 AtEOSubXact_PgStat(bool isCommit, int nestDepth)
2137 {
2138         PgStat_SubXactStatus *xact_state;
2139
2140         /*
2141          * Transfer transactional insert/update counts into the next higher
2142          * subtransaction state.
2143          */
2144         xact_state = pgStatXactStack;
2145         if (xact_state != NULL &&
2146                 xact_state->nest_level >= nestDepth)
2147         {
2148                 PgStat_TableXactStatus *trans;
2149                 PgStat_TableXactStatus *next_trans;
2150
2151                 /* delink xact_state from stack immediately to simplify reuse case */
2152                 pgStatXactStack = xact_state->prev;
2153
2154                 for (trans = xact_state->first; trans != NULL; trans = next_trans)
2155                 {
2156                         PgStat_TableStatus *tabstat;
2157
2158                         next_trans = trans->next;
2159                         Assert(trans->nest_level == nestDepth);
2160                         tabstat = trans->parent;
2161                         Assert(tabstat->trans == trans);
2162                         if (isCommit)
2163                         {
2164                                 if (trans->upper && trans->upper->nest_level == nestDepth - 1)
2165                                 {
2166                                         if (trans->truncated)
2167                                         {
2168                                                 /* propagate the truncate status one level up */
2169                                                 pgstat_truncate_save_counters(trans->upper);
2170                                                 /* replace upper xact stats with ours */
2171                                                 trans->upper->tuples_inserted = trans->tuples_inserted;
2172                                                 trans->upper->tuples_updated = trans->tuples_updated;
2173                                                 trans->upper->tuples_deleted = trans->tuples_deleted;
2174                                         }
2175                                         else
2176                                         {
2177                                                 trans->upper->tuples_inserted += trans->tuples_inserted;
2178                                                 trans->upper->tuples_updated += trans->tuples_updated;
2179                                                 trans->upper->tuples_deleted += trans->tuples_deleted;
2180                                         }
2181                                         tabstat->trans = trans->upper;
2182                                         pfree(trans);
2183                                 }
2184                                 else
2185                                 {
2186                                         /*
2187                                          * When there isn't an immediate parent state, we can just
2188                                          * reuse the record instead of going through a
2189                                          * palloc/pfree pushup (this works since it's all in
2190                                          * TopTransactionContext anyway).  We have to re-link it
2191                                          * into the parent level, though, and that might mean
2192                                          * pushing a new entry into the pgStatXactStack.
2193                                          */
2194                                         PgStat_SubXactStatus *upper_xact_state;
2195
2196                                         upper_xact_state = get_tabstat_stack_level(nestDepth - 1);
2197                                         trans->next = upper_xact_state->first;
2198                                         upper_xact_state->first = trans;
2199                                         trans->nest_level = nestDepth - 1;
2200                                 }
2201                         }
2202                         else
2203                         {
2204                                 /*
2205                                  * On abort, update top-level tabstat counts, then forget the
2206                                  * subtransaction
2207                                  */
2208
2209                                 /* first restore values obliterated by truncate */
2210                                 pgstat_truncate_restore_counters(trans);
2211                                 /* count attempted actions regardless of commit/abort */
2212                                 tabstat->t_counts.t_tuples_inserted += trans->tuples_inserted;
2213                                 tabstat->t_counts.t_tuples_updated += trans->tuples_updated;
2214                                 tabstat->t_counts.t_tuples_deleted += trans->tuples_deleted;
2215                                 /* inserted tuples are dead, deleted tuples are unaffected */
2216                                 tabstat->t_counts.t_delta_dead_tuples +=
2217                                         trans->tuples_inserted + trans->tuples_updated;
2218                                 tabstat->trans = trans->upper;
2219                                 pfree(trans);
2220                         }
2221                 }
2222                 pfree(xact_state);
2223         }
2224 }
2225
2226
2227 /*
2228  * AtPrepare_PgStat
2229  *              Save the transactional stats state at 2PC transaction prepare.
2230  *
2231  * In this phase we just generate 2PC records for all the pending
2232  * transaction-dependent stats work.
2233  */
2234 void
2235 AtPrepare_PgStat(void)
2236 {
2237         PgStat_SubXactStatus *xact_state;
2238
2239         xact_state = pgStatXactStack;
2240         if (xact_state != NULL)
2241         {
2242                 PgStat_TableXactStatus *trans;
2243
2244                 Assert(xact_state->nest_level == 1);
2245                 Assert(xact_state->prev == NULL);
2246                 for (trans = xact_state->first; trans != NULL; trans = trans->next)
2247                 {
2248                         PgStat_TableStatus *tabstat;
2249                         TwoPhasePgStatRecord record;
2250
2251                         Assert(trans->nest_level == 1);
2252                         Assert(trans->upper == NULL);
2253                         tabstat = trans->parent;
2254                         Assert(tabstat->trans == trans);
2255
2256                         record.tuples_inserted = trans->tuples_inserted;
2257                         record.tuples_updated = trans->tuples_updated;
2258                         record.tuples_deleted = trans->tuples_deleted;
2259                         record.inserted_pre_trunc = trans->inserted_pre_trunc;
2260                         record.updated_pre_trunc = trans->updated_pre_trunc;
2261                         record.deleted_pre_trunc = trans->deleted_pre_trunc;
2262                         record.t_id = tabstat->t_id;
2263                         record.t_shared = tabstat->t_shared;
2264                         record.t_truncated = trans->truncated;
2265
2266                         RegisterTwoPhaseRecord(TWOPHASE_RM_PGSTAT_ID, 0,
2267                                                                    &record, sizeof(TwoPhasePgStatRecord));
2268                 }
2269         }
2270 }
2271
2272 /*
2273  * PostPrepare_PgStat
2274  *              Clean up after successful PREPARE.
2275  *
2276  * All we need do here is unlink the transaction stats state from the
2277  * nontransactional state.  The nontransactional action counts will be
2278  * reported to the stats collector immediately, while the effects on live
2279  * and dead tuple counts are preserved in the 2PC state file.
2280  *
2281  * Note: AtEOXact_PgStat is not called during PREPARE.
2282  */
2283 void
2284 PostPrepare_PgStat(void)
2285 {
2286         PgStat_SubXactStatus *xact_state;
2287
2288         /*
2289          * We don't bother to free any of the transactional state, since it's all
2290          * in TopTransactionContext and will go away anyway.
2291          */
2292         xact_state = pgStatXactStack;
2293         if (xact_state != NULL)
2294         {
2295                 PgStat_TableXactStatus *trans;
2296
2297                 for (trans = xact_state->first; trans != NULL; trans = trans->next)
2298                 {
2299                         PgStat_TableStatus *tabstat;
2300
2301                         tabstat = trans->parent;
2302                         tabstat->trans = NULL;
2303                 }
2304         }
2305         pgStatXactStack = NULL;
2306
2307         /* Make sure any stats snapshot is thrown away */
2308         pgstat_clear_snapshot();
2309 }
2310
2311 /*
2312  * 2PC processing routine for COMMIT PREPARED case.
2313  *
2314  * Load the saved counts into our local pgstats state.
2315  */
2316 void
2317 pgstat_twophase_postcommit(TransactionId xid, uint16 info,
2318                                                    void *recdata, uint32 len)
2319 {
2320         TwoPhasePgStatRecord *rec = (TwoPhasePgStatRecord *) recdata;
2321         PgStat_TableStatus *pgstat_info;
2322
2323         /* Find or create a tabstat entry for the rel */
2324         pgstat_info = get_tabstat_entry(rec->t_id, rec->t_shared);
2325
2326         /* Same math as in AtEOXact_PgStat, commit case */
2327         pgstat_info->t_counts.t_tuples_inserted += rec->tuples_inserted;
2328         pgstat_info->t_counts.t_tuples_updated += rec->tuples_updated;
2329         pgstat_info->t_counts.t_tuples_deleted += rec->tuples_deleted;
2330         pgstat_info->t_counts.t_truncated = rec->t_truncated;
2331         if (rec->t_truncated)
2332         {
2333                 /* forget live/dead stats seen by backend thus far */
2334                 pgstat_info->t_counts.t_delta_live_tuples = 0;
2335                 pgstat_info->t_counts.t_delta_dead_tuples = 0;
2336         }
2337         pgstat_info->t_counts.t_delta_live_tuples +=
2338                 rec->tuples_inserted - rec->tuples_deleted;
2339         pgstat_info->t_counts.t_delta_dead_tuples +=
2340                 rec->tuples_updated + rec->tuples_deleted;
2341         pgstat_info->t_counts.t_changed_tuples +=
2342                 rec->tuples_inserted + rec->tuples_updated +
2343                 rec->tuples_deleted;
2344 }
2345
2346 /*
2347  * 2PC processing routine for ROLLBACK PREPARED case.
2348  *
2349  * Load the saved counts into our local pgstats state, but treat them
2350  * as aborted.
2351  */
2352 void
2353 pgstat_twophase_postabort(TransactionId xid, uint16 info,
2354                                                   void *recdata, uint32 len)
2355 {
2356         TwoPhasePgStatRecord *rec = (TwoPhasePgStatRecord *) recdata;
2357         PgStat_TableStatus *pgstat_info;
2358
2359         /* Find or create a tabstat entry for the rel */
2360         pgstat_info = get_tabstat_entry(rec->t_id, rec->t_shared);
2361
2362         /* Same math as in AtEOXact_PgStat, abort case */
2363         if (rec->t_truncated)
2364         {
2365                 rec->tuples_inserted = rec->inserted_pre_trunc;
2366                 rec->tuples_updated = rec->updated_pre_trunc;
2367                 rec->tuples_deleted = rec->deleted_pre_trunc;
2368         }
2369         pgstat_info->t_counts.t_tuples_inserted += rec->tuples_inserted;
2370         pgstat_info->t_counts.t_tuples_updated += rec->tuples_updated;
2371         pgstat_info->t_counts.t_tuples_deleted += rec->tuples_deleted;
2372         pgstat_info->t_counts.t_delta_dead_tuples +=
2373                 rec->tuples_inserted + rec->tuples_updated;
2374 }
2375
2376
2377 /* ----------
2378  * pgstat_fetch_stat_dbentry() -
2379  *
2380  *      Support function for the SQL-callable pgstat* functions. Returns
2381  *      the collected statistics for one database or NULL. NULL doesn't mean
2382  *      that the database doesn't exist, it is just not yet known by the
2383  *      collector, so the caller is better off to report ZERO instead.
2384  * ----------
2385  */
2386 PgStat_StatDBEntry *
2387 pgstat_fetch_stat_dbentry(Oid dbid)
2388 {
2389         /*
2390          * If not done for this transaction, read the statistics collector stats
2391          * file into some hash tables.
2392          */
2393         backend_read_statsfile();
2394
2395         /*
2396          * Lookup the requested database; return NULL if not found
2397          */
2398         return (PgStat_StatDBEntry *) hash_search(pgStatDBHash,
2399                                                                                           (void *) &dbid,
2400                                                                                           HASH_FIND, NULL);
2401 }
2402
2403
2404 /* ----------
2405  * pgstat_fetch_stat_tabentry() -
2406  *
2407  *      Support function for the SQL-callable pgstat* functions. Returns
2408  *      the collected statistics for one table or NULL. NULL doesn't mean
2409  *      that the table doesn't exist, it is just not yet known by the
2410  *      collector, so the caller is better off to report ZERO instead.
2411  * ----------
2412  */
2413 PgStat_StatTabEntry *
2414 pgstat_fetch_stat_tabentry(Oid relid)
2415 {
2416         Oid                     dbid;
2417         PgStat_StatDBEntry *dbentry;
2418         PgStat_StatTabEntry *tabentry;
2419
2420         /*
2421          * If not done for this transaction, read the statistics collector stats
2422          * file into some hash tables.
2423          */
2424         backend_read_statsfile();
2425
2426         /*
2427          * Lookup our database, then look in its table hash table.
2428          */
2429         dbid = MyDatabaseId;
2430         dbentry = (PgStat_StatDBEntry *) hash_search(pgStatDBHash,
2431                                                                                                  (void *) &dbid,
2432                                                                                                  HASH_FIND, NULL);
2433         if (dbentry != NULL && dbentry->tables != NULL)
2434         {
2435                 tabentry = (PgStat_StatTabEntry *) hash_search(dbentry->tables,
2436                                                                                                            (void *) &relid,
2437                                                                                                            HASH_FIND, NULL);
2438                 if (tabentry)
2439                         return tabentry;
2440         }
2441
2442         /*
2443          * If we didn't find it, maybe it's a shared table.
2444          */
2445         dbid = InvalidOid;
2446         dbentry = (PgStat_StatDBEntry *) hash_search(pgStatDBHash,
2447                                                                                                  (void *) &dbid,
2448                                                                                                  HASH_FIND, NULL);
2449         if (dbentry != NULL && dbentry->tables != NULL)
2450         {
2451                 tabentry = (PgStat_StatTabEntry *) hash_search(dbentry->tables,
2452                                                                                                            (void *) &relid,
2453                                                                                                            HASH_FIND, NULL);
2454                 if (tabentry)
2455                         return tabentry;
2456         }
2457
2458         return NULL;
2459 }
2460
2461
2462 /* ----------
2463  * pgstat_fetch_stat_funcentry() -
2464  *
2465  *      Support function for the SQL-callable pgstat* functions. Returns
2466  *      the collected statistics for one function or NULL.
2467  * ----------
2468  */
2469 PgStat_StatFuncEntry *
2470 pgstat_fetch_stat_funcentry(Oid func_id)
2471 {
2472         PgStat_StatDBEntry *dbentry;
2473         PgStat_StatFuncEntry *funcentry = NULL;
2474
2475         /* load the stats file if needed */
2476         backend_read_statsfile();
2477
2478         /* Lookup our database, then find the requested function.  */
2479         dbentry = pgstat_fetch_stat_dbentry(MyDatabaseId);
2480         if (dbentry != NULL && dbentry->functions != NULL)
2481         {
2482                 funcentry = (PgStat_StatFuncEntry *) hash_search(dbentry->functions,
2483                                                                                                                  (void *) &func_id,
2484                                                                                                                  HASH_FIND, NULL);
2485         }
2486
2487         return funcentry;
2488 }
2489
2490
2491 /* ----------
2492  * pgstat_fetch_stat_beentry() -
2493  *
2494  *      Support function for the SQL-callable pgstat* functions. Returns
2495  *      our local copy of the current-activity entry for one backend.
2496  *
2497  *      NB: caller is responsible for a check if the user is permitted to see
2498  *      this info (especially the querystring).
2499  * ----------
2500  */
2501 PgBackendStatus *
2502 pgstat_fetch_stat_beentry(int beid)
2503 {
2504         pgstat_read_current_status();
2505
2506         if (beid < 1 || beid > localNumBackends)
2507                 return NULL;
2508
2509         return &localBackendStatusTable[beid - 1].backendStatus;
2510 }
2511
2512
2513 /* ----------
2514  * pgstat_fetch_stat_local_beentry() -
2515  *
2516  *      Like pgstat_fetch_stat_beentry() but with locally computed additions (like
2517  *      xid and xmin values of the backend)
2518  *
2519  *      NB: caller is responsible for a check if the user is permitted to see
2520  *      this info (especially the querystring).
2521  * ----------
2522  */
2523 LocalPgBackendStatus *
2524 pgstat_fetch_stat_local_beentry(int beid)
2525 {
2526         pgstat_read_current_status();
2527
2528         if (beid < 1 || beid > localNumBackends)
2529                 return NULL;
2530
2531         return &localBackendStatusTable[beid - 1];
2532 }
2533
2534
2535 /* ----------
2536  * pgstat_fetch_stat_numbackends() -
2537  *
2538  *      Support function for the SQL-callable pgstat* functions. Returns
2539  *      the maximum current backend id.
2540  * ----------
2541  */
2542 int
2543 pgstat_fetch_stat_numbackends(void)
2544 {
2545         pgstat_read_current_status();
2546
2547         return localNumBackends;
2548 }
2549
2550 /*
2551  * ---------
2552  * pgstat_fetch_stat_archiver() -
2553  *
2554  *      Support function for the SQL-callable pgstat* functions. Returns
2555  *      a pointer to the archiver statistics struct.
2556  * ---------
2557  */
2558 PgStat_ArchiverStats *
2559 pgstat_fetch_stat_archiver(void)
2560 {
2561         backend_read_statsfile();
2562
2563         return &archiverStats;
2564 }
2565
2566
2567 /*
2568  * ---------
2569  * pgstat_fetch_global() -
2570  *
2571  *      Support function for the SQL-callable pgstat* functions. Returns
2572  *      a pointer to the global statistics struct.
2573  * ---------
2574  */
2575 PgStat_GlobalStats *
2576 pgstat_fetch_global(void)
2577 {
2578         backend_read_statsfile();
2579
2580         return &globalStats;
2581 }
2582
2583
2584 /* ------------------------------------------------------------
2585  * Functions for management of the shared-memory PgBackendStatus array
2586  * ------------------------------------------------------------
2587  */
2588
2589 static PgBackendStatus *BackendStatusArray = NULL;
2590 static PgBackendStatus *MyBEEntry = NULL;
2591 static char *BackendAppnameBuffer = NULL;
2592 static char *BackendClientHostnameBuffer = NULL;
2593 static char *BackendActivityBuffer = NULL;
2594 static Size BackendActivityBufferSize = 0;
2595 #ifdef USE_SSL
2596 static PgBackendSSLStatus *BackendSslStatusBuffer = NULL;
2597 #endif
2598
2599
2600 /*
2601  * Report shared-memory space needed by CreateSharedBackendStatus.
2602  */
2603 Size
2604 BackendStatusShmemSize(void)
2605 {
2606         Size            size;
2607
2608         /* BackendStatusArray: */
2609         size = mul_size(sizeof(PgBackendStatus), NumBackendStatSlots);
2610         /* BackendAppnameBuffer: */
2611         size = add_size(size,
2612                                         mul_size(NAMEDATALEN, NumBackendStatSlots));
2613         /* BackendClientHostnameBuffer: */
2614         size = add_size(size,
2615                                         mul_size(NAMEDATALEN, NumBackendStatSlots));
2616         /* BackendActivityBuffer: */
2617         size = add_size(size,
2618                                         mul_size(pgstat_track_activity_query_size, NumBackendStatSlots));
2619 #ifdef USE_SSL
2620         /* BackendSslStatusBuffer: */
2621         size = add_size(size,
2622                                         mul_size(sizeof(PgBackendSSLStatus), NumBackendStatSlots));
2623 #endif
2624         return size;
2625 }
2626
2627 /*
2628  * Initialize the shared status array and several string buffers
2629  * during postmaster startup.
2630  */
2631 void
2632 CreateSharedBackendStatus(void)
2633 {
2634         Size            size;
2635         bool            found;
2636         int                     i;
2637         char       *buffer;
2638
2639         /* Create or attach to the shared array */
2640         size = mul_size(sizeof(PgBackendStatus), NumBackendStatSlots);
2641         BackendStatusArray = (PgBackendStatus *)
2642                 ShmemInitStruct("Backend Status Array", size, &found);
2643
2644         if (!found)
2645         {
2646                 /*
2647                  * We're the first - initialize.
2648                  */
2649                 MemSet(BackendStatusArray, 0, size);
2650         }
2651
2652         /* Create or attach to the shared appname buffer */
2653         size = mul_size(NAMEDATALEN, MaxBackends);
2654         BackendAppnameBuffer = (char *)
2655                 ShmemInitStruct("Backend Application Name Buffer", size, &found);
2656
2657         if (!found)
2658         {
2659                 MemSet(BackendAppnameBuffer, 0, size);
2660
2661                 /* Initialize st_appname pointers. */
2662                 buffer = BackendAppnameBuffer;
2663                 for (i = 0; i < NumBackendStatSlots; i++)
2664                 {
2665                         BackendStatusArray[i].st_appname = buffer;
2666                         buffer += NAMEDATALEN;
2667                 }
2668         }
2669
2670         /* Create or attach to the shared client hostname buffer */
2671         size = mul_size(NAMEDATALEN, MaxBackends);
2672         BackendClientHostnameBuffer = (char *)
2673                 ShmemInitStruct("Backend Client Host Name Buffer", size, &found);
2674
2675         if (!found)
2676         {
2677                 MemSet(BackendClientHostnameBuffer, 0, size);
2678
2679                 /* Initialize st_clienthostname pointers. */
2680                 buffer = BackendClientHostnameBuffer;
2681                 for (i = 0; i < NumBackendStatSlots; i++)
2682                 {
2683                         BackendStatusArray[i].st_clienthostname = buffer;
2684                         buffer += NAMEDATALEN;
2685                 }
2686         }
2687
2688         /* Create or attach to the shared activity buffer */
2689         BackendActivityBufferSize = mul_size(pgstat_track_activity_query_size,
2690                                                                                  NumBackendStatSlots);
2691         BackendActivityBuffer = (char *)
2692                 ShmemInitStruct("Backend Activity Buffer",
2693                                                 BackendActivityBufferSize,
2694                                                 &found);
2695
2696         if (!found)
2697         {
2698                 MemSet(BackendActivityBuffer, 0, size);
2699
2700                 /* Initialize st_activity pointers. */
2701                 buffer = BackendActivityBuffer;
2702                 for (i = 0; i < NumBackendStatSlots; i++)
2703                 {
2704                         BackendStatusArray[i].st_activity = buffer;
2705                         buffer += pgstat_track_activity_query_size;
2706                 }
2707         }
2708
2709 #ifdef USE_SSL
2710         /* Create or attach to the shared SSL status buffer */
2711         size = mul_size(sizeof(PgBackendSSLStatus), NumBackendStatSlots);
2712         BackendSslStatusBuffer = (PgBackendSSLStatus *)
2713                 ShmemInitStruct("Backend SSL Status Buffer", size, &found);
2714
2715         if (!found)
2716         {
2717                 PgBackendSSLStatus *ptr;
2718
2719                 MemSet(BackendSslStatusBuffer, 0, size);
2720
2721                 /* Initialize st_sslstatus pointers. */
2722                 ptr = BackendSslStatusBuffer;
2723                 for (i = 0; i < NumBackendStatSlots; i++)
2724                 {
2725                         BackendStatusArray[i].st_sslstatus = ptr;
2726                         ptr++;
2727                 }
2728         }
2729 #endif
2730 }
2731
2732
2733 /* ----------
2734  * pgstat_initialize() -
2735  *
2736  *      Initialize pgstats state, and set up our on-proc-exit hook.
2737  *      Called from InitPostgres and AuxiliaryProcessMain. For auxiliary process,
2738  *      MyBackendId is invalid. Otherwise, MyBackendId must be set,
2739  *      but we must not have started any transaction yet (since the
2740  *      exit hook must run after the last transaction exit).
2741  *      NOTE: MyDatabaseId isn't set yet; so the shutdown hook has to be careful.
2742  * ----------
2743  */
2744 void
2745 pgstat_initialize(void)
2746 {
2747         /* Initialize MyBEEntry */
2748         if (MyBackendId != InvalidBackendId)
2749         {
2750                 Assert(MyBackendId >= 1 && MyBackendId <= MaxBackends);
2751                 MyBEEntry = &BackendStatusArray[MyBackendId - 1];
2752         }
2753         else
2754         {
2755                 /* Must be an auxiliary process */
2756                 Assert(MyAuxProcType != NotAnAuxProcess);
2757
2758                 /*
2759                  * Assign the MyBEEntry for an auxiliary process.  Since it doesn't
2760                  * have a BackendId, the slot is statically allocated based on the
2761                  * auxiliary process type (MyAuxProcType).  Backends use slots indexed
2762                  * in the range from 1 to MaxBackends (inclusive), so we use
2763                  * MaxBackends + AuxBackendType + 1 as the index of the slot for an
2764                  * auxiliary process.
2765                  */
2766                 MyBEEntry = &BackendStatusArray[MaxBackends + MyAuxProcType];
2767         }
2768
2769         /* Set up a process-exit hook to clean up */
2770         on_shmem_exit(pgstat_beshutdown_hook, 0);
2771 }
2772
2773 /* ----------
2774  * pgstat_bestart() -
2775  *
2776  *      Initialize this backend's entry in the PgBackendStatus array.
2777  *      Called from InitPostgres.
2778  *
2779  *      Apart from auxiliary processes, MyBackendId, MyDatabaseId,
2780  *      session userid, and application_name must be set for a
2781  *      backend (hence, this cannot be combined with pgstat_initialize).
2782  * ----------
2783  */
2784 void
2785 pgstat_bestart(void)
2786 {
2787         TimestampTz proc_start_timestamp;
2788         SockAddr        clientaddr;
2789         volatile PgBackendStatus *beentry;
2790
2791         /*
2792          * To minimize the time spent modifying the PgBackendStatus entry, fetch
2793          * all the needed data first.
2794          *
2795          * If we have a MyProcPort, use its session start time (for consistency,
2796          * and to save a kernel call).
2797          */
2798         if (MyProcPort)
2799                 proc_start_timestamp = MyProcPort->SessionStartTime;
2800         else
2801                 proc_start_timestamp = GetCurrentTimestamp();
2802
2803         /*
2804          * We may not have a MyProcPort (eg, if this is the autovacuum process).
2805          * If so, use all-zeroes client address, which is dealt with specially in
2806          * pg_stat_get_backend_client_addr and pg_stat_get_backend_client_port.
2807          */
2808         if (MyProcPort)
2809                 memcpy(&clientaddr, &MyProcPort->raddr, sizeof(clientaddr));
2810         else
2811                 MemSet(&clientaddr, 0, sizeof(clientaddr));
2812
2813         /*
2814          * Initialize my status entry, following the protocol of bumping
2815          * st_changecount before and after; and make sure it's even afterwards. We
2816          * use a volatile pointer here to ensure the compiler doesn't try to get
2817          * cute.
2818          */
2819         beentry = MyBEEntry;
2820
2821         /* pgstats state must be initialized from pgstat_initialize() */
2822         Assert(beentry != NULL);
2823
2824         if (MyBackendId != InvalidBackendId)
2825         {
2826                 if (IsAutoVacuumLauncherProcess())
2827                 {
2828                         /* Autovacuum Launcher */
2829                         beentry->st_backendType = B_AUTOVAC_LAUNCHER;
2830                 }
2831                 else if (IsAutoVacuumWorkerProcess())
2832                 {
2833                         /* Autovacuum Worker */
2834                         beentry->st_backendType = B_AUTOVAC_WORKER;
2835                 }
2836                 else if (am_walsender)
2837                 {
2838                         /* Wal sender */
2839                         beentry->st_backendType = B_WAL_SENDER;
2840                 }
2841                 else if (IsBackgroundWorker)
2842                 {
2843                         /* bgworker */
2844                         beentry->st_backendType = B_BG_WORKER;
2845                 }
2846                 else
2847                 {
2848                         /* client-backend */
2849                         beentry->st_backendType = B_BACKEND;
2850                 }
2851         }
2852         else
2853         {
2854                 /* Must be an auxiliary process */
2855                 Assert(MyAuxProcType != NotAnAuxProcess);
2856                 switch (MyAuxProcType)
2857                 {
2858                         case StartupProcess:
2859                                 beentry->st_backendType = B_STARTUP;
2860                                 break;
2861                         case BgWriterProcess:
2862                                 beentry->st_backendType = B_BG_WRITER;
2863                                 break;
2864                         case CheckpointerProcess:
2865                                 beentry->st_backendType = B_CHECKPOINTER;
2866                                 break;
2867                         case WalWriterProcess:
2868                                 beentry->st_backendType = B_WAL_WRITER;
2869                                 break;
2870                         case WalReceiverProcess:
2871                                 beentry->st_backendType = B_WAL_RECEIVER;
2872                                 break;
2873                         default:
2874                                 elog(FATAL, "unrecognized process type: %d",
2875                                          (int) MyAuxProcType);
2876                                 proc_exit(1);
2877                 }
2878         }
2879
2880         do
2881         {
2882                 pgstat_increment_changecount_before(beentry);
2883         } while ((beentry->st_changecount & 1) == 0);
2884
2885         beentry->st_procpid = MyProcPid;
2886         beentry->st_proc_start_timestamp = proc_start_timestamp;
2887         beentry->st_activity_start_timestamp = 0;
2888         beentry->st_state_start_timestamp = 0;
2889         beentry->st_xact_start_timestamp = 0;
2890         beentry->st_databaseid = MyDatabaseId;
2891
2892         /* We have userid for client-backends, wal-sender and bgworker processes */
2893         if (beentry->st_backendType == B_BACKEND
2894                 || beentry->st_backendType == B_WAL_SENDER
2895                 || beentry->st_backendType == B_BG_WORKER)
2896                 beentry->st_userid = GetSessionUserId();
2897         else
2898                 beentry->st_userid = InvalidOid;
2899
2900         beentry->st_clientaddr = clientaddr;
2901         if (MyProcPort && MyProcPort->remote_hostname)
2902                 strlcpy(beentry->st_clienthostname, MyProcPort->remote_hostname,
2903                                 NAMEDATALEN);
2904         else
2905                 beentry->st_clienthostname[0] = '\0';
2906 #ifdef USE_SSL
2907         if (MyProcPort && MyProcPort->ssl != NULL)
2908         {
2909                 beentry->st_ssl = true;
2910                 beentry->st_sslstatus->ssl_bits = be_tls_get_cipher_bits(MyProcPort);
2911                 beentry->st_sslstatus->ssl_compression = be_tls_get_compression(MyProcPort);
2912                 be_tls_get_version(MyProcPort, beentry->st_sslstatus->ssl_version, NAMEDATALEN);
2913                 be_tls_get_cipher(MyProcPort, beentry->st_sslstatus->ssl_cipher, NAMEDATALEN);
2914                 be_tls_get_peerdn_name(MyProcPort, beentry->st_sslstatus->ssl_clientdn, NAMEDATALEN);
2915         }
2916         else
2917         {
2918                 beentry->st_ssl = false;
2919         }
2920 #else
2921         beentry->st_ssl = false;
2922 #endif
2923         beentry->st_state = STATE_UNDEFINED;
2924         beentry->st_appname[0] = '\0';
2925         beentry->st_activity[0] = '\0';
2926         /* Also make sure the last byte in each string area is always 0 */
2927         beentry->st_clienthostname[NAMEDATALEN - 1] = '\0';
2928         beentry->st_appname[NAMEDATALEN - 1] = '\0';
2929         beentry->st_activity[pgstat_track_activity_query_size - 1] = '\0';
2930         beentry->st_progress_command = PROGRESS_COMMAND_INVALID;
2931         beentry->st_progress_command_target = InvalidOid;
2932
2933         /*
2934          * we don't zero st_progress_param here to save cycles; nobody should
2935          * examine it until st_progress_command has been set to something other
2936          * than PROGRESS_COMMAND_INVALID
2937          */
2938
2939         pgstat_increment_changecount_after(beentry);
2940
2941         /* Update app name to current GUC setting */
2942         if (application_name)
2943                 pgstat_report_appname(application_name);
2944 }
2945
2946 /*
2947  * Shut down a single backend's statistics reporting at process exit.
2948  *
2949  * Flush any remaining statistics counts out to the collector.
2950  * Without this, operations triggered during backend exit (such as
2951  * temp table deletions) won't be counted.
2952  *
2953  * Lastly, clear out our entry in the PgBackendStatus array.
2954  */
2955 static void
2956 pgstat_beshutdown_hook(int code, Datum arg)
2957 {
2958         volatile PgBackendStatus *beentry = MyBEEntry;
2959
2960         /*
2961          * If we got as far as discovering our own database ID, we can report what
2962          * we did to the collector.  Otherwise, we'd be sending an invalid
2963          * database ID, so forget it.  (This means that accesses to pg_database
2964          * during failed backend starts might never get counted.)
2965          */
2966         if (OidIsValid(MyDatabaseId))
2967                 pgstat_report_stat(true);
2968
2969         /*
2970          * Clear my status entry, following the protocol of bumping st_changecount
2971          * before and after.  We use a volatile pointer here to ensure the
2972          * compiler doesn't try to get cute.
2973          */
2974         pgstat_increment_changecount_before(beentry);
2975
2976         beentry->st_procpid = 0;        /* mark invalid */
2977
2978         pgstat_increment_changecount_after(beentry);
2979 }
2980
2981
2982 /* ----------
2983  * pgstat_report_activity() -
2984  *
2985  *      Called from tcop/postgres.c to report what the backend is actually doing
2986  *      (but note cmd_str can be NULL for certain cases).
2987  *
2988  * All updates of the status entry follow the protocol of bumping
2989  * st_changecount before and after.  We use a volatile pointer here to
2990  * ensure the compiler doesn't try to get cute.
2991  * ----------
2992  */
2993 void
2994 pgstat_report_activity(BackendState state, const char *cmd_str)
2995 {
2996         volatile PgBackendStatus *beentry = MyBEEntry;
2997         TimestampTz start_timestamp;
2998         TimestampTz current_timestamp;
2999         int                     len = 0;
3000
3001         TRACE_POSTGRESQL_STATEMENT_STATUS(cmd_str);
3002
3003         if (!beentry)
3004                 return;
3005
3006         if (!pgstat_track_activities)
3007         {
3008                 if (beentry->st_state != STATE_DISABLED)
3009                 {
3010                         volatile PGPROC *proc = MyProc;
3011
3012                         /*
3013                          * track_activities is disabled, but we last reported a
3014                          * non-disabled state.  As our final update, change the state and
3015                          * clear fields we will not be updating anymore.
3016                          */
3017                         pgstat_increment_changecount_before(beentry);
3018                         beentry->st_state = STATE_DISABLED;
3019                         beentry->st_state_start_timestamp = 0;
3020                         beentry->st_activity[0] = '\0';
3021                         beentry->st_activity_start_timestamp = 0;
3022                         /* st_xact_start_timestamp and wait_event_info are also disabled */
3023                         beentry->st_xact_start_timestamp = 0;
3024                         proc->wait_event_info = 0;
3025                         pgstat_increment_changecount_after(beentry);
3026                 }
3027                 return;
3028         }
3029
3030         /*
3031          * To minimize the time spent modifying the entry, fetch all the needed
3032          * data first.
3033          */
3034         start_timestamp = GetCurrentStatementStartTimestamp();
3035         if (cmd_str != NULL)
3036         {
3037                 len = pg_mbcliplen(cmd_str, strlen(cmd_str),
3038                                                    pgstat_track_activity_query_size - 1);
3039         }
3040         current_timestamp = GetCurrentTimestamp();
3041
3042         /*
3043          * Now update the status entry
3044          */
3045         pgstat_increment_changecount_before(beentry);
3046
3047         beentry->st_state = state;
3048         beentry->st_state_start_timestamp = current_timestamp;
3049
3050         if (cmd_str != NULL)
3051         {
3052                 memcpy((char *) beentry->st_activity, cmd_str, len);
3053                 beentry->st_activity[len] = '\0';
3054                 beentry->st_activity_start_timestamp = start_timestamp;
3055         }
3056
3057         pgstat_increment_changecount_after(beentry);
3058 }
3059
3060 /*-----------
3061  * pgstat_progress_start_command() -
3062  *
3063  * Set st_progress_command (and st_progress_command_target) in own backend
3064  * entry.  Also, zero-initialize st_progress_param array.
3065  *-----------
3066  */
3067 void
3068 pgstat_progress_start_command(ProgressCommandType cmdtype, Oid relid)
3069 {
3070         volatile PgBackendStatus *beentry = MyBEEntry;
3071
3072         if (!beentry || !pgstat_track_activities)
3073                 return;
3074
3075         pgstat_increment_changecount_before(beentry);
3076         beentry->st_progress_command = cmdtype;
3077         beentry->st_progress_command_target = relid;
3078         MemSet(&beentry->st_progress_param, 0, sizeof(beentry->st_progress_param));
3079         pgstat_increment_changecount_after(beentry);
3080 }
3081
3082 /*-----------
3083  * pgstat_progress_update_param() -
3084  *
3085  * Update index'th member in st_progress_param[] of own backend entry.
3086  *-----------
3087  */
3088 void
3089 pgstat_progress_update_param(int index, int64 val)
3090 {
3091         volatile PgBackendStatus *beentry = MyBEEntry;
3092
3093         Assert(index >= 0 && index < PGSTAT_NUM_PROGRESS_PARAM);
3094
3095         if (!beentry || !pgstat_track_activities)
3096                 return;
3097
3098         pgstat_increment_changecount_before(beentry);
3099         beentry->st_progress_param[index] = val;
3100         pgstat_increment_changecount_after(beentry);
3101 }
3102
3103 /*-----------
3104  * pgstat_progress_update_multi_param() -
3105  *
3106  * Update multiple members in st_progress_param[] of own backend entry.
3107  * This is atomic; readers won't see intermediate states.
3108  *-----------
3109  */
3110 void
3111 pgstat_progress_update_multi_param(int nparam, const int *index,
3112                                                                    const int64 *val)
3113 {
3114         volatile PgBackendStatus *beentry = MyBEEntry;
3115         int                     i;
3116
3117         if (!beentry || !pgstat_track_activities || nparam == 0)
3118                 return;
3119
3120         pgstat_increment_changecount_before(beentry);
3121
3122         for (i = 0; i < nparam; ++i)
3123         {
3124                 Assert(index[i] >= 0 && index[i] < PGSTAT_NUM_PROGRESS_PARAM);
3125
3126                 beentry->st_progress_param[index[i]] = val[i];
3127         }
3128
3129         pgstat_increment_changecount_after(beentry);
3130 }
3131
3132 /*-----------
3133  * pgstat_progress_end_command() -
3134  *
3135  * Reset st_progress_command (and st_progress_command_target) in own backend
3136  * entry.  This signals the end of the command.
3137  *-----------
3138  */
3139 void
3140 pgstat_progress_end_command(void)
3141 {
3142         volatile PgBackendStatus *beentry = MyBEEntry;
3143
3144         if (!beentry)
3145                 return;
3146         if (!pgstat_track_activities
3147                 && beentry->st_progress_command == PROGRESS_COMMAND_INVALID)
3148                 return;
3149
3150         pgstat_increment_changecount_before(beentry);
3151         beentry->st_progress_command = PROGRESS_COMMAND_INVALID;
3152         beentry->st_progress_command_target = InvalidOid;
3153         pgstat_increment_changecount_after(beentry);
3154 }
3155
3156 /* ----------
3157  * pgstat_report_appname() -
3158  *
3159  *      Called to update our application name.
3160  * ----------
3161  */
3162 void
3163 pgstat_report_appname(const char *appname)
3164 {
3165         volatile PgBackendStatus *beentry = MyBEEntry;
3166         int                     len;
3167
3168         if (!beentry)
3169                 return;
3170
3171         /* This should be unnecessary if GUC did its job, but be safe */
3172         len = pg_mbcliplen(appname, strlen(appname), NAMEDATALEN - 1);
3173
3174         /*
3175          * Update my status entry, following the protocol of bumping
3176          * st_changecount before and after.  We use a volatile pointer here to
3177          * ensure the compiler doesn't try to get cute.
3178          */
3179         pgstat_increment_changecount_before(beentry);
3180
3181         memcpy((char *) beentry->st_appname, appname, len);
3182         beentry->st_appname[len] = '\0';
3183
3184         pgstat_increment_changecount_after(beentry);
3185 }
3186
3187 /*
3188  * Report current transaction start timestamp as the specified value.
3189  * Zero means there is no active transaction.
3190  */
3191 void
3192 pgstat_report_xact_timestamp(TimestampTz tstamp)
3193 {
3194         volatile PgBackendStatus *beentry = MyBEEntry;
3195
3196         if (!pgstat_track_activities || !beentry)
3197                 return;
3198
3199         /*
3200          * Update my status entry, following the protocol of bumping
3201          * st_changecount before and after.  We use a volatile pointer here to
3202          * ensure the compiler doesn't try to get cute.
3203          */
3204         pgstat_increment_changecount_before(beentry);
3205         beentry->st_xact_start_timestamp = tstamp;
3206         pgstat_increment_changecount_after(beentry);
3207 }
3208
3209 /* ----------
3210  * pgstat_read_current_status() -
3211  *
3212  *      Copy the current contents of the PgBackendStatus array to local memory,
3213  *      if not already done in this transaction.
3214  * ----------
3215  */
3216 static void
3217 pgstat_read_current_status(void)
3218 {
3219         volatile PgBackendStatus *beentry;
3220         LocalPgBackendStatus *localtable;
3221         LocalPgBackendStatus *localentry;
3222         char       *localappname,
3223                            *localactivity;
3224 #ifdef USE_SSL
3225         PgBackendSSLStatus *localsslstatus;
3226 #endif
3227         int                     i;
3228
3229         Assert(!pgStatRunningInCollector);
3230         if (localBackendStatusTable)
3231                 return;                                 /* already done */
3232
3233         pgstat_setup_memcxt();
3234
3235         localtable = (LocalPgBackendStatus *)
3236                 MemoryContextAlloc(pgStatLocalContext,
3237                                                    sizeof(LocalPgBackendStatus) * NumBackendStatSlots);
3238         localappname = (char *)
3239                 MemoryContextAlloc(pgStatLocalContext,
3240                                                    NAMEDATALEN * NumBackendStatSlots);
3241         localactivity = (char *)
3242                 MemoryContextAlloc(pgStatLocalContext,
3243                                                    pgstat_track_activity_query_size * NumBackendStatSlots);
3244 #ifdef USE_SSL
3245         localsslstatus = (PgBackendSSLStatus *)
3246                 MemoryContextAlloc(pgStatLocalContext,
3247                                                    sizeof(PgBackendSSLStatus) * NumBackendStatSlots);
3248 #endif
3249
3250         localNumBackends = 0;
3251
3252         beentry = BackendStatusArray;
3253         localentry = localtable;
3254         for (i = 1; i <= NumBackendStatSlots; i++)
3255         {
3256                 /*
3257                  * Follow the protocol of retrying if st_changecount changes while we
3258                  * copy the entry, or if it's odd.  (The check for odd is needed to
3259                  * cover the case where we are able to completely copy the entry while
3260                  * the source backend is between increment steps.)      We use a volatile
3261                  * pointer here to ensure the compiler doesn't try to get cute.
3262                  */
3263                 for (;;)
3264                 {
3265                         int                     before_changecount;
3266                         int                     after_changecount;
3267
3268                         pgstat_save_changecount_before(beentry, before_changecount);
3269
3270                         localentry->backendStatus.st_procpid = beentry->st_procpid;
3271                         if (localentry->backendStatus.st_procpid > 0)
3272                         {
3273                                 memcpy(&localentry->backendStatus, (char *) beentry, sizeof(PgBackendStatus));
3274
3275                                 /*
3276                                  * strcpy is safe even if the string is modified concurrently,
3277                                  * because there's always a \0 at the end of the buffer.
3278                                  */
3279                                 strcpy(localappname, (char *) beentry->st_appname);
3280                                 localentry->backendStatus.st_appname = localappname;
3281                                 strcpy(localactivity, (char *) beentry->st_activity);
3282                                 localentry->backendStatus.st_activity = localactivity;
3283                                 localentry->backendStatus.st_ssl = beentry->st_ssl;
3284 #ifdef USE_SSL
3285                                 if (beentry->st_ssl)
3286                                 {
3287                                         memcpy(localsslstatus, beentry->st_sslstatus, sizeof(PgBackendSSLStatus));
3288                                         localentry->backendStatus.st_sslstatus = localsslstatus;
3289                                 }
3290 #endif
3291                         }
3292
3293                         pgstat_save_changecount_after(beentry, after_changecount);
3294                         if (before_changecount == after_changecount &&
3295                                 (before_changecount & 1) == 0)
3296                                 break;
3297
3298                         /* Make sure we can break out of loop if stuck... */
3299                         CHECK_FOR_INTERRUPTS();
3300                 }
3301
3302                 beentry++;
3303                 /* Only valid entries get included into the local array */
3304                 if (localentry->backendStatus.st_procpid > 0)
3305                 {
3306                         BackendIdGetTransactionIds(i,
3307                                                                            &localentry->backend_xid,
3308                                                                            &localentry->backend_xmin);
3309
3310                         localentry++;
3311                         localappname += NAMEDATALEN;
3312                         localactivity += pgstat_track_activity_query_size;
3313 #ifdef USE_SSL
3314                         localsslstatus++;
3315 #endif
3316                         localNumBackends++;
3317                 }
3318         }
3319
3320         /* Set the pointer only after completion of a valid table */
3321         localBackendStatusTable = localtable;
3322 }
3323
3324 /* ----------
3325  * pgstat_get_wait_event_type() -
3326  *
3327  *      Return a string representing the current wait event type, backend is
3328  *      waiting on.
3329  */
3330 const char *
3331 pgstat_get_wait_event_type(uint32 wait_event_info)
3332 {
3333         uint32          classId;
3334         const char *event_type;
3335
3336         /* report process as not waiting. */
3337         if (wait_event_info == 0)
3338                 return NULL;
3339
3340         classId = wait_event_info & 0xFF000000;
3341
3342         switch (classId)
3343         {
3344                 case PG_WAIT_LWLOCK:
3345                         event_type = "LWLock";
3346                         break;
3347                 case PG_WAIT_LOCK:
3348                         event_type = "Lock";
3349                         break;
3350                 case PG_WAIT_BUFFER_PIN:
3351                         event_type = "BufferPin";
3352                         break;
3353                 case PG_WAIT_ACTIVITY:
3354                         event_type = "Activity";
3355                         break;
3356                 case PG_WAIT_CLIENT:
3357                         event_type = "Client";
3358                         break;
3359                 case PG_WAIT_EXTENSION:
3360                         event_type = "Extension";
3361                         break;
3362                 case PG_WAIT_IPC:
3363                         event_type = "IPC";
3364                         break;
3365                 case PG_WAIT_TIMEOUT:
3366                         event_type = "Timeout";
3367                         break;
3368                 case PG_WAIT_IO:
3369                         event_type = "IO";
3370                         break;
3371                 default:
3372                         event_type = "???";
3373                         break;
3374         }
3375
3376         return event_type;
3377 }
3378
3379 /* ----------
3380  * pgstat_get_wait_event() -
3381  *
3382  *      Return a string representing the current wait event, backend is
3383  *      waiting on.
3384  */
3385 const char *
3386 pgstat_get_wait_event(uint32 wait_event_info)
3387 {
3388         uint32          classId;
3389         uint16          eventId;
3390         const char *event_name;
3391
3392         /* report process as not waiting. */
3393         if (wait_event_info == 0)
3394                 return NULL;
3395
3396         classId = wait_event_info & 0xFF000000;
3397         eventId = wait_event_info & 0x0000FFFF;
3398
3399         switch (classId)
3400         {
3401                 case PG_WAIT_LWLOCK:
3402                         event_name = GetLWLockIdentifier(classId, eventId);
3403                         break;
3404                 case PG_WAIT_LOCK:
3405                         event_name = GetLockNameFromTagType(eventId);
3406                         break;
3407                 case PG_WAIT_BUFFER_PIN:
3408                         event_name = "BufferPin";
3409                         break;
3410                 case PG_WAIT_ACTIVITY:
3411                         {
3412                                 WaitEventActivity w = (WaitEventActivity) wait_event_info;
3413
3414                                 event_name = pgstat_get_wait_activity(w);
3415                                 break;
3416                         }
3417                 case PG_WAIT_CLIENT:
3418                         {
3419                                 WaitEventClient w = (WaitEventClient) wait_event_info;
3420
3421                                 event_name = pgstat_get_wait_client(w);
3422                                 break;
3423                         }
3424                 case PG_WAIT_EXTENSION:
3425                         event_name = "Extension";
3426                         break;
3427                 case PG_WAIT_IPC:
3428                         {
3429                                 WaitEventIPC w = (WaitEventIPC) wait_event_info;
3430
3431                                 event_name = pgstat_get_wait_ipc(w);
3432                                 break;
3433                         }
3434                 case PG_WAIT_TIMEOUT:
3435                         {
3436                                 WaitEventTimeout w = (WaitEventTimeout) wait_event_info;
3437
3438                                 event_name = pgstat_get_wait_timeout(w);
3439                                 break;
3440                         }
3441                 case PG_WAIT_IO:
3442                         {
3443                                 WaitEventIO w = (WaitEventIO) wait_event_info;
3444
3445                                 event_name = pgstat_get_wait_io(w);
3446                                 break;
3447                         }
3448                 default:
3449                         event_name = "unknown wait event";
3450                         break;
3451         }
3452
3453         return event_name;
3454 }
3455
3456 /* ----------
3457  * pgstat_get_wait_activity() -
3458  *
3459  * Convert WaitEventActivity to string.
3460  * ----------
3461  */
3462 static const char *
3463 pgstat_get_wait_activity(WaitEventActivity w)
3464 {
3465         const char *event_name = "unknown wait event";
3466
3467         switch (w)
3468         {
3469                 case WAIT_EVENT_ARCHIVER_MAIN:
3470                         event_name = "ArchiverMain";
3471                         break;
3472                 case WAIT_EVENT_AUTOVACUUM_MAIN:
3473                         event_name = "AutoVacuumMain";
3474                         break;
3475                 case WAIT_EVENT_BGWRITER_HIBERNATE:
3476                         event_name = "BgWriterHibernate";
3477                         break;
3478                 case WAIT_EVENT_BGWRITER_MAIN:
3479                         event_name = "BgWriterMain";
3480                         break;
3481                 case WAIT_EVENT_CHECKPOINTER_MAIN:
3482                         event_name = "CheckpointerMain";
3483                         break;
3484                 case WAIT_EVENT_LOGICAL_LAUNCHER_MAIN:
3485                         event_name = "LogicalLauncherMain";
3486                         break;
3487                 case WAIT_EVENT_LOGICAL_APPLY_MAIN:
3488                         event_name = "LogicalApplyMain";
3489                         break;
3490                 case WAIT_EVENT_PGSTAT_MAIN:
3491                         event_name = "PgStatMain";
3492                         break;
3493                 case WAIT_EVENT_RECOVERY_WAL_ALL:
3494                         event_name = "RecoveryWalAll";
3495                         break;
3496                 case WAIT_EVENT_RECOVERY_WAL_STREAM:
3497                         event_name = "RecoveryWalStream";
3498                         break;
3499                 case WAIT_EVENT_SYSLOGGER_MAIN:
3500                         event_name = "SysLoggerMain";
3501                         break;
3502                 case WAIT_EVENT_WAL_RECEIVER_MAIN:
3503                         event_name = "WalReceiverMain";
3504                         break;
3505                 case WAIT_EVENT_WAL_SENDER_MAIN:
3506                         event_name = "WalSenderMain";
3507                         break;
3508                 case WAIT_EVENT_WAL_WRITER_MAIN:
3509                         event_name = "WalWriterMain";
3510                         break;
3511                         /* no default case, so that compiler will warn */
3512         }
3513
3514         return event_name;
3515 }
3516
3517 /* ----------
3518  * pgstat_get_wait_client() -
3519  *
3520  * Convert WaitEventClient to string.
3521  * ----------
3522  */
3523 static const char *
3524 pgstat_get_wait_client(WaitEventClient w)
3525 {
3526         const char *event_name = "unknown wait event";
3527
3528         switch (w)
3529         {
3530                 case WAIT_EVENT_CLIENT_READ:
3531                         event_name = "ClientRead";
3532                         break;
3533                 case WAIT_EVENT_CLIENT_WRITE:
3534                         event_name = "ClientWrite";
3535                         break;
3536                 case WAIT_EVENT_LIBPQWALRECEIVER_CONNECT:
3537                         event_name = "LibPQWalReceiverConnect";
3538                         break;
3539                 case WAIT_EVENT_LIBPQWALRECEIVER_RECEIVE:
3540                         event_name = "LibPQWalReceiverReceive";
3541                         break;
3542                 case WAIT_EVENT_SSL_OPEN_SERVER:
3543                         event_name = "SSLOpenServer";
3544                         break;
3545                 case WAIT_EVENT_WAL_RECEIVER_WAIT_START:
3546                         event_name = "WalReceiverWaitStart";
3547                         break;
3548                 case WAIT_EVENT_WAL_SENDER_WAIT_WAL:
3549                         event_name = "WalSenderWaitForWAL";
3550                         break;
3551                 case WAIT_EVENT_WAL_SENDER_WRITE_DATA:
3552                         event_name = "WalSenderWriteData";
3553                         break;
3554                         /* no default case, so that compiler will warn */
3555         }
3556
3557         return event_name;
3558 }
3559
3560 /* ----------
3561  * pgstat_get_wait_ipc() -
3562  *
3563  * Convert WaitEventIPC to string.
3564  * ----------
3565  */
3566 static const char *
3567 pgstat_get_wait_ipc(WaitEventIPC w)
3568 {
3569         const char *event_name = "unknown wait event";
3570
3571         switch (w)
3572         {
3573                 case WAIT_EVENT_BGWORKER_SHUTDOWN:
3574                         event_name = "BgWorkerShutdown";
3575                         break;
3576                 case WAIT_EVENT_BGWORKER_STARTUP:
3577                         event_name = "BgWorkerStartup";
3578                         break;
3579                 case WAIT_EVENT_BTREE_PAGE:
3580                         event_name = "BtreePage";
3581                         break;
3582                 case WAIT_EVENT_EXECUTE_GATHER:
3583                         event_name = "ExecuteGather";
3584                         break;
3585                 case WAIT_EVENT_LOGICAL_SYNC_DATA:
3586                         event_name = "LogicalSyncData";
3587                         break;
3588                 case WAIT_EVENT_LOGICAL_SYNC_STATE_CHANGE:
3589                         event_name = "LogicalSyncStateChange";
3590                         break;
3591                 case WAIT_EVENT_MQ_INTERNAL:
3592                         event_name = "MessageQueueInternal";
3593                         break;
3594                 case WAIT_EVENT_MQ_PUT_MESSAGE:
3595                         event_name = "MessageQueuePutMessage";
3596                         break;
3597                 case WAIT_EVENT_MQ_RECEIVE:
3598                         event_name = "MessageQueueReceive";
3599                         break;
3600                 case WAIT_EVENT_MQ_SEND:
3601                         event_name = "MessageQueueSend";
3602                         break;
3603                 case WAIT_EVENT_PARALLEL_FINISH:
3604                         event_name = "ParallelFinish";
3605                         break;
3606                 case WAIT_EVENT_PARALLEL_BITMAP_SCAN:
3607                         event_name = "ParallelBitmapScan";
3608                         break;
3609                 case WAIT_EVENT_PROCARRAY_GROUP_UPDATE:
3610                         event_name = "ProcArrayGroupUpdate";
3611                         break;
3612                 case WAIT_EVENT_CLOG_GROUP_UPDATE:
3613                         event_name = "ClogGroupUpdate";
3614                         break;
3615                 case WAIT_EVENT_REPLICATION_ORIGIN_DROP:
3616                         event_name = "ReplicationOriginDrop";
3617                         break;
3618                 case WAIT_EVENT_REPLICATION_SLOT_DROP:
3619                         event_name = "ReplicationSlotDrop";
3620                         break;
3621                 case WAIT_EVENT_SAFE_SNAPSHOT:
3622                         event_name = "SafeSnapshot";
3623                         break;
3624                 case WAIT_EVENT_SYNC_REP:
3625                         event_name = "SyncRep";
3626                         break;
3627                         /* no default case, so that compiler will warn */
3628         }
3629
3630         return event_name;
3631 }
3632
3633 /* ----------
3634  * pgstat_get_wait_timeout() -
3635  *
3636  * Convert WaitEventTimeout to string.
3637  * ----------
3638  */
3639 static const char *
3640 pgstat_get_wait_timeout(WaitEventTimeout w)
3641 {
3642         const char *event_name = "unknown wait event";
3643
3644         switch (w)
3645         {
3646                 case WAIT_EVENT_BASE_BACKUP_THROTTLE:
3647                         event_name = "BaseBackupThrottle";
3648                         break;
3649                 case WAIT_EVENT_PG_SLEEP:
3650                         event_name = "PgSleep";
3651                         break;
3652                 case WAIT_EVENT_RECOVERY_APPLY_DELAY:
3653                         event_name = "RecoveryApplyDelay";
3654                         break;
3655                         /* no default case, so that compiler will warn */
3656         }
3657
3658         return event_name;
3659 }
3660
3661 /* ----------
3662  * pgstat_get_wait_io() -
3663  *
3664  * Convert WaitEventIO to string.
3665  * ----------
3666  */
3667 static const char *
3668 pgstat_get_wait_io(WaitEventIO w)
3669 {
3670         const char *event_name = "unknown wait event";
3671
3672         switch (w)
3673         {
3674                 case WAIT_EVENT_BUFFILE_READ:
3675                         event_name = "BufFileRead";
3676                         break;
3677                 case WAIT_EVENT_BUFFILE_WRITE:
3678                         event_name = "BufFileWrite";
3679                         break;
3680                 case WAIT_EVENT_CONTROL_FILE_READ:
3681                         event_name = "ControlFileRead";
3682                         break;
3683                 case WAIT_EVENT_CONTROL_FILE_SYNC:
3684                         event_name = "ControlFileSync";
3685                         break;
3686                 case WAIT_EVENT_CONTROL_FILE_SYNC_UPDATE:
3687                         event_name = "ControlFileSyncUpdate";
3688                         break;
3689                 case WAIT_EVENT_CONTROL_FILE_WRITE:
3690                         event_name = "ControlFileWrite";
3691                         break;
3692                 case WAIT_EVENT_CONTROL_FILE_WRITE_UPDATE:
3693                         event_name = "ControlFileWriteUpdate";
3694                         break;
3695                 case WAIT_EVENT_COPY_FILE_READ:
3696                         event_name = "CopyFileRead";
3697                         break;
3698                 case WAIT_EVENT_COPY_FILE_WRITE:
3699                         event_name = "CopyFileWrite";
3700                         break;
3701                 case WAIT_EVENT_DATA_FILE_EXTEND:
3702                         event_name = "DataFileExtend";
3703                         break;
3704                 case WAIT_EVENT_DATA_FILE_FLUSH:
3705                         event_name = "DataFileFlush";
3706                         break;
3707                 case WAIT_EVENT_DATA_FILE_IMMEDIATE_SYNC:
3708                         event_name = "DataFileImmediateSync";
3709                         break;
3710                 case WAIT_EVENT_DATA_FILE_PREFETCH:
3711                         event_name = "DataFilePrefetch";
3712                         break;
3713                 case WAIT_EVENT_DATA_FILE_READ:
3714                         event_name = "DataFileRead";
3715                         break;
3716                 case WAIT_EVENT_DATA_FILE_SYNC:
3717                         event_name = "DataFileSync";
3718                         break;
3719                 case WAIT_EVENT_DATA_FILE_TRUNCATE:
3720                         event_name = "DataFileTruncate";
3721                         break;
3722                 case WAIT_EVENT_DATA_FILE_WRITE:
3723                         event_name = "DataFileWrite";
3724                         break;
3725                 case WAIT_EVENT_DSM_FILL_ZERO_WRITE:
3726                         event_name = "DSMFillZeroWrite";
3727                         break;
3728                 case WAIT_EVENT_LOCK_FILE_ADDTODATADIR_READ:
3729                         event_name = "LockFileAddToDataDirRead";
3730                         break;
3731                 case WAIT_EVENT_LOCK_FILE_ADDTODATADIR_SYNC:
3732                         event_name = "LockFileAddToDataDirSync";
3733                         break;
3734                 case WAIT_EVENT_LOCK_FILE_ADDTODATADIR_WRITE:
3735                         event_name = "LockFileAddToDataDirWrite";
3736                         break;
3737                 case WAIT_EVENT_LOCK_FILE_CREATE_READ:
3738                         event_name = "LockFileCreateRead";
3739                         break;
3740                 case WAIT_EVENT_LOCK_FILE_CREATE_SYNC:
3741                         event_name = "LockFileCreateSync";
3742                         break;
3743                 case WAIT_EVENT_LOCK_FILE_CREATE_WRITE:
3744                         event_name = "LockFileCreateWRITE";
3745                         break;
3746                 case WAIT_EVENT_LOCK_FILE_RECHECKDATADIR_READ:
3747                         event_name = "LockFileReCheckDataDirRead";
3748                         break;
3749                 case WAIT_EVENT_LOGICAL_REWRITE_CHECKPOINT_SYNC:
3750                         event_name = "LogicalRewriteCheckpointSync";
3751                         break;
3752                 case WAIT_EVENT_LOGICAL_REWRITE_MAPPING_SYNC:
3753                         event_name = "LogicalRewriteMappingSync";
3754                         break;
3755                 case WAIT_EVENT_LOGICAL_REWRITE_MAPPING_WRITE:
3756                         event_name = "LogicalRewriteMappingWrite";
3757                         break;
3758                 case WAIT_EVENT_LOGICAL_REWRITE_SYNC:
3759                         event_name = "LogicalRewriteSync";
3760                         break;
3761                 case WAIT_EVENT_LOGICAL_REWRITE_TRUNCATE:
3762                         event_name = "LogicalRewriteTruncate";
3763                         break;
3764                 case WAIT_EVENT_LOGICAL_REWRITE_WRITE:
3765                         event_name = "LogicalRewriteWrite";
3766                         break;
3767                 case WAIT_EVENT_RELATION_MAP_READ:
3768                         event_name = "RelationMapRead";
3769                         break;
3770                 case WAIT_EVENT_RELATION_MAP_SYNC:
3771                         event_name = "RelationMapSync";
3772                         break;
3773                 case WAIT_EVENT_RELATION_MAP_WRITE:
3774                         event_name = "RelationMapWrite";
3775                         break;
3776                 case WAIT_EVENT_REORDER_BUFFER_READ:
3777                         event_name = "ReorderBufferRead";
3778                         break;
3779                 case WAIT_EVENT_REORDER_BUFFER_WRITE:
3780                         event_name = "ReorderBufferWrite";
3781                         break;
3782                 case WAIT_EVENT_REORDER_LOGICAL_MAPPING_READ:
3783                         event_name = "ReorderLogicalMappingRead";
3784                         break;
3785                 case WAIT_EVENT_REPLICATION_SLOT_READ:
3786                         event_name = "ReplicationSlotRead";
3787                         break;
3788                 case WAIT_EVENT_REPLICATION_SLOT_RESTORE_SYNC:
3789                         event_name = "ReplicationSlotRestoreSync";
3790                         break;
3791                 case WAIT_EVENT_REPLICATION_SLOT_SYNC:
3792                         event_name = "ReplicationSlotSync";
3793                         break;
3794                 case WAIT_EVENT_REPLICATION_SLOT_WRITE:
3795                         event_name = "ReplicationSlotWrite";
3796                         break;
3797                 case WAIT_EVENT_SLRU_FLUSH_SYNC:
3798                         event_name = "SLRUFlushSync";
3799                         break;
3800                 case WAIT_EVENT_SLRU_READ:
3801                         event_name = "SLRURead";
3802                         break;
3803                 case WAIT_EVENT_SLRU_SYNC:
3804                         event_name = "SLRUSync";
3805                         break;
3806                 case WAIT_EVENT_SLRU_WRITE:
3807                         event_name = "SLRUWrite";
3808                         break;
3809                 case WAIT_EVENT_SNAPBUILD_READ:
3810                         event_name = "SnapbuildRead";
3811                         break;
3812                 case WAIT_EVENT_SNAPBUILD_SYNC:
3813                         event_name = "SnapbuildSync";
3814                         break;
3815                 case WAIT_EVENT_SNAPBUILD_WRITE:
3816                         event_name = "SnapbuildWrite";
3817                         break;
3818                 case WAIT_EVENT_TIMELINE_HISTORY_FILE_SYNC:
3819                         event_name = "TimelineHistoryFileSync";
3820                         break;
3821                 case WAIT_EVENT_TIMELINE_HISTORY_FILE_WRITE:
3822                         event_name = "TimelineHistoryFileWrite";
3823                         break;
3824                 case WAIT_EVENT_TIMELINE_HISTORY_READ:
3825                         event_name = "TimelineHistoryRead";
3826                         break;
3827                 case WAIT_EVENT_TIMELINE_HISTORY_SYNC:
3828                         event_name = "TimelineHistorySync";
3829                         break;
3830                 case WAIT_EVENT_TIMELINE_HISTORY_WRITE:
3831                         event_name = "TimelineHistoryWrite";
3832                         break;
3833                 case WAIT_EVENT_TWOPHASE_FILE_READ:
3834                         event_name = "TwophaseFileRead";
3835                         break;
3836                 case WAIT_EVENT_TWOPHASE_FILE_SYNC:
3837                         event_name = "TwophaseFileSync";
3838                         break;
3839                 case WAIT_EVENT_TWOPHASE_FILE_WRITE:
3840                         event_name = "TwophaseFileWrite";
3841                         break;
3842                 case WAIT_EVENT_WALSENDER_TIMELINE_HISTORY_READ:
3843                         event_name = "WALSenderTimelineHistoryRead";
3844                         break;
3845                 case WAIT_EVENT_WAL_BOOTSTRAP_SYNC:
3846                         event_name = "WALBootstrapSync";
3847                         break;
3848                 case WAIT_EVENT_WAL_BOOTSTRAP_WRITE:
3849                         event_name = "WALBootstrapWrite";
3850                         break;
3851                 case WAIT_EVENT_WAL_COPY_READ:
3852                         event_name = "WALCopyRead";
3853                         break;
3854                 case WAIT_EVENT_WAL_COPY_SYNC:
3855                         event_name = "WALCopySync";
3856                         break;
3857                 case WAIT_EVENT_WAL_COPY_WRITE:
3858                         event_name = "WALCopyWrite";
3859                         break;
3860                 case WAIT_EVENT_WAL_INIT_SYNC:
3861                         event_name = "WALInitSync";
3862                         break;
3863                 case WAIT_EVENT_WAL_INIT_WRITE:
3864                         event_name = "WALInitWrite";
3865                         break;
3866                 case WAIT_EVENT_WAL_READ:
3867                         event_name = "WALRead";
3868                         break;
3869                 case WAIT_EVENT_WAL_SYNC_METHOD_ASSIGN:
3870                         event_name = "WALSyncMethodAssign";
3871                         break;
3872                 case WAIT_EVENT_WAL_WRITE:
3873                         event_name = "WALWrite";
3874                         break;
3875
3876                         /* no default case, so that compiler will warn */
3877         }
3878
3879         return event_name;
3880 }
3881
3882
3883 /* ----------
3884  * pgstat_get_backend_current_activity() -
3885  *
3886  *      Return a string representing the current activity of the backend with
3887  *      the specified PID.  This looks directly at the BackendStatusArray,
3888  *      and so will provide current information regardless of the age of our
3889  *      transaction's snapshot of the status array.
3890  *
3891  *      It is the caller's responsibility to invoke this only for backends whose
3892  *      state is expected to remain stable while the result is in use.  The
3893  *      only current use is in deadlock reporting, where we can expect that
3894  *      the target backend is blocked on a lock.  (There are corner cases
3895  *      where the target's wait could get aborted while we are looking at it,
3896  *      but the very worst consequence is to return a pointer to a string
3897  *      that's been changed, so we won't worry too much.)
3898  *
3899  *      Note: return strings for special cases match pg_stat_get_backend_activity.
3900  * ----------
3901  */
3902 const char *
3903 pgstat_get_backend_current_activity(int pid, bool checkUser)
3904 {
3905         PgBackendStatus *beentry;
3906         int                     i;
3907
3908         beentry = BackendStatusArray;
3909         for (i = 1; i <= MaxBackends; i++)
3910         {
3911                 /*
3912                  * Although we expect the target backend's entry to be stable, that
3913                  * doesn't imply that anyone else's is.  To avoid identifying the
3914                  * wrong backend, while we check for a match to the desired PID we
3915                  * must follow the protocol of retrying if st_changecount changes
3916                  * while we examine the entry, or if it's odd.  (This might be
3917                  * unnecessary, since fetching or storing an int is almost certainly
3918                  * atomic, but let's play it safe.)  We use a volatile pointer here to
3919                  * ensure the compiler doesn't try to get cute.
3920                  */
3921                 volatile PgBackendStatus *vbeentry = beentry;
3922                 bool            found;
3923
3924                 for (;;)
3925                 {
3926                         int                     before_changecount;
3927                         int                     after_changecount;
3928
3929                         pgstat_save_changecount_before(vbeentry, before_changecount);
3930
3931                         found = (vbeentry->st_procpid == pid);
3932
3933                         pgstat_save_changecount_after(vbeentry, after_changecount);
3934
3935                         if (before_changecount == after_changecount &&
3936                                 (before_changecount & 1) == 0)
3937                                 break;
3938
3939                         /* Make sure we can break out of loop if stuck... */
3940                         CHECK_FOR_INTERRUPTS();
3941                 }
3942
3943                 if (found)
3944                 {
3945                         /* Now it is safe to use the non-volatile pointer */
3946                         if (checkUser && !superuser() && beentry->st_userid != GetUserId())
3947                                 return "<insufficient privilege>";
3948                         else if (*(beentry->st_activity) == '\0')
3949                                 return "<command string not enabled>";
3950                         else
3951                                 return beentry->st_activity;
3952                 }
3953
3954                 beentry++;
3955         }
3956
3957         /* If we get here, caller is in error ... */
3958         return "<backend information not available>";
3959 }
3960
3961 /* ----------
3962  * pgstat_get_crashed_backend_activity() -
3963  *
3964  *      Return a string representing the current activity of the backend with
3965  *      the specified PID.  Like the function above, but reads shared memory with
3966  *      the expectation that it may be corrupt.  On success, copy the string
3967  *      into the "buffer" argument and return that pointer.  On failure,
3968  *      return NULL.
3969  *
3970  *      This function is only intended to be used by the postmaster to report the
3971  *      query that crashed a backend.  In particular, no attempt is made to
3972  *      follow the correct concurrency protocol when accessing the
3973  *      BackendStatusArray.  But that's OK, in the worst case we'll return a
3974  *      corrupted message.  We also must take care not to trip on ereport(ERROR).
3975  * ----------
3976  */
3977 const char *
3978 pgstat_get_crashed_backend_activity(int pid, char *buffer, int buflen)
3979 {
3980         volatile PgBackendStatus *beentry;
3981         int                     i;
3982
3983         beentry = BackendStatusArray;
3984
3985         /*
3986          * We probably shouldn't get here before shared memory has been set up,
3987          * but be safe.
3988          */
3989         if (beentry == NULL || BackendActivityBuffer == NULL)
3990                 return NULL;
3991
3992         for (i = 1; i <= MaxBackends; i++)
3993         {
3994                 if (beentry->st_procpid == pid)
3995                 {
3996                         /* Read pointer just once, so it can't change after validation */
3997                         const char *activity = beentry->st_activity;
3998                         const char *activity_last;
3999
4000                         /*
4001                          * We mustn't access activity string before we verify that it
4002                          * falls within the BackendActivityBuffer. To make sure that the
4003                          * entire string including its ending is contained within the
4004                          * buffer, subtract one activity length from the buffer size.
4005                          */
4006                         activity_last = BackendActivityBuffer + BackendActivityBufferSize
4007                                 - pgstat_track_activity_query_size;
4008
4009                         if (activity < BackendActivityBuffer ||
4010                                 activity > activity_last)
4011                                 return NULL;
4012
4013                         /* If no string available, no point in a report */
4014                         if (activity[0] == '\0')
4015                                 return NULL;
4016
4017                         /*
4018                          * Copy only ASCII-safe characters so we don't run into encoding
4019                          * problems when reporting the message; and be sure not to run off
4020                          * the end of memory.
4021                          */
4022                         ascii_safe_strlcpy(buffer, activity,
4023                                                            Min(buflen, pgstat_track_activity_query_size));
4024
4025                         return buffer;
4026                 }
4027
4028                 beentry++;
4029         }
4030
4031         /* PID not found */
4032         return NULL;
4033 }
4034
4035 const char *
4036 pgstat_get_backend_desc(BackendType backendType)
4037 {
4038         const char *backendDesc = "unknown process type";
4039
4040         switch (backendType)
4041         {
4042                 case B_AUTOVAC_LAUNCHER:
4043                         backendDesc = "autovacuum launcher";
4044                         break;
4045                 case B_AUTOVAC_WORKER:
4046                         backendDesc = "autovacuum worker";
4047                         break;
4048                 case B_BACKEND:
4049                         backendDesc = "client backend";
4050                         break;
4051                 case B_BG_WORKER:
4052                         backendDesc = "background worker";
4053                         break;
4054                 case B_BG_WRITER:
4055                         backendDesc = "background writer";
4056                         break;
4057                 case B_CHECKPOINTER:
4058                         backendDesc = "checkpointer";
4059                         break;
4060                 case B_STARTUP:
4061                         backendDesc = "startup";
4062                         break;
4063                 case B_WAL_RECEIVER:
4064                         backendDesc = "walreceiver";
4065                         break;
4066                 case B_WAL_SENDER:
4067                         backendDesc = "walsender";
4068                         break;
4069                 case B_WAL_WRITER:
4070                         backendDesc = "walwriter";
4071                         break;
4072         }
4073
4074         return backendDesc;
4075 }
4076
4077 /* ------------------------------------------------------------
4078  * Local support functions follow
4079  * ------------------------------------------------------------
4080  */
4081
4082
4083 /* ----------
4084  * pgstat_setheader() -
4085  *
4086  *              Set common header fields in a statistics message
4087  * ----------
4088  */
4089 static void
4090 pgstat_setheader(PgStat_MsgHdr *hdr, StatMsgType mtype)
4091 {
4092         hdr->m_type = mtype;
4093 }
4094
4095
4096 /* ----------
4097  * pgstat_send() -
4098  *
4099  *              Send out one statistics message to the collector
4100  * ----------
4101  */
4102 static void
4103 pgstat_send(void *msg, int len)
4104 {
4105         int                     rc;
4106
4107         if (pgStatSock == PGINVALID_SOCKET)
4108                 return;
4109
4110         ((PgStat_MsgHdr *) msg)->m_size = len;
4111
4112         /* We'll retry after EINTR, but ignore all other failures */
4113         do
4114         {
4115                 rc = send(pgStatSock, msg, len, 0);
4116         } while (rc < 0 && errno == EINTR);
4117
4118 #ifdef USE_ASSERT_CHECKING
4119         /* In debug builds, log send failures ... */
4120         if (rc < 0)
4121                 elog(LOG, "could not send to statistics collector: %m");
4122 #endif
4123 }
4124
4125 /* ----------
4126  * pgstat_send_archiver() -
4127  *
4128  *      Tell the collector about the WAL file that we successfully
4129  *      archived or failed to archive.
4130  * ----------
4131  */
4132 void
4133 pgstat_send_archiver(const char *xlog, bool failed)
4134 {
4135         PgStat_MsgArchiver msg;
4136
4137         /*
4138          * Prepare and send the message
4139          */
4140         pgstat_setheader(&msg.m_hdr, PGSTAT_MTYPE_ARCHIVER);
4141         msg.m_failed = failed;
4142         StrNCpy(msg.m_xlog, xlog, sizeof(msg.m_xlog));
4143         msg.m_timestamp = GetCurrentTimestamp();
4144         pgstat_send(&msg, sizeof(msg));
4145 }
4146
4147 /* ----------
4148  * pgstat_send_bgwriter() -
4149  *
4150  *              Send bgwriter statistics to the collector
4151  * ----------
4152  */
4153 void
4154 pgstat_send_bgwriter(void)
4155 {
4156         /* We assume this initializes to zeroes */
4157         static const PgStat_MsgBgWriter all_zeroes;
4158
4159         /*
4160          * This function can be called even if nothing at all has happened. In
4161          * this case, avoid sending a completely empty message to the stats
4162          * collector.
4163          */
4164         if (memcmp(&BgWriterStats, &all_zeroes, sizeof(PgStat_MsgBgWriter)) == 0)
4165                 return;
4166
4167         /*
4168          * Prepare and send the message
4169          */
4170         pgstat_setheader(&BgWriterStats.m_hdr, PGSTAT_MTYPE_BGWRITER);
4171         pgstat_send(&BgWriterStats, sizeof(BgWriterStats));
4172
4173         /*
4174          * Clear out the statistics buffer, so it can be re-used.
4175          */
4176         MemSet(&BgWriterStats, 0, sizeof(BgWriterStats));
4177 }
4178
4179
4180 /* ----------
4181  * PgstatCollectorMain() -
4182  *
4183  *      Start up the statistics collector process.  This is the body of the
4184  *      postmaster child process.
4185  *
4186  *      The argc/argv parameters are valid only in EXEC_BACKEND case.
4187  * ----------
4188  */
4189 NON_EXEC_STATIC void
4190 PgstatCollectorMain(int argc, char *argv[])
4191 {
4192         int                     len;
4193         PgStat_Msg      msg;
4194         int                     wr;
4195
4196         /*
4197          * Ignore all signals usually bound to some action in the postmaster,
4198          * except SIGHUP and SIGQUIT.  Note we don't need a SIGUSR1 handler to
4199          * support latch operations, because we only use a local latch.
4200          */
4201         pqsignal(SIGHUP, pgstat_sighup_handler);
4202         pqsignal(SIGINT, SIG_IGN);
4203         pqsignal(SIGTERM, SIG_IGN);
4204         pqsignal(SIGQUIT, pgstat_exit);
4205         pqsignal(SIGALRM, SIG_IGN);
4206         pqsignal(SIGPIPE, SIG_IGN);
4207         pqsignal(SIGUSR1, SIG_IGN);
4208         pqsignal(SIGUSR2, SIG_IGN);
4209         pqsignal(SIGCHLD, SIG_DFL);
4210         pqsignal(SIGTTIN, SIG_DFL);
4211         pqsignal(SIGTTOU, SIG_DFL);
4212         pqsignal(SIGCONT, SIG_DFL);
4213         pqsignal(SIGWINCH, SIG_DFL);
4214         PG_SETMASK(&UnBlockSig);
4215
4216         /*
4217          * Identify myself via ps
4218          */
4219         init_ps_display("stats collector process", "", "", "");
4220
4221         /*
4222          * Read in existing stats files or initialize the stats to zero.
4223          */
4224         pgStatRunningInCollector = true;
4225         pgStatDBHash = pgstat_read_statsfiles(InvalidOid, true, true);
4226
4227         /*
4228          * Loop to process messages until we get SIGQUIT or detect ungraceful
4229          * death of our parent postmaster.
4230          *
4231          * For performance reasons, we don't want to do ResetLatch/WaitLatch after
4232          * every message; instead, do that only after a recv() fails to obtain a
4233          * message.  (This effectively means that if backends are sending us stuff
4234          * like mad, we won't notice postmaster death until things slack off a
4235          * bit; which seems fine.)      To do that, we have an inner loop that
4236          * iterates as long as recv() succeeds.  We do recognize got_SIGHUP inside
4237          * the inner loop, which means that such interrupts will get serviced but
4238          * the latch won't get cleared until next time there is a break in the
4239          * action.
4240          */
4241         for (;;)
4242         {
4243                 /* Clear any already-pending wakeups */
4244                 ResetLatch(MyLatch);
4245
4246                 /*
4247                  * Quit if we get SIGQUIT from the postmaster.
4248                  */
4249                 if (need_exit)
4250                         break;
4251
4252                 /*
4253                  * Inner loop iterates as long as we keep getting messages, or until
4254                  * need_exit becomes set.
4255                  */
4256                 while (!need_exit)
4257                 {
4258                         /*
4259                          * Reload configuration if we got SIGHUP from the postmaster.
4260                          */
4261                         if (got_SIGHUP)
4262                         {
4263                                 got_SIGHUP = false;
4264                                 ProcessConfigFile(PGC_SIGHUP);
4265                         }
4266
4267                         /*
4268                          * Write the stats file(s) if a new request has arrived that is
4269                          * not satisfied by existing file(s).
4270                          */
4271                         if (pgstat_write_statsfile_needed())
4272                                 pgstat_write_statsfiles(false, false);
4273
4274                         /*
4275                          * Try to receive and process a message.  This will not block,
4276                          * since the socket is set to non-blocking mode.
4277                          *
4278                          * XXX On Windows, we have to force pgwin32_recv to cooperate,
4279                          * despite the previous use of pg_set_noblock() on the socket.
4280                          * This is extremely broken and should be fixed someday.
4281                          */
4282 #ifdef WIN32
4283                         pgwin32_noblock = 1;
4284 #endif
4285
4286                         len = recv(pgStatSock, (char *) &msg,
4287                                            sizeof(PgStat_Msg), 0);
4288
4289 #ifdef WIN32
4290                         pgwin32_noblock = 0;
4291 #endif
4292
4293                         if (len < 0)
4294                         {
4295                                 if (errno == EAGAIN || errno == EWOULDBLOCK || errno == EINTR)
4296                                         break;          /* out of inner loop */
4297                                 ereport(ERROR,
4298                                                 (errcode_for_socket_access(),
4299                                                  errmsg("could not read statistics message: %m")));
4300                         }
4301
4302                         /*
4303                          * We ignore messages that are smaller than our common header
4304                          */
4305                         if (len < sizeof(PgStat_MsgHdr))
4306                                 continue;
4307
4308                         /*
4309                          * The received length must match the length in the header
4310                          */
4311                         if (msg.msg_hdr.m_size != len)
4312                                 continue;
4313
4314                         /*
4315                          * O.K. - we accept this message.  Process it.
4316                          */
4317                         switch (msg.msg_hdr.m_type)
4318                         {
4319                                 case PGSTAT_MTYPE_DUMMY:
4320                                         break;
4321
4322                                 case PGSTAT_MTYPE_INQUIRY:
4323                                         pgstat_recv_inquiry((PgStat_MsgInquiry *) &msg, len);
4324                                         break;
4325
4326                                 case PGSTAT_MTYPE_TABSTAT:
4327                                         pgstat_recv_tabstat((PgStat_MsgTabstat *) &msg, len);
4328                                         break;
4329
4330                                 case PGSTAT_MTYPE_TABPURGE:
4331                                         pgstat_recv_tabpurge((PgStat_MsgTabpurge *) &msg, len);
4332                                         break;
4333
4334                                 case PGSTAT_MTYPE_DROPDB:
4335                                         pgstat_recv_dropdb((PgStat_MsgDropdb *) &msg, len);
4336                                         break;
4337
4338                                 case PGSTAT_MTYPE_RESETCOUNTER:
4339                                         pgstat_recv_resetcounter((PgStat_MsgResetcounter *) &msg,
4340                                                                                          len);
4341                                         break;
4342
4343                                 case PGSTAT_MTYPE_RESETSHAREDCOUNTER:
4344                                         pgstat_recv_resetsharedcounter(
4345                                                                                                    (PgStat_MsgResetsharedcounter *) &msg,
4346                                                                                                    len);
4347                                         break;
4348
4349                                 case PGSTAT_MTYPE_RESETSINGLECOUNTER:
4350                                         pgstat_recv_resetsinglecounter(
4351                                                                                                    (PgStat_MsgResetsinglecounter *) &msg,
4352                                                                                                    len);
4353                                         break;
4354
4355                                 case PGSTAT_MTYPE_AUTOVAC_START:
4356                                         pgstat_recv_autovac((PgStat_MsgAutovacStart *) &msg, len);
4357                                         break;
4358
4359                                 case PGSTAT_MTYPE_VACUUM:
4360                                         pgstat_recv_vacuum((PgStat_MsgVacuum *) &msg, len);
4361                                         break;
4362
4363                                 case PGSTAT_MTYPE_ANALYZE:
4364                                         pgstat_recv_analyze((PgStat_MsgAnalyze *) &msg, len);
4365                                         break;
4366
4367                                 case PGSTAT_MTYPE_ARCHIVER:
4368                                         pgstat_recv_archiver((PgStat_MsgArchiver *) &msg, len);
4369                                         break;
4370
4371                                 case PGSTAT_MTYPE_BGWRITER:
4372                                         pgstat_recv_bgwriter((PgStat_MsgBgWriter *) &msg, len);
4373                                         break;
4374
4375                                 case PGSTAT_MTYPE_FUNCSTAT:
4376                                         pgstat_recv_funcstat((PgStat_MsgFuncstat *) &msg, len);
4377                                         break;
4378
4379                                 case PGSTAT_MTYPE_FUNCPURGE:
4380                                         pgstat_recv_funcpurge((PgStat_MsgFuncpurge *) &msg, len);
4381                                         break;
4382
4383                                 case PGSTAT_MTYPE_RECOVERYCONFLICT:
4384                                         pgstat_recv_recoveryconflict((PgStat_MsgRecoveryConflict *) &msg, len);
4385                                         break;
4386
4387                                 case PGSTAT_MTYPE_DEADLOCK:
4388                                         pgstat_recv_deadlock((PgStat_MsgDeadlock *) &msg, len);
4389                                         break;
4390
4391                                 case PGSTAT_MTYPE_TEMPFILE:
4392                                         pgstat_recv_tempfile((PgStat_MsgTempFile *) &msg, len);
4393                                         break;
4394
4395                                 default:
4396                                         break;
4397                         }
4398                 }                                               /* end of inner message-processing loop */
4399
4400                 /* Sleep until there's something to do */
4401 #ifndef WIN32
4402                 wr = WaitLatchOrSocket(MyLatch,
4403                                                            WL_LATCH_SET | WL_POSTMASTER_DEATH | WL_SOCKET_READABLE,
4404                                                            pgStatSock, -1L,
4405                                                            WAIT_EVENT_PGSTAT_MAIN);
4406 #else
4407
4408                 /*
4409                  * Windows, at least in its Windows Server 2003 R2 incarnation,
4410                  * sometimes loses FD_READ events.  Waking up and retrying the recv()
4411                  * fixes that, so don't sleep indefinitely.  This is a crock of the
4412                  * first water, but until somebody wants to debug exactly what's
4413                  * happening there, this is the best we can do.  The two-second
4414                  * timeout matches our pre-9.2 behavior, and needs to be short enough
4415                  * to not provoke "using stale statistics" complaints from
4416                  * backend_read_statsfile.
4417                  */
4418                 wr = WaitLatchOrSocket(MyLatch,
4419                                                            WL_LATCH_SET | WL_POSTMASTER_DEATH | WL_SOCKET_READABLE | WL_TIMEOUT,
4420                                                            pgStatSock,
4421                                                            2 * 1000L /* msec */ ,
4422                                                            WAIT_EVENT_PGSTAT_MAIN);
4423 #endif
4424
4425                 /*
4426                  * Emergency bailout if postmaster has died.  This is to avoid the
4427                  * necessity for manual cleanup of all postmaster children.
4428                  */
4429                 if (wr & WL_POSTMASTER_DEATH)
4430                         break;
4431         }                                                       /* end of outer loop */
4432
4433         /*
4434          * Save the final stats to reuse at next startup.
4435          */
4436         pgstat_write_statsfiles(true, true);
4437
4438         exit(0);
4439 }
4440
4441
4442 /* SIGQUIT signal handler for collector process */
4443 static void
4444 pgstat_exit(SIGNAL_ARGS)
4445 {
4446         int                     save_errno = errno;
4447
4448         need_exit = true;
4449         SetLatch(MyLatch);
4450
4451         errno = save_errno;
4452 }
4453
4454 /* SIGHUP handler for collector process */
4455 static void
4456 pgstat_sighup_handler(SIGNAL_ARGS)
4457 {
4458         int                     save_errno = errno;
4459
4460         got_SIGHUP = true;
4461         SetLatch(MyLatch);
4462
4463         errno = save_errno;
4464 }
4465
4466 /*
4467  * Subroutine to clear stats in a database entry
4468  *
4469  * Tables and functions hashes are initialized to empty.
4470  */
4471 static void
4472 reset_dbentry_counters(PgStat_StatDBEntry *dbentry)
4473 {
4474         HASHCTL         hash_ctl;
4475
4476         dbentry->n_xact_commit = 0;
4477         dbentry->n_xact_rollback = 0;
4478         dbentry->n_blocks_fetched = 0;
4479         dbentry->n_blocks_hit = 0;
4480         dbentry->n_tuples_returned = 0;
4481         dbentry->n_tuples_fetched = 0;
4482         dbentry->n_tuples_inserted = 0;
4483         dbentry->n_tuples_updated = 0;
4484         dbentry->n_tuples_deleted = 0;
4485         dbentry->last_autovac_time = 0;
4486         dbentry->n_conflict_tablespace = 0;
4487         dbentry->n_conflict_lock = 0;
4488         dbentry->n_conflict_snapshot = 0;
4489         dbentry->n_conflict_bufferpin = 0;
4490         dbentry->n_conflict_startup_deadlock = 0;
4491         dbentry->n_temp_files = 0;
4492         dbentry->n_temp_bytes = 0;
4493         dbentry->n_deadlocks = 0;
4494         dbentry->n_block_read_time = 0;
4495         dbentry->n_block_write_time = 0;
4496
4497         dbentry->stat_reset_timestamp = GetCurrentTimestamp();
4498         dbentry->stats_timestamp = 0;
4499
4500         memset(&hash_ctl, 0, sizeof(hash_ctl));
4501         hash_ctl.keysize = sizeof(Oid);
4502         hash_ctl.entrysize = sizeof(PgStat_StatTabEntry);
4503         dbentry->tables = hash_create("Per-database table",
4504                                                                   PGSTAT_TAB_HASH_SIZE,
4505                                                                   &hash_ctl,
4506                                                                   HASH_ELEM | HASH_BLOBS);
4507
4508         hash_ctl.keysize = sizeof(Oid);
4509         hash_ctl.entrysize = sizeof(PgStat_StatFuncEntry);
4510         dbentry->functions = hash_create("Per-database function",
4511                                                                          PGSTAT_FUNCTION_HASH_SIZE,
4512                                                                          &hash_ctl,
4513                                                                          HASH_ELEM | HASH_BLOBS);
4514 }
4515
4516 /*
4517  * Lookup the hash table entry for the specified database. If no hash
4518  * table entry exists, initialize it, if the create parameter is true.
4519  * Else, return NULL.
4520  */
4521 static PgStat_StatDBEntry *
4522 pgstat_get_db_entry(Oid databaseid, bool create)
4523 {
4524         PgStat_StatDBEntry *result;
4525         bool            found;
4526         HASHACTION      action = (create ? HASH_ENTER : HASH_FIND);
4527
4528         /* Lookup or create the hash table entry for this database */
4529         result = (PgStat_StatDBEntry *) hash_search(pgStatDBHash,
4530                                                                                                 &databaseid,
4531                                                                                                 action, &found);
4532
4533         if (!create && !found)
4534                 return NULL;
4535
4536         /*
4537          * If not found, initialize the new one.  This creates empty hash tables
4538          * for tables and functions, too.
4539          */
4540         if (!found)
4541                 reset_dbentry_counters(result);
4542
4543         return result;
4544 }
4545
4546
4547 /*
4548  * Lookup the hash table entry for the specified table. If no hash
4549  * table entry exists, initialize it, if the create parameter is true.
4550  * Else, return NULL.
4551  */
4552 static PgStat_StatTabEntry *
4553 pgstat_get_tab_entry(PgStat_StatDBEntry *dbentry, Oid tableoid, bool create)
4554 {
4555         PgStat_StatTabEntry *result;
4556         bool            found;
4557         HASHACTION      action = (create ? HASH_ENTER : HASH_FIND);
4558
4559         /* Lookup or create the hash table entry for this table */
4560         result = (PgStat_StatTabEntry *) hash_search(dbentry->tables,
4561                                                                                                  &tableoid,
4562                                                                                                  action, &found);
4563
4564         if (!create && !found)
4565                 return NULL;
4566
4567         /* If not found, initialize the new one. */
4568         if (!found)
4569         {
4570                 result->numscans = 0;
4571                 result->tuples_returned = 0;
4572                 result->tuples_fetched = 0;
4573                 result->tuples_inserted = 0;
4574                 result->tuples_updated = 0;
4575                 result->tuples_deleted = 0;
4576                 result->tuples_hot_updated = 0;
4577                 result->n_live_tuples = 0;
4578                 result->n_dead_tuples = 0;
4579                 result->changes_since_analyze = 0;
4580                 result->blocks_fetched = 0;
4581                 result->blocks_hit = 0;
4582                 result->vacuum_timestamp = 0;
4583                 result->vacuum_count = 0;
4584                 result->autovac_vacuum_timestamp = 0;
4585                 result->autovac_vacuum_count = 0;
4586                 result->analyze_timestamp = 0;
4587                 result->analyze_count = 0;
4588                 result->autovac_analyze_timestamp = 0;
4589                 result->autovac_analyze_count = 0;
4590         }
4591
4592         return result;
4593 }
4594
4595
4596 /* ----------
4597  * pgstat_write_statsfiles() -
4598  *              Write the global statistics file, as well as requested DB files.
4599  *
4600  *      'permanent' specifies writing to the permanent files not temporary ones.
4601  *      When true (happens only when the collector is shutting down), also remove
4602  *      the temporary files so that backends starting up under a new postmaster
4603  *      can't read old data before the new collector is ready.
4604  *
4605  *      When 'allDbs' is false, only the requested databases (listed in
4606  *      pending_write_requests) will be written; otherwise, all databases
4607  *      will be written.
4608  * ----------
4609  */
4610 static void
4611 pgstat_write_statsfiles(bool permanent, bool allDbs)
4612 {
4613         HASH_SEQ_STATUS hstat;
4614         PgStat_StatDBEntry *dbentry;
4615         FILE       *fpout;
4616         int32           format_id;
4617         const char *tmpfile = permanent ? PGSTAT_STAT_PERMANENT_TMPFILE : pgstat_stat_tmpname;
4618         const char *statfile = permanent ? PGSTAT_STAT_PERMANENT_FILENAME : pgstat_stat_filename;
4619         int                     rc;
4620
4621         elog(DEBUG2, "writing stats file \"%s\"", statfile);
4622
4623         /*
4624          * Open the statistics temp file to write out the current values.
4625          */
4626         fpout = AllocateFile(tmpfile, PG_BINARY_W);
4627         if (fpout == NULL)
4628         {
4629                 ereport(LOG,
4630                                 (errcode_for_file_access(),
4631                                  errmsg("could not open temporary statistics file \"%s\": %m",
4632                                                 tmpfile)));
4633                 return;
4634         }
4635
4636         /*
4637          * Set the timestamp of the stats file.
4638          */
4639         globalStats.stats_timestamp = GetCurrentTimestamp();
4640
4641         /*
4642          * Write the file header --- currently just a format ID.
4643          */
4644         format_id = PGSTAT_FILE_FORMAT_ID;
4645         rc = fwrite(&format_id, sizeof(format_id), 1, fpout);
4646         (void) rc;                                      /* we'll check for error with ferror */
4647
4648         /*
4649          * Write global stats struct
4650          */
4651         rc = fwrite(&globalStats, sizeof(globalStats), 1, fpout);
4652         (void) rc;                                      /* we'll check for error with ferror */
4653
4654         /*
4655          * Write archiver stats struct
4656          */
4657         rc = fwrite(&archiverStats, sizeof(archiverStats), 1, fpout);
4658         (void) rc;                                      /* we'll check for error with ferror */
4659
4660         /*
4661          * Walk through the database table.
4662          */
4663         hash_seq_init(&hstat, pgStatDBHash);
4664         while ((dbentry = (PgStat_StatDBEntry *) hash_seq_search(&hstat)) != NULL)
4665         {
4666                 /*
4667                  * Write out the table and function stats for this DB into the
4668                  * appropriate per-DB stat file, if required.
4669                  */
4670                 if (allDbs || pgstat_db_requested(dbentry->databaseid))
4671                 {
4672                         /* Make DB's timestamp consistent with the global stats */
4673                         dbentry->stats_timestamp = globalStats.stats_timestamp;
4674
4675                         pgstat_write_db_statsfile(dbentry, permanent);
4676                 }
4677
4678                 /*
4679                  * Write out the DB entry. We don't write the tables or functions
4680                  * pointers, since they're of no use to any other process.
4681                  */
4682                 fputc('D', fpout);
4683                 rc = fwrite(dbentry, offsetof(PgStat_StatDBEntry, tables), 1, fpout);
4684                 (void) rc;                              /* we'll check for error with ferror */
4685         }
4686
4687         /*
4688          * No more output to be done. Close the temp file and replace the old
4689          * pgstat.stat with it.  The ferror() check replaces testing for error
4690          * after each individual fputc or fwrite above.
4691          */
4692         fputc('E', fpout);
4693
4694         if (ferror(fpout))
4695         {
4696                 ereport(LOG,
4697                                 (errcode_for_file_access(),
4698                                  errmsg("could not write temporary statistics file \"%s\": %m",
4699                                                 tmpfile)));
4700                 FreeFile(fpout);
4701                 unlink(tmpfile);
4702         }
4703         else if (FreeFile(fpout) < 0)
4704         {
4705                 ereport(LOG,
4706                                 (errcode_for_file_access(),
4707                                  errmsg("could not close temporary statistics file \"%s\": %m",
4708                                                 tmpfile)));
4709                 unlink(tmpfile);
4710         }
4711         else if (rename(tmpfile, statfile) < 0)
4712         {
4713                 ereport(LOG,
4714                                 (errcode_for_file_access(),
4715                                  errmsg("could not rename temporary statistics file \"%s\" to \"%s\": %m",
4716                                                 tmpfile, statfile)));
4717                 unlink(tmpfile);
4718         }
4719
4720         if (permanent)
4721                 unlink(pgstat_stat_filename);
4722
4723         /*
4724          * Now throw away the list of requests.  Note that requests sent after we
4725          * started the write are still waiting on the network socket.
4726          */
4727         list_free(pending_write_requests);
4728         pending_write_requests = NIL;
4729 }
4730
4731 /*
4732  * return the filename for a DB stat file; filename is the output buffer,
4733  * of length len.
4734  */
4735 static void
4736 get_dbstat_filename(bool permanent, bool tempname, Oid databaseid,
4737                                         char *filename, int len)
4738 {
4739         int                     printed;
4740
4741         /* NB -- pgstat_reset_remove_files knows about the pattern this uses */
4742         printed = snprintf(filename, len, "%s/db_%u.%s",
4743                                            permanent ? PGSTAT_STAT_PERMANENT_DIRECTORY :
4744                                            pgstat_stat_directory,
4745                                            databaseid,
4746                                            tempname ? "tmp" : "stat");
4747         if (printed > len)
4748                 elog(ERROR, "overlength pgstat path");
4749 }
4750
4751 /* ----------
4752  * pgstat_write_db_statsfile() -
4753  *              Write the stat file for a single database.
4754  *
4755  *      If writing to the permanent file (happens when the collector is
4756  *      shutting down only), remove the temporary file so that backends
4757  *      starting up under a new postmaster can't read the old data before
4758  *      the new collector is ready.
4759  * ----------
4760  */
4761 static void
4762 pgstat_write_db_statsfile(PgStat_StatDBEntry *dbentry, bool permanent)
4763 {
4764         HASH_SEQ_STATUS tstat;
4765         HASH_SEQ_STATUS fstat;
4766         PgStat_StatTabEntry *tabentry;
4767         PgStat_StatFuncEntry *funcentry;
4768         FILE       *fpout;
4769         int32           format_id;
4770         Oid                     dbid = dbentry->databaseid;
4771         int                     rc;
4772         char            tmpfile[MAXPGPATH];
4773         char            statfile[MAXPGPATH];
4774
4775         get_dbstat_filename(permanent, true, dbid, tmpfile, MAXPGPATH);
4776         get_dbstat_filename(permanent, false, dbid, statfile, MAXPGPATH);
4777
4778         elog(DEBUG2, "writing stats file \"%s\"", statfile);
4779
4780         /*
4781          * Open the statistics temp file to write out the current values.
4782          */
4783         fpout = AllocateFile(tmpfile, PG_BINARY_W);
4784         if (fpout == NULL)
4785         {
4786                 ereport(LOG,
4787                                 (errcode_for_file_access(),
4788                                  errmsg("could not open temporary statistics file \"%s\": %m",
4789                                                 tmpfile)));
4790                 return;
4791         }
4792
4793         /*
4794          * Write the file header --- currently just a format ID.
4795          */
4796         format_id = PGSTAT_FILE_FORMAT_ID;
4797         rc = fwrite(&format_id, sizeof(format_id), 1, fpout);
4798         (void) rc;                                      /* we'll check for error with ferror */
4799
4800         /*
4801          * Walk through the database's access stats per table.
4802          */
4803         hash_seq_init(&tstat, dbentry->tables);
4804         while ((tabentry = (PgStat_StatTabEntry *) hash_seq_search(&tstat)) != NULL)
4805         {
4806                 fputc('T', fpout);
4807                 rc = fwrite(tabentry, sizeof(PgStat_StatTabEntry), 1, fpout);
4808                 (void) rc;                              /* we'll check for error with ferror */
4809         }
4810
4811         /*
4812          * Walk through the database's function stats table.
4813          */
4814         hash_seq_init(&fstat, dbentry->functions);
4815         while ((funcentry = (PgStat_StatFuncEntry *) hash_seq_search(&fstat)) != NULL)
4816         {
4817                 fputc('F', fpout);
4818                 rc = fwrite(funcentry, sizeof(PgStat_StatFuncEntry), 1, fpout);
4819                 (void) rc;                              /* we'll check for error with ferror */
4820         }
4821
4822         /*
4823          * No more output to be done. Close the temp file and replace the old
4824          * pgstat.stat with it.  The ferror() check replaces testing for error
4825          * after each individual fputc or fwrite above.
4826          */
4827         fputc('E', fpout);
4828
4829         if (ferror(fpout))
4830         {
4831                 ereport(LOG,
4832                                 (errcode_for_file_access(),
4833                                  errmsg("could not write temporary statistics file \"%s\": %m",
4834                                                 tmpfile)));
4835                 FreeFile(fpout);
4836                 unlink(tmpfile);
4837         }
4838         else if (FreeFile(fpout) < 0)
4839         {
4840                 ereport(LOG,
4841                                 (errcode_for_file_access(),
4842                                  errmsg("could not close temporary statistics file \"%s\": %m",
4843                                                 tmpfile)));
4844                 unlink(tmpfile);
4845         }
4846         else if (rename(tmpfile, statfile) < 0)
4847         {
4848                 ereport(LOG,
4849                                 (errcode_for_file_access(),
4850                                  errmsg("could not rename temporary statistics file \"%s\" to \"%s\": %m",
4851                                                 tmpfile, statfile)));
4852                 unlink(tmpfile);
4853         }
4854
4855         if (permanent)
4856         {
4857                 get_dbstat_filename(false, false, dbid, statfile, MAXPGPATH);
4858
4859                 elog(DEBUG2, "removing temporary stats file \"%s\"", statfile);
4860                 unlink(statfile);
4861         }
4862 }
4863
4864 /* ----------
4865  * pgstat_read_statsfiles() -
4866  *
4867  *      Reads in some existing statistics collector files and returns the
4868  *      databases hash table that is the top level of the data.
4869  *
4870  *      If 'onlydb' is not InvalidOid, it means we only want data for that DB
4871  *      plus the shared catalogs ("DB 0").  We'll still populate the DB hash
4872  *      table for all databases, but we don't bother even creating table/function
4873  *      hash tables for other databases.
4874  *
4875  *      'permanent' specifies reading from the permanent files not temporary ones.
4876  *      When true (happens only when the collector is starting up), remove the
4877  *      files after reading; the in-memory status is now authoritative, and the
4878  *      files would be out of date in case somebody else reads them.
4879  *
4880  *      If a 'deep' read is requested, table/function stats are read, otherwise
4881  *      the table/function hash tables remain empty.
4882  * ----------
4883  */
4884 static HTAB *
4885 pgstat_read_statsfiles(Oid onlydb, bool permanent, bool deep)
4886 {
4887         PgStat_StatDBEntry *dbentry;
4888         PgStat_StatDBEntry dbbuf;
4889         HASHCTL         hash_ctl;
4890         HTAB       *dbhash;
4891         FILE       *fpin;
4892         int32           format_id;
4893         bool            found;
4894         const char *statfile = permanent ? PGSTAT_STAT_PERMANENT_FILENAME : pgstat_stat_filename;
4895
4896         /*
4897          * The tables will live in pgStatLocalContext.
4898          */
4899         pgstat_setup_memcxt();
4900
4901         /*
4902          * Create the DB hashtable
4903          */
4904         memset(&hash_ctl, 0, sizeof(hash_ctl));
4905         hash_ctl.keysize = sizeof(Oid);
4906         hash_ctl.entrysize = sizeof(PgStat_StatDBEntry);
4907         hash_ctl.hcxt = pgStatLocalContext;
4908         dbhash = hash_create("Databases hash", PGSTAT_DB_HASH_SIZE, &hash_ctl,
4909                                                  HASH_ELEM | HASH_BLOBS | HASH_CONTEXT);
4910
4911         /*
4912          * Clear out global and archiver statistics so they start from zero in
4913          * case we can't load an existing statsfile.
4914          */
4915         memset(&globalStats, 0, sizeof(globalStats));
4916         memset(&archiverStats, 0, sizeof(archiverStats));
4917
4918         /*
4919          * Set the current timestamp (will be kept only in case we can't load an
4920          * existing statsfile).
4921          */
4922         globalStats.stat_reset_timestamp = GetCurrentTimestamp();
4923         archiverStats.stat_reset_timestamp = globalStats.stat_reset_timestamp;
4924
4925         /*
4926          * Try to open the stats file. If it doesn't exist, the backends simply
4927          * return zero for anything and the collector simply starts from scratch
4928          * with empty counters.
4929          *
4930          * ENOENT is a possibility if the stats collector is not running or has
4931          * not yet written the stats file the first time.  Any other failure
4932          * condition is suspicious.
4933          */
4934         if ((fpin = AllocateFile(statfile, PG_BINARY_R)) == NULL)
4935         {
4936                 if (errno != ENOENT)
4937                         ereport(pgStatRunningInCollector ? LOG : WARNING,
4938                                         (errcode_for_file_access(),
4939                                          errmsg("could not open statistics file \"%s\": %m",
4940                                                         statfile)));
4941                 return dbhash;
4942         }
4943
4944         /*
4945          * Verify it's of the expected format.
4946          */
4947         if (fread(&format_id, 1, sizeof(format_id), fpin) != sizeof(format_id) ||
4948                 format_id != PGSTAT_FILE_FORMAT_ID)
4949         {
4950                 ereport(pgStatRunningInCollector ? LOG : WARNING,
4951                                 (errmsg("corrupted statistics file \"%s\"", statfile)));
4952                 goto done;
4953         }
4954
4955         /*
4956          * Read global stats struct
4957          */
4958         if (fread(&globalStats, 1, sizeof(globalStats), fpin) != sizeof(globalStats))
4959         {
4960                 ereport(pgStatRunningInCollector ? LOG : WARNING,
4961                                 (errmsg("corrupted statistics file \"%s\"", statfile)));
4962                 memset(&globalStats, 0, sizeof(globalStats));
4963                 goto done;
4964         }
4965
4966         /*
4967          * In the collector, disregard the timestamp we read from the permanent
4968          * stats file; we should be willing to write a temp stats file immediately
4969          * upon the first request from any backend.  This only matters if the old
4970          * file's timestamp is less than PGSTAT_STAT_INTERVAL ago, but that's not
4971          * an unusual scenario.
4972          */
4973         if (pgStatRunningInCollector)
4974                 globalStats.stats_timestamp = 0;
4975
4976         /*
4977          * Read archiver stats struct
4978          */
4979         if (fread(&archiverStats, 1, sizeof(archiverStats), fpin) != sizeof(archiverStats))
4980         {
4981                 ereport(pgStatRunningInCollector ? LOG : WARNING,
4982                                 (errmsg("corrupted statistics file \"%s\"", statfile)));
4983                 memset(&archiverStats, 0, sizeof(archiverStats));
4984                 goto done;
4985         }
4986
4987         /*
4988          * We found an existing collector stats file. Read it and put all the
4989          * hashtable entries into place.
4990          */
4991         for (;;)
4992         {
4993                 switch (fgetc(fpin))
4994                 {
4995                                 /*
4996                                  * 'D'  A PgStat_StatDBEntry struct describing a database
4997                                  * follows.
4998                                  */
4999                         case 'D':
5000                                 if (fread(&dbbuf, 1, offsetof(PgStat_StatDBEntry, tables),
5001                                                   fpin) != offsetof(PgStat_StatDBEntry, tables))
5002                                 {
5003                                         ereport(pgStatRunningInCollector ? LOG : WARNING,
5004                                                         (errmsg("corrupted statistics file \"%s\"",
5005                                                                         statfile)));
5006                                         goto done;
5007                                 }
5008
5009                                 /*
5010                                  * Add to the DB hash
5011                                  */
5012                                 dbentry = (PgStat_StatDBEntry *) hash_search(dbhash,
5013                                                                                                                          (void *) &dbbuf.databaseid,
5014                                                                                                                          HASH_ENTER,
5015                                                                                                                          &found);
5016                                 if (found)
5017                                 {
5018                                         ereport(pgStatRunningInCollector ? LOG : WARNING,
5019                                                         (errmsg("corrupted statistics file \"%s\"",
5020                                                                         statfile)));
5021                                         goto done;
5022                                 }
5023
5024                                 memcpy(dbentry, &dbbuf, sizeof(PgStat_StatDBEntry));
5025                                 dbentry->tables = NULL;
5026                                 dbentry->functions = NULL;
5027
5028                                 /*
5029                                  * In the collector, disregard the timestamp we read from the
5030                                  * permanent stats file; we should be willing to write a temp
5031                                  * stats file immediately upon the first request from any
5032                                  * backend.
5033                                  */
5034                                 if (pgStatRunningInCollector)
5035                                         dbentry->stats_timestamp = 0;
5036
5037                                 /*
5038                                  * Don't create tables/functions hashtables for uninteresting
5039                                  * databases.
5040                                  */
5041                                 if (onlydb != InvalidOid)
5042                                 {
5043                                         if (dbbuf.databaseid != onlydb &&
5044                                                 dbbuf.databaseid != InvalidOid)
5045                                                 break;
5046                                 }
5047
5048                                 memset(&hash_ctl, 0, sizeof(hash_ctl));
5049                                 hash_ctl.keysize = sizeof(Oid);
5050                                 hash_ctl.entrysize = sizeof(PgStat_StatTabEntry);
5051                                 hash_ctl.hcxt = pgStatLocalContext;
5052                                 dbentry->tables = hash_create("Per-database table",
5053                                                                                           PGSTAT_TAB_HASH_SIZE,
5054                                                                                           &hash_ctl,
5055                                                                                           HASH_ELEM | HASH_BLOBS | HASH_CONTEXT);
5056
5057                                 hash_ctl.keysize = sizeof(Oid);
5058                                 hash_ctl.entrysize = sizeof(PgStat_StatFuncEntry);
5059                                 hash_ctl.hcxt = pgStatLocalContext;
5060                                 dbentry->functions = hash_create("Per-database function",
5061                                                                                                  PGSTAT_FUNCTION_HASH_SIZE,
5062                                                                                                  &hash_ctl,
5063                                                                                                  HASH_ELEM | HASH_BLOBS | HASH_CONTEXT);
5064
5065                                 /*
5066                                  * If requested, read the data from the database-specific
5067                                  * file.  Otherwise we just leave the hashtables empty.
5068                                  */
5069                                 if (deep)
5070                                         pgstat_read_db_statsfile(dbentry->databaseid,
5071                                                                                          dbentry->tables,
5072                                                                                          dbentry->functions,
5073                                                                                          permanent);
5074
5075                                 break;
5076
5077                         case 'E':
5078                                 goto done;
5079
5080                         default:
5081                                 ereport(pgStatRunningInCollector ? LOG : WARNING,
5082                                                 (errmsg("corrupted statistics file \"%s\"",
5083                                                                 statfile)));
5084                                 goto done;
5085                 }
5086         }
5087
5088 done:
5089         FreeFile(fpin);
5090
5091         /* If requested to read the permanent file, also get rid of it. */
5092         if (permanent)
5093         {
5094                 elog(DEBUG2, "removing permanent stats file \"%s\"", statfile);
5095                 unlink(statfile);
5096         }
5097
5098         return dbhash;
5099 }
5100
5101
5102 /* ----------
5103  * pgstat_read_db_statsfile() -
5104  *
5105  *      Reads in the existing statistics collector file for the given database,
5106  *      filling the passed-in tables and functions hash tables.
5107  *
5108  *      As in pgstat_read_statsfiles, if the permanent file is requested, it is
5109  *      removed after reading.
5110  *
5111  *      Note: this code has the ability to skip storing per-table or per-function
5112  *      data, if NULL is passed for the corresponding hashtable.  That's not used
5113  *      at the moment though.
5114  * ----------
5115  */
5116 static void
5117 pgstat_read_db_statsfile(Oid databaseid, HTAB *tabhash, HTAB *funchash,
5118                                                  bool permanent)
5119 {
5120         PgStat_StatTabEntry *tabentry;
5121         PgStat_StatTabEntry tabbuf;
5122         PgStat_StatFuncEntry funcbuf;
5123         PgStat_StatFuncEntry *funcentry;
5124         FILE       *fpin;
5125         int32           format_id;
5126         bool            found;
5127         char            statfile[MAXPGPATH];
5128
5129         get_dbstat_filename(permanent, false, databaseid, statfile, MAXPGPATH);
5130
5131         /*
5132          * Try to open the stats file. If it doesn't exist, the backends simply
5133          * return zero for anything and the collector simply starts from scratch
5134          * with empty counters.
5135          *
5136          * ENOENT is a possibility if the stats collector is not running or has
5137          * not yet written the stats file the first time.  Any other failure
5138          * condition is suspicious.
5139          */
5140         if ((fpin = AllocateFile(statfile, PG_BINARY_R)) == NULL)
5141         {
5142                 if (errno != ENOENT)
5143                         ereport(pgStatRunningInCollector ? LOG : WARNING,
5144                                         (errcode_for_file_access(),
5145                                          errmsg("could not open statistics file \"%s\": %m",
5146                                                         statfile)));
5147                 return;
5148         }
5149
5150         /*
5151          * Verify it's of the expected format.
5152          */
5153         if (fread(&format_id, 1, sizeof(format_id), fpin) != sizeof(format_id) ||
5154                 format_id != PGSTAT_FILE_FORMAT_ID)
5155         {
5156                 ereport(pgStatRunningInCollector ? LOG : WARNING,
5157                                 (errmsg("corrupted statistics file \"%s\"", statfile)));
5158                 goto done;
5159         }
5160
5161         /*
5162          * We found an existing collector stats file. Read it and put all the
5163          * hashtable entries into place.
5164          */
5165         for (;;)
5166         {
5167                 switch (fgetc(fpin))
5168                 {
5169                                 /*
5170                                  * 'T'  A PgStat_StatTabEntry follows.
5171                                  */
5172                         case 'T':
5173                                 if (fread(&tabbuf, 1, sizeof(PgStat_StatTabEntry),
5174                                                   fpin) != sizeof(PgStat_StatTabEntry))
5175                                 {
5176                                         ereport(pgStatRunningInCollector ? LOG : WARNING,
5177                                                         (errmsg("corrupted statistics file \"%s\"",
5178                                                                         statfile)));
5179                                         goto done;
5180                                 }
5181
5182                                 /*
5183                                  * Skip if table data not wanted.
5184                                  */
5185                                 if (tabhash == NULL)
5186                                         break;
5187
5188                                 tabentry = (PgStat_StatTabEntry *) hash_search(tabhash,
5189                                                                                                                            (void *) &tabbuf.tableid,
5190                                                                                                                            HASH_ENTER, &found);
5191
5192                                 if (found)
5193                                 {
5194                                         ereport(pgStatRunningInCollector ? LOG : WARNING,
5195                                                         (errmsg("corrupted statistics file \"%s\"",
5196                                                                         statfile)));
5197                                         goto done;
5198                                 }
5199
5200                                 memcpy(tabentry, &tabbuf, sizeof(tabbuf));
5201                                 break;
5202
5203                                 /*
5204                                  * 'F'  A PgStat_StatFuncEntry follows.
5205                                  */
5206                         case 'F':
5207                                 if (fread(&funcbuf, 1, sizeof(PgStat_StatFuncEntry),
5208                                                   fpin) != sizeof(PgStat_StatFuncEntry))
5209                                 {
5210                                         ereport(pgStatRunningInCollector ? LOG : WARNING,
5211                                                         (errmsg("corrupted statistics file \"%s\"",
5212                                                                         statfile)));
5213                                         goto done;
5214                                 }
5215
5216                                 /*
5217                                  * Skip if function data not wanted.
5218                                  */
5219                                 if (funchash == NULL)
5220                                         break;
5221
5222                                 funcentry = (PgStat_StatFuncEntry *) hash_search(funchash,
5223                                                                                                                                  (void *) &funcbuf.functionid,
5224                                                                                                                                  HASH_ENTER, &found);
5225
5226                                 if (found)
5227                                 {
5228                                         ereport(pgStatRunningInCollector ? LOG : WARNING,
5229                                                         (errmsg("corrupted statistics file \"%s\"",
5230                                                                         statfile)));
5231                                         goto done;
5232                                 }
5233
5234                                 memcpy(funcentry, &funcbuf, sizeof(funcbuf));
5235                                 break;
5236
5237                                 /*
5238                                  * 'E'  The EOF marker of a complete stats file.
5239                                  */
5240                         case 'E':
5241                                 goto done;
5242
5243                         default:
5244                                 ereport(pgStatRunningInCollector ? LOG : WARNING,
5245                                                 (errmsg("corrupted statistics file \"%s\"",
5246                                                                 statfile)));
5247                                 goto done;
5248                 }
5249         }
5250
5251 done:
5252         FreeFile(fpin);
5253
5254         if (permanent)
5255         {
5256                 elog(DEBUG2, "removing permanent stats file \"%s\"", statfile);
5257                 unlink(statfile);
5258         }
5259 }
5260
5261 /* ----------
5262  * pgstat_read_db_statsfile_timestamp() -
5263  *
5264  *      Attempt to determine the timestamp of the last db statfile write.
5265  *      Returns TRUE if successful; the timestamp is stored in *ts.
5266  *
5267  *      This needs to be careful about handling databases for which no stats file
5268  *      exists, such as databases without a stat entry or those not yet written:
5269  *
5270  *      - if there's a database entry in the global file, return the corresponding
5271  *      stats_timestamp value.
5272  *
5273  *      - if there's no db stat entry (e.g. for a new or inactive database),
5274  *      there's no stats_timestamp value, but also nothing to write so we return
5275  *      the timestamp of the global statfile.
5276  * ----------
5277  */
5278 static bool
5279 pgstat_read_db_statsfile_timestamp(Oid databaseid, bool permanent,
5280                                                                    TimestampTz *ts)
5281 {
5282         PgStat_StatDBEntry dbentry;
5283         PgStat_GlobalStats myGlobalStats;
5284         PgStat_ArchiverStats myArchiverStats;
5285         FILE       *fpin;
5286         int32           format_id;
5287         const char *statfile = permanent ? PGSTAT_STAT_PERMANENT_FILENAME : pgstat_stat_filename;
5288
5289         /*
5290          * Try to open the stats file.  As above, anything but ENOENT is worthy of
5291          * complaining about.
5292          */
5293         if ((fpin = AllocateFile(statfile, PG_BINARY_R)) == NULL)
5294         {
5295                 if (errno != ENOENT)
5296                         ereport(pgStatRunningInCollector ? LOG : WARNING,
5297                                         (errcode_for_file_access(),
5298                                          errmsg("could not open statistics file \"%s\": %m",
5299                                                         statfile)));
5300                 return false;
5301         }
5302
5303         /*
5304          * Verify it's of the expected format.
5305          */
5306         if (fread(&format_id, 1, sizeof(format_id), fpin) != sizeof(format_id) ||
5307                 format_id != PGSTAT_FILE_FORMAT_ID)
5308         {
5309                 ereport(pgStatRunningInCollector ? LOG : WARNING,
5310                                 (errmsg("corrupted statistics file \"%s\"", statfile)));
5311                 FreeFile(fpin);
5312                 return false;
5313         }
5314
5315         /*
5316          * Read global stats struct
5317          */
5318         if (fread(&myGlobalStats, 1, sizeof(myGlobalStats),
5319                           fpin) != sizeof(myGlobalStats))
5320         {
5321                 ereport(pgStatRunningInCollector ? LOG : WARNING,
5322                                 (errmsg("corrupted statistics file \"%s\"", statfile)));
5323                 FreeFile(fpin);
5324                 return false;
5325         }
5326
5327         /*
5328          * Read archiver stats struct
5329          */
5330         if (fread(&myArchiverStats, 1, sizeof(myArchiverStats),
5331                           fpin) != sizeof(myArchiverStats))
5332         {
5333                 ereport(pgStatRunningInCollector ? LOG : WARNING,
5334                                 (errmsg("corrupted statistics file \"%s\"", statfile)));
5335                 FreeFile(fpin);
5336                 return false;
5337         }
5338
5339         /* By default, we're going to return the timestamp of the global file. */
5340         *ts = myGlobalStats.stats_timestamp;
5341
5342         /*
5343          * We found an existing collector stats file.  Read it and look for a
5344          * record for the requested database.  If found, use its timestamp.
5345          */
5346         for (;;)
5347         {
5348                 switch (fgetc(fpin))
5349                 {
5350                                 /*
5351                                  * 'D'  A PgStat_StatDBEntry struct describing a database
5352                                  * follows.
5353                                  */
5354                         case 'D':
5355                                 if (fread(&dbentry, 1, offsetof(PgStat_StatDBEntry, tables),
5356                                                   fpin) != offsetof(PgStat_StatDBEntry, tables))
5357                                 {
5358                                         ereport(pgStatRunningInCollector ? LOG : WARNING,
5359                                                         (errmsg("corrupted statistics file \"%s\"",
5360                                                                         statfile)));
5361                                         goto done;
5362                                 }
5363
5364                                 /*
5365                                  * If this is the DB we're looking for, save its timestamp and
5366                                  * we're done.
5367                                  */
5368                                 if (dbentry.databaseid == databaseid)
5369                                 {
5370                                         *ts = dbentry.stats_timestamp;
5371                                         goto done;
5372                                 }
5373
5374                                 break;
5375
5376                         case 'E':
5377                                 goto done;
5378
5379                         default:
5380                                 ereport(pgStatRunningInCollector ? LOG : WARNING,
5381                                                 (errmsg("corrupted statistics file \"%s\"",
5382                                                                 statfile)));
5383                                 goto done;
5384                 }
5385         }
5386
5387 done:
5388         FreeFile(fpin);
5389         return true;
5390 }
5391
5392 /*
5393  * If not already done, read the statistics collector stats file into
5394  * some hash tables.  The results will be kept until pgstat_clear_snapshot()
5395  * is called (typically, at end of transaction).
5396  */
5397 static void
5398 backend_read_statsfile(void)
5399 {
5400         TimestampTz min_ts = 0;
5401         TimestampTz ref_ts = 0;
5402         Oid                     inquiry_db;
5403         int                     count;
5404
5405         /* already read it? */
5406         if (pgStatDBHash)
5407                 return;
5408         Assert(!pgStatRunningInCollector);
5409
5410         /*
5411          * In a normal backend, we check staleness of the data for our own DB, and
5412          * so we send MyDatabaseId in inquiry messages.  In the autovac launcher,
5413          * check staleness of the shared-catalog data, and send InvalidOid in
5414          * inquiry messages so as not to force writing unnecessary data.
5415          */
5416         if (IsAutoVacuumLauncherProcess())
5417                 inquiry_db = InvalidOid;
5418         else
5419                 inquiry_db = MyDatabaseId;
5420
5421         /*
5422          * Loop until fresh enough stats file is available or we ran out of time.
5423          * The stats inquiry message is sent repeatedly in case collector drops
5424          * it; but not every single time, as that just swamps the collector.
5425          */
5426         for (count = 0; count < PGSTAT_POLL_LOOP_COUNT; count++)
5427         {
5428                 bool            ok;
5429                 TimestampTz file_ts = 0;
5430                 TimestampTz cur_ts;
5431
5432                 CHECK_FOR_INTERRUPTS();
5433
5434                 ok = pgstat_read_db_statsfile_timestamp(inquiry_db, false, &file_ts);
5435
5436                 cur_ts = GetCurrentTimestamp();
5437                 /* Calculate min acceptable timestamp, if we didn't already */
5438                 if (count == 0 || cur_ts < ref_ts)
5439                 {
5440                         /*
5441                          * We set the minimum acceptable timestamp to PGSTAT_STAT_INTERVAL
5442                          * msec before now.  This indirectly ensures that the collector
5443                          * needn't write the file more often than PGSTAT_STAT_INTERVAL. In
5444                          * an autovacuum worker, however, we want a lower delay to avoid
5445                          * using stale data, so we use PGSTAT_RETRY_DELAY (since the
5446                          * number of workers is low, this shouldn't be a problem).
5447                          *
5448                          * We don't recompute min_ts after sleeping, except in the
5449                          * unlikely case that cur_ts went backwards.  So we might end up
5450                          * accepting a file a bit older than PGSTAT_STAT_INTERVAL.  In
5451                          * practice that shouldn't happen, though, as long as the sleep
5452                          * time is less than PGSTAT_STAT_INTERVAL; and we don't want to
5453                          * tell the collector that our cutoff time is less than what we'd
5454                          * actually accept.
5455                          */
5456                         ref_ts = cur_ts;
5457                         if (IsAutoVacuumWorkerProcess())
5458                                 min_ts = TimestampTzPlusMilliseconds(ref_ts,
5459                                                                                                          -PGSTAT_RETRY_DELAY);
5460                         else
5461                                 min_ts = TimestampTzPlusMilliseconds(ref_ts,
5462                                                                                                          -PGSTAT_STAT_INTERVAL);
5463                 }
5464
5465                 /*
5466                  * If the file timestamp is actually newer than cur_ts, we must have
5467                  * had a clock glitch (system time went backwards) or there is clock
5468                  * skew between our processor and the stats collector's processor.
5469                  * Accept the file, but send an inquiry message anyway to make
5470                  * pgstat_recv_inquiry do a sanity check on the collector's time.
5471                  */
5472                 if (ok && file_ts > cur_ts)
5473                 {
5474                         /*
5475                          * A small amount of clock skew between processors isn't terribly
5476                          * surprising, but a large difference is worth logging.  We
5477                          * arbitrarily define "large" as 1000 msec.
5478                          */
5479                         if (file_ts >= TimestampTzPlusMilliseconds(cur_ts, 1000))
5480                         {
5481                                 char       *filetime;
5482                                 char       *mytime;
5483
5484                                 /* Copy because timestamptz_to_str returns a static buffer */
5485                                 filetime = pstrdup(timestamptz_to_str(file_ts));
5486                                 mytime = pstrdup(timestamptz_to_str(cur_ts));
5487                                 elog(LOG, "stats collector's time %s is later than backend local time %s",
5488                                          filetime, mytime);
5489                                 pfree(filetime);
5490                                 pfree(mytime);
5491                         }
5492
5493                         pgstat_send_inquiry(cur_ts, min_ts, inquiry_db);
5494                         break;
5495                 }
5496
5497                 /* Normal acceptance case: file is not older than cutoff time */
5498                 if (ok && file_ts >= min_ts)
5499                         break;
5500
5501                 /* Not there or too old, so kick the collector and wait a bit */
5502                 if ((count % PGSTAT_INQ_LOOP_COUNT) == 0)
5503                         pgstat_send_inquiry(cur_ts, min_ts, inquiry_db);
5504
5505                 pg_usleep(PGSTAT_RETRY_DELAY * 1000L);
5506         }
5507
5508         if (count >= PGSTAT_POLL_LOOP_COUNT)
5509                 ereport(LOG,
5510                                 (errmsg("using stale statistics instead of current ones "
5511                                                 "because stats collector is not responding")));
5512
5513         /*
5514          * Autovacuum launcher wants stats about all databases, but a shallow read
5515          * is sufficient.  Regular backends want a deep read for just the tables
5516          * they can see (MyDatabaseId + shared catalogs).
5517          */
5518         if (IsAutoVacuumLauncherProcess())
5519                 pgStatDBHash = pgstat_read_statsfiles(InvalidOid, false, false);
5520         else
5521                 pgStatDBHash = pgstat_read_statsfiles(MyDatabaseId, false, true);
5522 }
5523
5524
5525 /* ----------
5526  * pgstat_setup_memcxt() -
5527  *
5528  *      Create pgStatLocalContext, if not already done.
5529  * ----------
5530  */
5531 static void
5532 pgstat_setup_memcxt(void)
5533 {
5534         if (!pgStatLocalContext)
5535                 pgStatLocalContext = AllocSetContextCreate(TopMemoryContext,
5536                                                                                                    "Statistics snapshot",
5537                                                                                                    ALLOCSET_SMALL_SIZES);
5538 }
5539
5540
5541 /* ----------
5542  * pgstat_clear_snapshot() -
5543  *
5544  *      Discard any data collected in the current transaction.  Any subsequent
5545  *      request will cause new snapshots to be read.
5546  *
5547  *      This is also invoked during transaction commit or abort to discard
5548  *      the no-longer-wanted snapshot.
5549  * ----------
5550  */
5551 void
5552 pgstat_clear_snapshot(void)
5553 {
5554         /* Release memory, if any was allocated */
5555         if (pgStatLocalContext)
5556                 MemoryContextDelete(pgStatLocalContext);
5557
5558         /* Reset variables */
5559         pgStatLocalContext = NULL;
5560         pgStatDBHash = NULL;
5561         localBackendStatusTable = NULL;
5562         localNumBackends = 0;
5563 }
5564
5565
5566 /* ----------
5567  * pgstat_recv_inquiry() -
5568  *
5569  *      Process stat inquiry requests.
5570  * ----------
5571  */
5572 static void
5573 pgstat_recv_inquiry(PgStat_MsgInquiry *msg, int len)
5574 {
5575         PgStat_StatDBEntry *dbentry;
5576
5577         elog(DEBUG2, "received inquiry for database %u", msg->databaseid);
5578
5579         /*
5580          * If there's already a write request for this DB, there's nothing to do.
5581          *
5582          * Note that if a request is found, we return early and skip the below
5583          * check for clock skew.  This is okay, since the only way for a DB
5584          * request to be present in the list is that we have been here since the
5585          * last write round.  It seems sufficient to check for clock skew once per
5586          * write round.
5587          */
5588         if (list_member_oid(pending_write_requests, msg->databaseid))
5589                 return;
5590
5591         /*
5592          * Check to see if we last wrote this database at a time >= the requested
5593          * cutoff time.  If so, this is a stale request that was generated before
5594          * we updated the DB file, and we don't need to do so again.
5595          *
5596          * If the requestor's local clock time is older than stats_timestamp, we
5597          * should suspect a clock glitch, ie system time going backwards; though
5598          * the more likely explanation is just delayed message receipt.  It is
5599          * worth expending a GetCurrentTimestamp call to be sure, since a large
5600          * retreat in the system clock reading could otherwise cause us to neglect
5601          * to update the stats file for a long time.
5602          */
5603         dbentry = pgstat_get_db_entry(msg->databaseid, false);
5604         if (dbentry == NULL)
5605         {
5606                 /*
5607                  * We have no data for this DB.  Enter a write request anyway so that
5608                  * the global stats will get updated.  This is needed to prevent
5609                  * backend_read_statsfile from waiting for data that we cannot supply,
5610                  * in the case of a new DB that nobody has yet reported any stats for.
5611                  * See the behavior of pgstat_read_db_statsfile_timestamp.
5612                  */
5613         }
5614         else if (msg->clock_time < dbentry->stats_timestamp)
5615         {
5616                 TimestampTz cur_ts = GetCurrentTimestamp();
5617
5618                 if (cur_ts < dbentry->stats_timestamp)
5619                 {
5620                         /*
5621                          * Sure enough, time went backwards.  Force a new stats file write
5622                          * to get back in sync; but first, log a complaint.
5623                          */
5624                         char       *writetime;
5625                         char       *mytime;
5626
5627                         /* Copy because timestamptz_to_str returns a static buffer */
5628                         writetime = pstrdup(timestamptz_to_str(dbentry->stats_timestamp));
5629                         mytime = pstrdup(timestamptz_to_str(cur_ts));
5630                         elog(LOG,
5631                                  "stats_timestamp %s is later than collector's time %s for database %u",
5632                                  writetime, mytime, dbentry->databaseid);
5633                         pfree(writetime);
5634                         pfree(mytime);
5635                 }
5636                 else
5637                 {
5638                         /*
5639                          * Nope, it's just an old request.  Assuming msg's clock_time is
5640                          * >= its cutoff_time, it must be stale, so we can ignore it.
5641                          */
5642                         return;
5643                 }
5644         }
5645         else if (msg->cutoff_time <= dbentry->stats_timestamp)
5646         {
5647                 /* Stale request, ignore it */
5648                 return;
5649         }
5650
5651         /*
5652          * We need to write this DB, so create a request.
5653          */
5654         pending_write_requests = lappend_oid(pending_write_requests,
5655                                                                                  msg->databaseid);
5656 }
5657
5658
5659 /* ----------
5660  * pgstat_recv_tabstat() -
5661  *
5662  *      Count what the backend has done.
5663  * ----------
5664  */
5665 static void
5666 pgstat_recv_tabstat(PgStat_MsgTabstat *msg, int len)
5667 {
5668         PgStat_StatDBEntry *dbentry;
5669         PgStat_StatTabEntry *tabentry;
5670         int                     i;
5671         bool            found;
5672
5673         dbentry = pgstat_get_db_entry(msg->m_databaseid, true);
5674
5675         /*
5676          * Update database-wide stats.
5677          */
5678         dbentry->n_xact_commit += (PgStat_Counter) (msg->m_xact_commit);
5679         dbentry->n_xact_rollback += (PgStat_Counter) (msg->m_xact_rollback);
5680         dbentry->n_block_read_time += msg->m_block_read_time;
5681         dbentry->n_block_write_time += msg->m_block_write_time;
5682
5683         /*
5684          * Process all table entries in the message.
5685          */
5686         for (i = 0; i < msg->m_nentries; i++)
5687         {
5688                 PgStat_TableEntry *tabmsg = &(msg->m_entry[i]);
5689
5690                 tabentry = (PgStat_StatTabEntry *) hash_search(dbentry->tables,
5691                                                                                                            (void *) &(tabmsg->t_id),
5692                                                                                                            HASH_ENTER, &found);
5693
5694                 if (!found)
5695                 {
5696                         /*
5697                          * If it's a new table entry, initialize counters to the values we
5698                          * just got.
5699                          */
5700                         tabentry->numscans = tabmsg->t_counts.t_numscans;
5701                         tabentry->tuples_returned = tabmsg->t_counts.t_tuples_returned;
5702                         tabentry->tuples_fetched = tabmsg->t_counts.t_tuples_fetched;
5703                         tabentry->tuples_inserted = tabmsg->t_counts.t_tuples_inserted;
5704                         tabentry->tuples_updated = tabmsg->t_counts.t_tuples_updated;
5705                         tabentry->tuples_deleted = tabmsg->t_counts.t_tuples_deleted;
5706                         tabentry->tuples_hot_updated = tabmsg->t_counts.t_tuples_hot_updated;
5707                         tabentry->n_live_tuples = tabmsg->t_counts.t_delta_live_tuples;
5708                         tabentry->n_dead_tuples = tabmsg->t_counts.t_delta_dead_tuples;
5709                         tabentry->changes_since_analyze = tabmsg->t_counts.t_changed_tuples;
5710                         tabentry->blocks_fetched = tabmsg->t_counts.t_blocks_fetched;
5711                         tabentry->blocks_hit = tabmsg->t_counts.t_blocks_hit;
5712
5713                         tabentry->vacuum_timestamp = 0;
5714                         tabentry->vacuum_count = 0;
5715                         tabentry->autovac_vacuum_timestamp = 0;
5716                         tabentry->autovac_vacuum_count = 0;
5717                         tabentry->analyze_timestamp = 0;
5718                         tabentry->analyze_count = 0;
5719                         tabentry->autovac_analyze_timestamp = 0;
5720                         tabentry->autovac_analyze_count = 0;
5721                 }
5722                 else
5723                 {
5724                         /*
5725                          * Otherwise add the values to the existing entry.
5726                          */
5727                         tabentry->numscans += tabmsg->t_counts.t_numscans;
5728                         tabentry->tuples_returned += tabmsg->t_counts.t_tuples_returned;
5729                         tabentry->tuples_fetched += tabmsg->t_counts.t_tuples_fetched;
5730                         tabentry->tuples_inserted += tabmsg->t_counts.t_tuples_inserted;
5731                         tabentry->tuples_updated += tabmsg->t_counts.t_tuples_updated;
5732                         tabentry->tuples_deleted += tabmsg->t_counts.t_tuples_deleted;
5733                         tabentry->tuples_hot_updated += tabmsg->t_counts.t_tuples_hot_updated;
5734                         /* If table was truncated, first reset the live/dead counters */
5735                         if (tabmsg->t_counts.t_truncated)
5736                         {
5737                                 tabentry->n_live_tuples = 0;
5738                                 tabentry->n_dead_tuples = 0;
5739                         }
5740                         tabentry->n_live_tuples += tabmsg->t_counts.t_delta_live_tuples;
5741                         tabentry->n_dead_tuples += tabmsg->t_counts.t_delta_dead_tuples;
5742                         tabentry->changes_since_analyze += tabmsg->t_counts.t_changed_tuples;
5743                         tabentry->blocks_fetched += tabmsg->t_counts.t_blocks_fetched;
5744                         tabentry->blocks_hit += tabmsg->t_counts.t_blocks_hit;
5745                 }
5746
5747                 /* Clamp n_live_tuples in case of negative delta_live_tuples */
5748                 tabentry->n_live_tuples = Max(tabentry->n_live_tuples, 0);
5749                 /* Likewise for n_dead_tuples */
5750                 tabentry->n_dead_tuples = Max(tabentry->n_dead_tuples, 0);
5751
5752                 /*
5753                  * Add per-table stats to the per-database entry, too.
5754                  */
5755                 dbentry->n_tuples_returned += tabmsg->t_counts.t_tuples_returned;
5756                 dbentry->n_tuples_fetched += tabmsg->t_counts.t_tuples_fetched;
5757                 dbentry->n_tuples_inserted += tabmsg->t_counts.t_tuples_inserted;
5758                 dbentry->n_tuples_updated += tabmsg->t_counts.t_tuples_updated;
5759                 dbentry->n_tuples_deleted += tabmsg->t_counts.t_tuples_deleted;
5760                 dbentry->n_blocks_fetched += tabmsg->t_counts.t_blocks_fetched;
5761                 dbentry->n_blocks_hit += tabmsg->t_counts.t_blocks_hit;
5762         }
5763 }
5764
5765
5766 /* ----------
5767  * pgstat_recv_tabpurge() -
5768  *
5769  *      Arrange for dead table removal.
5770  * ----------
5771  */
5772 static void
5773 pgstat_recv_tabpurge(PgStat_MsgTabpurge *msg, int len)
5774 {
5775         PgStat_StatDBEntry *dbentry;
5776         int                     i;
5777
5778         dbentry = pgstat_get_db_entry(msg->m_databaseid, false);
5779
5780         /*
5781          * No need to purge if we don't even know the database.
5782          */
5783         if (!dbentry || !dbentry->tables)
5784                 return;
5785
5786         /*
5787          * Process all table entries in the message.
5788          */
5789         for (i = 0; i < msg->m_nentries; i++)
5790         {
5791                 /* Remove from hashtable if present; we don't care if it's not. */
5792                 (void) hash_search(dbentry->tables,
5793                                                    (void *) &(msg->m_tableid[i]),
5794                                                    HASH_REMOVE, NULL);
5795         }
5796 }
5797
5798
5799 /* ----------
5800  * pgstat_recv_dropdb() -
5801  *
5802  *      Arrange for dead database removal
5803  * ----------
5804  */
5805 static void
5806 pgstat_recv_dropdb(PgStat_MsgDropdb *msg, int len)
5807 {
5808         Oid                     dbid = msg->m_databaseid;
5809         PgStat_StatDBEntry *dbentry;
5810
5811         /*
5812          * Lookup the database in the hashtable.
5813          */
5814         dbentry = pgstat_get_db_entry(dbid, false);
5815
5816         /*
5817          * If found, remove it (along with the db statfile).
5818          */
5819         if (dbentry)
5820         {
5821                 char            statfile[MAXPGPATH];
5822
5823                 get_dbstat_filename(false, false, dbid, statfile, MAXPGPATH);
5824
5825                 elog(DEBUG2, "removing stats file \"%s\"", statfile);
5826                 unlink(statfile);
5827
5828                 if (dbentry->tables != NULL)
5829                         hash_destroy(dbentry->tables);
5830                 if (dbentry->functions != NULL)
5831                         hash_destroy(dbentry->functions);
5832
5833                 if (hash_search(pgStatDBHash,
5834                                                 (void *) &dbid,
5835                                                 HASH_REMOVE, NULL) == NULL)
5836                         ereport(ERROR,
5837                                         (errmsg("database hash table corrupted during cleanup --- abort")));
5838         }
5839 }
5840
5841
5842 /* ----------
5843  * pgstat_recv_resetcounter() -
5844  *
5845  *      Reset the statistics for the specified database.
5846  * ----------
5847  */
5848 static void
5849 pgstat_recv_resetcounter(PgStat_MsgResetcounter *msg, int len)
5850 {
5851         PgStat_StatDBEntry *dbentry;
5852
5853         /*
5854          * Lookup the database in the hashtable.  Nothing to do if not there.
5855          */
5856         dbentry = pgstat_get_db_entry(msg->m_databaseid, false);
5857
5858         if (!dbentry)
5859                 return;
5860
5861         /*
5862          * We simply throw away all the database's table entries by recreating a
5863          * new hash table for them.
5864          */
5865         if (dbentry->tables != NULL)
5866                 hash_destroy(dbentry->tables);
5867         if (dbentry->functions != NULL)
5868                 hash_destroy(dbentry->functions);
5869
5870         dbentry->tables = NULL;
5871         dbentry->functions = NULL;
5872
5873         /*
5874          * Reset database-level stats, too.  This creates empty hash tables for
5875          * tables and functions.
5876          */
5877         reset_dbentry_counters(dbentry);
5878 }
5879
5880 /* ----------
5881  * pgstat_recv_resetshared() -
5882  *
5883  *      Reset some shared statistics of the cluster.
5884  * ----------
5885  */
5886 static void
5887 pgstat_recv_resetsharedcounter(PgStat_MsgResetsharedcounter *msg, int len)
5888 {
5889         if (msg->m_resettarget == RESET_BGWRITER)
5890         {
5891                 /* Reset the global background writer statistics for the cluster. */
5892                 memset(&globalStats, 0, sizeof(globalStats));
5893                 globalStats.stat_reset_timestamp = GetCurrentTimestamp();
5894         }
5895         else if (msg->m_resettarget == RESET_ARCHIVER)
5896         {
5897                 /* Reset the archiver statistics for the cluster. */
5898                 memset(&archiverStats, 0, sizeof(archiverStats));
5899                 archiverStats.stat_reset_timestamp = GetCurrentTimestamp();
5900         }
5901
5902         /*
5903          * Presumably the sender of this message validated the target, don't
5904          * complain here if it's not valid
5905          */
5906 }
5907
5908 /* ----------
5909  * pgstat_recv_resetsinglecounter() -
5910  *
5911  *      Reset a statistics for a single object
5912  * ----------
5913  */
5914 static void
5915 pgstat_recv_resetsinglecounter(PgStat_MsgResetsinglecounter *msg, int len)
5916 {
5917         PgStat_StatDBEntry *dbentry;
5918
5919         dbentry = pgstat_get_db_entry(msg->m_databaseid, false);
5920
5921         if (!dbentry)
5922                 return;
5923
5924         /* Set the reset timestamp for the whole database */
5925         dbentry->stat_reset_timestamp = GetCurrentTimestamp();
5926
5927         /* Remove object if it exists, ignore it if not */
5928         if (msg->m_resettype == RESET_TABLE)
5929                 (void) hash_search(dbentry->tables, (void *) &(msg->m_objectid),
5930                                                    HASH_REMOVE, NULL);
5931         else if (msg->m_resettype == RESET_FUNCTION)
5932                 (void) hash_search(dbentry->functions, (void *) &(msg->m_objectid),
5933                                                    HASH_REMOVE, NULL);
5934 }
5935
5936 /* ----------
5937  * pgstat_recv_autovac() -
5938  *
5939  *      Process an autovacuum signalling message.
5940  * ----------
5941  */
5942 static void
5943 pgstat_recv_autovac(PgStat_MsgAutovacStart *msg, int len)
5944 {
5945         PgStat_StatDBEntry *dbentry;
5946
5947         /*
5948          * Store the last autovacuum time in the database's hashtable entry.
5949          */
5950         dbentry = pgstat_get_db_entry(msg->m_databaseid, true);
5951
5952         dbentry->last_autovac_time = msg->m_start_time;
5953 }
5954
5955 /* ----------
5956  * pgstat_recv_vacuum() -
5957  *
5958  *      Process a VACUUM message.
5959  * ----------
5960  */
5961 static void
5962 pgstat_recv_vacuum(PgStat_MsgVacuum *msg, int len)
5963 {
5964         PgStat_StatDBEntry *dbentry;
5965         PgStat_StatTabEntry *tabentry;
5966
5967         /*
5968          * Store the data in the table's hashtable entry.
5969          */
5970         dbentry = pgstat_get_db_entry(msg->m_databaseid, true);
5971
5972         tabentry = pgstat_get_tab_entry(dbentry, msg->m_tableoid, true);
5973
5974         tabentry->n_live_tuples = msg->m_live_tuples;
5975         tabentry->n_dead_tuples = msg->m_dead_tuples;
5976
5977         if (msg->m_autovacuum)
5978         {
5979                 tabentry->autovac_vacuum_timestamp = msg->m_vacuumtime;
5980                 tabentry->autovac_vacuum_count++;
5981         }
5982         else
5983         {
5984                 tabentry->vacuum_timestamp = msg->m_vacuumtime;
5985                 tabentry->vacuum_count++;
5986         }
5987 }
5988
5989 /* ----------
5990  * pgstat_recv_analyze() -
5991  *
5992  *      Process an ANALYZE message.
5993  * ----------
5994  */
5995 static void
5996 pgstat_recv_analyze(PgStat_MsgAnalyze *msg, int len)
5997 {
5998         PgStat_StatDBEntry *dbentry;
5999         PgStat_StatTabEntry *tabentry;
6000
6001         /*
6002          * Store the data in the table's hashtable entry.
6003          */
6004         dbentry = pgstat_get_db_entry(msg->m_databaseid, true);
6005
6006         tabentry = pgstat_get_tab_entry(dbentry, msg->m_tableoid, true);
6007
6008         tabentry->n_live_tuples = msg->m_live_tuples;
6009         tabentry->n_dead_tuples = msg->m_dead_tuples;
6010
6011         /*
6012          * If commanded, reset changes_since_analyze to zero.  This forgets any
6013          * changes that were committed while the ANALYZE was in progress, but we
6014          * have no good way to estimate how many of those there were.
6015          */
6016         if (msg->m_resetcounter)
6017                 tabentry->changes_since_analyze = 0;
6018
6019         if (msg->m_autovacuum)
6020         {
6021                 tabentry->autovac_analyze_timestamp = msg->m_analyzetime;
6022                 tabentry->autovac_analyze_count++;
6023         }
6024         else
6025         {
6026                 tabentry->analyze_timestamp = msg->m_analyzetime;
6027                 tabentry->analyze_count++;
6028         }
6029 }
6030
6031
6032 /* ----------
6033  * pgstat_recv_archiver() -
6034  *
6035  *      Process a ARCHIVER message.
6036  * ----------
6037  */
6038 static void
6039 pgstat_recv_archiver(PgStat_MsgArchiver *msg, int len)
6040 {
6041         if (msg->m_failed)
6042         {
6043                 /* Failed archival attempt */
6044                 ++archiverStats.failed_count;
6045                 memcpy(archiverStats.last_failed_wal, msg->m_xlog,
6046                            sizeof(archiverStats.last_failed_wal));
6047                 archiverStats.last_failed_timestamp = msg->m_timestamp;
6048         }
6049         else
6050         {
6051                 /* Successful archival operation */
6052                 ++archiverStats.archived_count;
6053                 memcpy(archiverStats.last_archived_wal, msg->m_xlog,
6054                            sizeof(archiverStats.last_archived_wal));
6055                 archiverStats.last_archived_timestamp = msg->m_timestamp;
6056         }
6057 }
6058
6059 /* ----------
6060  * pgstat_recv_bgwriter() -
6061  *
6062  *      Process a BGWRITER message.
6063  * ----------
6064  */
6065 static void
6066 pgstat_recv_bgwriter(PgStat_MsgBgWriter *msg, int len)
6067 {
6068         globalStats.timed_checkpoints += msg->m_timed_checkpoints;
6069         globalStats.requested_checkpoints += msg->m_requested_checkpoints;
6070         globalStats.checkpoint_write_time += msg->m_checkpoint_write_time;
6071         globalStats.checkpoint_sync_time += msg->m_checkpoint_sync_time;
6072         globalStats.buf_written_checkpoints += msg->m_buf_written_checkpoints;
6073         globalStats.buf_written_clean += msg->m_buf_written_clean;
6074         globalStats.maxwritten_clean += msg->m_maxwritten_clean;
6075         globalStats.buf_written_backend += msg->m_buf_written_backend;
6076         globalStats.buf_fsync_backend += msg->m_buf_fsync_backend;
6077         globalStats.buf_alloc += msg->m_buf_alloc;
6078 }
6079
6080 /* ----------
6081  * pgstat_recv_recoveryconflict() -
6082  *
6083  *      Process a RECOVERYCONFLICT message.
6084  * ----------
6085  */
6086 static void
6087 pgstat_recv_recoveryconflict(PgStat_MsgRecoveryConflict *msg, int len)
6088 {
6089         PgStat_StatDBEntry *dbentry;
6090
6091         dbentry = pgstat_get_db_entry(msg->m_databaseid, true);
6092
6093         switch (msg->m_reason)
6094         {
6095                 case PROCSIG_RECOVERY_CONFLICT_DATABASE:
6096
6097                         /*
6098                          * Since we drop the information about the database as soon as it
6099                          * replicates, there is no point in counting these conflicts.
6100                          */
6101                         break;
6102                 case PROCSIG_RECOVERY_CONFLICT_TABLESPACE:
6103                         dbentry->n_conflict_tablespace++;
6104                         break;
6105                 case PROCSIG_RECOVERY_CONFLICT_LOCK:
6106                         dbentry->n_conflict_lock++;
6107                         break;
6108                 case PROCSIG_RECOVERY_CONFLICT_SNAPSHOT:
6109                         dbentry->n_conflict_snapshot++;
6110                         break;
6111                 case PROCSIG_RECOVERY_CONFLICT_BUFFERPIN:
6112                         dbentry->n_conflict_bufferpin++;
6113                         break;
6114                 case PROCSIG_RECOVERY_CONFLICT_STARTUP_DEADLOCK:
6115                         dbentry->n_conflict_startup_deadlock++;
6116                         break;
6117         }
6118 }
6119
6120 /* ----------
6121  * pgstat_recv_deadlock() -
6122  *
6123  *      Process a DEADLOCK message.
6124  * ----------
6125  */
6126 static void
6127 pgstat_recv_deadlock(PgStat_MsgDeadlock *msg, int len)
6128 {
6129         PgStat_StatDBEntry *dbentry;
6130
6131         dbentry = pgstat_get_db_entry(msg->m_databaseid, true);
6132
6133         dbentry->n_deadlocks++;
6134 }
6135
6136 /* ----------
6137  * pgstat_recv_tempfile() -
6138  *
6139  *      Process a TEMPFILE message.
6140  * ----------
6141  */
6142 static void
6143 pgstat_recv_tempfile(PgStat_MsgTempFile *msg, int len)
6144 {
6145         PgStat_StatDBEntry *dbentry;
6146
6147         dbentry = pgstat_get_db_entry(msg->m_databaseid, true);
6148
6149         dbentry->n_temp_bytes += msg->m_filesize;
6150         dbentry->n_temp_files += 1;
6151 }
6152
6153 /* ----------
6154  * pgstat_recv_funcstat() -
6155  *
6156  *      Count what the backend has done.
6157  * ----------
6158  */
6159 static void
6160 pgstat_recv_funcstat(PgStat_MsgFuncstat *msg, int len)
6161 {
6162         PgStat_FunctionEntry *funcmsg = &(msg->m_entry[0]);
6163         PgStat_StatDBEntry *dbentry;
6164         PgStat_StatFuncEntry *funcentry;
6165         int                     i;
6166         bool            found;
6167
6168         dbentry = pgstat_get_db_entry(msg->m_databaseid, true);
6169
6170         /*
6171          * Process all function entries in the message.
6172          */
6173         for (i = 0; i < msg->m_nentries; i++, funcmsg++)
6174         {
6175                 funcentry = (PgStat_StatFuncEntry *) hash_search(dbentry->functions,
6176                                                                                                                  (void *) &(funcmsg->f_id),
6177                                                                                                                  HASH_ENTER, &found);
6178
6179                 if (!found)
6180                 {
6181                         /*
6182                          * If it's a new function entry, initialize counters to the values
6183                          * we just got.
6184                          */
6185                         funcentry->f_numcalls = funcmsg->f_numcalls;
6186                         funcentry->f_total_time = funcmsg->f_total_time;
6187                         funcentry->f_self_time = funcmsg->f_self_time;
6188                 }
6189                 else
6190                 {
6191                         /*
6192                          * Otherwise add the values to the existing entry.
6193                          */
6194                         funcentry->f_numcalls += funcmsg->f_numcalls;
6195                         funcentry->f_total_time += funcmsg->f_total_time;
6196                         funcentry->f_self_time += funcmsg->f_self_time;
6197                 }
6198         }
6199 }
6200
6201 /* ----------
6202  * pgstat_recv_funcpurge() -
6203  *
6204  *      Arrange for dead function removal.
6205  * ----------
6206  */
6207 static void
6208 pgstat_recv_funcpurge(PgStat_MsgFuncpurge *msg, int len)
6209 {
6210         PgStat_StatDBEntry *dbentry;
6211         int                     i;
6212
6213         dbentry = pgstat_get_db_entry(msg->m_databaseid, false);
6214
6215         /*
6216          * No need to purge if we don't even know the database.
6217          */
6218         if (!dbentry || !dbentry->functions)
6219                 return;
6220
6221         /*
6222          * Process all function entries in the message.
6223          */
6224         for (i = 0; i < msg->m_nentries; i++)
6225         {
6226                 /* Remove from hashtable if present; we don't care if it's not. */
6227                 (void) hash_search(dbentry->functions,
6228                                                    (void *) &(msg->m_functionid[i]),
6229                                                    HASH_REMOVE, NULL);
6230         }
6231 }
6232
6233 /* ----------
6234  * pgstat_write_statsfile_needed() -
6235  *
6236  *      Do we need to write out any stats files?
6237  * ----------
6238  */
6239 static bool
6240 pgstat_write_statsfile_needed(void)
6241 {
6242         if (pending_write_requests != NIL)
6243                 return true;
6244
6245         /* Everything was written recently */
6246         return false;
6247 }
6248
6249 /* ----------
6250  * pgstat_db_requested() -
6251  *
6252  *      Checks whether stats for a particular DB need to be written to a file.
6253  * ----------
6254  */
6255 static bool
6256 pgstat_db_requested(Oid databaseid)
6257 {
6258         /*
6259          * If any requests are outstanding at all, we should write the stats for
6260          * shared catalogs (the "database" with OID 0).  This ensures that
6261          * backends will see up-to-date stats for shared catalogs, even though
6262          * they send inquiry messages mentioning only their own DB.
6263          */
6264         if (databaseid == InvalidOid && pending_write_requests != NIL)
6265                 return true;
6266
6267         /* Search to see if there's an open request to write this database. */
6268         if (list_member_oid(pending_write_requests, databaseid))
6269                 return true;
6270
6271         return false;
6272 }