test_resowner_basic.c File Reference

#include "postgres.h"
#include "fmgr.h"
#include "lib/ilist.h"
#include "utils/memutils.h"
#include "utils/resowner.h"

Include dependency graph for test_resowner_basic.c:

Go to the source code of this file.

Functions
static void	ReleaseString (Datum res)
static char *	PrintString (Datum res)
	PG_FUNCTION_INFO_V1 (test_resowner_priorities)
Datum	test_resowner_priorities (PG_FUNCTION_ARGS)
	PG_FUNCTION_INFO_V1 (test_resowner_leak)
Datum	test_resowner_leak (PG_FUNCTION_ARGS)
	PG_FUNCTION_INFO_V1 (test_resowner_remember_between_phases)
Datum	test_resowner_remember_between_phases (PG_FUNCTION_ARGS)
	PG_FUNCTION_INFO_V1 (test_resowner_forget_between_phases)
Datum	test_resowner_forget_between_phases (PG_FUNCTION_ARGS)

Variables
	PG_MODULE_MAGIC
static const ResourceOwnerDesc	string_desc

Function Documentation

PG_FUNCTION_INFO_V1() [1/4]

PG_FUNCTION_INFO_V1

(

test_resowner_forget_between_phases

)

PG_FUNCTION_INFO_V1() [2/4]

PG_FUNCTION_INFO_V1

(

test_resowner_leak

)

PG_FUNCTION_INFO_V1() [3/4]

PG_FUNCTION_INFO_V1

(

test_resowner_priorities

)

PG_FUNCTION_INFO_V1() [4/4]

PG_FUNCTION_INFO_V1

(

test_resowner_remember_between_phases

)

PrintString()

static char * PrintString ( Datum  res )

static

Definition at line 44 of file test_resowner_basic.c.

{
    return psprintf("test string \"%s\"", DatumGetPointer(res));
}

References DatumGetPointer(), psprintf(), and res.

Referenced by test_resowner_priorities().

ReleaseString()

static void ReleaseString ( Datum  res )

static

Definition at line 38 of file test_resowner_basic.c.

{
    elog(NOTICE, "releasing string: %s", DatumGetPointer(res));
}

References DatumGetPointer(), elog, NOTICE, and res.

Referenced by test_resowner_priorities().

test_resowner_forget_between_phases()

Datum test_resowner_forget_between_phases

(

PG_FUNCTION_ARGS

)

Definition at line 188 of file test_resowner_basic.c.

{
    ResourceOwner resowner;
    Datum       str_resource;
 
    resowner = ResourceOwnerCreate(CurrentResourceOwner, "TestOwner");
 
    ResourceOwnerEnlarge(resowner);
    str_resource = CStringGetDatum("my string");
    ResourceOwnerRemember(resowner, str_resource, &string_desc);
 
    ResourceOwnerRelease(resowner, RESOURCE_RELEASE_BEFORE_LOCKS, true, false);
 
    /*
     * Try to forget the resource that was remembered earlier.  Fails because
     * we already called ResourceOwnerRelease.
     */
    ResourceOwnerForget(resowner, str_resource, &string_desc);
 
    /* unreachable */
    elog(ERROR, "ResourceOwnerForget should have errored out");
 
    PG_RETURN_VOID();
}

References CStringGetDatum(), CurrentResourceOwner, elog, ERROR, PG_RETURN_VOID, RESOURCE_RELEASE_BEFORE_LOCKS, ResourceOwnerCreate(), ResourceOwnerEnlarge(), ResourceOwnerForget(), ResourceOwnerRelease(), ResourceOwnerRemember(), and string_desc.

test_resowner_leak()

Datum test_resowner_leak

(

PG_FUNCTION_ARGS

)

Definition at line 142 of file test_resowner_basic.c.

{
    ResourceOwner resowner;
 
    resowner = ResourceOwnerCreate(CurrentResourceOwner, "TestOwner");
 
    ResourceOwnerEnlarge(resowner);
 
    ResourceOwnerRemember(resowner, CStringGetDatum("my string"), &string_desc);
 
    /* don't call ResourceOwnerForget, so that it is leaked */
 
    ResourceOwnerRelease(resowner, RESOURCE_RELEASE_BEFORE_LOCKS, true, false);
    ResourceOwnerRelease(resowner, RESOURCE_RELEASE_LOCKS, true, false);
    ResourceOwnerRelease(resowner, RESOURCE_RELEASE_AFTER_LOCKS, true, false);
 
    ResourceOwnerDelete(resowner);
 
    PG_RETURN_VOID();
}

References CStringGetDatum(), CurrentResourceOwner, PG_RETURN_VOID, RESOURCE_RELEASE_AFTER_LOCKS, RESOURCE_RELEASE_BEFORE_LOCKS, RESOURCE_RELEASE_LOCKS, ResourceOwnerCreate(), ResourceOwnerDelete(), ResourceOwnerEnlarge(), ResourceOwnerRelease(), ResourceOwnerRemember(), and string_desc.

test_resowner_priorities()

Datum test_resowner_priorities

(

PG_FUNCTION_ARGS

)

Definition at line 52 of file test_resowner_basic.c.

{
    int32       nkinds = PG_GETARG_INT32(0);
    int32       nresources = PG_GETARG_INT32(1);
    ResourceOwner parent,
                child;
    ResourceOwnerDesc *before_desc;
    ResourceOwnerDesc *after_desc;
 
    if (nkinds <= 0)
        elog(ERROR, "nkinds must be greater than zero");
    if (nresources <= 0)
        elog(ERROR, "nresources must be greater than zero");
 
    parent = ResourceOwnerCreate(CurrentResourceOwner, "test parent");
    child = ResourceOwnerCreate(parent, "test child");
 
    before_desc = palloc(nkinds * sizeof(ResourceOwnerDesc));
    for (int i = 0; i < nkinds; i++)
    {
        before_desc[i].name = psprintf("test resource before locks %d", i);
        before_desc[i].release_phase = RESOURCE_RELEASE_BEFORE_LOCKS;
        before_desc[i].release_priority = RELEASE_PRIO_FIRST + i;
        before_desc[i].ReleaseResource = ReleaseString;
        before_desc[i].DebugPrint = PrintString;
    }
    after_desc = palloc(nkinds * sizeof(ResourceOwnerDesc));
    for (int i = 0; i < nkinds; i++)
    {
        after_desc[i].name = psprintf("test resource after locks %d", i);
        after_desc[i].release_phase = RESOURCE_RELEASE_AFTER_LOCKS;
        after_desc[i].release_priority = RELEASE_PRIO_FIRST + i;
        after_desc[i].ReleaseResource = ReleaseString;
        after_desc[i].DebugPrint = PrintString;
    }
 
    /* Add a bunch of resources to child, with different priorities */
    for (int i = 0; i < nresources; i++)
    {
        ResourceOwnerDesc *kind = &before_desc[i % nkinds];
 
        ResourceOwnerEnlarge(child);
        ResourceOwnerRemember(child,
                              CStringGetDatum(psprintf("child before locks priority %d", kind->release_priority)),
                              kind);
    }
    for (int i = 0; i < nresources; i++)
    {
        ResourceOwnerDesc *kind = &after_desc[i % nkinds];
 
        ResourceOwnerEnlarge(child);
        ResourceOwnerRemember(child,
                              CStringGetDatum(psprintf("child after locks priority %d", kind->release_priority)),
                              kind);
    }
 
    /* And also to the parent */
    for (int i = 0; i < nresources; i++)
    {
        ResourceOwnerDesc *kind = &after_desc[i % nkinds];
 
        ResourceOwnerEnlarge(parent);
        ResourceOwnerRemember(parent,
                              CStringGetDatum(psprintf("parent after locks priority %d", kind->release_priority)),
                              kind);
    }
    for (int i = 0; i < nresources; i++)
    {
        ResourceOwnerDesc *kind = &before_desc[i % nkinds];
 
        ResourceOwnerEnlarge(parent);
        ResourceOwnerRemember(parent,
                              CStringGetDatum(psprintf("parent before locks priority %d", kind->release_priority)),
                              kind);
    }
 
    elog(NOTICE, "releasing resources before locks");
    ResourceOwnerRelease(parent, RESOURCE_RELEASE_BEFORE_LOCKS, false, false);
    elog(NOTICE, "releasing locks");
    ResourceOwnerRelease(parent, RESOURCE_RELEASE_LOCKS, false, false);
    elog(NOTICE, "releasing resources after locks");
    ResourceOwnerRelease(parent, RESOURCE_RELEASE_AFTER_LOCKS, false, false);
 
    ResourceOwnerDelete(parent);
 
    PG_RETURN_VOID();
}

References CStringGetDatum(), CurrentResourceOwner, ResourceOwnerDesc::DebugPrint, elog, ERROR, i, ResourceOwnerDesc::name, NOTICE, palloc(), PG_GETARG_INT32, PG_RETURN_VOID, PrintString(), psprintf(), ResourceOwnerDesc::release_phase, RELEASE_PRIO_FIRST, ResourceOwnerDesc::release_priority, ResourceOwnerDesc::ReleaseResource, ReleaseString(), RESOURCE_RELEASE_AFTER_LOCKS, RESOURCE_RELEASE_BEFORE_LOCKS, RESOURCE_RELEASE_LOCKS, ResourceOwnerCreate(), ResourceOwnerDelete(), ResourceOwnerEnlarge(), ResourceOwnerRelease(), and ResourceOwnerRemember().

test_resowner_remember_between_phases()

Datum test_resowner_remember_between_phases

(

PG_FUNCTION_ARGS

)

Definition at line 165 of file test_resowner_basic.c.

{
    ResourceOwner resowner;
 
    resowner = ResourceOwnerCreate(CurrentResourceOwner, "TestOwner");
 
    ResourceOwnerRelease(resowner, RESOURCE_RELEASE_BEFORE_LOCKS, true, false);
 
    /*
     * Try to remember a new resource.  Fails because we already called
     * ResourceOwnerRelease.
     */
    ResourceOwnerEnlarge(resowner);
    ResourceOwnerRemember(resowner, CStringGetDatum("my string"), &string_desc);
 
    /* unreachable */
    elog(ERROR, "ResourceOwnerEnlarge should have errored out");
 
    PG_RETURN_VOID();
}

References CStringGetDatum(), CurrentResourceOwner, elog, ERROR, PG_RETURN_VOID, RESOURCE_RELEASE_BEFORE_LOCKS, ResourceOwnerCreate(), ResourceOwnerEnlarge(), ResourceOwnerRelease(), ResourceOwnerRemember(), and string_desc.

Variable Documentation

PG_MODULE_MAGIC

PG_MODULE_MAGIC

Definition at line 20 of file test_resowner_basic.c.

string_desc

const ResourceOwnerDesc string_desc

static

Initial value:

= {
    .name = "test resource",
    .release_phase = RESOURCE_RELEASE_AFTER_LOCKS,
    .release_priority = RELEASE_PRIO_FIRST,
    .ReleaseResource = ReleaseString,
    .DebugPrint = PrintString
}

Definition at line 29 of file test_resowner_basic.c.

Referenced by test_resowner_forget_between_phases(), test_resowner_leak(), and test_resowner_remember_between_phases().