protected void setSourcePartitionId( BaseNode source, BuildContext context, RuleBasePartitionId partitionId) { if (this.partitionId == partitionId) { return; } this.partitionId = partitionId; if (source.getPartitionId() == RuleBasePartitionId.MAIN_PARTITION) { setPartitionIdWithSinks(partitionId); } else { source.setPartitionId(context, partitionId); } }