autoprewarm.c File Reference

#include "postgres.h"
#include <unistd.h>
#include "access/relation.h"
#include "access/xact.h"
#include "pgstat.h"
#include "postmaster/bgworker.h"
#include "postmaster/interrupt.h"
#include "storage/buf_internals.h"
#include "storage/dsm.h"
#include "storage/dsm_registry.h"
#include "storage/fd.h"
#include "storage/ipc.h"
#include "storage/latch.h"
#include "storage/lwlock.h"
#include "storage/procsignal.h"
#include "storage/read_stream.h"
#include "storage/smgr.h"
#include "tcop/tcopprot.h"
#include "utils/guc.h"
#include "utils/rel.h"
#include "utils/relfilenumbermap.h"
#include "utils/timestamp.h"
#include "utils/wait_event.h"

Include dependency graph for autoprewarm.c:

Go to the source code of this file.

Data Structures
struct	BlockInfoRecord
struct	AutoPrewarmSharedState
struct	AutoPrewarmReadStreamData

Macros
#define	AUTOPREWARM_FILE   "autoprewarm.blocks"
#define	cmp_member_elem(fld)

Typedefs
typedef struct BlockInfoRecord	BlockInfoRecord
typedef struct AutoPrewarmSharedState	AutoPrewarmSharedState
typedef struct AutoPrewarmReadStreamData	AutoPrewarmReadStreamData

Functions
PGDLLEXPORT void	autoprewarm_main (Datum main_arg)
PGDLLEXPORT void	autoprewarm_database_main (Datum main_arg)
	PG_FUNCTION_INFO_V1 (autoprewarm_start_worker)
	PG_FUNCTION_INFO_V1 (autoprewarm_dump_now)
static void	apw_load_buffers (void)
static int	apw_dump_now (bool is_bgworker, bool dump_unlogged)
static void	apw_start_leader_worker (void)
static void	apw_start_database_worker (void)
static bool	apw_init_shmem (void)
static void	apw_detach_shmem (int code, Datum arg)
static int	apw_compare_blockinfo (const void *p, const void *q)
void	_PG_init (void)
static BlockNumber	apw_read_stream_next_block (ReadStream *stream, void *callback_private_data, void *per_buffer_data)
Datum	autoprewarm_start_worker (PG_FUNCTION_ARGS)
Datum	autoprewarm_dump_now (PG_FUNCTION_ARGS)
static void	apw_init_state (void *ptr, void *arg)

Variables
static AutoPrewarmSharedState *	apw_state = NULL
static bool	autoprewarm = true
static int	autoprewarm_interval = 300

Macro Definition Documentation

AUTOPREWARM_FILE

#define AUTOPREWARM_FILE   "autoprewarm.blocks"

Definition at line 53 of file autoprewarm.c.

cmp_member_elem

#define cmp_member_elem

(

fld

)

Value:

do { \
	if (a->fld < b->fld)       \
        return -1;              \
    else if (a->fld > b->fld)   \
        return 1;               \
} while(0)

Definition at line 979 of file autoprewarm.c.

   { \
	if (a->fld < b->fld)       \
        return -1;              \
    else if (a->fld > b->fld)   \
        return 1;               \
} while(0)

Typedef Documentation

AutoPrewarmReadStreamData

typedef struct AutoPrewarmReadStreamData AutoPrewarmReadStreamData

AutoPrewarmSharedState

typedef struct AutoPrewarmSharedState AutoPrewarmSharedState

BlockInfoRecord

typedef struct BlockInfoRecord BlockInfoRecord

Function Documentation

_PG_init()

void _PG_init

(

void

)

Definition at line 127 of file autoprewarm.c.

{
    DefineCustomIntVariable("pg_prewarm.autoprewarm_interval",
                            "Sets the interval between dumps of shared buffers",
                            "If set to zero, time-based dumping is disabled.",
                            &autoprewarm_interval,
                            300,
                            0, INT_MAX / 1000,
                            PGC_SIGHUP,
                            GUC_UNIT_S,
                            NULL,
                            NULL,
                            NULL);
 
    if (!process_shared_preload_libraries_in_progress)
        return;
 
    /* can't define PGC_POSTMASTER variable after startup */
    DefineCustomBoolVariable("pg_prewarm.autoprewarm",
                             "Starts the autoprewarm worker.",
                             NULL,
                             &autoprewarm,
                             true,
                             PGC_POSTMASTER,
                             0,
                             NULL,
                             NULL,
                             NULL);
 
    MarkGUCPrefixReserved("pg_prewarm");
 
    /* Register autoprewarm worker, if enabled. */
    if (autoprewarm)
        apw_start_leader_worker();
}

References apw_start_leader_worker(), autoprewarm, autoprewarm_interval, DefineCustomBoolVariable(), DefineCustomIntVariable(), fb(), GUC_UNIT_S, MarkGUCPrefixReserved(), PGC_POSTMASTER, PGC_SIGHUP, and process_shared_preload_libraries_in_progress.

apw_compare_blockinfo()

static int apw_compare_blockinfo ( const void *  p, const void *  q  )

static

Definition at line 997 of file autoprewarm.c.

{
    const BlockInfoRecord *a = (const BlockInfoRecord *) p;
    const BlockInfoRecord *b = (const BlockInfoRecord *) q;
 
    cmp_member_elem(database);
    cmp_member_elem(tablespace);
    cmp_member_elem(filenumber);
    cmp_member_elem(forknum);
    cmp_member_elem(blocknum);
 
    return 0;
}

References a, b, cmp_member_elem, and tablespace.

Referenced by apw_load_buffers().

apw_detach_shmem()

static void apw_detach_shmem ( int  code, Datum  arg  )

static

Definition at line 893 of file autoprewarm.c.

{
    LWLockAcquire(&apw_state->lock, LW_EXCLUSIVE);
    if (apw_state->pid_using_dumpfile == MyProcPid)
        apw_state->pid_using_dumpfile = InvalidPid;
    if (apw_state->bgworker_pid == MyProcPid)
        apw_state->bgworker_pid = InvalidPid;
    LWLockRelease(&apw_state->lock);
}

References apw_state, AutoPrewarmSharedState::bgworker_pid, InvalidPid, AutoPrewarmSharedState::lock, LW_EXCLUSIVE, LWLockAcquire(), LWLockRelease(), MyProcPid, and AutoPrewarmSharedState::pid_using_dumpfile.

Referenced by autoprewarm_dump_now(), and autoprewarm_main().

apw_dump_now()

static int apw_dump_now ( bool  is_bgworker, bool  dump_unlogged  )

static

Definition at line 665 of file autoprewarm.c.

{
    int         num_blocks;
    int         i;
    int         ret;
    BlockInfoRecord *block_info_array;
    BufferDesc *bufHdr;
    FILE       *file;
    char        transient_dump_file_path[MAXPGPATH];
    pid_t       pid;
 
    LWLockAcquire(&apw_state->lock, LW_EXCLUSIVE);
    pid = apw_state->pid_using_dumpfile;
    if (apw_state->pid_using_dumpfile == InvalidPid)
        apw_state->pid_using_dumpfile = MyProcPid;
    LWLockRelease(&apw_state->lock);
 
    if (pid != InvalidPid)
    {
        if (!is_bgworker)
            ereport(ERROR,
                    (errmsg("could not perform block dump because dump file is being used by PID %d",
                            (int) apw_state->pid_using_dumpfile)));
 
        ereport(LOG,
                (errmsg("skipping block dump because it is already being performed by PID %d",
                        (int) apw_state->pid_using_dumpfile)));
        return 0;
    }
 
    /*
     * With sufficiently large shared_buffers, allocation will exceed 1GB, so
     * allow for a huge allocation to prevent outright failure.
     *
     * (In the future, it might be a good idea to redesign this to use a more
     * memory-efficient data structure.)
     */
    block_info_array = (BlockInfoRecord *)
        palloc_extended((sizeof(BlockInfoRecord) * NBuffers), MCXT_ALLOC_HUGE);
 
    for (num_blocks = 0, i = 0; i < NBuffers; i++)
    {
        uint64      buf_state;
 
        CHECK_FOR_INTERRUPTS();
 
        bufHdr = GetBufferDescriptor(i);
 
        /* Lock each buffer header before inspecting. */
        buf_state = LockBufHdr(bufHdr);
 
        /*
         * Unlogged tables will be automatically truncated after a crash or
         * unclean shutdown. In such cases we need not prewarm them. Dump them
         * only if requested by caller.
         */
        if (buf_state & BM_TAG_VALID &&
            ((buf_state & BM_PERMANENT) || dump_unlogged))
        {
            block_info_array[num_blocks].database = bufHdr->tag.dbOid;
            block_info_array[num_blocks].tablespace = bufHdr->tag.spcOid;
            block_info_array[num_blocks].filenumber =
                BufTagGetRelNumber(&bufHdr->tag);
            block_info_array[num_blocks].forknum =
                BufTagGetForkNum(&bufHdr->tag);
            block_info_array[num_blocks].blocknum = bufHdr->tag.blockNum;
            ++num_blocks;
        }
 
        UnlockBufHdr(bufHdr);
    }
 
    snprintf(transient_dump_file_path, MAXPGPATH, "%s.tmp", AUTOPREWARM_FILE);
    file = AllocateFile(transient_dump_file_path, "w");
    if (!file)
        ereport(ERROR,
                (errcode_for_file_access(),
                 errmsg("could not open file \"%s\": %m",
                        transient_dump_file_path)));
 
    ret = fprintf(file, "<<%d>>\n", num_blocks);
    if (ret < 0)
    {
        int         save_errno = errno;
 
        FreeFile(file);
        unlink(transient_dump_file_path);
        errno = save_errno;
        ereport(ERROR,
                (errcode_for_file_access(),
                 errmsg("could not write to file \"%s\": %m",
                        transient_dump_file_path)));
    }
 
    for (i = 0; i < num_blocks; i++)
    {
        CHECK_FOR_INTERRUPTS();
 
        ret = fprintf(file, "%u,%u,%u,%u,%u\n",
                      block_info_array[i].database,
                      block_info_array[i].tablespace,
                      block_info_array[i].filenumber,
                      (uint32) block_info_array[i].forknum,
                      block_info_array[i].blocknum);
        if (ret < 0)
        {
            int         save_errno = errno;
 
            FreeFile(file);
            unlink(transient_dump_file_path);
            errno = save_errno;
            ereport(ERROR,
                    (errcode_for_file_access(),
                     errmsg("could not write to file \"%s\": %m",
                            transient_dump_file_path)));
        }
    }
 
    pfree(block_info_array);
 
    /*
     * Rename transient_dump_file_path to AUTOPREWARM_FILE to make things
     * permanent.
     */
    ret = FreeFile(file);
    if (ret != 0)
    {
        int         save_errno = errno;
 
        unlink(transient_dump_file_path);
        errno = save_errno;
        ereport(ERROR,
                (errcode_for_file_access(),
                 errmsg("could not close file \"%s\": %m",
                        transient_dump_file_path)));
    }
 
    (void) durable_rename(transient_dump_file_path, AUTOPREWARM_FILE, ERROR);
    apw_state->pid_using_dumpfile = InvalidPid;
 
    ereport(DEBUG1,
            (errmsg_internal("wrote block details for %d blocks", num_blocks)));
    return num_blocks;
}

References AllocateFile(), apw_state, AUTOPREWARM_FILE, BM_PERMANENT, BM_TAG_VALID, BufTagGetForkNum(), BufTagGetRelNumber(), CHECK_FOR_INTERRUPTS, DEBUG1, durable_rename(), ereport, errcode_for_file_access(), errmsg, errmsg_internal(), ERROR, fb(), fprintf, FreeFile(), GetBufferDescriptor(), i, InvalidPid, AutoPrewarmSharedState::lock, LockBufHdr(), LOG, LW_EXCLUSIVE, LWLockAcquire(), LWLockRelease(), MAXPGPATH, MCXT_ALLOC_HUGE, MyProcPid, NBuffers, palloc_extended(), pfree(), AutoPrewarmSharedState::pid_using_dumpfile, snprintf, tablespace, and UnlockBufHdr().

Referenced by autoprewarm_dump_now(), and autoprewarm_main().

apw_init_shmem()

static bool apw_init_shmem ( void  )

static

Definition at line 877 of file autoprewarm.c.

{
    bool        found;
 
    apw_state = GetNamedDSMSegment("autoprewarm",
                                   sizeof(AutoPrewarmSharedState),
                                   apw_init_state,
                                   &found, NULL);
 
    return found;
}

References apw_init_state(), apw_state, fb(), and GetNamedDSMSegment().

Referenced by autoprewarm_database_main(), autoprewarm_dump_now(), autoprewarm_main(), and autoprewarm_start_worker().

apw_init_state()

static void apw_init_state ( void *  ptr, void *  arg  )

static

Definition at line 862 of file autoprewarm.c.

{
    AutoPrewarmSharedState *state = (AutoPrewarmSharedState *) ptr;
 
    LWLockInitialize(&state->lock, LWLockNewTrancheId("autoprewarm"));
    state->bgworker_pid = InvalidPid;
    state->pid_using_dumpfile = InvalidPid;
}

References InvalidPid, LWLockInitialize(), and LWLockNewTrancheId().

Referenced by apw_init_shmem().

apw_load_buffers()

static void apw_load_buffers ( void  )

static

Definition at line 293 of file autoprewarm.c.

{
    FILE       *file = NULL;
    int         num_elements,
                i;
    BlockInfoRecord *blkinfo;
    dsm_segment *seg;
 
    /*
     * Skip the prewarm if the dump file is in use; otherwise, prevent any
     * other process from writing it while we're using it.
     */
    LWLockAcquire(&apw_state->lock, LW_EXCLUSIVE);
    if (apw_state->pid_using_dumpfile == InvalidPid)
        apw_state->pid_using_dumpfile = MyProcPid;
    else
    {
        LWLockRelease(&apw_state->lock);
        ereport(LOG,
                (errmsg("skipping prewarm because block dump file is being written by PID %d",
                        (int) apw_state->pid_using_dumpfile)));
        return;
    }
    LWLockRelease(&apw_state->lock);
 
    /*
     * Open the block dump file.  Exit quietly if it doesn't exist, but report
     * any other error.
     */
    file = AllocateFile(AUTOPREWARM_FILE, "r");
    if (!file)
    {
        if (errno == ENOENT)
        {
            LWLockAcquire(&apw_state->lock, LW_EXCLUSIVE);
            apw_state->pid_using_dumpfile = InvalidPid;
            LWLockRelease(&apw_state->lock);
            return;             /* No file to load. */
        }
        ereport(ERROR,
                (errcode_for_file_access(),
                 errmsg("could not read file \"%s\": %m",
                        AUTOPREWARM_FILE)));
    }
 
    /* First line of the file is a record count. */
    if (fscanf(file, "<<%d>>\n", &num_elements) != 1)
        ereport(ERROR,
                (errcode_for_file_access(),
                 errmsg("could not read from file \"%s\": %m",
                        AUTOPREWARM_FILE)));
 
    /* Allocate a dynamic shared memory segment to store the record data. */
    seg = dsm_create(sizeof(BlockInfoRecord) * num_elements, 0);
    blkinfo = (BlockInfoRecord *) dsm_segment_address(seg);
 
    /* Read records, one per line. */
    for (i = 0; i < num_elements; i++)
    {
        unsigned    forknum;
 
        if (fscanf(file, "%u,%u,%u,%u,%u\n", &blkinfo[i].database,
                   &blkinfo[i].tablespace, &blkinfo[i].filenumber,
                   &forknum, &blkinfo[i].blocknum) != 5)
            ereport(ERROR,
                    (errmsg("autoprewarm block dump file is corrupted at line %d",
                            i + 1)));
        blkinfo[i].forknum = forknum;
    }
 
    FreeFile(file);
 
    /* Sort the blocks to be loaded. */
    qsort(blkinfo, num_elements, sizeof(BlockInfoRecord),
          apw_compare_blockinfo);
 
    /* Populate shared memory state. */
    apw_state->block_info_handle = dsm_segment_handle(seg);
    apw_state->prewarm_start_idx = apw_state->prewarm_stop_idx = 0;
    apw_state->prewarmed_blocks = 0;
 
    /* Don't prewarm more than we can fit. */
    if (num_elements > NBuffers)
    {
        num_elements = NBuffers;
        ereport(LOG,
                (errmsg("autoprewarm capping prewarmed blocks to %d (shared_buffers size)",
                        NBuffers)));
    }
 
    /* Get the info position of the first block of the next database. */
    while (apw_state->prewarm_start_idx < num_elements)
    {
        int         j = apw_state->prewarm_start_idx;
        Oid         current_db = blkinfo[j].database;
 
        /*
         * Advance the prewarm_stop_idx to the first BlockInfoRecord that does
         * not belong to this database.
         */
        j++;
        while (j < num_elements)
        {
            if (current_db != blkinfo[j].database)
            {
                /*
                 * Combine BlockInfoRecords for global objects with those of
                 * the database.
                 */
                if (current_db != InvalidOid)
                    break;
                current_db = blkinfo[j].database;
            }
 
            j++;
        }
 
        /*
         * If we reach this point with current_db == InvalidOid, then only
         * BlockInfoRecords belonging to global objects exist.  We can't
         * prewarm without a database connection, so just bail out.
         */
        if (current_db == InvalidOid)
            break;
 
        /* Configure stop point and database for next per-database worker. */
        apw_state->prewarm_stop_idx = j;
        apw_state->database = current_db;
        Assert(apw_state->prewarm_start_idx < apw_state->prewarm_stop_idx);
 
        /*
         * Likewise, don't launch if we've already been told to shut down.
         * (The launch would fail anyway, but we might as well skip it.)
         */
        if (ShutdownRequestPending)
            break;
 
        /*
         * Start a per-database worker to load blocks for this database; this
         * function will return once the per-database worker exits.
         */
        apw_start_database_worker();
 
        /* Prepare for next database. */
        apw_state->prewarm_start_idx = apw_state->prewarm_stop_idx;
    }
 
    /* Clean up. */
    dsm_detach(seg);
    LWLockAcquire(&apw_state->lock, LW_EXCLUSIVE);
    apw_state->block_info_handle = DSM_HANDLE_INVALID;
    apw_state->pid_using_dumpfile = InvalidPid;
    LWLockRelease(&apw_state->lock);
 
    /* Report our success, if we were able to finish. */
    if (!ShutdownRequestPending)
        ereport(LOG,
                (errmsg("autoprewarm successfully prewarmed %d of %d previously-loaded blocks",
                        apw_state->prewarmed_blocks, num_elements)));
}

References AllocateFile(), apw_compare_blockinfo(), apw_start_database_worker(), apw_state, Assert, AUTOPREWARM_FILE, AutoPrewarmSharedState::block_info_handle, AutoPrewarmSharedState::database, dsm_create(), dsm_detach(), DSM_HANDLE_INVALID, dsm_segment_address(), dsm_segment_handle(), ereport, errcode_for_file_access(), errmsg, ERROR, fb(), FreeFile(), i, InvalidOid, InvalidPid, j, AutoPrewarmSharedState::lock, LOG, LW_EXCLUSIVE, LWLockAcquire(), LWLockRelease(), MyProcPid, NBuffers, AutoPrewarmSharedState::pid_using_dumpfile, AutoPrewarmSharedState::prewarm_start_idx, AutoPrewarmSharedState::prewarm_stop_idx, AutoPrewarmSharedState::prewarmed_blocks, qsort, ShutdownRequestPending, and tablespace.

Referenced by autoprewarm_main().

apw_read_stream_next_block()

static BlockNumber apw_read_stream_next_block ( ReadStream *  stream, void *  callback_private_data, void *  per_buffer_data  )

static

Definition at line 459 of file autoprewarm.c.

{
    AutoPrewarmReadStreamData *p = callback_private_data;
 
    CHECK_FOR_INTERRUPTS();
 
    while (p->pos < apw_state->prewarm_stop_idx)
    {
        BlockInfoRecord blk = p->block_info[p->pos];
 
        if (blk.tablespace != p->tablespace)
            return InvalidBlockNumber;
 
        if (blk.filenumber != p->filenumber)
            return InvalidBlockNumber;
 
        if (blk.forknum != p->forknum)
            return InvalidBlockNumber;
 
        p->pos++;
 
        /*
         * Check whether blocknum is valid and within fork file size.
         * Fast-forward through any invalid blocks. We want p->pos to reflect
         * the location of the next relation or fork before ending the stream.
         */
        if (blk.blocknum >= p->nblocks)
            continue;
 
        return blk.blocknum;
    }
 
    return InvalidBlockNumber;
}

References apw_state, AutoPrewarmReadStreamData::block_info, CHECK_FOR_INTERRUPTS, fb(), AutoPrewarmReadStreamData::filenumber, AutoPrewarmReadStreamData::forknum, InvalidBlockNumber, AutoPrewarmReadStreamData::nblocks, AutoPrewarmReadStreamData::pos, AutoPrewarmSharedState::prewarm_stop_idx, and AutoPrewarmReadStreamData::tablespace.

Referenced by autoprewarm_database_main().

apw_start_database_worker()

static void apw_start_database_worker ( void  )

static

Definition at line 948 of file autoprewarm.c.

{
    BackgroundWorker worker = {0};
    BackgroundWorkerHandle *handle;
 
    worker.bgw_flags =
        BGWORKER_SHMEM_ACCESS | BGWORKER_BACKEND_DATABASE_CONNECTION;
    worker.bgw_start_time = BgWorkerStart_ConsistentState;
    worker.bgw_restart_time = BGW_NEVER_RESTART;
    strcpy(worker.bgw_library_name, "pg_prewarm");
    strcpy(worker.bgw_function_name, "autoprewarm_database_main");
    strcpy(worker.bgw_name, "autoprewarm worker");
    strcpy(worker.bgw_type, "autoprewarm worker");
 
    /* must set notify PID to wait for shutdown */
    worker.bgw_notify_pid = MyProcPid;
 
    if (!RegisterDynamicBackgroundWorker(&worker, &handle))
        ereport(ERROR,
                (errcode(ERRCODE_INSUFFICIENT_RESOURCES),
                 errmsg("registering dynamic bgworker autoprewarm failed"),
                 errhint("Consider increasing the configuration parameter \"%s\".", "max_worker_processes")));
 
    /*
     * Ignore return value; if it fails, postmaster has died, but we have
     * checks for that elsewhere.
     */
    WaitForBackgroundWorkerShutdown(handle);
}

References BackgroundWorker::bgw_flags, BackgroundWorker::bgw_function_name, BackgroundWorker::bgw_library_name, BackgroundWorker::bgw_name, BGW_NEVER_RESTART, BackgroundWorker::bgw_notify_pid, BackgroundWorker::bgw_restart_time, BackgroundWorker::bgw_start_time, BackgroundWorker::bgw_type, BGWORKER_BACKEND_DATABASE_CONNECTION, BGWORKER_SHMEM_ACCESS, BgWorkerStart_ConsistentState, ereport, errcode(), errhint(), errmsg, ERROR, fb(), MyProcPid, RegisterDynamicBackgroundWorker(), and WaitForBackgroundWorkerShutdown().

Referenced by apw_load_buffers().

apw_start_leader_worker()

static void apw_start_leader_worker ( void  )

static

Definition at line 907 of file autoprewarm.c.

{
    BackgroundWorker worker = {0};
    BackgroundWorkerHandle *handle;
    BgwHandleStatus status;
    pid_t       pid;
 
    worker.bgw_flags = BGWORKER_SHMEM_ACCESS;
    worker.bgw_start_time = BgWorkerStart_ConsistentState;
    strcpy(worker.bgw_library_name, "pg_prewarm");
    strcpy(worker.bgw_function_name, "autoprewarm_main");
    strcpy(worker.bgw_name, "autoprewarm leader");
    strcpy(worker.bgw_type, "autoprewarm leader");
 
    if (process_shared_preload_libraries_in_progress)
    {
        RegisterBackgroundWorker(&worker);
        return;
    }
 
    /* must set notify PID to wait for startup */
    worker.bgw_notify_pid = MyProcPid;
 
    if (!RegisterDynamicBackgroundWorker(&worker, &handle))
        ereport(ERROR,
                (errcode(ERRCODE_INSUFFICIENT_RESOURCES),
                 errmsg("could not register background process"),
                 errhint("You may need to increase \"max_worker_processes\".")));
 
    status = WaitForBackgroundWorkerStartup(handle, &pid);
    if (status != BGWH_STARTED)
        ereport(ERROR,
                (errcode(ERRCODE_INSUFFICIENT_RESOURCES),
                 errmsg("could not start background process"),
                 errhint("More details may be available in the server log.")));
}

References BackgroundWorker::bgw_flags, BackgroundWorker::bgw_function_name, BackgroundWorker::bgw_library_name, BackgroundWorker::bgw_name, BackgroundWorker::bgw_notify_pid, BackgroundWorker::bgw_start_time, BackgroundWorker::bgw_type, BGWH_STARTED, BGWORKER_SHMEM_ACCESS, BgWorkerStart_ConsistentState, ereport, errcode(), errhint(), errmsg, ERROR, fb(), MyProcPid, process_shared_preload_libraries_in_progress, RegisterBackgroundWorker(), RegisterDynamicBackgroundWorker(), and WaitForBackgroundWorkerStartup().

Referenced by _PG_init(), and autoprewarm_start_worker().

autoprewarm_database_main()

void autoprewarm_database_main

(

Datum

main_arg

)

Definition at line 501 of file autoprewarm.c.

{
    BlockInfoRecord *block_info;
    int         i;
    BlockInfoRecord blk;
    dsm_segment *seg;
 
    /* Establish signal handlers; once that's done, unblock signals. */
    pqsignal(SIGTERM, die);
    BackgroundWorkerUnblockSignals();
 
    /* Connect to correct database and get block information. */
    apw_init_shmem();
    seg = dsm_attach(apw_state->block_info_handle);
    if (seg == NULL)
        ereport(ERROR,
                (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                 errmsg("could not map dynamic shared memory segment")));
    BackgroundWorkerInitializeConnectionByOid(apw_state->database, InvalidOid, 0);
    block_info = (BlockInfoRecord *) dsm_segment_address(seg);
 
    i = apw_state->prewarm_start_idx;
    blk = block_info[i];
 
    /*
     * Loop until we run out of blocks to prewarm or until we run out of
     * buffers.
     */
    while (i < apw_state->prewarm_stop_idx)
    {
        Oid         tablespace = blk.tablespace;
        RelFileNumber filenumber = blk.filenumber;
        Oid         reloid;
        Relation    rel;
 
        /*
         * All blocks between prewarm_start_idx and prewarm_stop_idx should
         * belong either to global objects or the same database.
         */
        Assert(blk.database == apw_state->database || blk.database == 0);
 
        StartTransactionCommand();
 
        reloid = RelidByRelfilenumber(blk.tablespace, blk.filenumber);
        if (!OidIsValid(reloid) ||
            (rel = try_relation_open(reloid, AccessShareLock)) == NULL)
        {
            /* We failed to open the relation, so there is nothing to close. */
            CommitTransactionCommand();
 
            /*
             * Fast-forward to the next relation. We want to skip all of the
             * other records referencing this relation since we know we can't
             * open it. That way, we avoid repeatedly trying and failing to
             * open the same relation.
             */
            for (; i < apw_state->prewarm_stop_idx; i++)
            {
                blk = block_info[i];
                if (blk.tablespace != tablespace ||
                    blk.filenumber != filenumber)
                    break;
            }
 
            /* Time to try and open our newfound relation */
            continue;
        }
 
        /*
         * We have a relation; now let's loop until we find a valid fork of
         * the relation or we run out of buffers. Once we've read from all
         * valid forks or run out of options, we'll close the relation and
         * move on.
         */
        while (i < apw_state->prewarm_stop_idx &&
               blk.tablespace == tablespace &&
               blk.filenumber == filenumber)
        {
            ForkNumber  forknum = blk.forknum;
            BlockNumber nblocks;
            struct AutoPrewarmReadStreamData p;
            ReadStream *stream;
            Buffer      buf;
 
            /*
             * smgrexists is not safe for illegal forknum, hence check whether
             * the passed forknum is valid before using it in smgrexists.
             */
            if (blk.forknum <= InvalidForkNumber ||
                blk.forknum > MAX_FORKNUM ||
                !smgrexists(RelationGetSmgr(rel), blk.forknum))
            {
                /*
                 * Fast-forward to the next fork. We want to skip all of the
                 * other records referencing this fork since we already know
                 * it's not valid.
                 */
                for (; i < apw_state->prewarm_stop_idx; i++)
                {
                    blk = block_info[i];
                    if (blk.tablespace != tablespace ||
                        blk.filenumber != filenumber ||
                        blk.forknum != forknum)
                        break;
                }
 
                /* Time to check if this newfound fork is valid */
                continue;
            }
 
            nblocks = RelationGetNumberOfBlocksInFork(rel, blk.forknum);
 
            p = (struct AutoPrewarmReadStreamData)
            {
                .block_info = block_info,
                    .pos = i,
                    .tablespace = tablespace,
                    .filenumber = filenumber,
                    .forknum = forknum,
                    .nblocks = nblocks,
            };
 
            stream = read_stream_begin_relation(READ_STREAM_MAINTENANCE |
                                                READ_STREAM_DEFAULT |
                                                READ_STREAM_USE_BATCHING,
                                                NULL,
                                                rel,
                                                p.forknum,
                                                apw_read_stream_next_block,
                                                &p,
                                                0);
 
            /*
             * Loop until we've prewarmed all the blocks from this fork. The
             * read stream callback will check that we still have free buffers
             * before requesting each block from the read stream API.
             */
            while ((buf = read_stream_next_buffer(stream, NULL)) != InvalidBuffer)
            {
                apw_state->prewarmed_blocks++;
                ReleaseBuffer(buf);
            }
 
            read_stream_end(stream);
 
            /* Advance i past all the blocks just prewarmed. */
            i = p.pos;
            blk = block_info[i];
        }
 
        relation_close(rel, AccessShareLock);
        CommitTransactionCommand();
    }
 
    dsm_detach(seg);
}

References AccessShareLock, apw_init_shmem(), apw_read_stream_next_block(), apw_state, Assert, BackgroundWorkerInitializeConnectionByOid(), BackgroundWorkerUnblockSignals(), AutoPrewarmReadStreamData::block_info, AutoPrewarmSharedState::block_info_handle, buf, CommitTransactionCommand(), AutoPrewarmSharedState::database, die, dsm_attach(), dsm_detach(), dsm_segment_address(), ereport, errcode(), errmsg, ERROR, fb(), AutoPrewarmReadStreamData::filenumber, BlockInfoRecord::forknum, AutoPrewarmReadStreamData::forknum, i, InvalidBuffer, InvalidForkNumber, InvalidOid, MAX_FORKNUM, AutoPrewarmReadStreamData::nblocks, OidIsValid, AutoPrewarmReadStreamData::pos, pqsignal, AutoPrewarmSharedState::prewarm_start_idx, AutoPrewarmSharedState::prewarm_stop_idx, AutoPrewarmSharedState::prewarmed_blocks, read_stream_begin_relation(), READ_STREAM_DEFAULT, read_stream_end(), READ_STREAM_MAINTENANCE, read_stream_next_buffer(), READ_STREAM_USE_BATCHING, relation_close(), RelationGetNumberOfBlocksInFork(), RelationGetSmgr(), ReleaseBuffer(), RelidByRelfilenumber(), smgrexists(), StartTransactionCommand(), BlockInfoRecord::tablespace, AutoPrewarmReadStreamData::tablespace, tablespace, and try_relation_open().

autoprewarm_dump_now()

Datum autoprewarm_dump_now

(

PG_FUNCTION_ARGS

)

Definition at line 846 of file autoprewarm.c.

{
    int         num_blocks;
 
    apw_init_shmem();
 
    PG_ENSURE_ERROR_CLEANUP(apw_detach_shmem, 0);
    {
        num_blocks = apw_dump_now(false, true);
    }
    PG_END_ENSURE_ERROR_CLEANUP(apw_detach_shmem, 0);
 
    PG_RETURN_INT64((int64) num_blocks);
}

References apw_detach_shmem(), apw_dump_now(), apw_init_shmem(), PG_END_ENSURE_ERROR_CLEANUP, PG_ENSURE_ERROR_CLEANUP, and PG_RETURN_INT64.

autoprewarm_main()

void autoprewarm_main

(

Datum

main_arg

)

Definition at line 168 of file autoprewarm.c.

{
    bool        first_time = true;
    bool        final_dump_allowed = true;
    TimestampTz last_dump_time = 0;
 
    /* Establish signal handlers; once that's done, unblock signals. */
    pqsignal(SIGTERM, SignalHandlerForShutdownRequest);
    pqsignal(SIGHUP, SignalHandlerForConfigReload);
    pqsignal(SIGUSR1, procsignal_sigusr1_handler);
    BackgroundWorkerUnblockSignals();
 
    /* Create (if necessary) and attach to our shared memory area. */
    if (apw_init_shmem())
        first_time = false;
 
    /*
     * Set on-detach hook so that our PID will be cleared on exit.
     *
     * NB: Autoprewarm's state is stored in a DSM segment, and DSM segments
     * are detached before calling the on_shmem_exit callbacks, so we must put
     * apw_detach_shmem in the before_shmem_exit callback list.
     */
    before_shmem_exit(apw_detach_shmem, 0);
 
    /*
     * Store our PID in the shared memory area --- unless there's already
     * another worker running, in which case just exit.
     */
    LWLockAcquire(&apw_state->lock, LW_EXCLUSIVE);
    if (apw_state->bgworker_pid != InvalidPid)
    {
        LWLockRelease(&apw_state->lock);
        ereport(LOG,
                (errmsg("autoprewarm worker is already running under PID %d",
                        (int) apw_state->bgworker_pid)));
        return;
    }
    apw_state->bgworker_pid = MyProcPid;
    LWLockRelease(&apw_state->lock);
 
    /*
     * Preload buffers from the dump file only if we just created the shared
     * memory region.  Otherwise, it's either already been done or shouldn't
     * be done - e.g. because the old dump file has been overwritten since the
     * server was started.
     *
     * There's not much point in performing a dump immediately after we finish
     * preloading; so, if we do end up preloading, consider the last dump time
     * to be equal to the current time.
     *
     * If apw_load_buffers() is terminated early by a shutdown request,
     * prevent dumping out our state below the loop, because we'd effectively
     * just truncate the saved state to however much we'd managed to preload.
     */
    if (first_time)
    {
        apw_load_buffers();
        final_dump_allowed = !ShutdownRequestPending;
        last_dump_time = GetCurrentTimestamp();
    }
 
    /* Periodically dump buffers until terminated. */
    while (!ShutdownRequestPending)
    {
        /* In case of a SIGHUP, just reload the configuration. */
        if (ConfigReloadPending)
        {
            ConfigReloadPending = false;
            ProcessConfigFile(PGC_SIGHUP);
        }
 
        if (autoprewarm_interval <= 0)
        {
            /* We're only dumping at shutdown, so just wait forever. */
            (void) WaitLatch(MyLatch,
                             WL_LATCH_SET | WL_EXIT_ON_PM_DEATH,
                             -1L,
                             PG_WAIT_EXTENSION);
        }
        else
        {
            TimestampTz next_dump_time;
            long        delay_in_ms;
 
            /* Compute the next dump time. */
            next_dump_time =
                TimestampTzPlusMilliseconds(last_dump_time,
                                            autoprewarm_interval * 1000);
            delay_in_ms =
                TimestampDifferenceMilliseconds(GetCurrentTimestamp(),
                                                next_dump_time);
 
            /* Perform a dump if it's time. */
            if (delay_in_ms <= 0)
            {
                last_dump_time = GetCurrentTimestamp();
                apw_dump_now(true, false);
                continue;
            }
 
            /* Sleep until the next dump time. */
            (void) WaitLatch(MyLatch,
                             WL_LATCH_SET | WL_TIMEOUT | WL_EXIT_ON_PM_DEATH,
                             delay_in_ms,
                             PG_WAIT_EXTENSION);
        }
 
        /* Reset the latch, loop. */
        ResetLatch(MyLatch);
    }
 
    /*
     * Dump one last time.  We assume this is probably the result of a system
     * shutdown, although it's possible that we've merely been terminated.
     */
    if (final_dump_allowed)
        apw_dump_now(true, true);
}

References apw_detach_shmem(), apw_dump_now(), apw_init_shmem(), apw_load_buffers(), apw_state, autoprewarm_interval, BackgroundWorkerUnblockSignals(), before_shmem_exit(), AutoPrewarmSharedState::bgworker_pid, ConfigReloadPending, ereport, errmsg, fb(), GetCurrentTimestamp(), InvalidPid, AutoPrewarmSharedState::lock, LOG, LW_EXCLUSIVE, LWLockAcquire(), LWLockRelease(), MyLatch, MyProcPid, PG_WAIT_EXTENSION, PGC_SIGHUP, pqsignal, ProcessConfigFile(), procsignal_sigusr1_handler(), ResetLatch(), ShutdownRequestPending, SIGHUP, SignalHandlerForConfigReload(), SignalHandlerForShutdownRequest(), SIGUSR1, TimestampDifferenceMilliseconds(), TimestampTzPlusMilliseconds, WaitLatch(), WL_EXIT_ON_PM_DEATH, WL_LATCH_SET, and WL_TIMEOUT.

autoprewarm_start_worker()

Datum autoprewarm_start_worker

(

PG_FUNCTION_ARGS

)

Definition at line 814 of file autoprewarm.c.

{
    pid_t       pid;
 
    if (!autoprewarm)
        ereport(ERROR,
                (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                 errmsg("autoprewarm is disabled")));
 
    apw_init_shmem();
    LWLockAcquire(&apw_state->lock, LW_EXCLUSIVE);
    pid = apw_state->bgworker_pid;
    LWLockRelease(&apw_state->lock);
 
    if (pid != InvalidPid)
        ereport(ERROR,
                (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
                 errmsg("autoprewarm worker is already running under PID %d",
                        (int) pid)));
 
    apw_start_leader_worker();
 
    PG_RETURN_VOID();
}

References apw_init_shmem(), apw_start_leader_worker(), apw_state, autoprewarm, AutoPrewarmSharedState::bgworker_pid, ereport, errcode(), errmsg, ERROR, fb(), InvalidPid, AutoPrewarmSharedState::lock, LW_EXCLUSIVE, LWLockAcquire(), LWLockRelease(), and PG_RETURN_VOID.

PG_FUNCTION_INFO_V1() [1/2]

PG_FUNCTION_INFO_V1

(

autoprewarm_dump_now

)

PG_FUNCTION_INFO_V1() [2/2]

PG_FUNCTION_INFO_V1

(

autoprewarm_start_worker

)

Variable Documentation

apw_state

AutoPrewarmSharedState* apw_state = NULL

static

Definition at line 117 of file autoprewarm.c.

Referenced by apw_detach_shmem(), apw_dump_now(), apw_init_shmem(), apw_load_buffers(), apw_read_stream_next_block(), autoprewarm_database_main(), autoprewarm_main(), and autoprewarm_start_worker().

autoprewarm

bool autoprewarm = true

static

Definition at line 120 of file autoprewarm.c.

Referenced by _PG_init(), and autoprewarm_start_worker().

autoprewarm_interval

int autoprewarm_interval = 300

static

Definition at line 121 of file autoprewarm.c.

Referenced by _PG_init(), and autoprewarm_main().