From 6af1ece5470db542d18c6dd7c6760ca18663b145 Mon Sep 17 00:00:00 2001
From: Onur Tirtir <onurcantirtir@gmail.com>
Date: Fri, 25 Dec 2020 10:25:16 +0300
Subject: [PATCH] Cover ADD COLUMN commands defining fkeys

---
 src/backend/distributed/commands/table.c | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/backend/distributed/commands/table.c b/src/backend/distributed/commands/table.c
index bde809b43..6079d6e95 100644
--- a/src/backend/distributed/commands/table.c
+++ b/src/backend/distributed/commands/table.c
@@ -1033,6 +1033,19 @@ AlterTableCmdAddsFKey(AlterTableCmd *command, Oid relationId)
 			return true;
 		}
 	}
+	else if (alterTableType == AT_AddColumn)
+	{
+		ColumnDef *columnDef = (ColumnDef *) command->def;
+		List *constraints = columnDef->constraints;
+		Constraint *constraint = NULL;
+		foreach_ptr(constraint, constraints)
+		{
+			if (constraint->contype == CONSTR_FOREIGN)
+			{
+				return true;
+			}
+		}
+	}
 
 	return false;
 }