-

danolivo · danolivo · commit e8d526b93a50 · 2019-05-24T15:56:20.000+05:00
diff --git a/contrib/pg_exchange/expath.c b/contrib/pg_exchange/expath.c
@@ -0,0 +1,31 @@
+/*
+ * expath.c
+ *
+ */
+
+#include "postgres.h"
+
+#include "optimizer/pathnode.h"
+
+#include "expath.h"
+
+
+/*
+ * FDW paths and EXCHANGE paths are incompatible and can't be combined at a plan.
+ * We need to construct two non-intersecting path branches across all plan.
+ * Costs of this plans is not an indicator of path quality at intermediate
+ * stages of a plan building. We need bypass add_path() path checking procedure.
+ */
+void
+force_add_path(RelOptInfo *rel, Path *path)
+{
+	List *pathlist = rel->pathlist;
+
+	rel->pathlist = NIL;
+	rel->cheapest_parameterized_paths = NIL;
+	rel->cheapest_startup_path = rel->cheapest_total_path =
+											rel->cheapest_unique_path = NULL;
+	add_path(rel, path);
+	rel->pathlist = list_concat(rel->pathlist, pathlist);
+	set_cheapest(rel);
+}
diff --git a/contrib/pg_exchange/expath.h b/contrib/pg_exchange/expath.h
@@ -0,0 +1,13 @@
+/*
+ * expath.h
+ *
+ */
+
+#ifndef EXPATH_H_
+#define EXPATH_H_
+
+#include "nodes/relation.h"
+
+void force_add_path(RelOptInfo *rel, Path *path);
+
+#endif /* EXPATH_H_ */
diff --git a/contrib/pg_exchange/sbuf.c b/contrib/pg_exchange/sbuf.c
@@ -0,0 +1,14 @@
+/*
+ * sbuf.c
+ *
+ */
+
+#include "sbuf.h"
+
+void
+initTupleBuffer(TupleBuffer *tbuf, size_t mem_size)
+{
+	tbuf->curptr = &tbuf->data;
+	/* Will corrected before send to DMQ for 'trim tails' purpose. */
+	tbuf->size = mem_size;
+}
diff --git a/contrib/pg_exchange/sbuf.h b/contrib/pg_exchange/sbuf.h
@@ -0,0 +1,22 @@
+/*
+ * sbuf.h
+ *
+ */
+
+#ifndef SBUF_H_
+#define SBUF_H_
+
+#include "postgres.h"
+
+typedef struct TupleBuffer
+{
+	size_t size;
+	void *curptr;
+	char data[FLEXIBLE_ARRAY_MEMBER];
+} TupleBuffer;
+
+#define DEFAULT_TUPLEBUF_SIZE	(BLCKSZ * 2)
+
+extern void initTupleBuffer(TupleBuffer *tbuf, size_t mem_size);
+
+#endif /* SBUF_H_ */
