Support Table and Index partitioning in Babelfish

src/backend/access/common/reloptions.c

@@ -564,6 +564,8 @@ static void initialize_reloptions(void);
 static void parse_one_reloption(relopt_value *option, char *text_str,
 								int text_len, bool validate);
 
+pltsql_partitioned_table_reloptions_hook_type pltsql_partitioned_table_reloptions_hook = NULL;
+
 /*
  * Get the length of a string reloption (either default or the user-defined
  * value).  This is used for allocation purposes when building a set of
@@ -2001,10 +2003,16 @@ bytea *
 partitioned_table_reloptions(Datum reloptions, bool validate)
 {
 	if (validate && reloptions)
+	{
+		/* For babelfish partitioned table, allow usage of reloptions in TSQL dialect. */
+		if (pltsql_partitioned_table_reloptions_hook && pltsql_partitioned_table_reloptions_hook(reloptions))
+			return NULL;
+
 		ereport(ERROR,
 				errcode(ERRCODE_WRONG_OBJECT_TYPE),
 				errmsg("cannot specify storage parameters for a partitioned table"),
 				errhint("Specify storage parameters for its leaf partitions instead."));
+	}
 	return NULL;
 }
 

src/include/access/reloptions.h

@@ -243,5 +243,7 @@ extern bytea *index_reloptions(amoptions_function amoptions, Datum reloptions,
 extern bytea *attribute_reloptions(Datum reloptions, bool validate);
 extern bytea *tablespace_reloptions(Datum reloptions, bool validate);
 extern LOCKMODE AlterTableGetRelOptionsLockLevel(List *defList);
+typedef bool (*pltsql_partitioned_table_reloptions_hook_type)();
+extern PGDLLEXPORT pltsql_partitioned_table_reloptions_hook_type pltsql_partitioned_table_reloptions_hook;
 
 #endif							/* RELOPTIONS_H */

src/include/nodes/parsenodes.h

@@ -882,6 +882,7 @@ typedef struct PartitionSpec
 	PartitionStrategy strategy;
 	List	   *partParams;		/* List of PartitionElems */
 	int			location;		/* token location, or -1 if unknown */
+	char		*tsql_partition_scheme;		/* tsql partition scheme name */
 } PartitionSpec;
 
 /*