test_dsm_registry.c File Reference

#include "postgres.h"
#include "fmgr.h"
#include "storage/dsm_registry.h"
#include "storage/lwlock.h"

Include dependency graph for test_dsm_registry.c:

Go to the source code of this file.

Data Structures
struct	TestDSMRegistryStruct

Typedefs
typedef struct TestDSMRegistryStruct	TestDSMRegistryStruct

Functions
static void	tdr_init_shmem (void *ptr)
static void	tdr_attach_shmem (void)
	PG_FUNCTION_INFO_V1 (set_val_in_shmem)
Datum	set_val_in_shmem (PG_FUNCTION_ARGS)
	PG_FUNCTION_INFO_V1 (get_val_in_shmem)
Datum	get_val_in_shmem (PG_FUNCTION_ARGS)

Variables
	PG_MODULE_MAGIC
static TestDSMRegistryStruct *	tdr_state

Typedef Documentation

TestDSMRegistryStruct

typedef struct TestDSMRegistryStruct TestDSMRegistryStruct

Function Documentation

get_val_in_shmem()

Datum get_val_in_shmem

(

PG_FUNCTION_ARGS

)

Definition at line 65 of file test_dsm_registry.c.

{
    int         ret;
 
    tdr_attach_shmem();
 
    LWLockAcquire(&tdr_state->lck, LW_SHARED);
    ret = tdr_state->val;
    LWLockRelease(&tdr_state->lck);
 
    PG_RETURN_UINT32(ret);
}

References TestDSMRegistryStruct::lck, LW_SHARED, LWLockAcquire(), LWLockRelease(), PG_RETURN_UINT32, tdr_attach_shmem(), tdr_state, and TestDSMRegistryStruct::val.

PG_FUNCTION_INFO_V1() [1/2]

PG_FUNCTION_INFO_V1

(

get_val_in_shmem

)

PG_FUNCTION_INFO_V1() [2/2]

PG_FUNCTION_INFO_V1

(

set_val_in_shmem

)

set_val_in_shmem()

Datum set_val_in_shmem

(

PG_FUNCTION_ARGS

)

Definition at line 52 of file test_dsm_registry.c.

{
    tdr_attach_shmem();
 
    LWLockAcquire(&tdr_state->lck, LW_EXCLUSIVE);
    tdr_state->val = PG_GETARG_UINT32(0);
    LWLockRelease(&tdr_state->lck);
 
    PG_RETURN_VOID();
}

References TestDSMRegistryStruct::lck, LW_EXCLUSIVE, LWLockAcquire(), LWLockRelease(), PG_GETARG_UINT32, PG_RETURN_VOID, tdr_attach_shmem(), tdr_state, and TestDSMRegistryStruct::val.

tdr_attach_shmem()

static void tdr_attach_shmem ( void  )

static

Definition at line 39 of file test_dsm_registry.c.

{
    bool        found;
 
    tdr_state = GetNamedDSMSegment("test_dsm_registry",
                                   sizeof(TestDSMRegistryStruct),
                                   tdr_init_shmem,
                                   &found);
    LWLockRegisterTranche(tdr_state->lck.tranche, "test_dsm_registry");
}

References GetNamedDSMSegment(), TestDSMRegistryStruct::lck, LWLockRegisterTranche(), tdr_init_shmem(), tdr_state, and LWLock::tranche.

Referenced by get_val_in_shmem(), and set_val_in_shmem().

tdr_init_shmem()

static void tdr_init_shmem ( void *  ptr )

static

Definition at line 30 of file test_dsm_registry.c.

{
    TestDSMRegistryStruct *state = (TestDSMRegistryStruct *) ptr;
 
    LWLockInitialize(&state->lck, LWLockNewTrancheId());
    state->val = 0;
}

References LWLockInitialize(), and LWLockNewTrancheId().

Referenced by tdr_attach_shmem().

Variable Documentation

PG_MODULE_MAGIC

PG_MODULE_MAGIC

Definition at line 19 of file test_dsm_registry.c.

tdr_state

TestDSMRegistryStruct* tdr_state

static

Definition at line 27 of file test_dsm_registry.c.

Referenced by get_val_in_shmem(), set_val_in_shmem(), and tdr_attach_shmem().