Refactor and cleanup of AbstractMappingPartitionAssigner