refactoring for PartitionFilter custom node (extract ResultRelInfo cache), introduce copy_stmt_hooking subsystem

Author: funbringer

Makefile

@@ -3,7 +3,8 @@
 MODULE_big = pg_pathman
 OBJS = src/init.o src/relation_info.o src/utils.o src/partition_filter.o src/runtimeappend.o \
 	src/runtime_merge_append.o src/pg_pathman.o src/dsm_array.o src/rangeset.o src/pl_funcs.o \
-	src/pathman_workers.o src/hooks.o src/nodes_common.o src/xact_handling.o $(WIN32RES)
+	src/pathman_workers.o src/hooks.o src/nodes_common.o src/xact_handling.o src/copy_stmt_hooking.o \
+	$(WIN32RES)
 
 EXTENSION = pg_pathman
 EXTVERSION = 1.0

src/copy_stmt_hooking.c

@@ -0,0 +1,33 @@
+#include "copy_stmt_hooking.h"
+#include "relation_info.h"
+
+#include "catalog/namespace.h"
+#include "commands/copy.h"
+
+
+/*
+ * Is pg_pathman supposed to handle this COPY stmt?
+ */
+bool
+is_pathman_related_copy(Node *parsetree)
+{
+	CopyStmt   *copy_stmt = (CopyStmt *) parsetree;
+	Oid			partitioned_table;
+
+	/* Check that it's a CopyStmt */
+	if (!IsA(parsetree, CopyStmt))
+		return false;
+
+	/* Also check that stmt->relation exists */
+	if (!copy_stmt->relation)
+		return false;
+
+	/* TODO: select appropriate lock for COPY */
+	partitioned_table = RangeVarGetRelid(copy_stmt->relation, NoLock, false);
+
+	/* Check that relation is partitioned */
+	if (get_pathman_relation_info(partitioned_table))
+		return true;
+
+	return false;
+}

src/copy_stmt_hooking.h

@@ -0,0 +1,12 @@
+#ifndef COPY_STMT_HOOKING_H
+#define COPY_STMT_HOOKING_H
+
+
+#include "postgres.h"
+#include "commands/copy.h"
+#include "nodes/nodes.h"
+
+
+bool is_pathman_related_copy(Node *parsetree);
+
+#endif

src/hooks.c

@@ -8,6 +8,7 @@
  * ------------------------------------------------------------------------
  */
 
+#include "copy_stmt_hooking.h"
 #include "hooks.h"
 #include "init.h"
 #include "partition_filter.h"
@@ -27,6 +28,7 @@ set_rel_pathlist_hook_type		set_rel_pathlist_hook_next = NULL;
 planner_hook_type				planner_hook_next = NULL;
 post_parse_analyze_hook_type	post_parse_analyze_hook_next = NULL;
 shmem_startup_hook_type			shmem_startup_hook_next = NULL;
+ProcessUtility_hook_type		process_utility_hook_next = NULL;
 
 
 /* Take care of joins */
@@ -574,3 +576,32 @@ pathman_relcache_hook(Datum arg, Oid relid)
 			break;
 	}
 }
+
+/*
+ * Utility function invoker hook.
+ */
+void
+pathman_process_utility_hook(Node *parsetree,
+							 const char *queryString,
+							 ProcessUtilityContext context,
+							 ParamListInfo params,
+							 DestReceiver *dest,
+							 char *completionTag)
+{
+	/* Call hooks set by other extensions */
+	if (process_utility_hook_next)
+		process_utility_hook_next(parsetree, queryString,
+								  context, params,
+								  dest, completionTag);
+
+	/* Override standard COPY statements if needed */
+	if (is_pathman_related_copy(parsetree))
+	{
+		elog(INFO, "copy!");
+	}
+
+	/* Call internal implementation */
+	standard_ProcessUtility(parsetree, queryString,
+							context, params,
+							dest, completionTag);
+}

src/hooks.h

@@ -16,13 +16,15 @@
 #include "optimizer/paths.h"
 #include "parser/analyze.h"
 #include "storage/ipc.h"
+#include "tcop/utility.h"
 
 
 extern set_join_pathlist_hook_type		set_join_pathlist_next;
 extern set_rel_pathlist_hook_type		set_rel_pathlist_hook_next;
 extern planner_hook_type				planner_hook_next;
 extern post_parse_analyze_hook_type		post_parse_analyze_hook_next;
 extern shmem_startup_hook_type			shmem_startup_hook_next;
+extern ProcessUtility_hook_type			process_utility_hook_next;
 
 
 void pathman_join_pathlist_hook(PlannerInfo *root,
@@ -50,4 +52,11 @@ void pathman_shmem_startup_hook(void);
 
 void pathman_relcache_hook(Datum arg, Oid relid);
 
+void pathman_process_utility_hook(Node *parsetree,
+								  const char *queryString,
+								  ProcessUtilityContext context,
+								  ParamListInfo params,
+								  DestReceiver *dest,
+								  char *completionTag);
+
 #endif