#include "postgres.h"
#include "fmgr.h"
#include "miscadmin.h"
#include "storage/lwlock.h"
#include "utils/builtins.h"
#include "utils/wait_classes.h"

Include dependency graph for test_lwlock_tranches.c:

Variables
	PG_MODULE_MAGIC

static shmem_request_hook_type	prev_shmem_request_hook

Function Documentation

◆ _PG_init()

void _PG_init ( void )

Definition at line 28 of file test_lwlock_tranches.c.

{
    prev_shmem_request_hook = shmem_request_hook;
    shmem_request_hook = test_lwlock_tranches_shmem_request;
}

References prev_shmem_request_hook, shmem_request_hook, and test_lwlock_tranches_shmem_request().

◆ PG_FUNCTION_INFO_V1() [1/5]

PG_FUNCTION_INFO_V1 ( test_lwlock_get_lwlock_identifier )

◆ PG_FUNCTION_INFO_V1() [2/5]

PG_FUNCTION_INFO_V1 ( test_lwlock_initialize )

◆ PG_FUNCTION_INFO_V1() [3/5]

PG_FUNCTION_INFO_V1 ( test_lwlock_tranche_create )

◆ PG_FUNCTION_INFO_V1() [4/5]

PG_FUNCTION_INFO_V1 ( test_lwlock_tranche_lookup )

◆ PG_FUNCTION_INFO_V1() [5/5]

PG_FUNCTION_INFO_V1 ( test_startup_lwlocks )

◆ test_lwlock_get_lwlock_identifier()

Datum test_lwlock_get_lwlock_identifier ( PG_FUNCTION_ARGS )

Definition at line 115 of file test_lwlock_tranches.c.

{
    int         eventId = PG_GETARG_INT32(0);
    const char *tranche_name;
 
    tranche_name = GetLWLockIdentifier(PG_WAIT_LWLOCK, eventId);
 
    if (tranche_name)
        PG_RETURN_TEXT_P(cstring_to_text(tranche_name));
    else
        PG_RETURN_NULL();
}

References cstring_to_text(), fb(), GetLWLockIdentifier(), PG_GETARG_INT32, PG_RETURN_NULL, PG_RETURN_TEXT_P, and PG_WAIT_LWLOCK.

◆ test_lwlock_initialize()

Datum test_lwlock_initialize ( PG_FUNCTION_ARGS )

Definition at line 133 of file test_lwlock_tranches.c.

{
    int         tranche_id = PG_GETARG_INT32(0);
    LWLock      lock;
 
    LWLockInitialize(&lock, tranche_id);
 
    PG_RETURN_VOID();
}

References LWLockInitialize(), PG_GETARG_INT32, and PG_RETURN_VOID.

◆ test_lwlock_tranche_create()

Datum test_lwlock_tranche_create ( PG_FUNCTION_ARGS )

Definition at line 89 of file test_lwlock_tranches.c.

{
    char       *tranche_name = PG_ARGISNULL(0) ? NULL : TextDatumGetCString(PG_GETARG_DATUM(0));
    int         tranche_id;
 
    tranche_id = LWLockNewTrancheId(tranche_name);
 
    PG_RETURN_INT32(tranche_id);
}

References fb(), LWLockNewTrancheId(), PG_ARGISNULL, PG_GETARG_DATUM, PG_RETURN_INT32, and TextDatumGetCString.

◆ test_lwlock_tranche_lookup()

Datum test_lwlock_tranche_lookup ( PG_FUNCTION_ARGS )

Definition at line 104 of file test_lwlock_tranches.c.

{
    char       *tranche_name = TextDatumGetCString(PG_GETARG_DATUM(0));
 
    (void) GetNamedLWLockTranche(tranche_name);
 
    PG_RETURN_VOID();
}

References fb(), GetNamedLWLockTranche(), PG_GETARG_DATUM, PG_RETURN_VOID, and TextDatumGetCString.

◆ test_lwlock_tranches_shmem_request()

static void test_lwlock_tranches_shmem_request ( void )

static

Definition at line 35 of file test_lwlock_tranches.c.

{
    if (prev_shmem_request_hook)
        prev_shmem_request_hook();
 
    /*
     * Request some tranches with RequestNamedLWLockTranche() for
     * test_startup_lwlocks()
     */
    RequestNamedLWLockTranche("test_lwlock_tranches_startup", 1);
    RequestNamedLWLockTranche("test_lwlock_tranches_startup10", 10);
}

References prev_shmem_request_hook, and RequestNamedLWLockTranche().

Referenced by _PG_init().

◆ test_startup_lwlocks()

Datum test_startup_lwlocks ( PG_FUNCTION_ARGS )

Definition at line 53 of file test_lwlock_tranches.c.

{
    LWLockPadded *lwlock_startup;
    LWLockPadded *lwlock_startup10;
 
    /* Check that the locks can be used */
    lwlock_startup = GetNamedLWLockTranche("test_lwlock_tranches_startup");
    lwlock_startup10 = GetNamedLWLockTranche("test_lwlock_tranches_startup10");
 
    LWLockAcquire(&lwlock_startup->lock, LW_EXCLUSIVE);
    for (int i = 0; i < 10; i++)
        LWLockAcquire(&lwlock_startup10[i].lock, LW_EXCLUSIVE);
 
    LWLockRelease(&lwlock_startup->lock);
    for (int i = 0; i < 10; i++)
        LWLockRelease(&lwlock_startup10[i].lock);
 
    /*
     * Check that GetLWLockIdentifier() returns the expected value for
     * tranches
     */
    if (strcmp("test_lwlock_tranches_startup",
               GetLWLockIdentifier(PG_WAIT_LWLOCK, LWTRANCHE_FIRST_USER_DEFINED)) != 0)
        elog(ERROR, "incorrect startup lock tranche name");
    if (strcmp("test_lwlock_tranches_startup10",
               GetLWLockIdentifier(PG_WAIT_LWLOCK, LWTRANCHE_FIRST_USER_DEFINED + 1)) != 0)
        elog(ERROR, "incorrect startup lock tranche name");
 
    PG_RETURN_VOID();
}

References elog, ERROR, fb(), GetLWLockIdentifier(), GetNamedLWLockTranche(), i, LW_EXCLUSIVE, LWLockAcquire(), LWLockRelease(), LWTRANCHE_FIRST_USER_DEFINED, PG_RETURN_VOID, and PG_WAIT_LWLOCK.

Variable Documentation

◆ PG_MODULE_MAGIC

PG_MODULE_MAGIC

Definition at line 22 of file test_lwlock_tranches.c.

◆ prev_shmem_request_hook

shmem_request_hook_type prev_shmem_request_hook