Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

Commit2f5804f

Browse files
committed
refactoring for PartitionFilter custom node (extract ResultRelInfo cache), introduce copy_stmt_hooking subsystem
1 parent9476927 commit2f5804f

File tree

8 files changed

+284
-137
lines changed

8 files changed

+284
-137
lines changed

‎Makefile

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,8 @@
33
MODULE_big = pg_pathman
44
OBJS = src/init.o src/relation_info.o src/utils.o src/partition_filter.o src/runtimeappend.o\
55
src/runtime_merge_append.o src/pg_pathman.o src/dsm_array.o src/rangeset.o src/pl_funcs.o\
6-
src/pathman_workers.o src/hooks.o src/nodes_common.o src/xact_handling.o$(WIN32RES)
6+
src/pathman_workers.o src/hooks.o src/nodes_common.o src/xact_handling.o src/copy_stmt_hooking.o\
7+
$(WIN32RES)
78

89
EXTENSION = pg_pathman
910
EXTVERSION = 1.0

‎src/copy_stmt_hooking.c

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
#include"copy_stmt_hooking.h"
2+
#include"relation_info.h"
3+
4+
#include"catalog/namespace.h"
5+
#include"commands/copy.h"
6+
7+
8+
/*
9+
* Is pg_pathman supposed to handle this COPY stmt?
10+
*/
11+
bool
12+
is_pathman_related_copy(Node*parsetree)
13+
{
14+
CopyStmt*copy_stmt= (CopyStmt*)parsetree;
15+
Oidpartitioned_table;
16+
17+
/* Check that it's a CopyStmt */
18+
if (!IsA(parsetree,CopyStmt))
19+
return false;
20+
21+
/* Also check that stmt->relation exists */
22+
if (!copy_stmt->relation)
23+
return false;
24+
25+
/* TODO: select appropriate lock for COPY */
26+
partitioned_table=RangeVarGetRelid(copy_stmt->relation,NoLock, false);
27+
28+
/* Check that relation is partitioned */
29+
if (get_pathman_relation_info(partitioned_table))
30+
return true;
31+
32+
return false;
33+
}

‎src/copy_stmt_hooking.h

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
#ifndefCOPY_STMT_HOOKING_H
2+
#defineCOPY_STMT_HOOKING_H
3+
4+
5+
#include"postgres.h"
6+
#include"commands/copy.h"
7+
#include"nodes/nodes.h"
8+
9+
10+
boolis_pathman_related_copy(Node*parsetree);
11+
12+
#endif

‎src/hooks.c

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
* ------------------------------------------------------------------------
99
*/
1010

11+
#include"copy_stmt_hooking.h"
1112
#include"hooks.h"
1213
#include"init.h"
1314
#include"partition_filter.h"
@@ -27,6 +28,7 @@ set_rel_pathlist_hook_typeset_rel_pathlist_hook_next = NULL;
2728
planner_hook_typeplanner_hook_next=NULL;
2829
post_parse_analyze_hook_typepost_parse_analyze_hook_next=NULL;
2930
shmem_startup_hook_typeshmem_startup_hook_next=NULL;
31+
ProcessUtility_hook_typeprocess_utility_hook_next=NULL;
3032

3133

3234
/* Take care of joins */
@@ -574,3 +576,32 @@ pathman_relcache_hook(Datum arg, Oid relid)
574576
break;
575577
}
576578
}
579+
580+
/*
581+
* Utility function invoker hook.
582+
*/
583+
void
584+
pathman_process_utility_hook(Node*parsetree,
585+
constchar*queryString,
586+
ProcessUtilityContextcontext,
587+
ParamListInfoparams,
588+
DestReceiver*dest,
589+
char*completionTag)
590+
{
591+
/* Call hooks set by other extensions */
592+
if (process_utility_hook_next)
593+
process_utility_hook_next(parsetree,queryString,
594+
context,params,
595+
dest,completionTag);
596+
597+
/* Override standard COPY statements if needed */
598+
if (is_pathman_related_copy(parsetree))
599+
{
600+
elog(INFO,"copy!");
601+
}
602+
603+
/* Call internal implementation */
604+
standard_ProcessUtility(parsetree,queryString,
605+
context,params,
606+
dest,completionTag);
607+
}

‎src/hooks.h

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,13 +16,15 @@
1616
#include"optimizer/paths.h"
1717
#include"parser/analyze.h"
1818
#include"storage/ipc.h"
19+
#include"tcop/utility.h"
1920

2021

2122
externset_join_pathlist_hook_typeset_join_pathlist_next;
2223
externset_rel_pathlist_hook_typeset_rel_pathlist_hook_next;
2324
externplanner_hook_typeplanner_hook_next;
2425
externpost_parse_analyze_hook_typepost_parse_analyze_hook_next;
2526
externshmem_startup_hook_typeshmem_startup_hook_next;
27+
externProcessUtility_hook_typeprocess_utility_hook_next;
2628

2729

2830
voidpathman_join_pathlist_hook(PlannerInfo*root,
@@ -50,4 +52,11 @@ void pathman_shmem_startup_hook(void);
5052

5153
voidpathman_relcache_hook(Datumarg,Oidrelid);
5254

55+
voidpathman_process_utility_hook(Node*parsetree,
56+
constchar*queryString,
57+
ProcessUtilityContextcontext,
58+
ParamListInfoparams,
59+
DestReceiver*dest,
60+
char*completionTag);
61+
5362
#endif

0 commit comments

Comments
 (0)
[8]ページ先頭
©2009-2025 Movatter.jp