#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:

Macros
#define	STARTUP_TRANCHE_NAME "test_lwlock_tranches_startup"

#define	DYNAMIC_TRANCHE_NAME "test_lwlock_tranches_dynamic"

#define	NUM_STARTUP_TRANCHES (32)

#define	NUM_DYNAMIC_TRANCHES (256 - NUM_STARTUP_TRANCHES)

#define	GET_TRANCHE_NAME(a) GetLWLockIdentifier(PG_WAIT_LWLOCK, (a))

Variables
	PG_MODULE_MAGIC

static shmem_request_hook_type	prev_shmem_request_hook

Macro Definition Documentation

◆ DYNAMIC_TRANCHE_NAME

#define DYNAMIC_TRANCHE_NAME "test_lwlock_tranches_dynamic"

Definition at line 25 of file test_lwlock_tranches.c.

◆ GET_TRANCHE_NAME

#define GET_TRANCHE_NAME ( a ) GetLWLockIdentifier(PG_WAIT_LWLOCK, (a))

Definition at line 30 of file test_lwlock_tranches.c.

◆ NUM_DYNAMIC_TRANCHES

#define NUM_DYNAMIC_TRANCHES (256 - NUM_STARTUP_TRANCHES)

Definition at line 28 of file test_lwlock_tranches.c.

◆ NUM_STARTUP_TRANCHES

#define NUM_STARTUP_TRANCHES (32)

Definition at line 27 of file test_lwlock_tranches.c.

◆ STARTUP_TRANCHE_NAME

#define STARTUP_TRANCHE_NAME "test_lwlock_tranches_startup"

Definition at line 24 of file test_lwlock_tranches.c.

Function Documentation

◆ _PG_init()

void _PG_init ( void )

Definition at line 36 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/4]

PG_FUNCTION_INFO_V1 ( test_lwlock_initialize )

◆ PG_FUNCTION_INFO_V1() [2/4]

PG_FUNCTION_INFO_V1 ( test_lwlock_tranche_creation )

◆ PG_FUNCTION_INFO_V1() [3/4]

PG_FUNCTION_INFO_V1 ( test_lwlock_tranche_lookup )

◆ PG_FUNCTION_INFO_V1() [4/4]

PG_FUNCTION_INFO_V1 ( test_lwlock_tranches )

◆ test_lwlock_initialize()

Datum test_lwlock_initialize ( PG_FUNCTION_ARGS )

Definition at line 115 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_creation()

Datum test_lwlock_tranche_creation ( PG_FUNCTION_ARGS )

Definition at line 87 of file test_lwlock_tranches.c.

{
    char       *tranche_name = PG_ARGISNULL(0) ? NULL : TextDatumGetCString(PG_GETARG_DATUM(0));
 
    (void) LWLockNewTrancheId(tranche_name);
 
    PG_RETURN_VOID();
}

References LWLockNewTrancheId(), PG_ARGISNULL, PG_GETARG_DATUM, PG_RETURN_VOID, and TextDatumGetCString.

◆ test_lwlock_tranche_lookup()

Datum test_lwlock_tranche_lookup ( PG_FUNCTION_ARGS )

Definition at line 101 of file test_lwlock_tranches.c.

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

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

◆ test_lwlock_tranches()

Datum test_lwlock_tranches ( PG_FUNCTION_ARGS )

Definition at line 58 of file test_lwlock_tranches.c.

{
    int         dynamic_tranches[NUM_DYNAMIC_TRANCHES];
 
    for (int i = 0; i < NUM_DYNAMIC_TRANCHES; i++)
        dynamic_tranches[i] = LWLockNewTrancheId(DYNAMIC_TRANCHE_NAME);
 
    for (int i = 0; i < NUM_STARTUP_TRANCHES; i++)
    {
        if (strcmp(GET_TRANCHE_NAME(LWTRANCHE_FIRST_USER_DEFINED + i),
                   STARTUP_TRANCHE_NAME) != 0)
            elog(ERROR, "incorrect startup lock tranche name");
    }
 
    for (int i = 0; i < NUM_DYNAMIC_TRANCHES; i++)
    {
        if (strcmp(GET_TRANCHE_NAME(dynamic_tranches[i]),
                   DYNAMIC_TRANCHE_NAME) != 0)
            elog(ERROR, "incorrect dynamic lock tranche name");
    }
 
    PG_RETURN_VOID();
}

References DYNAMIC_TRANCHE_NAME, elog, ERROR, GET_TRANCHE_NAME, i, LWLockNewTrancheId(), LWTRANCHE_FIRST_USER_DEFINED, NUM_DYNAMIC_TRANCHES, NUM_STARTUP_TRANCHES, PG_RETURN_VOID, and STARTUP_TRANCHE_NAME.

◆ test_lwlock_tranches_shmem_request()

static void test_lwlock_tranches_shmem_request ( void )

static

Definition at line 43 of file test_lwlock_tranches.c.

{
    if (prev_shmem_request_hook)
        prev_shmem_request_hook();
 
    for (int i = 0; i < NUM_STARTUP_TRANCHES; i++)
        RequestNamedLWLockTranche(STARTUP_TRANCHE_NAME, 1);
}

References i, NUM_STARTUP_TRANCHES, prev_shmem_request_hook, RequestNamedLWLockTranche(), and STARTUP_TRANCHE_NAME.

Referenced by _PG_init().

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