Support local COPY ultility

knizhnik · knizhnik · commit 484e3d884d9a · 2017-08-10T12:25:23.000+03:00
diff --git a/contrib/mmts/multimaster.c b/contrib/mmts/multimaster.c
@@ -5441,6 +5441,7 @@ static void MtmProcessUtility(Node *parsetree, const char *queryString,
 			CopyStmt *copyStatement = (CopyStmt *) parsetree;
 			skipCommand = true;
 			if (copyStatement->is_from) {
+				ListCell *opt;
 				RangeVar *relation = copyStatement->relation;
 
 				if (relation != NULL)
@@ -5455,6 +5456,17 @@ static void MtmProcessUtility(Node *parsetree, const char *queryString,
 						heap_close(rel, ShareLock);
 					}
 				}
+
+				foreach(opt, copyStatement->options)
+				{
+					DefElem	*elem = lfirst(opt);
+					if (strcmp("local", elem->defname) == 0) {
+						MtmTx.isDistributed = false; /* Skip */
+						MtmTx.snapshot = INVALID_CSN;
+						MtmTx.containsDML = false;
+						break;
+					}
+				}
 			}
 		    case T_CreateFunctionStmt:
 		    {
diff --git a/src/backend/commands/copy.c b/src/backend/commands/copy.c
@@ -1191,7 +1191,7 @@ ProcessCopyOptions(CopyState cstate,
 						 errmsg("argument to option \"%s\" must be a valid encoding name",
 								defel->defname)));
 		}
-		else
+		else if (strcmp(defel->defname, "local") != 0)
 			ereport(ERROR,
 					(errcode(ERRCODE_SYNTAX_ERROR),
 					 errmsg("option \"%s\" not recognized",
diff --git a/src/backend/parser/gram.y b/src/backend/parser/gram.y
@@ -2856,6 +2856,10 @@ copy_opt_item:
 				{
 					$$ = makeDefElem("format", (Node *)makeString("csv"));
 				}
+			| LOCAL
+				{
+					$$ = makeDefElem("local", (Node *)makeInteger(TRUE));
+				}
 			| HEADER_P
 				{
 					$$ = makeDefElem("header", (Node *)makeInteger(TRUE));

Original file line number	Diff line number	Diff line change
`@@ -5441,6 +5441,7 @@ static void MtmProcessUtility(Node parsetree, const char queryString,`
`5441`	`5441`	`CopyStmt copyStatement = (CopyStmt ) parsetree;`
`5442`	`5442`	`skipCommand = true;`
`5443`	`5443`	`if (copyStatement->is_from) {`
	`5444`	`+ ListCell *opt;`
`5444`	`5445`	`RangeVar *relation = copyStatement->relation;`
`5445`	`5446`
`5446`	`5447`	`if (relation != NULL)`
`@@ -5455,6 +5456,17 @@ static void MtmProcessUtility(Node parsetree, const char queryString,`
`5455`	`5456`	`heap_close(rel, ShareLock);`
`5456`	`5457`	`}`
`5457`	`5458`	`}`
	`5459`	`+`
	`5460`	`+ foreach(opt, copyStatement->options)`
	`5461`	`+ {`
	`5462`	`+ DefElem *elem = lfirst(opt);`
	`5463`	`+ if (strcmp("local", elem->defname) == 0) {`
	`5464`	`+ MtmTx.isDistributed = false; /* Skip */`
	`5465`	`+ MtmTx.snapshot = INVALID_CSN;`
	`5466`	`+ MtmTx.containsDML = false;`
	`5467`	`+ break;`
	`5468`	`+ }`
	`5469`	`+ }`
`5458`	`5470`	`}`
`5459`	`5471`	`case T_CreateFunctionStmt:`
`5460`	`5472`	`{`
Original file line number	Diff line number	Diff line change
`@@ -1191,7 +1191,7 @@ ProcessCopyOptions(CopyState cstate,`
`1191`	`1191`	`errmsg("argument to option \"%s\" must be a valid encoding name",`
`1192`	`1192`	`defel->defname)));`
`1193`	`1193`	`}`
`1194`		`- else`
	`1194`	`+ else if (strcmp(defel->defname, "local") != 0)`
`1195`	`1195`	`ereport(ERROR,`
`1196`	`1196`	`(errcode(ERRCODE_SYNTAX_ERROR),`
`1197`	`1197`	`errmsg("option \"%s\" not recognized",`
Original file line number	Diff line number	Diff line change
`@@ -2856,6 +2856,10 @@ copy_opt_item:`
`2856`	`2856`	`{`
`2857`	`2857`	`$$ = makeDefElem("format", (Node *)makeString("csv"));`
`2858`	`2858`	`}`
	`2859`	`+ \| LOCAL`
	`2860`	`+ {`
	`2861`	`+ $$ = makeDefElem("local", (Node *)makeInteger(TRUE));`
	`2862`	`+ }`
`2859`	`2863`	`\| HEADER_P`
`2860`	`2864`	`{`
`2861`	`2865`	`$$ = makeDefElem("header", (Node *)makeInteger(TRUE));`