Github user hbdeshmukh commented on a diff in the pull request:
https://github.com/apache/incubator-quickstep/pull/181#discussion_r104721479
--- Diff: query_optimizer/ExecutionGenerator.cpp ---
@@ -740,6 +805,202 @@ void ExecutionGenerator::convertHashJoin(const
P::HashJoinPtr &physical_plan) {
key_types.push_back(&left_attribute_type);
}
+ const CatalogRelationInfo *build_relation_info =
+ findRelationInfoOutputByPhysical(build_physical);
+ const CatalogRelationInfo *probe_operator_info =
+ findRelationInfoOutputByPhysical(probe_physical);
+
+ const CatalogRelation *build_relation = build_relation_info->relation;
+ const CatalogRelation *probe_relation = probe_operator_info->relation;
+
+ // FIXME(quickstep-team): Add support for self-join.
+ if (build_relation == probe_relation) {
+ THROW_SQL_ERROR() << "Self-join is not supported";
+ }
+
+ const PartitionScheme *build_partition_scheme =
build_relation->getPartitionScheme();
+ const PartitionScheme *probe_partition_scheme =
probe_relation->getPartitionScheme();
+
+ bool build_needs_repartition = false;
+ bool probe_needs_repartition = false;
+ bool needs_swap = false;
+ if (build_partition_scheme && probe_partition_scheme) {
+ const PartitionSchemeHeader &build_partition_scheme_header =
build_partition_scheme->getPartitionSchemeHeader();
+ const PartitionSchemeHeader &probe_partition_scheme_header =
probe_partition_scheme->getPartitionSchemeHeader();
+
+ switch (build_partition_scheme_header.getPartitionType()) {
+ case PartitionSchemeHeader::PartitionType::kRange:
+ build_needs_repartition = true;
+
+ switch (probe_partition_scheme_header.getPartitionType()) {
+ case PartitionSchemeHeader::PartitionType::kRange:
+ probe_needs_repartition = true;
+ break;
+ case PartitionSchemeHeader::PartitionType::kHash: {
+ const attribute_id probe_partition_attr =
probe_partition_scheme_header.getPartitionAttributeId();
+ if (find(probe_attribute_ids.begin(),
probe_attribute_ids.end(), probe_partition_attr) !=
+ probe_attribute_ids.end()) {
+ needs_swap = true;
+ } else {
+ probe_needs_repartition = true;
+ }
+ break;
+ }
+ }
+ break;
+ case PartitionSchemeHeader::PartitionType::kHash: {
+ const attribute_id build_partition_attr =
build_partition_scheme_header.getPartitionAttributeId();
+ if (find(build_attribute_ids.begin(), build_attribute_ids.end(),
build_partition_attr) !=
+ build_attribute_ids.end()) {
+ // BuildRelation has an useful partition.
+ switch (probe_partition_scheme_header.getPartitionType()) {
+ case PartitionSchemeHeader::PartitionType::kRange:
+ probe_needs_repartition = true;
+ break;
+ case PartitionSchemeHeader::PartitionType::kHash: {
--- End diff --
Are you sure that the compiler will throw an error saying that we didn't
handle all the cases in the switch statement? I would recommend a safe approach
where we add a default case that throws a fatal error.
---
If your project is set up for it, you can reply to this email and have your
reply appear on GitHub as well. If your project does not have this feature
enabled and wishes so, or if the feature is enabled but not working, please
contact infrastructure at infrastruct...@apache.org or file a JIRA ticket
with INFRA.
---
