external
/
citus
mirror of https://github.com/citusdata/citus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Refactor EnsureLocalTableEmptyIfNecessary (#3830)

pull/3832/head
Onur Tirtir 2020-05-15 14:20:33 +03:00 committed by GitHub
parent 8f3373c702
commit 249550b815
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 40 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
src/backend/distributed/commands/create_distributed_table.c
View File

@ -102,7 +102,9 @@ static void EnsureTableNotDistributed(Oid relationId);
static Oid SupportFunctionForColumn(Var *partitionColumn, Oid accessMethodId,
int16 supportFunctionNumber);
static void EnsureLocalTableEmptyIfNecessary(Oid relationId, char distributionMethod,
bool viaDepracatedAPI);
bool viaDeprecatedAPI);
static bool ShouldLocalTableBeEmpty(Oid relationId, char distributionMethod, bool
viaDeprecatedAPI);
static bool LocalTableEmpty(Oid tableId);
static void CopyLocalDataIntoShards(Oid relationId);
static List * TupleDescColumnNameList(TupleDesc tupleDescriptor);
@ -849,33 +851,57 @@ EnsureTableCanBeColocatedWith(Oid relationId, char replicationModel,
/*
* EnsureLocalTableEmptyIfNecessary only checks for emptiness if only an empty
* relation can be distributed in given configuration.
*
* In some cases, it is possible and safe to send local data to shards while
* distributing the table. In those cases, we can distribute non-empty local
* tables. This function checks the distributionMethod and relation kind to
* see whether we need to be ensure emptiness of local table. If we need to
* be sure, this function calls EnsureLocalTableEmpty function to ensure
* that local table does not contain any data.
* EnsureLocalTableEmptyIfNecessary errors out if the function should be empty
* according to ShouldLocalTableBeEmpty but it is not.
*/
static void
EnsureLocalTableEmptyIfNecessary(Oid relationId, char distributionMethod,
bool viaDepracatedAPI)
bool viaDeprecatedAPI)
{
if (viaDepracatedAPI)
if (ShouldLocalTableBeEmpty(relationId, distributionMethod, viaDeprecatedAPI))
{
EnsureLocalTableEmpty(relationId);
}
}
/*
* ShouldLocalTableBeEmpty returns true if the local table should be empty
* before creating a citus table.
* In some cases, it is possible and safe to send local data to shards while
* distributing the table. In those cases, we can distribute non-empty local
* tables. This function checks the distributionMethod and relation kind to
* see whether we need to be ensure emptiness of local table.
*/
static bool
ShouldLocalTableBeEmpty(Oid relationId, char distributionMethod,
bool viaDeprecatedAPI)
{
bool shouldLocalTableBeEmpty = false;
if (viaDeprecatedAPI)
{
/* we don't support copying local data via deprecated API */
shouldLocalTableBeEmpty = true;
}
else if (distributionMethod != DISTRIBUTE_BY_HASH &&
distributionMethod != DISTRIBUTE_BY_NONE)
{
EnsureLocalTableEmpty(relationId);
/*
* We only support hash distributed tables and reference tables
* for initial data loading
*/
shouldLocalTableBeEmpty = true;
}
else if (!RegularTable(relationId))
{
EnsureLocalTableEmpty(relationId);
/*
* We only support tables and partitioned tables for initial
* data loading
*/
shouldLocalTableBeEmpty = true;
}
return shouldLocalTableBeEmpty;
}