parallel_slot.h File Reference

#include "fe_utils/connect_utils.h"
#include "libpq-fe.h"

Include dependency graph for parallel_slot.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Data Structures
struct	ParallelSlot
struct	ParallelSlotArray

Typedefs
typedef bool(*	ParallelSlotResultHandler) (PGresult *res, PGconn *conn, void *context)
typedef struct ParallelSlot	ParallelSlot
typedef struct ParallelSlotArray	ParallelSlotArray

Functions
static void	ParallelSlotSetHandler (ParallelSlot *slot, ParallelSlotResultHandler handler, void *context)
static void	ParallelSlotClearHandler (ParallelSlot *slot)
static void	ParallelSlotSetIdle (ParallelSlot *slot)
ParallelSlot *	ParallelSlotsGetIdle (ParallelSlotArray *sa, const char *dbname)
ParallelSlotArray *	ParallelSlotsSetup (int numslots, ConnParams *cparams, const char *progname, bool echo, const char *initcmd)
void	ParallelSlotsAdoptConn (ParallelSlotArray *sa, PGconn *conn)
void	ParallelSlotsTerminate (ParallelSlotArray *sa)
bool	ParallelSlotsWaitCompletion (ParallelSlotArray *sa)
bool	TableCommandResultHandler (PGresult *res, PGconn *conn, void *context)

Typedef Documentation

ParallelSlot

typedef struct ParallelSlot ParallelSlot

ParallelSlotArray

typedef struct ParallelSlotArray ParallelSlotArray

ParallelSlotResultHandler

typedef bool(* ParallelSlotResultHandler) (PGresult *res, PGconn *conn, void *context)

Definition at line 18 of file parallel_slot.h.

Function Documentation

ParallelSlotClearHandler()

static void ParallelSlotClearHandler ( ParallelSlot *  slot )

inlinestatic

Definition at line 55 of file parallel_slot.h.

{
    slot->handler = NULL;
    slot->handler_context = NULL;
}

References fb(), ParallelSlot::handler, and ParallelSlot::handler_context.

Referenced by ParallelSlotSetIdle().

ParallelSlotsAdoptConn()

void ParallelSlotsAdoptConn ( ParallelSlotArray *  sa, PGconn *  conn  )

extern

Definition at line 459 of file parallel_slot.c.

{
    int         offset;
 
    offset = find_unconnected_slot(sa);
    if (offset >= 0)
        sa->slots[offset].connection = conn;
    else
        disconnectDatabase(conn);
}

References conn, disconnectDatabase(), and find_unconnected_slot().

Referenced by main(), reindex_one_database(), and vacuum_one_database().

ParallelSlotSetHandler()

static void ParallelSlotSetHandler ( ParallelSlot *  slot, ParallelSlotResultHandler  handler, void *  context  )

inlinestatic

Definition at line 47 of file parallel_slot.h.

{
    slot->handler = handler;
    slot->handler_context = context;
}

References ParallelSlot::handler, and ParallelSlot::handler_context.

Referenced by main(), reindex_one_database(), and vacuum_one_database().

ParallelSlotSetIdle()

static void ParallelSlotSetIdle ( ParallelSlot *  slot )

inlinestatic

Definition at line 62 of file parallel_slot.h.

{
    slot->inUse = false;
    ParallelSlotClearHandler(slot);
}

References ParallelSlot::inUse, and ParallelSlotClearHandler().

Referenced by ParallelSlotsWaitCompletion(), run_vacuum_command(), and wait_on_slots().

ParallelSlotsGetIdle()

ParallelSlot * ParallelSlotsGetIdle ( ParallelSlotArray *  sa, const char *  dbname  )

extern

Definition at line 370 of file parallel_slot.c.

{
    int         offset;
 
    Assert(sa);
    Assert(sa->numslots > 0);
 
    while (1)
    {
        /* First choice: a slot already connected to the desired database. */
        offset = find_matching_idle_slot(sa, dbname);
        if (offset >= 0)
        {
            sa->slots[offset].inUse = true;
            return &sa->slots[offset];
        }
 
        /* Second choice: a slot not connected to any database. */
        offset = find_unconnected_slot(sa);
        if (offset >= 0)
        {
            connect_slot(sa, offset, dbname);
            sa->slots[offset].inUse = true;
            return &sa->slots[offset];
        }
 
        /* Third choice: a slot connected to the wrong database. */
        offset = find_any_idle_slot(sa);
        if (offset >= 0)
        {
            disconnectDatabase(sa->slots[offset].connection);
            sa->slots[offset].connection = NULL;
            connect_slot(sa, offset, dbname);
            sa->slots[offset].inUse = true;
            return &sa->slots[offset];
        }
 
        /*
         * Fourth choice: block until one or more slots become available. If
         * any slots hit a fatal error, we'll find out about that here and
         * return NULL.
         */
        if (!wait_on_slots(sa))
            return NULL;
    }
}

References Assert, connect_slot(), dbname, disconnectDatabase(), fb(), find_any_idle_slot(), find_matching_idle_slot(), find_unconnected_slot(), and wait_on_slots().

Referenced by main(), reindex_one_database(), and vacuum_one_database().

ParallelSlotsSetup()

ParallelSlotArray * ParallelSlotsSetup ( int  numslots, ConnParams *  cparams, const char *  progname, bool  echo, const char *  initcmd  )

extern

Definition at line 427 of file parallel_slot.c.

{
    ParallelSlotArray *sa;
 
    Assert(numslots > 0);
    Assert(cparams != NULL);
    Assert(progname != NULL);
 
    sa = (ParallelSlotArray *) palloc0(offsetof(ParallelSlotArray, slots) +
                                       numslots * sizeof(ParallelSlot));
 
    sa->numslots = numslots;
    sa->cparams = cparams;
    sa->progname = progname;
    sa->echo = echo;
    sa->initcmd = initcmd;
 
    return sa;
}

References Assert, fb(), palloc0(), and progname.

Referenced by main(), reindex_one_database(), and vacuum_one_database().

ParallelSlotsTerminate()

void ParallelSlotsTerminate ( ParallelSlotArray *  sa )

extern

Definition at line 478 of file parallel_slot.c.

{
    int         i;
 
    for (i = 0; i < sa->numslots; i++)
    {
        PGconn     *conn = sa->slots[i].connection;
 
        if (conn == NULL)
            continue;
 
        disconnectDatabase(conn);
    }
}

References conn, disconnectDatabase(), fb(), and i.

Referenced by main(), reindex_one_database(), and vacuum_one_database().

ParallelSlotsWaitCompletion()

bool ParallelSlotsWaitCompletion ( ParallelSlotArray *  sa )

extern

Definition at line 500 of file parallel_slot.c.

{
    int         i;
 
    for (i = 0; i < sa->numslots; i++)
    {
        if (sa->slots[i].connection == NULL)
            continue;
        if (!consumeQueryResult(&sa->slots[i]))
            return false;
        /* Mark connection as idle */
        ParallelSlotSetIdle(&sa->slots[i]);
    }
 
    return true;
}

References consumeQueryResult(), fb(), i, and ParallelSlotSetIdle().

Referenced by main(), reindex_one_database(), and vacuum_one_database().

TableCommandResultHandler()

bool TableCommandResultHandler ( PGresult *  res, PGconn *  conn, void *  context  )

extern

Definition at line 538 of file parallel_slot.c.

{
    Assert(res != NULL);
    Assert(conn != NULL);
 
    /*
     * If it's an error, report it.  Errors about a missing table are harmless
     * so we continue processing; but die for other errors.
     */
    if (PQresultStatus(res) != PGRES_COMMAND_OK)
    {
        char       *sqlState = PQresultErrorField(res, PG_DIAG_SQLSTATE);
 
        pg_log_error("processing of database \"%s\" failed: %s",
                     PQdb(conn), PQerrorMessage(conn));
 
        if (sqlState && strcmp(sqlState, ERRCODE_UNDEFINED_TABLE) != 0)
        {
            PQclear(res);
            return false;
        }
    }
 
    return true;
}

References Assert, conn, ERRCODE_UNDEFINED_TABLE, fb(), PG_DIAG_SQLSTATE, pg_log_error, PGRES_COMMAND_OK, PQclear, PQdb(), PQerrorMessage(), PQresultErrorField, and PQresultStatus.

Referenced by reindex_one_database(), and vacuum_one_database().