Fix building RUM after commit 9fa6f00b.

Alexander Korotkov · Alexander Korotkov · commit 002356c72631 · 2017-12-26T18:16:50.000+03:00
diff --git a/src/rum.h b/src/rum.h
@@ -1057,4 +1057,20 @@ extern Datum FunctionCall10Coll(FmgrInfo *flinfo, Oid collation,
 				   Datum arg6, Datum arg7, Datum arg8,
 				   Datum arg9, Datum arg10);
 
+/* PostgreSQL version-agnostic creation of memory context */
+#if PG_VERSION_NUM >= 110000
+	#define RumContextCreate(parent, name) \
+		AllocSetContextCreateExtended(parent, name, \
+									  MEMCONTEXT_COPY_NAME, \
+									  ALLOCSET_DEFAULT_MINSIZE, \
+									  ALLOCSET_DEFAULT_INITSIZE, \
+									  ALLOCSET_DEFAULT_MAXSIZE)
+#else
+	#define RumContextCreate(parent, name) \
+		AllocSetContextCreate(parent, name, \
+							  ALLOCSET_DEFAULT_MINSIZE, \
+							  ALLOCSET_DEFAULT_INITSIZE, \
+							  ALLOCSET_DEFAULT_MAXSIZE)
+#endif
+
 #endif   /* __RUM_H__ */
diff --git a/src/ruminsert.c b/src/ruminsert.c
@@ -633,17 +633,11 @@ rumbuild(Relation heap, Relation index, struct IndexInfo *indexInfo)
 	 * create a temporary memory context that is reset once for each tuple
 	 * inserted into the index
 	 */
-	buildstate.tmpCtx = AllocSetContextCreate(CurrentMemoryContext,
-											  "Rum build temporary context",
-											  ALLOCSET_DEFAULT_MINSIZE,
-											  ALLOCSET_DEFAULT_INITSIZE,
-											  ALLOCSET_DEFAULT_MAXSIZE);
-
-	buildstate.funcCtx = AllocSetContextCreate(CurrentMemoryContext,
-					 "Rum build temporary context for user-defined function",
-											   ALLOCSET_DEFAULT_MINSIZE,
-											   ALLOCSET_DEFAULT_INITSIZE,
-											   ALLOCSET_DEFAULT_MAXSIZE);
+	buildstate.tmpCtx = RumContextCreate(CurrentMemoryContext,
+										 "Rum build temporary context");
+
+	buildstate.funcCtx = RumContextCreate(CurrentMemoryContext,
+					 "Rum build temporary context for user-defined function");
 
 	buildstate.accum.rumstate = &buildstate.rumstate;
 	rumInitBA(&buildstate.accum);
@@ -813,11 +807,8 @@ ruminsert(Relation index, Datum *values, bool *isnull,
 	Datum		outerAddInfo = (Datum) 0;
 	bool		outerAddInfoIsNull = true;
 
-	insertCtx = AllocSetContextCreate(CurrentMemoryContext,
-									  "Rum insert temporary context",
-									  ALLOCSET_DEFAULT_MINSIZE,
-									  ALLOCSET_DEFAULT_INITSIZE,
-									  ALLOCSET_DEFAULT_MAXSIZE);
+	insertCtx = RumContextCreate(CurrentMemoryContext,
+								 "Rum insert temporary context");
 
 	oldCtx = MemoryContextSwitchTo(insertCtx);
 
diff --git a/src/rumscan.c b/src/rumscan.c
@@ -35,16 +35,10 @@ rumbeginscan(Relation rel, int nkeys, int norderbys)
 	so->firstCall = true;
 	so->totalentries = 0;
 	so->sortedEntries = NULL;
-	so->tempCtx = AllocSetContextCreate(CurrentMemoryContext,
-										"Rum scan temporary context",
-										ALLOCSET_DEFAULT_MINSIZE,
-										ALLOCSET_DEFAULT_INITSIZE,
-										ALLOCSET_DEFAULT_MAXSIZE);
-	so->keyCtx = AllocSetContextCreate(CurrentMemoryContext,
-									   "Rum scan key context",
-									   ALLOCSET_DEFAULT_MINSIZE,
-									   ALLOCSET_DEFAULT_INITSIZE,
-									   ALLOCSET_DEFAULT_MAXSIZE);
+	so->tempCtx = RumContextCreate(CurrentMemoryContext,
+								   "Rum scan temporary context");
+	so->keyCtx = RumContextCreate(CurrentMemoryContext,
+								  "Rum scan key context");
 
 	initRumState(&so->rumstate, scan->indexRelation);
 
diff --git a/src/rumsort.c b/src/rumsort.c
@@ -877,11 +877,7 @@ rum_tuplesort_begin_common(int workMem, bool randomAccess)
 	 * Create a working memory context for this sort operation. All data
 	 * needed by the sort will live inside this context.
 	 */
-	sortcontext = AllocSetContextCreate(CurrentMemoryContext,
-										"TupleSort",
-										ALLOCSET_DEFAULT_MINSIZE,
-										ALLOCSET_DEFAULT_INITSIZE,
-										ALLOCSET_DEFAULT_MAXSIZE);
+	sortcontext = RumContextCreate(CurrentMemoryContext, "TupleSort");
 
 	/*
 	 * Make the Tuplesortstate within the per-sort context.  This way, we
