injection_stats.c File Reference

#include "postgres.h"
#include "fmgr.h"
#include "common/hashfn.h"
#include "injection_stats.h"
#include "pgstat.h"
#include "utils/builtins.h"
#include "utils/pgstat_internal.h"

Include dependency graph for injection_stats.c:

Go to the source code of this file.

Data Structures
struct	PgStat_StatInjEntry
struct	PgStatShared_InjectionPoint

Macros
#define	PGSTAT_INJ_IDX(name)   hash_bytes_extended((const unsigned char *) name, strlen(name), 0)
#define	PGSTAT_KIND_INJECTION   129

Typedefs
typedef struct PgStat_StatInjEntry	PgStat_StatInjEntry
typedef struct PgStatShared_InjectionPoint	PgStatShared_InjectionPoint

Functions
static bool	injection_stats_flush_cb (PgStat_EntryRef *entry_ref, bool nowait)
static PgStat_StatInjEntry *	pgstat_fetch_stat_injentry (const char *name)
void	pgstat_register_inj (void)
void	pgstat_create_inj (const char *name)
void	pgstat_drop_inj (const char *name)
void	pgstat_report_inj (const char *name)
	PG_FUNCTION_INFO_V1 (injection_points_stats_numcalls)
Datum	injection_points_stats_numcalls (PG_FUNCTION_ARGS)

Variables
static const PgStat_KindInfo	injection_stats
static bool	inj_stats_loaded = false

Macro Definition Documentation

PGSTAT_INJ_IDX

#define PGSTAT_INJ_IDX

(

name

)

hash_bytes_extended((const unsigned char *) name, strlen(name), 0)

Definition at line 56 of file injection_stats.c.

PGSTAT_KIND_INJECTION

#define PGSTAT_KIND_INJECTION   129

Definition at line 61 of file injection_stats.c.

Typedef Documentation

PgStat_StatInjEntry

typedef struct PgStat_StatInjEntry PgStat_StatInjEntry

PgStatShared_InjectionPoint

typedef struct PgStatShared_InjectionPoint PgStatShared_InjectionPoint

Function Documentation

injection_points_stats_numcalls()

Datum injection_points_stats_numcalls

(

PG_FUNCTION_ARGS

)

Definition at line 188 of file injection_stats.c.

{
    char       *name = text_to_cstring(PG_GETARG_TEXT_PP(0));
    PgStat_StatInjEntry *entry = pgstat_fetch_stat_injentry(name);
 
    if (entry == NULL)
        PG_RETURN_NULL();
 
    PG_RETURN_INT64(entry->numcalls);
}

References name, PgStat_StatInjEntry::numcalls, PG_GETARG_TEXT_PP, PG_RETURN_INT64, PG_RETURN_NULL, pgstat_fetch_stat_injentry(), and text_to_cstring().

injection_stats_flush_cb()

static bool injection_stats_flush_cb ( PgStat_EntryRef *  entry_ref, bool  nowait  )

static

Definition at line 70 of file injection_stats.c.

{
    PgStat_StatInjEntry *localent;
    PgStatShared_InjectionPoint *shfuncent;
 
    localent = (PgStat_StatInjEntry *) entry_ref->pending;
    shfuncent = (PgStatShared_InjectionPoint *) entry_ref->shared_stats;
 
    if (!pgstat_lock_entry(entry_ref, nowait))
        return false;
 
    shfuncent->stats.numcalls += localent->numcalls;
    return true;
}

References PgStat_StatInjEntry::numcalls, PgStat_EntryRef::pending, pgstat_lock_entry(), PgStat_EntryRef::shared_stats, and PgStatShared_InjectionPoint::stats.

PG_FUNCTION_INFO_V1()

PG_FUNCTION_INFO_V1

(

injection_points_stats_numcalls

)

pgstat_create_inj()

void pgstat_create_inj

(

const char *

name

)

Definition at line 120 of file injection_stats.c.

{
    PgStat_EntryRef *entry_ref;
    PgStatShared_InjectionPoint *shstatent;
 
    /* leave if disabled */
    if (!inj_stats_loaded || !inj_stats_enabled)
        return;
 
    entry_ref = pgstat_get_entry_ref_locked(PGSTAT_KIND_INJECTION, InvalidOid,
                                            PGSTAT_INJ_IDX(name), false);
    shstatent = (PgStatShared_InjectionPoint *) entry_ref->shared_stats;
 
    /* initialize shared memory data */
    memset(&shstatent->stats, 0, sizeof(shstatent->stats));
    pgstat_unlock_entry(entry_ref);
}

References inj_stats_enabled, inj_stats_loaded, InvalidOid, name, pgstat_get_entry_ref_locked(), PGSTAT_INJ_IDX, PGSTAT_KIND_INJECTION, pgstat_unlock_entry(), PgStat_EntryRef::shared_stats, and PgStatShared_InjectionPoint::stats.

Referenced by injection_points_attach().

pgstat_drop_inj()

void pgstat_drop_inj

(

const char *

name

)

Definition at line 142 of file injection_stats.c.

{
    /* leave if disabled */
    if (!inj_stats_loaded || !inj_stats_enabled)
        return;
 
    if (!pgstat_drop_entry(PGSTAT_KIND_INJECTION, InvalidOid,
                           PGSTAT_INJ_IDX(name)))
        pgstat_request_entry_refs_gc();
}

References inj_stats_enabled, inj_stats_loaded, InvalidOid, name, pgstat_drop_entry(), PGSTAT_INJ_IDX, PGSTAT_KIND_INJECTION, and pgstat_request_entry_refs_gc().

Referenced by injection_points_cleanup(), and injection_points_detach().

pgstat_fetch_stat_injentry()

static PgStat_StatInjEntry* pgstat_fetch_stat_injentry ( const char *  name )

static

Definition at line 90 of file injection_stats.c.

{
    PgStat_StatInjEntry *entry = NULL;
 
    if (!inj_stats_loaded || !inj_stats_enabled)
        return NULL;
 
    /* Compile the lookup key as a hash of the point name */
    entry = (PgStat_StatInjEntry *) pgstat_fetch_entry(PGSTAT_KIND_INJECTION,
                                                       InvalidOid,
                                                       PGSTAT_INJ_IDX(name));
    return entry;
}

References inj_stats_enabled, inj_stats_loaded, InvalidOid, name, pgstat_fetch_entry(), PGSTAT_INJ_IDX, and PGSTAT_KIND_INJECTION.

Referenced by injection_points_stats_numcalls().

pgstat_register_inj()

void pgstat_register_inj

(

void

)

Definition at line 108 of file injection_stats.c.

{
    pgstat_register_kind(PGSTAT_KIND_INJECTION, &injection_stats);
 
    /* mark stats as loaded */
    inj_stats_loaded = true;
}

References inj_stats_loaded, injection_stats, PGSTAT_KIND_INJECTION, and pgstat_register_kind().

Referenced by _PG_init().

pgstat_report_inj()

void pgstat_report_inj

(

const char *

name

)

Definition at line 160 of file injection_stats.c.

{
    PgStat_EntryRef *entry_ref;
    PgStatShared_InjectionPoint *shstatent;
    PgStat_StatInjEntry *statent;
 
    /* leave if disabled */
    if (!inj_stats_loaded || !inj_stats_enabled)
        return;
 
    entry_ref = pgstat_get_entry_ref_locked(PGSTAT_KIND_INJECTION, InvalidOid,
                                            PGSTAT_INJ_IDX(name), false);
 
    shstatent = (PgStatShared_InjectionPoint *) entry_ref->shared_stats;
    statent = &shstatent->stats;
 
    /* Update the injection point statistics */
    statent->numcalls++;
 
    pgstat_unlock_entry(entry_ref);
}

References inj_stats_enabled, inj_stats_loaded, InvalidOid, name, PgStat_StatInjEntry::numcalls, pgstat_get_entry_ref_locked(), PGSTAT_INJ_IDX, PGSTAT_KIND_INJECTION, pgstat_unlock_entry(), PgStat_EntryRef::shared_stats, and PgStatShared_InjectionPoint::stats.

Referenced by injection_error(), injection_notice(), and injection_wait().

Variable Documentation

inj_stats_loaded

bool inj_stats_loaded = false

static

Definition at line 64 of file injection_stats.c.

Referenced by pgstat_create_inj(), pgstat_drop_inj(), pgstat_fetch_stat_injentry(), pgstat_register_inj(), and pgstat_report_inj().

injection_stats

const PgStat_KindInfo injection_stats

static

Initial value:

= {
    .name = "injection_points",
    .fixed_amount = false,      
 
    
    .accessed_across_databases = true,
 
    .shared_size = sizeof(PgStatShared_InjectionPoint),
    .shared_data_off = offsetof(PgStatShared_InjectionPoint, stats),
    .shared_data_len = sizeof(((PgStatShared_InjectionPoint *) 0)->stats),
    .pending_size = sizeof(PgStat_StatInjEntry),
    .flush_pending_cb = injection_stats_flush_cb,
}

Definition at line 39 of file injection_stats.c.

Referenced by pgstat_register_inj().