test_dsm_registry.c File Reference

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

Include dependency graph for test_dsm_registry.c:

Go to the source code of this file.

Data Structures
struct	TestDSMRegistryStruct
struct	TestDSMRegistryHashEntry

Typedefs
typedef struct TestDSMRegistryStruct	TestDSMRegistryStruct
typedef struct TestDSMRegistryHashEntry	TestDSMRegistryHashEntry

Functions
static void	init_tdr_dsm (void *ptr, void *arg)
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)
	PG_FUNCTION_INFO_V1 (set_val_in_hash)
Datum	set_val_in_hash (PG_FUNCTION_ARGS)
	PG_FUNCTION_INFO_V1 (get_val_in_hash)
Datum	get_val_in_hash (PG_FUNCTION_ARGS)

Variables
	PG_MODULE_MAGIC
static TestDSMRegistryStruct *	tdr_dsm
static dsa_area *	tdr_dsa
static dshash_table *	tdr_hash
static const dshash_parameters	dsh_params

Typedef Documentation

TestDSMRegistryHashEntry

typedef struct TestDSMRegistryHashEntry TestDSMRegistryHashEntry

TestDSMRegistryStruct

typedef struct TestDSMRegistryStruct TestDSMRegistryStruct

Function Documentation

get_val_in_hash()

Datum get_val_in_hash

(

PG_FUNCTION_ARGS

)

Definition at line 132 of file test_dsm_registry.c.

{
    TestDSMRegistryHashEntry *entry;
    char       *key = TextDatumGetCString(PG_GETARG_DATUM(0));
    text       *val = NULL;
 
    tdr_attach_shmem();
 
    entry = dshash_find(tdr_hash, key, false);
    if (entry == NULL)
        PG_RETURN_NULL();
 
    val = cstring_to_text(dsa_get_address(tdr_dsa, entry->val));
 
    dshash_release_lock(tdr_hash, entry);
 
    PG_RETURN_TEXT_P(val);
}

References cstring_to_text(), dsa_get_address(), dshash_find(), dshash_release_lock(), fb(), PG_GETARG_DATUM, PG_RETURN_NULL, PG_RETURN_TEXT_P, tdr_attach_shmem(), tdr_dsa, tdr_hash, TextDatumGetCString, val, and TestDSMRegistryHashEntry::val.

get_val_in_shmem()

Datum get_val_in_shmem

(

PG_FUNCTION_ARGS

)

Definition at line 90 of file test_dsm_registry.c.

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

References TestDSMRegistryStruct::lck, LW_SHARED, LWLockAcquire(), LWLockRelease(), PG_RETURN_INT32, tdr_attach_shmem(), tdr_dsm, and TestDSMRegistryStruct::val.

init_tdr_dsm()

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

static

Definition at line 47 of file test_dsm_registry.c.

{
    TestDSMRegistryStruct *dsm = (TestDSMRegistryStruct *) ptr;
 
    if ((int) (intptr_t) arg != 5432)
        elog(ERROR, "unexpected arg value %d", (int) (intptr_t) arg);
 
    LWLockInitialize(&dsm->lck, LWLockNewTrancheId("test_dsm_registry"));
    dsm->val = 0;
}

References arg, elog, ERROR, fb(), TestDSMRegistryStruct::lck, LWLockInitialize(), LWLockNewTrancheId(), and TestDSMRegistryStruct::val.

Referenced by tdr_attach_shmem().

PG_FUNCTION_INFO_V1() [1/4]

PG_FUNCTION_INFO_V1

(

get_val_in_hash

)

PG_FUNCTION_INFO_V1() [2/4]

PG_FUNCTION_INFO_V1

(

get_val_in_shmem

)

PG_FUNCTION_INFO_V1() [3/4]

PG_FUNCTION_INFO_V1

(

set_val_in_hash

)

PG_FUNCTION_INFO_V1() [4/4]

PG_FUNCTION_INFO_V1

(

set_val_in_shmem

)

set_val_in_hash()

Datum set_val_in_hash

(

PG_FUNCTION_ARGS

)

Definition at line 105 of file test_dsm_registry.c.

{
    TestDSMRegistryHashEntry *entry;
    char       *key = TextDatumGetCString(PG_GETARG_DATUM(0));
    char       *val = TextDatumGetCString(PG_GETARG_DATUM(1));
    bool        found;
 
    if (strlen(key) >= offsetof(TestDSMRegistryHashEntry, val))
        ereport(ERROR,
                (errmsg("key too long")));
 
    tdr_attach_shmem();
 
    entry = dshash_find_or_insert(tdr_hash, key, &found);
    if (found)
        dsa_free(tdr_dsa, entry->val);
 
    entry->val = dsa_allocate(tdr_dsa, strlen(val) + 1);
    strcpy(dsa_get_address(tdr_dsa, entry->val), val);
 
    dshash_release_lock(tdr_hash, entry);
 
    PG_RETURN_VOID();
}

References dsa_allocate, dsa_free(), dsa_get_address(), dshash_find_or_insert(), dshash_release_lock(), ereport, errmsg(), ERROR, fb(), PG_GETARG_DATUM, PG_RETURN_VOID, tdr_attach_shmem(), tdr_dsa, tdr_hash, TextDatumGetCString, val, and TestDSMRegistryHashEntry::val.

set_val_in_shmem()

Datum set_val_in_shmem

(

PG_FUNCTION_ARGS

)

Definition at line 77 of file test_dsm_registry.c.

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

References TestDSMRegistryStruct::lck, LW_EXCLUSIVE, LWLockAcquire(), LWLockRelease(), PG_GETARG_INT32, PG_RETURN_VOID, tdr_attach_shmem(), tdr_dsm, and TestDSMRegistryStruct::val.

tdr_attach_shmem()

static void tdr_attach_shmem ( void  )

static

Definition at line 59 of file test_dsm_registry.c.

{
    bool        found;
 
    tdr_dsm = GetNamedDSMSegment("test_dsm_registry_dsm",
                                 sizeof(TestDSMRegistryStruct),
                                 init_tdr_dsm,
                                 &found, (void *) (intptr_t) 5432);
 
    if (tdr_dsa == NULL)
        tdr_dsa = GetNamedDSA("test_dsm_registry_dsa", &found);
 
    if (tdr_hash == NULL)
        tdr_hash = GetNamedDSHash("test_dsm_registry_hash", &dsh_params, &found);
}

References dsh_params, fb(), GetNamedDSA(), GetNamedDSHash(), GetNamedDSMSegment(), init_tdr_dsm(), tdr_dsa, tdr_dsm, and tdr_hash.

Referenced by get_val_in_hash(), get_val_in_shmem(), set_val_in_hash(), and set_val_in_shmem().

Variable Documentation

dsh_params

const dshash_parameters dsh_params

static

Initial value:

= {
    offsetof(TestDSMRegistryHashEntry, val),
    sizeof(TestDSMRegistryHashEntry),
    dshash_strcmp,
    dshash_strhash,
    dshash_strcpy
}

Definition at line 38 of file test_dsm_registry.c.

                                            {
    offsetof(TestDSMRegistryHashEntry, val),
    sizeof(TestDSMRegistryHashEntry),
    dshash_strcmp,
    dshash_strhash,
    dshash_strcpy
};

Referenced by tdr_attach_shmem().

PG_MODULE_MAGIC

PG_MODULE_MAGIC

Definition at line 20 of file test_dsm_registry.c.

tdr_dsa

dsa_area* tdr_dsa

static

Definition at line 35 of file test_dsm_registry.c.

Referenced by get_val_in_hash(), set_val_in_hash(), and tdr_attach_shmem().

tdr_dsm

TestDSMRegistryStruct* tdr_dsm

static

Definition at line 34 of file test_dsm_registry.c.

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

tdr_hash

dshash_table* tdr_hash

static

Definition at line 36 of file test_dsm_registry.c.

Referenced by get_val_in_hash(), set_val_in_hash(), and tdr_attach_shmem().