Skip to content

Commit 50db596

Browse files
MasaoFujiiMasaoFujii
committed
Move codes for pg_backend_memory_contexts from mmgr/mcxt.c to adt/mcxtfuncs.c.
Previously the codes for pg_backend_memory_contexts were in src/backend/utils/mmgr/mcxt.c. This commit moves them to src/backend/utils/adt/mcxtfuncs.c so that mcxt.c basically includes only the low-level interface for memory contexts. Author: Atsushi Torikoshi Reviewed-by: Michael Paquier, Fujii Masao Discussion: https://postgr.es/m/20200819135545.GC19121@paquier.xyz
1 parent 29dd6d8 commit 50db596

File tree

3 files changed

+158
-137
lines changed

3 files changed

+158
-137
lines changed

‎src/backend/utils/adt/Makefile‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,7 @@ OBJS = \
5757
lockfuncs.o \
5858
mac.o \
5959
mac8.o \
60+
mcxtfuncs.o \
6061
misc.o \
6162
name.o \
6263
network.o \

‎src/backend/utils/adt/mcxtfuncs.c‎

Lines changed: 157 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,157 @@
1+
/*-------------------------------------------------------------------------
2+
*
3+
* mcxtfuncs.c
4+
* Functions to show backend memory context.
5+
*
6+
* Portions Copyright (c) 1996-2020, PostgreSQL Global Development Group
7+
* Portions Copyright (c) 1994, Regents of the University of California
8+
*
9+
*
10+
* IDENTIFICATION
11+
* src/backend/utils/adt/mcxtfuncs.c
12+
*
13+
*-------------------------------------------------------------------------
14+
*/
15+
16+
#include "postgres.h"
17+
18+
#include "funcapi.h"
19+
#include "miscadmin.h"
20+
#include "mb/pg_wchar.h"
21+
#include "utils/builtins.h"
22+
23+
/* ----------
24+
* The max bytes for showing identifiers of MemoryContext.
25+
* ----------
26+
*/
27+
#define MEMORY_CONTEXT_IDENT_DISPLAY_SIZE 1024
28+
29+
/*
30+
* PutMemoryContextsStatsTupleStore
31+
* One recursion level for pg_get_backend_memory_contexts.
32+
*/
33+
static void
34+
PutMemoryContextsStatsTupleStore(Tuplestorestate *tupstore,
35+
TupleDesc tupdesc, MemoryContext context,
36+
const char *parent, int level)
37+
{
38+
#define PG_GET_BACKEND_MEMORY_CONTEXTS_COLS 9
39+
40+
Datum values[PG_GET_BACKEND_MEMORY_CONTEXTS_COLS];
41+
bool nulls[PG_GET_BACKEND_MEMORY_CONTEXTS_COLS];
42+
MemoryContextCounters stat;
43+
MemoryContext child;
44+
const char *name;
45+
const char *ident;
46+
47+
AssertArg(MemoryContextIsValid(context));
48+
49+
name = context->name;
50+
ident = context->ident;
51+
52+
/*
53+
* To be consistent with logging output, we label dynahash contexts
54+
* with just the hash table name as with MemoryContextStatsPrint().
55+
*/
56+
if (ident && strcmp(name, "dynahash") == 0)
57+
{
58+
name = ident;
59+
ident = NULL;
60+
}
61+
62+
/* Examine the context itself */
63+
memset(&stat, 0, sizeof(stat));
64+
(*context->methods->stats) (context, NULL, (void *) &level, &stat);
65+
66+
memset(values, 0, sizeof(values));
67+
memset(nulls, 0, sizeof(nulls));
68+
69+
if (name)
70+
values[0] = CStringGetTextDatum(name);
71+
else
72+
nulls[0] = true;
73+
74+
if (ident)
75+
{
76+
int idlen = strlen(ident);
77+
char clipped_ident[MEMORY_CONTEXT_IDENT_DISPLAY_SIZE];
78+
79+
/*
80+
* Some identifiers such as SQL query string can be very long,
81+
* truncate oversize identifiers.
82+
*/
83+
if (idlen >= MEMORY_CONTEXT_IDENT_DISPLAY_SIZE)
84+
idlen = pg_mbcliplen(ident, idlen, MEMORY_CONTEXT_IDENT_DISPLAY_SIZE - 1);
85+
86+
memcpy(clipped_ident, ident, idlen);
87+
clipped_ident[idlen] = '\0';
88+
values[1] = CStringGetTextDatum(clipped_ident);
89+
}
90+
else
91+
nulls[1] = true;
92+
93+
if (parent)
94+
values[2] = CStringGetTextDatum(parent);
95+
else
96+
nulls[2] = true;
97+
98+
values[3] = Int32GetDatum(level);
99+
values[4] = Int64GetDatum(stat.totalspace);
100+
values[5] = Int64GetDatum(stat.nblocks);
101+
values[6] = Int64GetDatum(stat.freespace);
102+
values[7] = Int64GetDatum(stat.freechunks);
103+
values[8] = Int64GetDatum(stat.totalspace - stat.freespace);
104+
tuplestore_putvalues(tupstore, tupdesc, values, nulls);
105+
106+
for (child = context->firstchild; child != NULL; child = child->nextchild)
107+
{
108+
PutMemoryContextsStatsTupleStore(tupstore, tupdesc,
109+
child, name, level + 1);
110+
}
111+
}
112+
113+
/*
114+
* pg_get_backend_memory_contexts
115+
* SQL SRF showing backend memory context.
116+
*/
117+
Datum
118+
pg_get_backend_memory_contexts(PG_FUNCTION_ARGS)
119+
{
120+
ReturnSetInfo *rsinfo = (ReturnSetInfo *) fcinfo->resultinfo;
121+
TupleDesc tupdesc;
122+
Tuplestorestate *tupstore;
123+
MemoryContext per_query_ctx;
124+
MemoryContext oldcontext;
125+
126+
/* check to see if caller supports us returning a tuplestore */
127+
if (rsinfo == NULL || !IsA(rsinfo, ReturnSetInfo))
128+
ereport(ERROR,
129+
(errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
130+
errmsg("set-valued function called in context that cannot accept a set")));
131+
if (!(rsinfo->allowedModes & SFRM_Materialize))
132+
ereport(ERROR,
133+
(errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
134+
errmsg("materialize mode required, but it is not allowed in this context")));
135+
136+
/* Build a tuple descriptor for our result type */
137+
if (get_call_result_type(fcinfo, NULL, &tupdesc) != TYPEFUNC_COMPOSITE)
138+
elog(ERROR, "return type must be a row type");
139+
140+
per_query_ctx = rsinfo->econtext->ecxt_per_query_memory;
141+
oldcontext = MemoryContextSwitchTo(per_query_ctx);
142+
143+
tupstore = tuplestore_begin_heap(true, false, work_mem);
144+
rsinfo->returnMode = SFRM_Materialize;
145+
rsinfo->setResult = tupstore;
146+
rsinfo->setDesc = tupdesc;
147+
148+
MemoryContextSwitchTo(oldcontext);
149+
150+
PutMemoryContextsStatsTupleStore(tupstore, tupdesc,
151+
TopMemoryContext, NULL, 0);
152+
153+
/* clean up and return the tuplestore */
154+
tuplestore_donestoring(tupstore);
155+
156+
return (Datum) 0;
157+
}

‎src/backend/utils/mmgr/mcxt.c‎

Lines changed: 0 additions & 137 deletions
Original file line numberDiff line numberDiff line change
@@ -21,10 +21,8 @@
2121

2222
#include "postgres.h"
2323

24-
#include "funcapi.h"
2524
#include "mb/pg_wchar.h"
2625
#include "miscadmin.h"
27-
#include "utils/builtins.h"
2826
#include "utils/memdebug.h"
2927
#include "utils/memutils.h"
3028

@@ -69,11 +67,6 @@ static void MemoryContextStatsPrint(MemoryContext context, void *passthru,
6967
#define AssertNotInCriticalSection(context) \
7068
Assert(CritSectionCount == 0 || (context)->allowInCritSection)
7169

72-
/* ----------
73-
* The max bytes for showing identifiers of MemoryContext.
74-
* ----------
75-
*/
76-
#define MEMORY_CONTEXT_IDENT_DISPLAY_SIZE 1024
7770

7871
/*****************************************************************************
7972
* EXPORTED ROUTINES *
@@ -1228,133 +1221,3 @@ pchomp(const char *in)
12281221
n--;
12291222
return pnstrdup(in, n);
12301223
}
1231-
1232-
/*
1233-
* PutMemoryContextsStatsTupleStore
1234-
* One recursion level for pg_get_backend_memory_contexts.
1235-
*/
1236-
static void
1237-
PutMemoryContextsStatsTupleStore(Tuplestorestate *tupstore,
1238-
TupleDesc tupdesc, MemoryContext context,
1239-
const char *parent, int level)
1240-
{
1241-
#define PG_GET_BACKEND_MEMORY_CONTEXTS_COLS 9
1242-
1243-
Datum values[PG_GET_BACKEND_MEMORY_CONTEXTS_COLS];
1244-
bool nulls[PG_GET_BACKEND_MEMORY_CONTEXTS_COLS];
1245-
MemoryContextCounters stat;
1246-
MemoryContext child;
1247-
const char *name;
1248-
const char *ident;
1249-
1250-
AssertArg(MemoryContextIsValid(context));
1251-
1252-
name = context->name;
1253-
ident = context->ident;
1254-
1255-
/*
1256-
* To be consistent with logging output, we label dynahash contexts
1257-
* with just the hash table name as with MemoryContextStatsPrint().
1258-
*/
1259-
if (ident && strcmp(name, "dynahash") == 0)
1260-
{
1261-
name = ident;
1262-
ident = NULL;
1263-
}
1264-
1265-
/* Examine the context itself */
1266-
memset(&stat, 0, sizeof(stat));
1267-
(*context->methods->stats) (context, NULL, (void *) &level, &stat);
1268-
1269-
memset(values, 0, sizeof(values));
1270-
memset(nulls, 0, sizeof(nulls));
1271-
1272-
if (name)
1273-
values[0] = CStringGetTextDatum(name);
1274-
else
1275-
nulls[0] = true;
1276-
1277-
if (ident)
1278-
{
1279-
int idlen = strlen(ident);
1280-
char clipped_ident[MEMORY_CONTEXT_IDENT_DISPLAY_SIZE];
1281-
1282-
/*
1283-
* Some identifiers such as SQL query string can be very long,
1284-
* truncate oversize identifiers.
1285-
*/
1286-
if (idlen >= MEMORY_CONTEXT_IDENT_DISPLAY_SIZE)
1287-
idlen = pg_mbcliplen(ident, idlen, MEMORY_CONTEXT_IDENT_DISPLAY_SIZE - 1);
1288-
1289-
memcpy(clipped_ident, ident, idlen);
1290-
clipped_ident[idlen] = '\0';
1291-
values[1] = CStringGetTextDatum(clipped_ident);
1292-
}
1293-
else
1294-
nulls[1] = true;
1295-
1296-
if (parent)
1297-
values[2] = CStringGetTextDatum(parent);
1298-
else
1299-
nulls[2] = true;
1300-
1301-
values[3] = Int32GetDatum(level);
1302-
values[4] = Int64GetDatum(stat.totalspace);
1303-
values[5] = Int64GetDatum(stat.nblocks);
1304-
values[6] = Int64GetDatum(stat.freespace);
1305-
values[7] = Int64GetDatum(stat.freechunks);
1306-
values[8] = Int64GetDatum(stat.totalspace - stat.freespace);
1307-
tuplestore_putvalues(tupstore, tupdesc, values, nulls);
1308-
1309-
for (child = context->firstchild; child != NULL; child = child->nextchild)
1310-
{
1311-
PutMemoryContextsStatsTupleStore(tupstore, tupdesc,
1312-
child, name, level + 1);
1313-
}
1314-
}
1315-
1316-
/*
1317-
* pg_get_backend_memory_contexts
1318-
* SQL SRF showing backend memory context.
1319-
*/
1320-
Datum
1321-
pg_get_backend_memory_contexts(PG_FUNCTION_ARGS)
1322-
{
1323-
ReturnSetInfo *rsinfo = (ReturnSetInfo *) fcinfo->resultinfo;
1324-
TupleDesc tupdesc;
1325-
Tuplestorestate *tupstore;
1326-
MemoryContext per_query_ctx;
1327-
MemoryContext oldcontext;
1328-
1329-
/* check to see if caller supports us returning a tuplestore */
1330-
if (rsinfo == NULL || !IsA(rsinfo, ReturnSetInfo))
1331-
ereport(ERROR,
1332-
(errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1333-
errmsg("set-valued function called in context that cannot accept a set")));
1334-
if (!(rsinfo->allowedModes & SFRM_Materialize))
1335-
ereport(ERROR,
1336-
(errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1337-
errmsg("materialize mode required, but it is not allowed in this context")));
1338-
1339-
/* Build a tuple descriptor for our result type */
1340-
if (get_call_result_type(fcinfo, NULL, &tupdesc) != TYPEFUNC_COMPOSITE)
1341-
elog(ERROR, "return type must be a row type");
1342-
1343-
per_query_ctx = rsinfo->econtext->ecxt_per_query_memory;
1344-
oldcontext = MemoryContextSwitchTo(per_query_ctx);
1345-
1346-
tupstore = tuplestore_begin_heap(true, false, work_mem);
1347-
rsinfo->returnMode = SFRM_Materialize;
1348-
rsinfo->setResult = tupstore;
1349-
rsinfo->setDesc = tupdesc;
1350-
1351-
MemoryContextSwitchTo(oldcontext);
1352-
1353-
PutMemoryContextsStatsTupleStore(tupstore, tupdesc,
1354-
TopMemoryContext, NULL, 0);
1355-
1356-
/* clean up and return the tuplestore */
1357-
tuplestore_donestoring(tupstore);
1358-
1359-
return (Datum) 0;
1360-
}

0 commit comments

Comments
 (0)