Exemplo n.º 1
0
    def __init__(self, context, options, workdir, analysis_tools,
                 compile_task_name, sources_predicate):
        super(JvmCompileGlobalStrategy,
              self).__init__(context, options, workdir, analysis_tools,
                             compile_task_name, sources_predicate)

        # Various working directories.
        # NB: These are grandfathered in with non-strategy-specific names, but to prevent
        # collisions within the buildcache, strategies should use strategy-specific subdirectories.
        self._analysis_dir = os.path.join(workdir, 'analysis')
        self._classes_dir = os.path.join(workdir, 'classes')

        self._analysis_file = os.path.join(self._analysis_dir,
                                           'global_analysis.valid')
        self._invalid_analysis_file = os.path.join(self._analysis_dir,
                                                   'global_analysis.invalid')

        self._target_sources_dir = os.path.join(workdir, 'target_sources')

        # The rough number of source files to build in each compiler pass.
        self._partition_size_hint = options.partition_size_hint

        # Set up dep checking if needed.
        def munge_flag(flag):
            flag_value = getattr(options, flag, None)
            return None if flag_value == 'off' else flag_value

        check_missing_deps = munge_flag('missing_deps')
        check_missing_direct_deps = munge_flag('missing_direct_deps')
        check_unnecessary_deps = munge_flag('unnecessary_deps')

        if check_missing_deps or check_missing_direct_deps or check_unnecessary_deps:
            target_whitelist = options.missing_deps_whitelist
            # Must init it here, so it can set requirements on the context.
            self._dep_analyzer = JvmDependencyAnalyzer(
                self.context, check_missing_deps, check_missing_direct_deps,
                check_unnecessary_deps, target_whitelist)
        else:
            self._dep_analyzer = None

        # Computed lazily as needed.
        self._upstream_class_to_path = None

        # If non-zero, and we have fewer than this number of locally-changed targets,
        # then we partition them separately, to preserve stability in the face of repeated
        # compilations.
        self._changed_targets_heuristic_limit = options.changed_targets_heuristic_limit

        # Sources (relative to buildroot) present in the last analysis that have since been deleted.
        # Populated in prepare_compile().
        self._deleted_sources = None
  def __init__(self, context, options, workdir, analysis_tools, language, sources_predicate):
    super(JvmCompileGlobalStrategy, self).__init__(context, options, workdir, analysis_tools,
                                                   language, sources_predicate)

    # Various working directories.
    # NB: These are grandfathered in with non-strategy-specific names, but to prevent
    # collisions within the buildcache, strategies should use strategy-specific subdirectories.
    self._analysis_dir = os.path.join(workdir, 'analysis')
    self._classes_dir = os.path.join(workdir, 'classes')

    self._delete_scratch = options.delete_scratch

    self._analysis_file = os.path.join(self._analysis_dir, 'global_analysis.valid')
    self._invalid_analysis_file = os.path.join(self._analysis_dir, 'global_analysis.invalid')

    self._target_sources_dir = os.path.join(workdir, 'target_sources')

    # A temporary, but well-known, dir in which to munge analysis/dependency files in before
    # caching. It must be well-known so we know where to find the files when we retrieve them from
    # the cache.
    self._analysis_tmpdir = os.path.join(self._analysis_dir, 'artifact_cache_tmpdir')

    # The rough number of source files to build in each compiler pass.
    self._partition_size_hint = options.partition_size_hint

    # Set up dep checking if needed.
    def munge_flag(flag):
      flag_value = getattr(options, flag, None)
      return None if flag_value == 'off' else flag_value

    check_missing_deps = munge_flag('missing_deps')
    check_missing_direct_deps = munge_flag('missing_direct_deps')
    check_unnecessary_deps = munge_flag('unnecessary_deps')

    if check_missing_deps or check_missing_direct_deps or check_unnecessary_deps:
      target_whitelist = options.missing_deps_whitelist
      # Must init it here, so it can set requirements on the context.
      self._dep_analyzer = JvmDependencyAnalyzer(self.context,
                                                 check_missing_deps,
                                                 check_missing_direct_deps,
                                                 check_unnecessary_deps,
                                                 target_whitelist)
    else:
      self._dep_analyzer = None

    # Computed lazily as needed.
    self._upstream_class_to_path = None

    # If non-zero, and we have fewer than this number of locally-changed targets,
    # then we partition them separately, to preserve stability in the face of repeated
    # compilations.
    self._changed_targets_heuristic_limit = options.changed_targets_heuristic_limit

    # Sources (relative to buildroot) present in the last analysis that have since been deleted.
    # Populated in prepare_compile().
    self._deleted_sources = None
Exemplo n.º 3
0
  def prepare(cls, options, round_manager):
    super(JvmCompile, cls).prepare(options, round_manager)

    # This task uses JvmDependencyAnalyzer as a helper, get its product needs
    JvmDependencyAnalyzer.prepare(options, round_manager)

    round_manager.require_data('compile_classpath')
    round_manager.require_data('ivy_resolve_symlink_map')

    # Require codegen we care about
    # TODO(John Sirois): roll this up in Task - if the list of labels we care about for a target
    # predicate to filter the full build graph is exposed, the requirement can be made automatic
    # and in turn codegen tasks could denote the labels they produce automating wiring of the
    # produce side
    round_manager.require_data('java')
    round_manager.require_data('scala')

    # Allow the deferred_sources_mapping to take place first
    round_manager.require_data('deferred_sources')
Exemplo n.º 4
0
    def prepare(cls, options, round_manager):
        super(JvmCompile, cls).prepare(options, round_manager)

        # This task uses JvmDependencyAnalyzer as a helper, get its product needs
        JvmDependencyAnalyzer.prepare(options, round_manager)

        round_manager.require_data('compile_classpath')
        round_manager.require_data('ivy_resolve_symlink_map')

        # Require codegen we care about
        # TODO(John Sirois): roll this up in Task - if the list of labels we care about for a target
        # predicate to filter the full build graph is exposed, the requirement can be made automatic
        # and in turn codegen tasks could denote the labels they produce automating wiring of the
        # produce side
        round_manager.require_data('java')
        round_manager.require_data('scala')

        # Allow the deferred_sources_mapping to take place first
        round_manager.require_data('deferred_sources')
Exemplo n.º 5
0
class JvmCompile(NailgunTaskBase, GroupMember, JvmToolTaskMixin):
    """A common framework for JVM compilation.

  To subclass for a specific JVM language, implement the static values and methods
  mentioned below under "Subclasses must implement".
  """

    @classmethod
    def setup_parser(cls, option_group, args, mkflag):
        super(JvmCompile, cls).setup_parser(option_group, args, mkflag)

        option_group.add_option(
            mkflag("warnings"),
            mkflag("warnings", negate=True),
            dest=cls._language + "_compile_warnings",
            default=True,
            action="callback",
            callback=mkflag.set_bool,
            help="[%default] Compile with all configured warnings enabled.",
        )

        option_group.add_option(
            mkflag("partition-size-hint"),
            dest=cls._language + "_partition_size_hint",
            action="store",
            type="int",
            default=-1,
            help="Roughly how many source files to attempt to compile together. "
            "Set to a large number to compile all sources together. Set this "
            "to 0 to compile target-by-target. Default is set in pants.ini.",
        )

        option_group.add_option(
            mkflag("missing-deps"),
            dest=cls._language + "_missing_deps",
            choices=["off", "warn", "fatal"],
            default="warn",
            help="[%default] One of off, warn, fatal. "
            "Check for missing dependencies in " + cls._language + "code. "
            "Reports actual dependencies A -> B where there is no "
            "transitive BUILD file dependency path from A to B."
            "If fatal, missing deps are treated as a build error.",
        )

        option_group.add_option(
            mkflag("missing-direct-deps"),
            dest=cls._language + "_missing_direct_deps",
            choices=["off", "warn", "fatal"],
            default="off",
            help="[%default] One of off, warn, fatal. "
            "Check for missing direct dependencies in "
            + cls._language
            + " code. Reports actual dependencies A -> B where there is no "
            "direct BUILD file dependency path from A to B. This is a very "
            "strict check, as in practice it is common to rely on transitive, "
            "non-direct dependencies, e.g., due to type inference or when the "
            "main target in a BUILD file is modified to depend on other "
            "targets in the same BUILD file as an implementation detail. It "
            "may still be useful to set it to fatal temorarily, to detect "
            "these.",
        )

        option_group.add_option(
            mkflag("unnecessary-deps"),
            dest=cls._language + "_unnecessary_deps",
            choices=["off", "warn", "fatal"],
            default="off",
            help="[%default] One of off, warn, fatal. Check for declared "
            "dependencies in " + cls._language + " code that are not "
            "needed. This is a very strict check. For example, generated code "
            "will often legitimately have BUILD dependencies that are unused "
            "in practice.",
        )

        option_group.add_option(
            mkflag("delete-scratch"),
            mkflag("delete-scratch", negate=True),
            dest=cls._language + "_delete_scratch",
            default=True,
            action="callback",
            callback=mkflag.set_bool,
            help="[%default] Leave intermediate scratch files around, " "for debugging build problems.",
        )

    # Subclasses must implement.
    # --------------------------
    _language = None
    _file_suffix = None
    _config_section = None

    @classmethod
    def name(cls):
        return cls._language

    @classmethod
    def product_types(cls):
        return ["classes_by_target", "classes_by_source"]

    def select(self, target):
        return target.has_sources(self._file_suffix)

    def create_analysis_tools(self):
        """Returns an AnalysisTools implementation.

    Subclasses must implement.
    """
        raise NotImplementedError()

    def compile(self, args, classpath, sources, classes_output_dir, analysis_file):
        """Invoke the compiler.

    Must raise TaskError on compile failure.

    Subclasses must implement."""
        raise NotImplementedError()

    # Subclasses may override.
    # ------------------------
    def extra_compile_time_classpath_elements(self):
        """Extra classpath elements common to all compiler invocations.

    E.g., jars for compiler plugins.
    """
        return []

    def extra_products(self, target):
        """Any extra, out-of-band products created for a target.

    E.g., targets that produce scala compiler plugins produce an info file.
    Returns a list of pairs (root, [absolute paths of files under root]).
    """
        return []

    def post_process(self, relevant_targets):
        """Any extra post-execute work."""
        pass

    # Common code.
    # ------------
    @staticmethod
    def _analysis_for_target(analysis_dir, target):
        return os.path.join(analysis_dir, target.id + ".analysis")

    @staticmethod
    def _portable_analysis_for_target(analysis_dir, target):
        return JvmCompile._analysis_for_target(analysis_dir, target) + ".portable"

    def _get_lang_specific_option(self, opt):
        full_opt_name = self._language + "_" + opt
        return getattr(self.context.options, full_opt_name, None)

    def __init__(self, *args, **kwargs):
        super(JvmCompile, self).__init__(*args, **kwargs)
        config_section = self.config_section

        # Global workdir.
        self._pants_workdir = self.context.config.getdefault("pants_workdir")

        # Various working directories.
        self._classes_dir = os.path.join(self.workdir, "classes")
        self._resources_dir = os.path.join(self.workdir, "resources")
        self._analysis_dir = os.path.join(self.workdir, "analysis")
        self._target_sources_dir = os.path.join(self.workdir, "target_sources")

        self._delete_scratch = self._get_lang_specific_option("delete_scratch")

        safe_mkdir(self._classes_dir)
        safe_mkdir(self._analysis_dir)
        safe_mkdir(self._target_sources_dir)

        self._analysis_file = os.path.join(self._analysis_dir, "global_analysis.valid")
        self._invalid_analysis_file = os.path.join(self._analysis_dir, "global_analysis.invalid")

        # A temporary, but well-known, dir in which to munge analysis/dependency files in before
        # caching. It must be well-known so we know where to find the files when we retrieve them from
        # the cache.
        self._analysis_tmpdir = os.path.join(self._analysis_dir, "artifact_cache_tmpdir")

        # We can't create analysis tools until after construction.
        self._lazy_analysis_tools = None

        # The rough number of source files to build in each compiler pass.
        self._partition_size_hint = self._get_lang_specific_option("partition_size_hint")
        if self._partition_size_hint == -1:
            self._partition_size_hint = self.context.config.getint(config_section, "partition_size_hint", default=1000)

        # JVM options for running the compiler.
        self._jvm_options = self.context.config.getlist(config_section, "jvm_args")

        # The ivy confs for which we're building.
        self._confs = self.context.config.getlist(config_section, "confs", default=["default"])

        # Set up dep checking if needed.
        def munge_flag(flag):
            return None if flag == "off" else flag

        check_missing_deps = munge_flag(self._get_lang_specific_option("missing_deps"))
        check_missing_direct_deps = munge_flag(self._get_lang_specific_option("missing_direct_deps"))
        check_unnecessary_deps = munge_flag(self._get_lang_specific_option("unnecessary_deps"))

        if check_missing_deps or check_missing_direct_deps or check_unnecessary_deps:
            target_whitelist = self.context.config.getlist("jvm", "missing_deps_target_whitelist", default=[])

            # Must init it here, so it can set requirements on the context.
            self._dep_analyzer = JvmDependencyAnalyzer(
                self.context, check_missing_deps, check_missing_direct_deps, check_unnecessary_deps, target_whitelist
            )
        else:
            self._dep_analyzer = None

        # If non-zero, and we have fewer than this number of locally-changed targets,
        # then we partition them separately, to preserve stability in the face of repeated
        # compilations.
        self._locally_changed_targets_heuristic_limit = self.context.config.getint(
            config_section, "locally_changed_targets_heuristic_limit", 0
        )

        self._upstream_class_to_path = None  # Computed lazily as needed.
        self.setup_artifact_cache_from_config(config_section=config_section)

        # Sources (relative to buildroot) present in the last analysis that have since been deleted.
        # Populated in prepare_execute().
        self._deleted_sources = None

        # Map of target -> list of sources (relative to buildroot), for all targets in all chunks.
        # Populated in prepare_execute().
        self._sources_by_target = None

    def configure_args(self, args_defaults=[], warning_defaults=[], no_warning_defaults=[]):
        """
   Setup the compiler command line arguments, optionally providing default values.  It is mandatory
   to call this from __init__() of your subclass.
   :param list args_default:  compiler flags that should be invoked for all invocations
   :param list warning_defaults: compiler flags to turn on warnings
   :param list no_warning_defaults:  compiler flags to turn off all warnings
   """
        self._args = self.context.config.getlist(self._config_section, "args", default=args_defaults)
        if self._get_lang_specific_option("compile_warnings"):
            self._args.extend(
                self.context.config.getlist(self._config_section, "warning_args", default=warning_defaults)
            )
        else:
            self._args.extend(
                self.context.config.getlist(self._config_section, "no_warning_args", default=no_warning_defaults)
            )

    def prepare(self, round_manager):
        # TODO(John Sirois): this is a fake requirement on 'ivy_jar_products' in order to force
        # resolve to run before this goal.  Require a new CompileClasspath product to be produced by
        # IvyResolve instead.
        round_manager.require_data("ivy_jar_products")
        round_manager.require_data("exclusives_groups")

        # Require codegen we care about
        # TODO(John Sirois): roll this up in Task - if the list of labels we care about for a target
        # predicate to filter the full build graph is exposed, the requirement can be made automatic
        # and in turn codegen tasks could denote the labels they produce automating wiring of the
        # produce side
        round_manager.require_data("java")
        round_manager.require_data("scala")

    def move(self, src, dst):
        if self._delete_scratch:
            shutil.move(src, dst)
        else:
            shutil.copy(src, dst)

    def _jvm_fingerprint_strategy(self):
        # Use a fingerprint strategy that allows us to also include java/scala versions.
        return JvmFingerprintStrategy(self.platform_version_info())

    def platform_version_info(self):
        """
    Provides extra platform information such as java version that will be used
    in the fingerprinter. This in turn ensures different platform versions create different
    cache artifacts.

    Sublclasses should override this and return a list of version info.
    """
        return None

    def pre_execute(self):
        # TODO(John Sirois): Ensuring requested product maps are available - if empty - should probably
        # be lifted to Task infra.

        # In case we have no relevant targets and return early create the requested product maps.
        self._create_empty_products()

    def prepare_execute(self, chunks):
        all_targets = list(itertools.chain(*chunks))

        # Update the classpath for us and for downstream tasks.
        egroups = self.context.products.get_data("exclusives_groups")
        all_group_ids = set()
        for t in all_targets:
            all_group_ids.add(egroups.get_group_key_for_target(t))

        for conf in self._confs:
            for group_id in all_group_ids:
                egroups.update_compatible_classpaths(group_id, [(conf, self._classes_dir)])
                egroups.update_compatible_classpaths(group_id, [(conf, self._resources_dir)])

        # Target -> sources (relative to buildroot).
        # TODO(benjy): Should sources_by_target be available in all Tasks?
        self._sources_by_target = self._compute_current_sources_by_target(all_targets)

        # Split the global analysis file into valid and invalid parts.
        cache_manager = self.create_cache_manager(
            invalidate_dependents=True, fingerprint_strategy=self._jvm_fingerprint_strategy()
        )
        invalidation_check = cache_manager.check(all_targets)
        if invalidation_check.invalid_vts:
            # The analysis for invalid and deleted sources is no longer valid.
            invalid_targets = [vt.target for vt in invalidation_check.invalid_vts]
            invalid_sources_by_target = {}
            for tgt in invalid_targets:
                invalid_sources_by_target[tgt] = self._sources_by_target[tgt]
            invalid_sources = list(itertools.chain.from_iterable(invalid_sources_by_target.values()))
            self._deleted_sources = self._compute_deleted_sources()

            self._ensure_analysis_tmpdir()
            tmpdir = os.path.join(self._analysis_tmpdir, str(uuid.uuid4()))
            os.mkdir(tmpdir)
            valid_analysis_tmp = os.path.join(tmpdir, "valid_analysis")
            newly_invalid_analysis_tmp = os.path.join(tmpdir, "newly_invalid_analysis")
            invalid_analysis_tmp = os.path.join(tmpdir, "invalid_analysis")
            if self._analysis_parser.is_nonempty_analysis(self._analysis_file):
                with self.context.new_workunit(name="prepare-analysis"):
                    self._analysis_tools.split_to_paths(
                        self._analysis_file,
                        [(invalid_sources + self._deleted_sources, newly_invalid_analysis_tmp)],
                        valid_analysis_tmp,
                    )
                    if self._analysis_parser.is_nonempty_analysis(self._invalid_analysis_file):
                        self._analysis_tools.merge_from_paths(
                            [self._invalid_analysis_file, newly_invalid_analysis_tmp], invalid_analysis_tmp
                        )
                    else:
                        invalid_analysis_tmp = newly_invalid_analysis_tmp

                    # Now it's OK to overwrite the main analysis files with the new state.
                    self.move(valid_analysis_tmp, self._analysis_file)
                    self.move(invalid_analysis_tmp, self._invalid_analysis_file)
        else:
            self._deleted_sources = []

    # TODO(benjy): Break this monstrosity up? Previous attempts to do so
    #              turned out to be more trouble than it was worth.
    def execute_chunk(self, relevant_targets):
        # TODO(benjy): Add a pre-execute goal for injecting deps into targets, so e.g.,
        # we can inject a dep on the scala runtime library and still have it ivy-resolve.

        if not relevant_targets:
            return

        # Get the exclusives group for the targets to compile.
        # Group guarantees that they'll be a single exclusives key for them.
        egroups = self.context.products.get_data("exclusives_groups")
        group_id = egroups.get_group_key_for_target(relevant_targets[0])

        # Get the classpath generated by upstream JVM tasks and our own prepare_execute().
        classpath = egroups.get_classpath_for_group(group_id)

        # Add any extra compile-time-only classpath elements.
        # TODO(benjy): Model compile-time vs. runtime classpaths more explicitly.
        for conf in self._confs:
            for jar in self.extra_compile_time_classpath_elements():
                classpath.insert(0, (conf, jar))

        # Target -> sources (relative to buildroot), for just this chunk's targets.
        sources_by_target = self._sources_for_targets(relevant_targets)

        # If needed, find targets that we've changed locally (as opposed to
        # changes synced in from the SCM).
        # TODO(benjy): Should locally_changed_targets be available in all Tasks?
        locally_changed_targets = None
        if self._locally_changed_targets_heuristic_limit:
            locally_changed_targets = self._find_locally_changed_targets(sources_by_target)
            if locally_changed_targets and len(locally_changed_targets) > self._locally_changed_targets_heuristic_limit:
                locally_changed_targets = None

        # Invalidation check. Everything inside the with block must succeed for the
        # invalid targets to become valid.
        with self.invalidated(
            relevant_targets,
            invalidate_dependents=True,
            partition_size_hint=self._partition_size_hint,
            locally_changed_targets=locally_changed_targets,
            fingerprint_strategy=self._jvm_fingerprint_strategy(),
        ) as invalidation_check:
            if invalidation_check.invalid_vts:
                # Find the invalid sources for this chunk.
                invalid_targets = [vt.target for vt in invalidation_check.invalid_vts]
                invalid_sources_by_target = self._sources_for_targets(invalid_targets)

                tmpdir = os.path.join(self._analysis_tmpdir, str(uuid.uuid4()))
                os.mkdir(tmpdir)

                # Register products for all the valid targets.
                # We register as we go, so dependency checking code can use this data.
                valid_targets = list(set(relevant_targets) - set(invalid_targets))
                self._register_products(valid_targets, self._analysis_file)

                # Figure out the sources and analysis belonging to each partition.
                partitions = []  # Each element is a triple (vts, sources_by_target, analysis).
                for vts in invalidation_check.invalid_vts_partitioned:
                    partition_tmpdir = os.path.join(tmpdir, Target.maybe_readable_identify(vts.targets))
                    os.mkdir(partition_tmpdir)
                    sources = list(
                        itertools.chain.from_iterable([invalid_sources_by_target.get(t, []) for t in vts.targets])
                    )
                    de_duped_sources = list(OrderedSet(sources))
                    if len(sources) != len(de_duped_sources):
                        counts = [(src, len(list(srcs))) for src, srcs in itertools.groupby(sorted(sources))]
                        self.context.log.warn(
                            "De-duped the following sources:\n\t%s"
                            % "\n\t".join(sorted("%d %s" % (cnt, src) for src, cnt in counts if cnt > 1))
                        )
                    analysis_file = os.path.join(partition_tmpdir, "analysis")
                    partitions.append((vts, de_duped_sources, analysis_file))

                # Split per-partition files out of the global invalid analysis.
                if self._analysis_parser.is_nonempty_analysis(self._invalid_analysis_file) and partitions:
                    with self.context.new_workunit(name="partition-analysis"):
                        splits = [(x[1], x[2]) for x in partitions]
                        # We have to pass the analysis for any deleted files through zinc, to give it
                        # a chance to delete the relevant class files.
                        if splits:
                            splits[0] = (splits[0][0] + self._deleted_sources, splits[0][1])
                        self._analysis_tools.split_to_paths(self._invalid_analysis_file, splits)

                # Now compile partitions one by one.
                for partition in partitions:
                    (vts, sources, analysis_file) = partition
                    cp_entries = [entry for conf, entry in classpath if conf in self._confs]
                    self._process_target_partition(partition, cp_entries)
                    # No exception was thrown, therefore the compile succeded and analysis_file is now valid.
                    if os.path.exists(analysis_file):  # The compilation created an analysis.
                        # Merge the newly-valid analysis with our global valid analysis.
                        new_valid_analysis = analysis_file + ".valid.new"
                        if self._analysis_parser.is_nonempty_analysis(self._analysis_file):
                            with self.context.new_workunit(name="update-upstream-analysis"):
                                self._analysis_tools.merge_from_paths(
                                    [self._analysis_file, analysis_file], new_valid_analysis
                                )
                        else:  # We need to keep analysis_file around. Background tasks may need it.
                            shutil.copy(analysis_file, new_valid_analysis)

                        # Move the merged valid analysis to its proper location.
                        # We do this before checking for missing dependencies, so that we can still
                        # enjoy an incremental compile after fixing missing deps.
                        self.move(new_valid_analysis, self._analysis_file)

                        # Update the products with the latest classes. Must happen before the
                        # missing dependencies check.
                        self._register_products(vts.targets, analysis_file)
                        if self._dep_analyzer:
                            # Check for missing dependencies.
                            actual_deps = self._analysis_parser.parse_deps_from_path(
                                analysis_file, lambda: self._compute_classpath_elements_by_class(cp_entries)
                            )
                            with self.context.new_workunit(name="find-missing-dependencies"):
                                self._dep_analyzer.check(sources, actual_deps)

                        # Kick off the background artifact cache write.
                        if self.artifact_cache_writes_enabled():
                            self._write_to_artifact_cache(analysis_file, vts, invalid_sources_by_target)

                    if self._analysis_parser.is_nonempty_analysis(self._invalid_analysis_file):
                        with self.context.new_workunit(name="trim-downstream-analysis"):
                            # Trim out the newly-valid sources from our global invalid analysis.
                            new_invalid_analysis = analysis_file + ".invalid.new"
                            discarded_invalid_analysis = analysis_file + ".invalid.discard"
                            self._analysis_tools.split_to_paths(
                                self._invalid_analysis_file,
                                [(sources, discarded_invalid_analysis)],
                                new_invalid_analysis,
                            )
                            self.move(new_invalid_analysis, self._invalid_analysis_file)

                    # Record the built target -> sources mapping for future use.
                    for target in vts.targets:
                        self._record_sources_by_target(target, sources_by_target.get(target, []))

                    # Now that all the analysis accounting is complete, and we have no missing deps,
                    # we can safely mark the targets as valid.
                    vts.update()
            else:
                # Nothing to build. Register products for all the targets in one go.
                self._register_products(relevant_targets, self._analysis_file)

        self.post_process(relevant_targets)

    def _process_target_partition(self, partition, classpath):
        """Needs invoking only on invalid targets.

    partition - a triple (vts, sources_by_target, analysis_file).
    classpath - a list of classpath entries.

    May be invoked concurrently on independent target sets.

    Postcondition: The individual targets in vts are up-to-date, as if each were
                   compiled individually.
    """
        (vts, sources, analysis_file) = partition

        if not sources:
            self.context.log.warn(
                "Skipping %s compile for targets with no sources:\n  %s" % (self._language, vts.targets)
            )
        else:
            # Do some reporting.
            self.context.log.info(
                "Compiling a partition containing ",
                items_to_report_element(sources, "source"),
                " in ",
                items_to_report_element([t.address.reference() for t in vts.targets], "target"),
                ".",
            )
            with self.context.new_workunit("compile"):
                # The compiler may delete classfiles, then later exit on a compilation error. Then if the
                # change triggering the error is reverted, we won't rebuild to restore the missing
                # classfiles. So we force-invalidate here, to be on the safe side.
                vts.force_invalidate()
                self.compile(self._args, classpath, sources, self._classes_dir, analysis_file)

    def check_artifact_cache(self, vts):
        # Special handling for scala analysis files. Class files are retrieved directly into their
        # final locations in the global classes dir.

        def post_process_cached_vts(cached_vts):
            # Get all the targets whose artifacts we found in the cache.
            cached_targets = []
            for vt in cached_vts:
                for target in vt.targets:
                    cached_targets.append(target)

            # The current global analysis may contain old data for modified targets for
            # which we got cache hits. We need to strip out this old analysis, to ensure
            # that the new data incoming from the cache doesn't collide with it during the merge.
            sources_to_strip = []
            if os.path.exists(self._analysis_file):
                for target in cached_targets:
                    sources_to_strip.extend(self._get_previous_sources_by_target(target))

            # Localize the cached analyses.
            analyses_to_merge = []
            for target in cached_targets:
                analysis_file = JvmCompile._analysis_for_target(self._analysis_tmpdir, target)
                portable_analysis_file = JvmCompile._portable_analysis_for_target(self._analysis_tmpdir, target)
                if os.path.exists(portable_analysis_file):
                    self._analysis_tools.localize(portable_analysis_file, analysis_file)
                if os.path.exists(analysis_file):
                    analyses_to_merge.append(analysis_file)

            # Merge them into the global analysis.
            if analyses_to_merge:
                with temporary_dir() as tmpdir:
                    if sources_to_strip:
                        throwaway = os.path.join(tmpdir, "throwaway")
                        trimmed_analysis = os.path.join(tmpdir, "trimmed")
                        self._analysis_tools.split_to_paths(
                            self._analysis_file, [(sources_to_strip, throwaway)], trimmed_analysis
                        )
                    else:
                        trimmed_analysis = self._analysis_file
                    if os.path.exists(trimmed_analysis):
                        analyses_to_merge.append(trimmed_analysis)
                    tmp_analysis = os.path.join(tmpdir, "analysis")
                    with self.context.new_workunit(name="merge_analysis"):
                        self._analysis_tools.merge_from_paths(analyses_to_merge, tmp_analysis)

                    sources_by_cached_target = self._sources_for_targets(cached_targets)

                    # Record the cached target -> sources mapping for future use.
                    for target, sources in sources_by_cached_target.items():
                        self._record_sources_by_target(target, sources)

                    # Everything's good so move the merged analysis to its final location.
                    if os.path.exists(tmp_analysis):
                        self.move(tmp_analysis, self._analysis_file)

        self._ensure_analysis_tmpdir()
        return self.do_check_artifact_cache(vts, post_process_cached_vts=post_process_cached_vts)

    def _write_to_artifact_cache(self, analysis_file, vts, sources_by_target):
        vt_by_target = dict([(vt.target, vt) for vt in vts.versioned_targets])

        split_analysis_files = [JvmCompile._analysis_for_target(self._analysis_tmpdir, t) for t in vts.targets]
        portable_split_analysis_files = [
            JvmCompile._portable_analysis_for_target(self._analysis_tmpdir, t) for t in vts.targets
        ]

        # Set up args for splitting the analysis into per-target files.
        splits = zip([sources_by_target.get(t, []) for t in vts.targets], split_analysis_files)
        splits_args_tuples = [(analysis_file, splits)]

        # Set up args for rebasing the splits.
        relativize_args_tuples = zip(split_analysis_files, portable_split_analysis_files)

        # Set up args for artifact cache updating.
        vts_artifactfiles_pairs = []
        classes_by_source = self._compute_classes_by_source(analysis_file)
        for target, sources in sources_by_target.items():
            artifacts = []
            for source in sources:
                artifacts.extend(classes_by_source.get(source, []))
            vt = vt_by_target.get(target)
            if vt is not None:
                # NOTE: analysis_file doesn't exist yet.
                vts_artifactfiles_pairs.append(
                    (vt, artifacts + [JvmCompile._portable_analysis_for_target(self._analysis_tmpdir, target)])
                )

        update_artifact_cache_work = self.get_update_artifact_cache_work(vts_artifactfiles_pairs)
        if update_artifact_cache_work:
            work_chain = [
                Work(self._analysis_tools.split_to_paths, splits_args_tuples, "split"),
                Work(self._analysis_tools.relativize, relativize_args_tuples, "relativize"),
                update_artifact_cache_work,
            ]
            self.context.submit_background_work_chain(work_chain, parent_workunit_name="cache")

    def _compute_classes_by_source(self, analysis_file=None):
        """Compute src->classes.

    Srcs are relative to buildroot. Classes are absolute paths.
    """
        if analysis_file is None:
            analysis_file = self._analysis_file

        if not os.path.exists(analysis_file):
            return {}
        buildroot = get_buildroot()
        products = self._analysis_parser.parse_products_from_path(analysis_file)
        classes_by_src = {}
        for src, classes in products.items():
            relsrc = os.path.relpath(src, buildroot)
            classes_by_src[relsrc] = classes
        return classes_by_src

    def _compute_deleted_sources(self):
        """Computes the list of sources present in the last analysis that have since been deleted.

    This is a global list. We have no way of associating them to individual targets.
    Paths are relative to buildroot.
    """
        with self.context.new_workunit("find-deleted-sources"):
            if os.path.exists(self._analysis_file):
                products = self._analysis_parser.parse_products_from_path(self._analysis_file)
                buildroot = get_buildroot()
                old_srcs = products.keys()  # Absolute paths.
                return [os.path.relpath(src, buildroot) for src in old_srcs if not os.path.exists(src)]
            else:
                return []

    def _get_previous_sources_by_target(self, target):
        """Returns the target's sources as recorded on the last successful build of target.

    Returns a list of absolute paths.
    """
        path = os.path.join(self._target_sources_dir, target.identifier)
        if os.path.exists(path):
            with open(path, "r") as infile:
                return [s.rstrip() for s in infile.readlines()]
        else:
            return []

    def _record_sources_by_target(self, target, sources):
        # Record target -> source mapping for future use.
        with open(os.path.join(self._target_sources_dir, target.identifier), "w") as outfile:
            for src in sources:
                outfile.write(os.path.join(get_buildroot(), src))
                outfile.write("\n")

    def _compute_current_sources_by_target(self, targets):
        """Returns map target -> list of sources (relative to buildroot)."""

        def calculate_sources(target):
            sources = [s for s in target.sources_relative_to_buildroot() if s.endswith(self._file_suffix)]
            # TODO: Make this less hacky. Ideally target.java_sources will point to sources, not targets.
            if hasattr(target, "java_sources") and target.java_sources:
                sources.extend(self._resolve_target_sources(target.java_sources, ".java"))
            return sources

        return dict([(t, calculate_sources(t)) for t in targets])

    def _find_locally_changed_targets(self, sources_by_target):
        """Finds the targets whose sources have been modified locally.

    Returns a list of targets, or None if no SCM is available.
    """
        # Compute the src->targets mapping. There should only be one target per source,
        # but that's not yet a hard requirement, so the value is a list of targets.
        # TODO(benjy): Might this inverse mapping be needed elsewhere too?
        targets_by_source = defaultdict(list)
        for tgt, srcs in sources_by_target.items():
            for src in srcs:
                targets_by_source[src].append(tgt)

        ret = OrderedSet()
        scm = get_scm()
        if not scm:
            return None
        changed_files = scm.changed_files(include_untracked=True, relative_to=get_buildroot())
        for f in changed_files:
            ret.update(targets_by_source.get(f, []))
        return list(ret)

    def _resolve_target_sources(self, target_sources, extension=None):
        """Given a list of pants targets, extract their sources as a list.

    Filters against the extension if given and optionally returns the paths relative to the target
    base.
    """
        resolved_sources = []
        for target in target_sources:
            if target.has_sources():
                resolved_sources.extend(target.sources_relative_to_buildroot())
        return resolved_sources

    def _compute_classpath_elements_by_class(self, classpath):
        # Don't consider loose classes dirs in our classes dir. Those will be considered
        # separately, by looking at products.
        def non_product(path):
            return path != self._classes_dir

        if self._upstream_class_to_path is None:
            self._upstream_class_to_path = {}
            classpath_entries = filter(non_product, classpath)
            for cp_entry in self.find_all_bootstrap_jars() + classpath_entries:
                # Per the classloading spec, a 'jar' in this context can also be a .zip file.
                if os.path.isfile(cp_entry) and ((cp_entry.endswith(".jar") or cp_entry.endswith(".zip"))):
                    with open_zip(cp_entry, "r") as jar:
                        for cls in jar.namelist():
                            # First jar with a given class wins, just like when classloading.
                            if cls.endswith(b".class") and not cls in self._upstream_class_to_path:
                                self._upstream_class_to_path[cls] = cp_entry
                elif os.path.isdir(cp_entry):
                    for dirpath, _, filenames in os.walk(cp_entry, followlinks=True):
                        for f in filter(lambda x: x.endswith(".class"), filenames):
                            cls = os.path.relpath(os.path.join(dirpath, f), cp_entry)
                            if not cls in self._upstream_class_to_path:
                                self._upstream_class_to_path[cls] = os.path.join(dirpath, f)
        return self._upstream_class_to_path

    def find_all_bootstrap_jars(self):
        def get_path(key):
            return self.context.java_sysprops.get(key, "").split(":")

        def find_jars_in_dirs(dirs):
            ret = []
            for d in dirs:
                if os.path.isdir(d):
                    ret.extend(filter(lambda s: s.endswith(".jar"), os.listdir(d)))
            return ret

        # Note: assumes HotSpot, or some JVM that supports sun.boot.class.path.
        # TODO: Support other JVMs? Not clear if there's a standard way to do so.
        # May include loose classes dirs.
        boot_classpath = get_path("sun.boot.class.path")

        # Note that per the specs, overrides and extensions must be in jars.
        # Loose class files will not be found by the JVM.
        override_jars = find_jars_in_dirs(get_path("java.endorsed.dirs"))
        extension_jars = find_jars_in_dirs(get_path("java.ext.dirs"))

        # Note that this order matters: it reflects the classloading order.
        bootstrap_jars = filter(os.path.isfile, override_jars + boot_classpath + extension_jars)
        return bootstrap_jars  # Technically, may include loose class dirs from boot_classpath.

    @property
    def _analysis_tools(self):
        if self._lazy_analysis_tools is None:
            self._lazy_analysis_tools = self.create_analysis_tools()
        return self._lazy_analysis_tools

    @property
    def _analysis_parser(self):
        return self._analysis_tools.parser

    def _sources_for_targets(self, targets):
        """Returns a map target->sources for the specified targets."""
        if self._sources_by_target is None:
            raise TaskError("self._sources_by_target not computed yet.")
        return dict((t, self._sources_by_target.get(t, [])) for t in targets)

    # Work in a tmpdir so we don't stomp the main analysis files on error.
    # The tmpdir is cleaned up in a shutdown hook, because background work
    # may need to access files we create there even after this method returns.
    def _ensure_analysis_tmpdir(self):
        # Do this lazily, so we don't trigger creation of a worker pool unless we need it.
        if not os.path.exists(self._analysis_tmpdir):
            os.makedirs(self._analysis_tmpdir)
            if self._delete_scratch:
                self.context.background_worker_pool().add_shutdown_hook(lambda: safe_rmtree(self._analysis_tmpdir))

    def _create_empty_products(self):
        make_products = lambda: defaultdict(MultipleRootedProducts)
        if self.context.products.is_required_data("classes_by_source"):
            self.context.products.safe_create_data("classes_by_source", make_products)
        if self.context.products.is_required_data("classes_by_target"):
            self.context.products.safe_create_data("classes_by_target", make_products)
        if self.context.products.is_required_data("resources_by_target"):
            self.context.products.safe_create_data("resources_by_target", make_products)

    def _register_products(self, targets, analysis_file):
        classes_by_source = self.context.products.get_data("classes_by_source")
        classes_by_target = self.context.products.get_data("classes_by_target")
        resources_by_target = self.context.products.get_data("resources_by_target")

        if classes_by_source is not None or classes_by_target is not None:
            computed_classes_by_source = self._compute_classes_by_source(analysis_file)
            for target in targets:
                target_products = classes_by_target[target] if classes_by_target is not None else None
                for source in self._sources_by_target.get(target, []):  # Source is relative to buildroot.
                    classes = computed_classes_by_source.get(source, [])  # Classes are absolute paths.
                    if classes_by_target is not None:
                        target_products.add_abs_paths(self._classes_dir, classes)
                    if classes_by_source is not None:
                        classes_by_source[source].add_abs_paths(self._classes_dir, classes)

        # TODO(pl): https://github.com/pantsbuild/pants/issues/206
        if resources_by_target is not None:
            for target in targets:
                target_resources = resources_by_target[target]
                for root, abs_paths in self.extra_products(target):
                    target_resources.add_abs_paths(root, abs_paths)
Exemplo n.º 6
0
    def __init__(self, *args, **kwargs):
        super(JvmCompile, self).__init__(*args, **kwargs)
        config_section = self.config_section

        # Global workdir.
        self._pants_workdir = self.context.config.getdefault("pants_workdir")

        # Various working directories.
        self._classes_dir = os.path.join(self.workdir, "classes")
        self._resources_dir = os.path.join(self.workdir, "resources")
        self._analysis_dir = os.path.join(self.workdir, "analysis")
        self._target_sources_dir = os.path.join(self.workdir, "target_sources")

        self._delete_scratch = self._get_lang_specific_option("delete_scratch")

        safe_mkdir(self._classes_dir)
        safe_mkdir(self._analysis_dir)
        safe_mkdir(self._target_sources_dir)

        self._analysis_file = os.path.join(self._analysis_dir, "global_analysis.valid")
        self._invalid_analysis_file = os.path.join(self._analysis_dir, "global_analysis.invalid")

        # A temporary, but well-known, dir in which to munge analysis/dependency files in before
        # caching. It must be well-known so we know where to find the files when we retrieve them from
        # the cache.
        self._analysis_tmpdir = os.path.join(self._analysis_dir, "artifact_cache_tmpdir")

        # We can't create analysis tools until after construction.
        self._lazy_analysis_tools = None

        # The rough number of source files to build in each compiler pass.
        self._partition_size_hint = self._get_lang_specific_option("partition_size_hint")
        if self._partition_size_hint == -1:
            self._partition_size_hint = self.context.config.getint(config_section, "partition_size_hint", default=1000)

        # JVM options for running the compiler.
        self._jvm_options = self.context.config.getlist(config_section, "jvm_args")

        # The ivy confs for which we're building.
        self._confs = self.context.config.getlist(config_section, "confs", default=["default"])

        # Set up dep checking if needed.
        def munge_flag(flag):
            return None if flag == "off" else flag

        check_missing_deps = munge_flag(self._get_lang_specific_option("missing_deps"))
        check_missing_direct_deps = munge_flag(self._get_lang_specific_option("missing_direct_deps"))
        check_unnecessary_deps = munge_flag(self._get_lang_specific_option("unnecessary_deps"))

        if check_missing_deps or check_missing_direct_deps or check_unnecessary_deps:
            target_whitelist = self.context.config.getlist("jvm", "missing_deps_target_whitelist", default=[])

            # Must init it here, so it can set requirements on the context.
            self._dep_analyzer = JvmDependencyAnalyzer(
                self.context, check_missing_deps, check_missing_direct_deps, check_unnecessary_deps, target_whitelist
            )
        else:
            self._dep_analyzer = None

        # If non-zero, and we have fewer than this number of locally-changed targets,
        # then we partition them separately, to preserve stability in the face of repeated
        # compilations.
        self._locally_changed_targets_heuristic_limit = self.context.config.getint(
            config_section, "locally_changed_targets_heuristic_limit", 0
        )

        self._upstream_class_to_path = None  # Computed lazily as needed.
        self.setup_artifact_cache_from_config(config_section=config_section)

        # Sources (relative to buildroot) present in the last analysis that have since been deleted.
        # Populated in prepare_execute().
        self._deleted_sources = None

        # Map of target -> list of sources (relative to buildroot), for all targets in all chunks.
        # Populated in prepare_execute().
        self._sources_by_target = None
class JvmCompileGlobalStrategy(JvmCompileStrategy):
  """A strategy for JVM compilation that uses a global classpath and analysis."""

  class InternalTargetPartitioningError(Exception):
    """Error partitioning targets by jvm platform settings."""

  @classmethod
  def register_options(cls, register, compile_task_name, supports_concurrent_execution):
    register('--missing-deps', advanced=True, choices=['off', 'warn', 'fatal'], default='warn',
             fingerprint=True,
             help='Check for missing dependencies in code compiled with {0}. Reports actual '
                  'dependencies A -> B where there is no transitive BUILD file dependency path '
                  'from A to B. If fatal, missing deps are treated as a build error.'.format(
               compile_task_name))

    register('--missing-direct-deps', advanced=True, choices=['off', 'warn', 'fatal'],
             default='off',
             fingerprint=True,
             help='Check for missing direct dependencies in code compiled with {0}. Reports actual '
                  'dependencies A -> B where there is no direct BUILD file dependency path from '
                  'A to B. This is a very strict check; In practice it is common to rely on '
                  'transitive, indirect dependencies, e.g., due to type inference or when the main '
                  'target in a BUILD file is modified to depend on other targets in the same BUILD '
                  'file, as an implementation detail. However it may still be useful to use this '
                  'on occasion. '.format(compile_task_name))

    register('--missing-deps-whitelist', advanced=True, type=list_option,
             fingerprint=True,
             help="Don't report these targets even if they have missing deps.")

    register('--unnecessary-deps', advanced=True, choices=['off', 'warn', 'fatal'], default='off',
             fingerprint=True,
             help='Check for declared dependencies in code compiled with {0} that are not needed. '
                  'This is a very strict check. For example, generated code will often '
                  'legitimately have BUILD dependencies that are unused in practice.'.format(
               compile_task_name))

    register('--changed-targets-heuristic-limit', advanced=True, type=int, default=0,
             help='If non-zero, and we have fewer than this number of locally-changed targets, '
                  'partition them separately, to preserve stability when compiling repeatedly.')

  def __init__(self, context, options, workdir, analysis_tools, compile_task_name,
               sources_predicate):
    super(JvmCompileGlobalStrategy, self).__init__(context, options, workdir, analysis_tools,
                                                   compile_task_name, sources_predicate)

    # Various working directories.
    # NB: These are grandfathered in with non-strategy-specific names, but to prevent
    # collisions within the buildcache, strategies should use strategy-specific subdirectories.
    self._analysis_dir = os.path.join(workdir, 'analysis')
    self._classes_dir = os.path.join(workdir, 'classes')

    self._analysis_file = os.path.join(self._analysis_dir, 'global_analysis.valid')
    self._invalid_analysis_file = os.path.join(self._analysis_dir, 'global_analysis.invalid')

    self._target_sources_dir = os.path.join(workdir, 'target_sources')

    # The rough number of source files to build in each compiler pass.
    self._partition_size_hint = options.partition_size_hint

    # Set up dep checking if needed.
    def munge_flag(flag):
      flag_value = getattr(options, flag, None)
      return None if flag_value == 'off' else flag_value

    check_missing_deps = munge_flag('missing_deps')
    check_missing_direct_deps = munge_flag('missing_direct_deps')
    check_unnecessary_deps = munge_flag('unnecessary_deps')

    if check_missing_deps or check_missing_direct_deps or check_unnecessary_deps:
      target_whitelist = options.missing_deps_whitelist
      # Must init it here, so it can set requirements on the context.
      self._dep_analyzer = JvmDependencyAnalyzer(self.context,
                                                 check_missing_deps,
                                                 check_missing_direct_deps,
                                                 check_unnecessary_deps,
                                                 target_whitelist)
    else:
      self._dep_analyzer = None

    # Computed lazily as needed.
    self._upstream_class_to_path = None

    # If non-zero, and we have fewer than this number of locally-changed targets,
    # then we partition them separately, to preserve stability in the face of repeated
    # compilations.
    self._changed_targets_heuristic_limit = options.changed_targets_heuristic_limit

    # Sources (relative to buildroot) present in the last analysis that have since been deleted.
    # Populated in prepare_compile().
    self._deleted_sources = None

  def name(self):
    return 'global'

  def compile_context(self, target):
    """Returns the default/stable compile context for the given target.

    Temporary compile contexts are private to the strategy.
    """
    return CompileContext(target,
                          self._analysis_file,
                          self._classes_dir,
                          self._sources_for_target(target))

  def move(self, src, dst):
    if self.delete_scratch:
      shutil.move(src, dst)
    else:
      shutil.copy(src, dst)

  def pre_compile(self):
    super(JvmCompileGlobalStrategy, self).pre_compile()

    # Only create these working dirs during execution phase, otherwise, they
    # would be wiped out by clean-all goal/task if it's specified.
    safe_mkdir(self._target_sources_dir)
    safe_mkdir(self._analysis_dir)
    safe_mkdir(self._classes_dir)

    # Look for invalid analysis files.
    for f in (self._invalid_analysis_file, self._analysis_file):
      self.validate_analysis(f)

  def prepare_compile(self, cache_manager, all_targets, relevant_targets):
    super(JvmCompileGlobalStrategy, self).prepare_compile(cache_manager, all_targets,
                                                          relevant_targets)

    # Update the classpath for us and for downstream tasks.
    compile_classpaths = self.context.products.get_data('compile_classpath')
    for conf in self._confs:
      compile_classpaths.add_for_targets(all_targets, [(conf, self._classes_dir)])

    # Split the global analysis file into valid and invalid parts.
    invalidation_check = cache_manager.check(relevant_targets)
    if invalidation_check.invalid_vts:
      # The analysis for invalid and deleted sources is no longer valid.
      invalid_targets = [vt.target for vt in invalidation_check.invalid_vts]
      invalid_sources_by_target = {}
      for tgt in invalid_targets:
        invalid_sources_by_target[tgt] = self._sources_for_target(tgt)
      invalid_sources = list(itertools.chain.from_iterable(invalid_sources_by_target.values()))
      self._deleted_sources = self._compute_deleted_sources()

      tmpdir = os.path.join(self.analysis_tmpdir, str(uuid.uuid4()))
      os.mkdir(tmpdir)
      valid_analysis_tmp = os.path.join(tmpdir, 'valid_analysis')
      newly_invalid_analysis_tmp = os.path.join(tmpdir, 'newly_invalid_analysis')
      invalid_analysis_tmp = os.path.join(tmpdir, 'invalid_analysis')
      if self._analysis_parser.is_nonempty_analysis(self._analysis_file):
        with self.context.new_workunit(name='prepare-analysis'):
          self._analysis_tools.split_to_paths(self._analysis_file,
              [(invalid_sources + self._deleted_sources, newly_invalid_analysis_tmp)],
              valid_analysis_tmp)
          if self._analysis_parser.is_nonempty_analysis(self._invalid_analysis_file):
            self._analysis_tools.merge_from_paths(
              [self._invalid_analysis_file, newly_invalid_analysis_tmp], invalid_analysis_tmp)
          else:
            invalid_analysis_tmp = newly_invalid_analysis_tmp

          # Now it's OK to overwrite the main analysis files with the new state.
          self.move(valid_analysis_tmp, self._analysis_file)
          self.move(invalid_analysis_tmp, self._invalid_analysis_file)
    else:
      self._deleted_sources = []

  def invalidation_hints(self, relevant_targets):
    # If needed, find targets that we've changed locally (as opposed to
    # changes synced in from the SCM).
    # TODO(benjy): Should locally_changed_targets be available in all Tasks?
    locally_changed_targets = None
    if self._changed_targets_heuristic_limit:
      locally_changed_targets = self._find_locally_changed_targets(relevant_targets)
      if (locally_changed_targets and
          len(locally_changed_targets) > self._changed_targets_heuristic_limit):
        locally_changed_targets = None

    return (self._partition_size_hint, locally_changed_targets)

  def ordered_compile_settings_and_targets(self, relevant_targets):
    """Groups the targets into ordered chunks, dependencies before dependees.

    Each chunk is of the form (compile_setting, targets). Attempts to create as few chunks as
    possible, under the constraint that targets with different compile settings cannot be in the
    same chunk, and dependencies must be in the same chunk or an earlier chunk than their
    dependees.

    Detects impossible combinations/dependency relationships with respect to the java target and
    source level, and raising errors as necessary (see targets_to_compile and
    infer_and_validate_java_target_levels).

    :return: a list of tuples of the form (compile_settings, list of targets)
    """
    relevant_targets = set(relevant_targets)

    def get_platform(target):
      return getattr(target, 'platform', None)

    # NB(gmalmquist): Short-circuit if we only have one platform. Asymptotically, this only gives us
    # O(|V|) time instead of O(|V|+|E|) if we have only one platform, which doesn't seem like much,
    # but in practice we save a lot of time because the runtime for the non-short-circuited code is
    # multiplied by a higher constant, because we have to iterate over all the targets several
    # times.
    platform_counts = defaultdict(int)
    for target in relevant_targets:
      platform_counts[target.platform] += 1
    if len(platform_counts) == 1:
      settings, = platform_counts
      return [(settings, relevant_targets)]

    # Map of target -> dependees.
    outgoing = defaultdict(set)
    # Map of target -> dependencies.
    incoming = defaultdict(set)

    transitive_targets = set()

    def add_edges(target):
      transitive_targets.add(target)
      if target.dependencies:
        for dependency in target.dependencies:
          outgoing[dependency].add(target)
          incoming[target].add(dependency)

    self.context.build_graph.walk_transitive_dependency_graph([t.address for t in relevant_targets],
                                                               work=add_edges)
    # Topological sort.
    sorted_targets = []
    frontier = defaultdict(set)

    def add_node(node):
      frontier[get_platform(node)].add(node)

    def next_node():
      next_setting = None
      if sorted_targets:
        # Prefer targets with the same settings as whatever we just added to the sorted list, to
        # greedily create chains that are as long as possible.
        next_setting = get_platform(sorted_targets[-1])
      if next_setting not in frontier:
        if None in frontier:
          # NB(gmalmquist): compile_settings=None indicates a target that is not actually a
          # jvm_target, which mean's it's an intermediate dependency. We want to expand these
          # whenever we can, because they give us more options we can use to create longer chains.
          next_setting = None
        else:
          next_setting = max(frontier.keys(), key=lambda setting: len(frontier[setting]))
      node = frontier[next_setting].pop()
      if not frontier[next_setting]:
        frontier.pop(next_setting)
      return node

    for target in transitive_targets:
      if not incoming[target]:
        add_node(target)

    while frontier:
      node = next_node()
      sorted_targets.append(node)
      if node in outgoing:
        for dependee in tuple(outgoing[node]):
          outgoing[node].remove(dependee)
          incoming[dependee].remove(node)
          if not incoming[dependee]:
            add_node(dependee)

    sorted_targets = [target for target in sorted_targets if target in relevant_targets]

    if set(sorted_targets) != relevant_targets:
      added = '\n  '.join(t.address.spec for t in (set(sorted_targets) - relevant_targets))
      removed = '\n  '.join(t.address.spec for t in (set(relevant_targets) - sorted_targets))
      raise self.InternalTargetPartitioningError(
        'Internal partitioning targets:\nSorted targets =/= original targets!\n'
        'Added:\n  {}\nRemoved:\n  {}'.format(added, removed)
      )

    unconsumed_edges = any(len(edges) > 0 for edges in outgoing.values())
    if unconsumed_edges:
      raise self.InternalTargetPartitioningError(
        'Cycle detected while ordering jvm_targets for compilation. This should have been detected '
        'when constructing the build_graph, so the presence of this error means there is probably '
        'a bug in this method.'
      )

    chunks = []
    for target in sorted_targets:
      if not isinstance(target, JvmTarget):
        continue
      if chunks and chunks[-1][0] == get_platform(target):
        chunks[-1][1].append(target)
      else:
        chunks.append((get_platform(target), [target]))
    return chunks

  def compile_chunk(self,
                    invalidation_check,
                    all_targets,
                    relevant_targets,
                    invalid_targets,
                    extra_compile_time_classpath_elements,
                    compile_vts,
                    register_vts,
                    update_artifact_cache_vts_work):
    assert invalid_targets, "compile_chunk should only be invoked if there are invalid targets."
    settings_and_targets = self.ordered_compile_settings_and_targets(invalid_targets)
    for settings, targets in settings_and_targets:
      if targets:
        self.compile_sub_chunk(invalidation_check,
                               all_targets,
                               targets,
                               extra_compile_time_classpath_elements,
                               compile_vts,
                               register_vts,
                               update_artifact_cache_vts_work,
                               settings)

  def compile_sub_chunk(self,
                        invalidation_check,
                        all_targets,
                        invalid_targets,
                        extra_compile_time_classpath_elements,
                        compile_vts,
                        register_vts,
                        update_artifact_cache_vts_work,
                        settings):
    """Executes compilations for the invalid targets contained in a single chunk.

    Has the side effects of populating:
    # valid/invalid analysis files
    # classes_by_source product
    # classes_by_target product
    # resources_by_target product
    """
    extra_classpath_tuples = self._compute_extra_classpath(extra_compile_time_classpath_elements)

    # Get the classpath generated by upstream JVM tasks and our own prepare_compile().
    # NB: The global strategy uses the aggregated classpath (for all targets) to compile each
    # chunk, which avoids needing to introduce compile-time dependencies between annotation
    # processors and the classes they annotate.
    compile_classpath = ClasspathUtil.compute_classpath(all_targets, self.context.products.get_data(
      'compile_classpath'), extra_classpath_tuples, self._confs)

    # Find the invalid sources for this chunk.
    invalid_sources_by_target = {t: self._sources_for_target(t) for t in invalid_targets}

    tmpdir = os.path.join(self.analysis_tmpdir, str(uuid.uuid4()))
    os.mkdir(tmpdir)

    # Figure out the sources and analysis belonging to each partition.
    partitions = []  # Each element is a triple (vts, sources_by_target, analysis).
    for vts in invalidation_check.invalid_vts_partitioned:
      partition_tmpdir = os.path.join(tmpdir, Target.maybe_readable_identify(vts.targets))
      os.mkdir(partition_tmpdir)
      sources = list(itertools.chain.from_iterable(
          [invalid_sources_by_target.get(t, []) for t in vts.targets]))
      de_duped_sources = list(OrderedSet(sources))
      if len(sources) != len(de_duped_sources):
        counts = [(src, len(list(srcs))) for src, srcs in itertools.groupby(sorted(sources))]
        self.context.log.warn(
            'De-duped the following sources:\n\t{}'
            .format('\n\t'.join(sorted('{} {}'.format(cnt, src) for src, cnt in counts if cnt > 1))))
      analysis_file = os.path.join(partition_tmpdir, 'analysis')
      partitions.append((vts, de_duped_sources, analysis_file))

    # Split per-partition files out of the global invalid analysis.
    if self._analysis_parser.is_nonempty_analysis(self._invalid_analysis_file) and partitions:
      with self.context.new_workunit(name='partition-analysis'):
        splits = [(x[1], x[2]) for x in partitions]
        # We have to pass the analysis for any deleted files through zinc, to give it
        # a chance to delete the relevant class files.
        if splits:
          splits[0] = (splits[0][0] + self._deleted_sources, splits[0][1])
        self._analysis_tools.split_to_paths(self._invalid_analysis_file, splits)

    # Now compile partitions one by one.
    for partition_index, partition in enumerate(partitions):
      (vts, sources, analysis_file) = partition

      progress_message = 'partition {} of {}'.format(partition_index + 1, len(partitions))
      # We have to treat the global output dir as an upstream element, so compilers can
      # find valid analysis for previous partitions. We use the global valid analysis
      # for the upstream.
      upstream_analysis = ({self._classes_dir: self._analysis_file}
                           if os.path.exists(self._analysis_file) else {})
      compile_vts(vts,
                  sources,
                  analysis_file,
                  upstream_analysis,
                  compile_classpath,
                  self._classes_dir,
                  None,
                  progress_message,
                  settings)

      # No exception was thrown, therefore the compile succeeded and analysis_file is now valid.
      if os.path.exists(analysis_file):  # The compilation created an analysis.
        # Merge the newly-valid analysis with our global valid analysis.
        new_valid_analysis = analysis_file + '.valid.new'
        if self._analysis_parser.is_nonempty_analysis(self._analysis_file):
          with self.context.new_workunit(name='update-upstream-analysis'):
            self._analysis_tools.merge_from_paths([self._analysis_file, analysis_file],
                                                  new_valid_analysis)
        else:  # We need to keep analysis_file around. Background tasks may need it.
          shutil.copy(analysis_file, new_valid_analysis)

        # Move the merged valid analysis to its proper location.
        # We do this before checking for missing dependencies, so that we can still
        # enjoy an incremental compile after fixing missing deps.
        self.move(new_valid_analysis, self._analysis_file)

        # Update the products with the latest classes. Must happen before the
        # missing dependencies check.
        register_vts([self.compile_context(t) for t in vts.targets])
        if self._dep_analyzer:
          # Check for missing dependencies.
          actual_deps = self._analysis_parser.parse_deps_from_path(analysis_file,
              lambda: self._compute_classpath_elements_by_class(compile_classpath), self._classes_dir)
          with self.context.new_workunit(name='find-missing-dependencies'):
            self._dep_analyzer.check(sources, actual_deps)

        # Kick off the background artifact cache write.
        if update_artifact_cache_vts_work:
          self._write_to_artifact_cache(analysis_file,
                                        vts,
                                        update_artifact_cache_vts_work)

      if self._analysis_parser.is_nonempty_analysis(self._invalid_analysis_file):
        with self.context.new_workunit(name='trim-downstream-analysis'):
          # Trim out the newly-valid sources from our global invalid analysis.
          new_invalid_analysis = analysis_file + '.invalid.new'
          discarded_invalid_analysis = analysis_file + '.invalid.discard'
          self._analysis_tools.split_to_paths(self._invalid_analysis_file,
            [(sources, discarded_invalid_analysis)], new_invalid_analysis)
          self.move(new_invalid_analysis, self._invalid_analysis_file)

      # Record the built target -> sources mapping for future use.
      for target, sources in self._sources_for_targets(vts.targets).items():
        self._record_previous_sources_by_target(target, sources)

      # Now that all the analysis accounting is complete, and we have no missing deps,
      # we can safely mark the targets as valid.
      vts.update()

  def compute_resource_mapping(self, compile_contexts):
    return ResourceMapping(self._classes_dir)

  def compute_classes_by_source(self, compile_contexts):
    if not compile_contexts:
      return {}

    # This implementation requires that all contexts use the same analysis file and global classes.
    analysis_file = None
    for compile_context in compile_contexts:
      if compile_context.classes_dir != self._classes_dir:
        raise TaskError('Unrecognized classes directory for the global strategy: {}'.format(
            compile_context.classes_dir))
      if not analysis_file:
        analysis_file = compile_context.analysis_file
      else:
        if compile_context.analysis_file != analysis_file:
          raise TaskError('Inconsistent analysis file for the global strategy: {} vs {}'.format(
              compile_context.analysis_file, analysis_file))

    classes_by_src_by_context = defaultdict(dict)
    if os.path.exists(analysis_file):
      # Parse the global analysis once.
      buildroot = get_buildroot()
      products = self._analysis_parser.parse_products_from_path(analysis_file,
                                                                self._classes_dir)

      # Then iterate over contexts (targets), and add the classes for their sources.
      for compile_context in compile_contexts:
        classes_by_src = classes_by_src_by_context[compile_context]
        for source in compile_context.sources:
          absolute_source = os.path.join(buildroot, source)
          classes_by_src[source] = products.get(absolute_source, [])
    return classes_by_src_by_context

  def post_process_cached_vts(self, cached_vts):
    """Special post processing for global scala analysis files.

    Class files are retrieved directly into their final locations in the global classes dir.
    """

    # Get all the targets whose artifacts we found in the cache.
    cached_targets = []
    for vt in cached_vts:
      for target in vt.targets:
        cached_targets.append(target)

    # The current global analysis may contain old data for modified targets for
    # which we got cache hits. We need to strip out this old analysis, to ensure
    # that the new data incoming from the cache doesn't collide with it during the merge.
    sources_to_strip = []
    if os.path.exists(self._analysis_file):
      for target in cached_targets:
        sources_to_strip.extend(self._get_previous_sources_by_target(target))

    # Localize the cached analyses.
    analyses_to_merge = []
    for target in cached_targets:
      analysis_file = JvmCompileStrategy._analysis_for_target(self.analysis_tmpdir, target)
      portable_analysis_file = JvmCompileStrategy._portable_analysis_for_target(
          self.analysis_tmpdir, target)
      if os.path.exists(portable_analysis_file):
        self._analysis_tools.localize(portable_analysis_file, analysis_file)
      if os.path.exists(analysis_file):
        analyses_to_merge.append(analysis_file)

    # Merge them into the global analysis.
    if analyses_to_merge:
      with temporary_dir() as tmpdir:
        if sources_to_strip:
          throwaway = os.path.join(tmpdir, 'throwaway')
          trimmed_analysis = os.path.join(tmpdir, 'trimmed')
          self._analysis_tools.split_to_paths(self._analysis_file,
                                          [(sources_to_strip, throwaway)],
                                          trimmed_analysis)
        else:
          trimmed_analysis = self._analysis_file
        if os.path.exists(trimmed_analysis):
          analyses_to_merge.append(trimmed_analysis)
        tmp_analysis = os.path.join(tmpdir, 'analysis')
        with self.context.new_workunit(name='merge_analysis'):
          self._analysis_tools.merge_from_paths(analyses_to_merge, tmp_analysis)

        sources_by_cached_target = self._sources_for_targets(cached_targets)

        # Record the cached target -> sources mapping for future use.
        for target, sources in sources_by_cached_target.items():
          self._record_previous_sources_by_target(target, sources)

        # Everything's good so move the merged analysis to its final location.
        if os.path.exists(tmp_analysis):
          self.move(tmp_analysis, self._analysis_file)

  def _write_to_artifact_cache(self, analysis_file, vts, get_update_artifact_cache_work):
    vt_by_target = dict([(vt.target, vt) for vt in vts.versioned_targets])

    vts_targets = [t for t in vts.targets if not t.has_label('no_cache')]

    # Determine locations for analysis files that will be split in the background.
    split_analysis_files = [
        JvmCompileStrategy._analysis_for_target(self.analysis_tmpdir, t) for t in vts_targets]
    portable_split_analysis_files = [
        JvmCompileStrategy._portable_analysis_for_target(self.analysis_tmpdir, t) for t in vts_targets]

    # Set up args for splitting the analysis into per-target files.
    splits = zip([self._sources_for_target(t) for t in vts_targets], split_analysis_files)
    splits_args_tuples = [(analysis_file, splits)]

    # Set up args for rebasing the splits.
    relativize_args_tuples = zip(split_analysis_files, portable_split_analysis_files)

    # Compute the classes and resources for each vts.
    compile_contexts = [self.compile_context(t) for t in vts_targets]
    vts_artifactfiles_pairs = []
    classes_by_source_by_context = self.compute_classes_by_source(compile_contexts)
    resources_by_target = self.context.products.get_data('resources_by_target')
    for compile_context in compile_contexts:
      target = compile_context.target
      if target.has_label('no_cache'):
        continue
      artifacts = []
      if resources_by_target is not None:
        for _, paths in resources_by_target[target].abs_paths():
          artifacts.extend(paths)
      classes_by_source = classes_by_source_by_context[compile_context]
      for source in compile_context.sources:
        classes = classes_by_source.get(source, [])
        artifacts.extend(classes)

      vt = vt_by_target.get(target)
      if vt is not None:
        # NOTE: analysis_file doesn't exist yet.
        vts_artifactfiles_pairs.append(
            (vt, artifacts + [JvmCompileStrategy._portable_analysis_for_target(
                self.analysis_tmpdir, target)]))

    update_artifact_cache_work = get_update_artifact_cache_work(vts_artifactfiles_pairs)
    if update_artifact_cache_work:
      work_chain = [
        Work(self._analysis_tools.split_to_paths, splits_args_tuples, 'split'),
        Work(self._analysis_tools.relativize, relativize_args_tuples, 'relativize'),
        update_artifact_cache_work
      ]
      self.context.submit_background_work_chain(work_chain, parent_workunit_name='cache')

  def _get_previous_sources_by_target(self, target):
    """Returns the target's sources as recorded on the last successful build of target.

    Returns a list of absolute paths.
    """
    path = os.path.join(self._target_sources_dir, target.identifier)
    if os.path.exists(path):
      with open(path, 'r') as infile:
        return [s.rstrip() for s in infile.readlines()]
    else:
      return []

  def _record_previous_sources_by_target(self, target, sources):
    # Record target -> source mapping for future use.
    with open(os.path.join(self._target_sources_dir, target.identifier), 'w') as outfile:
      for src in sources:
        outfile.write(os.path.join(get_buildroot(), src))
        outfile.write(b'\n')

  def _compute_deleted_sources(self):
    """Computes the list of sources present in the last analysis that have since been deleted.

    This is a global list. We have no way of associating them to individual targets.
    Paths are relative to buildroot.
    """
    with self.context.new_workunit('find-deleted-sources'):
      if os.path.exists(self._analysis_file):
        products = self._analysis_parser.parse_products_from_path(self._analysis_file,
                                                                  self._classes_dir)
        buildroot = get_buildroot()
        old_srcs = products.keys()  # Absolute paths.
        return [os.path.relpath(src, buildroot) for src in old_srcs if not os.path.exists(src)]
      else:
        return []

  def _find_locally_changed_targets(self, relevant_targets):
    """Finds the targets whose sources have been modified locally.

    Returns a list of targets, or None if no SCM is available.
    """
    # Compute the src->targets mapping. There should only be one target per source,
    # but that's not yet a hard requirement, so the value is a list of targets.
    # TODO(benjy): Might this inverse mapping be needed elsewhere too?
    targets_by_source = defaultdict(list)
    for tgt, srcs in self._sources_for_targets(relevant_targets).items():
      for src in srcs:
        targets_by_source[src].append(tgt)

    ret = OrderedSet()
    scm = get_scm()
    if not scm:
      return None
    changed_files = scm.changed_files(include_untracked=True, relative_to=get_buildroot())
    for f in changed_files:
      ret.update(targets_by_source.get(f, []))
    return list(ret)

  def _compute_classpath_elements_by_class(self, classpath):
    # Don't consider loose classes dirs in our classes dir. Those will be considered
    # separately, by looking at products.
    def non_product(path):
      return path != self._classes_dir

    if self._upstream_class_to_path is None:
      self._upstream_class_to_path = {}
      classpath_entries = filter(non_product, classpath)
      for cp_entry in self._find_all_bootstrap_jars() + classpath_entries:
        # Per the classloading spec, a 'jar' in this context can also be a .zip file.
        if os.path.isfile(cp_entry) and (cp_entry.endswith('.jar') or cp_entry.endswith('.zip')):
          with open_zip(cp_entry, 'r') as jar:
            for cls in jar.namelist():
              # First jar with a given class wins, just like when classloading.
              if cls.endswith(b'.class') and not cls in self._upstream_class_to_path:
                self._upstream_class_to_path[cls] = cp_entry
        elif os.path.isdir(cp_entry):
          for dirpath, _, filenames in safe_walk(cp_entry, followlinks=True):
            for f in filter(lambda x: x.endswith('.class'), filenames):
              cls = os.path.relpath(os.path.join(dirpath, f), cp_entry)
              if not cls in self._upstream_class_to_path:
                self._upstream_class_to_path[cls] = os.path.join(dirpath, f)
    return self._upstream_class_to_path

  def _find_all_bootstrap_jars(self):
    def get_path(key):
      return self.context.java_sysprops.get(key, '').split(':')

    def find_jars_in_dirs(dirs):
      ret = []
      for d in dirs:
        if os.path.isdir(d):
          ret.extend(filter(lambda s: s.endswith('.jar'), os.listdir(d)))
      return ret

    # Note: assumes HotSpot, or some JVM that supports sun.boot.class.path.
    # TODO: Support other JVMs? Not clear if there's a standard way to do so.
    # May include loose classes dirs.
    boot_classpath = get_path('sun.boot.class.path')

    # Note that per the specs, overrides and extensions must be in jars.
    # Loose class files will not be found by the JVM.
    override_jars = find_jars_in_dirs(get_path('java.endorsed.dirs'))
    extension_jars = find_jars_in_dirs(get_path('java.ext.dirs'))

    # Note that this order matters: it reflects the classloading order.
    bootstrap_jars = filter(os.path.isfile, override_jars + boot_classpath + extension_jars)
    return bootstrap_jars  # Technically, may include loose class dirs from boot_classpath.
Exemplo n.º 8
0
  def __init__(self, context, workdir, minimum_version=None, jdk=False):
    # TODO(John Sirois): XXX plumb minimum_version via config or flags
    super(JvmCompile, self).__init__(context, workdir, minimum_version=minimum_version, jdk=jdk)
    concrete_class = type(self)
    config_section = concrete_class._config_section

    def get_lang_specific_option(opt):
      full_opt_name = self._language + '_' + opt
      return getattr(context.options, full_opt_name, None)

    # Global workdir.
    self._pants_workdir = context.config.getdefault('pants_workdir')

    # Various working directories.
    self._classes_dir = os.path.join(self.workdir, 'classes')
    self._resources_dir = os.path.join(self.workdir, 'resources')
    self._analysis_dir = os.path.join(self.workdir, 'analysis')
    self._target_sources_dir = os.path.join(self.workdir, 'target_sources')

    self._delete_scratch = get_lang_specific_option('delete_scratch')

    safe_mkdir(self._classes_dir)
    safe_mkdir(self._analysis_dir)
    safe_mkdir(self._target_sources_dir)

    self._analysis_file = os.path.join(self._analysis_dir, 'global_analysis.valid')
    self._invalid_analysis_file = os.path.join(self._analysis_dir, 'global_analysis.invalid')

    # A temporary, but well-known, dir in which to munge analysis/dependency files in before
    # caching. It must be well-known so we know where to find the files when we retrieve them from
    # the cache.
    self._analysis_tmpdir = os.path.join(self._analysis_dir, 'artifact_cache_tmpdir')

    # We can't create analysis tools until after construction.
    self._lazy_analysis_tools = None

    # Compiler options.
    self._args = context.config.getlist(config_section, 'args')
    if get_lang_specific_option('compile_warnings'):
      self._args.extend(context.config.getlist(config_section, 'warning_args'))
    else:
      self._args.extend(context.config.getlist(config_section, 'no_warning_args'))

    # The rough number of source files to build in each compiler pass.
    self._partition_size_hint = get_lang_specific_option('partition_size_hint')
    if self._partition_size_hint == -1:
      self._partition_size_hint = context.config.getint(config_section, 'partition_size_hint',
                                                        default=1000)

    # JVM options for running the compiler.
    self._jvm_options = context.config.getlist(config_section, 'jvm_args')

    # The ivy confs for which we're building.
    self._confs = context.config.getlist(config_section, 'confs', default=['default'])

    # Runtime dependencies.
    runtime_deps = context.config.getlist(config_section, 'runtime-deps', default=[])
    if runtime_deps:
      self._runtime_deps_key = self._language + '-runtime-deps'
      self.register_jvm_tool(self._runtime_deps_key, runtime_deps)
    else:
      self._runtime_deps_key = None

    # Set up dep checking if needed.
    def munge_flag(flag):
      return None if flag == 'off' else flag
    check_missing_deps = munge_flag(get_lang_specific_option('missing_deps'))
    check_missing_direct_deps = munge_flag(get_lang_specific_option('missing_direct_deps'))
    check_unnecessary_deps = munge_flag(get_lang_specific_option('unnecessary_deps'))

    if check_missing_deps or check_missing_direct_deps or check_unnecessary_deps:
      # Must init it here, so it can set requirements on the context.
      self._dep_analyzer = JvmDependencyAnalyzer(self.context,
                                                 check_missing_deps,
                                                 check_missing_direct_deps,
                                                 check_unnecessary_deps)
    else:
      self._dep_analyzer = None

    # If non-zero, and we have fewer than this number of locally-changed targets,
    # then we partition them separately, to preserve stability in the face of repeated
    # compilations.
    self._locally_changed_targets_heuristic_limit = context.config.getint(config_section,
        'locally_changed_targets_heuristic_limit', 0)

    self._class_to_jarfile = None  # Computed lazily as needed.

    self.context.products.require_data('exclusives_groups')
    self.setup_artifact_cache_from_config(config_section=config_section)

    # Sources (relative to buildroot) present in the last analysis that have since been deleted.
    # Populated in prepare_execute().
    self._deleted_sources = None

    # Map of target -> list of sources (relative to buildroot), for all targets in all chunks.
    # Populated in prepare_execute().
    self._sources_by_target = None
Exemplo n.º 9
0
  def __init__(self, *args, **kwargs):
    super(JvmCompile, self).__init__(*args, **kwargs)
    config_section = self.config_section

    # Various working directories.
    self._classes_dir = os.path.join(self.workdir, 'classes')
    self._resources_dir = os.path.join(self.workdir, 'resources')
    self._analysis_dir = os.path.join(self.workdir, 'analysis')
    self._target_sources_dir = os.path.join(self.workdir, 'target_sources')

    self._delete_scratch = self.get_options().delete_scratch

    self._analysis_file = os.path.join(self._analysis_dir, 'global_analysis.valid')
    self._invalid_analysis_file = os.path.join(self._analysis_dir, 'global_analysis.invalid')

    # A temporary, but well-known, dir in which to munge analysis/dependency files in before
    # caching. It must be well-known so we know where to find the files when we retrieve them from
    # the cache.
    self._analysis_tmpdir = os.path.join(self._analysis_dir, 'artifact_cache_tmpdir')

    # We can't create analysis tools until after construction.
    self._lazy_analysis_tools = None

    # The rough number of source files to build in each compiler pass.
    self._partition_size_hint = self.get_options().partition_size_hint

    # JVM options for running the compiler.
    self._jvm_options = self.context.config.getlist(config_section, 'jvm_args')

    # The ivy confs for which we're building.
    self._confs = self.context.config.getlist(config_section, 'confs', default=['default'])

    self._args = list(self.get_options().args)
    if self.get_options().warnings:
      self._args.extend(self.get_options().warning_args)
    else:
      self._args.extend(self.get_options().no_warning_args)

    # Set up dep checking if needed.
    def munge_flag(flag):
      flag_value = getattr(self.get_options(), flag, None)
      return None if flag_value == 'off' else flag_value

    check_missing_deps = munge_flag('missing_deps')
    check_missing_direct_deps = munge_flag('missing_direct_deps')
    check_unnecessary_deps = munge_flag('unnecessary_deps')

    if check_missing_deps or check_missing_direct_deps or check_unnecessary_deps:
      target_whitelist = self.context.config.getlist('jvm', 'missing_deps_target_whitelist', default=[])

      # Must init it here, so it can set requirements on the context.
      self._dep_analyzer = JvmDependencyAnalyzer(self.context,
                                                 check_missing_deps,
                                                 check_missing_direct_deps,
                                                 check_unnecessary_deps,
                                                 target_whitelist)
    else:
      self._dep_analyzer = None

    # If non-zero, and we have fewer than this number of locally-changed targets,
    # then we partition them separately, to preserve stability in the face of repeated
    # compilations.
    self._locally_changed_targets_heuristic_limit = self.context.config.getint(config_section,
        'locally_changed_targets_heuristic_limit', 0)

    self._upstream_class_to_path = None  # Computed lazily as needed.
    self.setup_artifact_cache_from_config(config_section=config_section)

    # Sources (relative to buildroot) present in the last analysis that have since been deleted.
    # Populated in prepare_execute().
    self._deleted_sources = None

    # Map of target -> list of sources (relative to buildroot), for all targets in all chunks.
    # Populated in prepare_execute().
    self._sources_by_target = None
Exemplo n.º 10
0
class JvmCompile(NailgunTaskBase, GroupMember, JvmToolTaskMixin):
    """A common framework for JVM compilation.

  To subclass for a specific JVM language, implement the static values and methods
  mentioned below under "Subclasses must implement".
  """
    @classmethod
    def setup_parser(cls, option_group, args, mkflag):
        super(JvmCompile, cls).setup_parser(option_group, args, mkflag)

        option_group.add_option(
            mkflag('warnings'),
            mkflag('warnings', negate=True),
            dest=cls._language + '_compile_warnings',
            default=True,
            action='callback',
            callback=mkflag.set_bool,
            help='[%default] Compile with all configured warnings enabled.')

        option_group.add_option(
            mkflag('partition-size-hint'),
            dest=cls._language + '_partition_size_hint',
            action='store',
            type='int',
            default=-1,
            help='Roughly how many source files to attempt to compile together. '
            'Set to a large number to compile all sources together. Set this '
            'to 0 to compile target-by-target. Default is set in pants.ini.')

        option_group.add_option(
            mkflag('missing-deps'),
            dest=cls._language + '_missing_deps',
            choices=['off', 'warn', 'fatal'],
            default='warn',
            help='[%default] One of off, warn, fatal. '
            'Check for missing dependencies in ' + cls._language + 'code. '
            'Reports actual dependencies A -> B where there is no '
            'transitive BUILD file dependency path from A to B.'
            'If fatal, missing deps are treated as a build error.')

        option_group.add_option(
            mkflag('missing-direct-deps'),
            dest=cls._language + '_missing_direct_deps',
            choices=['off', 'warn', 'fatal'],
            default='off',
            help='[%default] One of off, warn, fatal. '
            'Check for missing direct dependencies in ' + cls._language +
            ' code. Reports actual dependencies A -> B where there is no '
            'direct BUILD file dependency path from A to B. This is a very '
            'strict check, as in practice it is common to rely on transitive, '
            'non-direct dependencies, e.g., due to type inference or when the '
            'main target in a BUILD file is modified to depend on other '
            'targets in the same BUILD file as an implementation detail. It '
            'may still be useful to set it to fatal temorarily, to detect '
            'these.')

        option_group.add_option(
            mkflag('unnecessary-deps'),
            dest=cls._language + '_unnecessary_deps',
            choices=['off', 'warn', 'fatal'],
            default='off',
            help='[%default] One of off, warn, fatal. Check for declared '
            'dependencies in ' + cls._language + ' code that are not '
            'needed. This is a very strict check. For example, generated code '
            'will often legitimately have BUILD dependencies that are unused '
            'in practice.')

        option_group.add_option(
            mkflag('delete-scratch'),
            mkflag('delete-scratch', negate=True),
            dest=cls._language + '_delete_scratch',
            default=True,
            action='callback',
            callback=mkflag.set_bool,
            help='[%default] Leave intermediate scratch files around, '
            'for debugging build problems.')

    # Subclasses must implement.
    # --------------------------
    _language = None
    _file_suffix = None
    _config_section = None

    @classmethod
    def name(cls):
        return cls._language

    def select(self, target):
        return target.has_sources(self._file_suffix)

    def create_analysis_tools(self):
        """Returns an AnalysisTools implementation.

    Subclasses must implement.
    """
        raise NotImplementedError()

    def compile(self, args, classpath, sources, classes_output_dir,
                analysis_file):
        """Invoke the compiler.

    Must raise TaskError on compile failure.

    Subclasses must implement."""
        raise NotImplementedError()

    # Subclasses may override.
    # ------------------------
    def extra_compile_time_classpath_elements(self):
        """Extra classpath elements common to all compiler invocations.

    E.g., jars for compiler plugins.
    """
        return []

    def extra_products(self, target):
        """Any extra, out-of-band products created for a target.

    E.g., targets that produce scala compiler plugins produce an info file.
    Returns a list of pairs (root, [absolute paths of files under root]).
    """
        return []

    def post_process(self, relevant_targets):
        """Any extra post-execute work."""
        pass

    # Common code.
    # ------------
    @staticmethod
    def _analysis_for_target(analysis_dir, target):
        return os.path.join(analysis_dir, target.id + '.analysis')

    @staticmethod
    def _portable_analysis_for_target(analysis_dir, target):
        return JvmCompile._analysis_for_target(analysis_dir,
                                               target) + '.portable'

    def __init__(self, context, workdir, minimum_version=None, jdk=False):
        # TODO(John Sirois): XXX plumb minimum_version via config or flags
        super(JvmCompile, self).__init__(context,
                                         workdir,
                                         minimum_version=minimum_version,
                                         jdk=jdk)
        concrete_class = type(self)
        config_section = concrete_class._config_section

        def get_lang_specific_option(opt):
            full_opt_name = self._language + '_' + opt
            return getattr(context.options, full_opt_name, None)

        # Global workdir.
        self._pants_workdir = context.config.getdefault('pants_workdir')

        # Various working directories.
        self._classes_dir = os.path.join(self.workdir, 'classes')
        self._resources_dir = os.path.join(self.workdir, 'resources')
        self._analysis_dir = os.path.join(self.workdir, 'analysis')
        self._target_sources_dir = os.path.join(self.workdir, 'target_sources')

        self._delete_scratch = get_lang_specific_option('delete_scratch')

        safe_mkdir(self._classes_dir)
        safe_mkdir(self._analysis_dir)
        safe_mkdir(self._target_sources_dir)

        self._analysis_file = os.path.join(self._analysis_dir,
                                           'global_analysis.valid')
        self._invalid_analysis_file = os.path.join(self._analysis_dir,
                                                   'global_analysis.invalid')

        # A temporary, but well-known, dir in which to munge analysis/dependency files in before
        # caching. It must be well-known so we know where to find the files when we retrieve them from
        # the cache.
        self._analysis_tmpdir = os.path.join(self._analysis_dir,
                                             'artifact_cache_tmpdir')

        # We can't create analysis tools until after construction.
        self._lazy_analysis_tools = None

        # Compiler options.
        self._args = context.config.getlist(config_section, 'args')
        if get_lang_specific_option('compile_warnings'):
            self._args.extend(
                context.config.getlist(config_section, 'warning_args'))
        else:
            self._args.extend(
                context.config.getlist(config_section, 'no_warning_args'))

        # The rough number of source files to build in each compiler pass.
        self._partition_size_hint = get_lang_specific_option(
            'partition_size_hint')
        if self._partition_size_hint == -1:
            self._partition_size_hint = context.config.getint(
                config_section, 'partition_size_hint', default=1000)

        # JVM options for running the compiler.
        self._jvm_options = context.config.getlist(config_section, 'jvm_args')

        # The ivy confs for which we're building.
        self._confs = context.config.getlist(config_section,
                                             'confs',
                                             default=['default'])

        # Runtime dependencies.
        runtime_deps = context.config.getlist(config_section,
                                              'runtime-deps',
                                              default=[])
        if runtime_deps:
            self._runtime_deps_key = self._language + '-runtime-deps'
            self.register_jvm_tool(self._runtime_deps_key, runtime_deps)
        else:
            self._runtime_deps_key = None

        # Set up dep checking if needed.
        def munge_flag(flag):
            return None if flag == 'off' else flag

        check_missing_deps = munge_flag(
            get_lang_specific_option('missing_deps'))
        check_missing_direct_deps = munge_flag(
            get_lang_specific_option('missing_direct_deps'))
        check_unnecessary_deps = munge_flag(
            get_lang_specific_option('unnecessary_deps'))

        if check_missing_deps or check_missing_direct_deps or check_unnecessary_deps:
            # Must init it here, so it can set requirements on the context.
            self._dep_analyzer = JvmDependencyAnalyzer(
                self.context, check_missing_deps, check_missing_direct_deps,
                check_unnecessary_deps)
        else:
            self._dep_analyzer = None

        # If non-zero, and we have fewer than this number of locally-changed targets,
        # then we partition them separately, to preserve stability in the face of repeated
        # compilations.
        self._locally_changed_targets_heuristic_limit = context.config.getint(
            config_section, 'locally_changed_targets_heuristic_limit', 0)

        self._class_to_jarfile = None  # Computed lazily as needed.

        self.context.products.require_data('exclusives_groups')
        self.setup_artifact_cache_from_config(config_section=config_section)

        # Sources (relative to buildroot) present in the last analysis that have since been deleted.
        # Populated in prepare_execute().
        self._deleted_sources = None

        # Map of target -> list of sources (relative to buildroot), for all targets in all chunks.
        # Populated in prepare_execute().
        self._sources_by_target = None

    def move(self, src, dst):
        if self._delete_scratch:
            shutil.move(src, dst)
        else:
            shutil.copy(src, dst)

    def prepare_execute(self, chunks):
        all_targets = list(itertools.chain(*chunks))

        # Update the classpath for us and for downstream tasks.
        egroups = self.context.products.get_data('exclusives_groups')
        all_group_ids = set()
        for t in all_targets:
            all_group_ids.add(egroups.get_group_key_for_target(t))

        runtime_deps = self.tool_classpath(
            self._runtime_deps_key) if self._runtime_deps_key else []

        for conf in self._confs:
            for group_id in all_group_ids:
                egroups.update_compatible_classpaths(
                    group_id, [(conf, self._classes_dir)])
                egroups.update_compatible_classpaths(
                    group_id, [(conf, self._resources_dir)])
                for dep in runtime_deps:
                    egroups.update_compatible_classpaths(
                        group_id, [(conf, dep)])

        # Target -> sources (relative to buildroot).
        # TODO(benjy): Should sources_by_target be available in all Tasks?
        self._sources_by_target = self._compute_current_sources_by_target(
            all_targets)

        # Split the global analysis file into valid and invalid parts.
        cache_manager = self.create_cache_manager(invalidate_dependents=True)
        invalidation_check = cache_manager.check(all_targets)
        if invalidation_check.invalid_vts:
            # The analysis for invalid and deleted sources is no longer valid.
            invalid_targets = [
                vt.target for vt in invalidation_check.invalid_vts
            ]
            invalid_sources_by_target = {}
            for tgt in invalid_targets:
                invalid_sources_by_target[tgt] = self._sources_by_target[tgt]
            invalid_sources = list(
                itertools.chain.from_iterable(
                    invalid_sources_by_target.values()))
            self._deleted_sources = self._compute_deleted_sources()

            self._ensure_analysis_tmpdir()
            tmpdir = os.path.join(self._analysis_tmpdir, str(uuid.uuid4()))
            os.mkdir(tmpdir)
            valid_analysis_tmp = os.path.join(tmpdir, 'valid_analysis')
            newly_invalid_analysis_tmp = os.path.join(
                tmpdir, 'newly_invalid_analysis')
            invalid_analysis_tmp = os.path.join(tmpdir, 'invalid_analysis')
            if self._analysis_parser.is_nonempty_analysis(self._analysis_file):
                with self.context.new_workunit(name='prepare-analysis'):
                    self._analysis_tools.split_to_paths(
                        self._analysis_file,
                        [(invalid_sources + self._deleted_sources,
                          newly_invalid_analysis_tmp)], valid_analysis_tmp)
                    if self._analysis_parser.is_nonempty_analysis(
                            self._invalid_analysis_file):
                        self._analysis_tools.merge_from_paths([
                            self._invalid_analysis_file,
                            newly_invalid_analysis_tmp
                        ], invalid_analysis_tmp)
                    else:
                        invalid_analysis_tmp = newly_invalid_analysis_tmp

                    # Now it's OK to overwrite the main analysis files with the new state.
                    self.move(valid_analysis_tmp, self._analysis_file)
                    self.move(invalid_analysis_tmp,
                              self._invalid_analysis_file)
        else:
            self._deleted_sources = []

    # TODO(benjy): Break this monstrosity up? Previous attempts to do so
    #              turned out to be more trouble than it was worth.
    def execute_chunk(self, relevant_targets):
        # TODO(benjy): Add a pre-execute phase for injecting deps into targets, so e.g.,
        # we can inject a dep on the scala runtime library and still have it ivy-resolve.

        # In case we have no relevant targets and return early.
        self._create_empty_products()

        if not relevant_targets:
            return

        # Get the exclusives group for the targets to compile.
        # Group guarantees that they'll be a single exclusives key for them.
        egroups = self.context.products.get_data('exclusives_groups')
        group_id = egroups.get_group_key_for_target(relevant_targets[0])

        # Get the classpath generated by upstream JVM tasks and our own prepare_execute().
        classpath = egroups.get_classpath_for_group(group_id)

        # Add any extra compile-time-only classpath elements.
        # TODO(benjy): Model compile-time vs. runtime classpaths more explicitly.
        for conf in self._confs:
            for jar in self.extra_compile_time_classpath_elements():
                classpath.insert(0, (conf, jar))

        # Target -> sources (relative to buildroot), for just this chunk's targets.
        sources_by_target = self._sources_for_targets(relevant_targets)

        # If needed, find targets that we've changed locally (as opposed to
        # changes synced in from the SCM).
        # TODO(benjy): Should locally_changed_targets be available in all Tasks?
        locally_changed_targets = None
        if self._locally_changed_targets_heuristic_limit:
            locally_changed_targets = self._find_locally_changed_targets(
                sources_by_target)
            if locally_changed_targets and \
                    len(locally_changed_targets) > self._locally_changed_targets_heuristic_limit:
                locally_changed_targets = None

        # Invalidation check. Everything inside the with block must succeed for the
        # invalid targets to become valid.
        with self.invalidated(relevant_targets,
                              invalidate_dependents=True,
                              partition_size_hint=self._partition_size_hint,
                              locally_changed_targets=locally_changed_targets
                              ) as invalidation_check:
            if invalidation_check.invalid_vts:
                # Find the invalid sources for this chunk.
                invalid_targets = [
                    vt.target for vt in invalidation_check.invalid_vts
                ]
                invalid_sources_by_target = self._sources_for_targets(
                    invalid_targets)

                tmpdir = os.path.join(self._analysis_tmpdir, str(uuid.uuid4()))
                os.mkdir(tmpdir)

                # Register products for all the valid targets.
                # We register as we go, so dependency checking code can use this data.
                valid_targets = list(
                    set(relevant_targets) - set(invalid_targets))
                self._register_products(valid_targets, self._analysis_file)

                # Figure out the sources and analysis belonging to each partition.
                partitions = [
                ]  # Each element is a triple (vts, sources_by_target, analysis).
                for vts in invalidation_check.invalid_vts_partitioned:
                    partition_tmpdir = os.path.join(
                        tmpdir, Target.maybe_readable_identify(vts.targets))
                    os.mkdir(partition_tmpdir)
                    sources = list(
                        itertools.chain.from_iterable([
                            invalid_sources_by_target.get(t, [])
                            for t in vts.targets
                        ]))
                    de_duped_sources = list(OrderedSet(sources))
                    if len(sources) != len(de_duped_sources):
                        counts = [
                            (src, len(list(srcs)))
                            for src, srcs in itertools.groupby(sorted(sources))
                        ]
                        self.context.log.warn(
                            'De-duped the following sources:\n\t%s' %
                            '\n\t'.join(
                                sorted('%d %s' % (cnt, src)
                                       for src, cnt in counts if cnt > 1)))
                    analysis_file = os.path.join(partition_tmpdir, 'analysis')
                    partitions.append((vts, de_duped_sources, analysis_file))

                # Split per-partition files out of the global invalid analysis.
                if self._analysis_parser.is_nonempty_analysis(
                        self._invalid_analysis_file) and partitions:
                    with self.context.new_workunit(name='partition-analysis'):
                        splits = [(x[1], x[2]) for x in partitions]
                        # We have to pass the analysis for any deleted files through zinc, to give it
                        # a chance to delete the relevant class files.
                        if splits:
                            splits[0] = (splits[0][0] + self._deleted_sources,
                                         splits[0][1])
                        self._analysis_tools.split_to_paths(
                            self._invalid_analysis_file, splits)

                # Now compile partitions one by one.
                for partition in partitions:
                    (vts, sources, analysis_file) = partition
                    cp_entries = [
                        entry for conf, entry in classpath
                        if conf in self._confs
                    ]
                    self._process_target_partition(partition, cp_entries)
                    # No exception was thrown, therefore the compile succeded and analysis_file is now valid.
                    if os.path.exists(
                            analysis_file
                    ):  # The compilation created an analysis.
                        # Merge the newly-valid analysis with our global valid analysis.
                        new_valid_analysis = analysis_file + '.valid.new'
                        if self._analysis_parser.is_nonempty_analysis(
                                self._analysis_file):
                            with self.context.new_workunit(
                                    name='update-upstream-analysis'):
                                self._analysis_tools.merge_from_paths(
                                    [self._analysis_file, analysis_file],
                                    new_valid_analysis)
                        else:  # We need to keep analysis_file around. Background tasks may need it.
                            shutil.copy(analysis_file, new_valid_analysis)

                        # Move the merged valid analysis to its proper location.
                        # We do this before checking for missing dependencies, so that we can still
                        # enjoy an incremental compile after fixing missing deps.
                        self.move(new_valid_analysis, self._analysis_file)

                        # Update the products with the latest classes. Must happen before the
                        # missing dependencies check.
                        self._register_products(vts.targets, analysis_file)
                        if self._dep_analyzer:
                            # Check for missing dependencies.
                            actual_deps = self._analysis_parser.parse_deps_from_path(
                                analysis_file, lambda: self.
                                _compute_classpath_elements_by_class(cp_entries
                                                                     ))
                            with self.context.new_workunit(
                                    name='find-missing-dependencies'):
                                self._dep_analyzer.check(sources, actual_deps)

                        # Kick off the background artifact cache write.
                        if self.artifact_cache_writes_enabled():
                            self._write_to_artifact_cache(
                                analysis_file, vts, invalid_sources_by_target)

                    if self._analysis_parser.is_nonempty_analysis(
                            self._invalid_analysis_file):
                        with self.context.new_workunit(
                                name='trim-downstream-analysis'):
                            # Trim out the newly-valid sources from our global invalid analysis.
                            new_invalid_analysis = analysis_file + '.invalid.new'
                            discarded_invalid_analysis = analysis_file + '.invalid.discard'
                            self._analysis_tools.split_to_paths(
                                self._invalid_analysis_file,
                                [(sources, discarded_invalid_analysis)],
                                new_invalid_analysis)
                            self.move(new_invalid_analysis,
                                      self._invalid_analysis_file)

                    # Record the built target -> sources mapping for future use.
                    for target in vts.targets:
                        self._record_sources_by_target(
                            target, sources_by_target.get(target, []))

                    # Now that all the analysis accounting is complete, and we have no missing deps,
                    # we can safely mark the targets as valid.
                    vts.update()
            else:
                # Nothing to build. Register products for all the targets in one go.
                self._register_products(relevant_targets, self._analysis_file)

        self.post_process(relevant_targets)

    def _process_target_partition(self, partition, classpath):
        """Needs invoking only on invalid targets.

    partition - a triple (vts, sources_by_target, analysis_file).
    classpath - a list of classpath entries.

    May be invoked concurrently on independent target sets.

    Postcondition: The individual targets in vts are up-to-date, as if each were
                   compiled individually.
    """
        (vts, sources, analysis_file) = partition

        if not sources:
            self.context.log.warn(
                'Skipping %s compile for targets with no sources:\n  %s' %
                (self._language, vts.targets))
        else:
            # Do some reporting.
            self.context.log.info(
                'Compiling a partition containing ',
                items_to_report_element(sources, 'source'), ' in ',
                items_to_report_element(
                    [t.address.reference() for t in vts.targets], 'target'),
                '.')
            with self.context.new_workunit('compile'):
                # The compiler may delete classfiles, then later exit on a compilation error. Then if the
                # change triggering the error is reverted, we won't rebuild to restore the missing
                # classfiles. So we force-invalidate here, to be on the safe side.
                vts.force_invalidate()
                self.compile(self._args, classpath, sources, self._classes_dir,
                             analysis_file)

    def check_artifact_cache(self, vts):
        # Special handling for scala analysis files. Class files are retrieved directly into their
        # final locations in the global classes dir.

        def post_process_cached_vts(cached_vts):
            # Get all the targets whose artifacts we found in the cache.
            cached_targets = []
            for vt in cached_vts:
                for target in vt.targets:
                    cached_targets.append(target)

            # The current global analysis may contain old data for modified targets for
            # which we got cache hits. We need to strip out this old analysis, to ensure
            # that the new data incoming from the cache doesn't collide with it during the merge.
            sources_to_strip = []
            if os.path.exists(self._analysis_file):
                for target in cached_targets:
                    sources_to_strip.extend(
                        self._get_previous_sources_by_target(target))

            # Localize the cached analyses.
            analyses_to_merge = []
            for target in cached_targets:
                analysis_file = JvmCompile._analysis_for_target(
                    self._analysis_tmpdir, target)
                portable_analysis_file = JvmCompile._portable_analysis_for_target(
                    self._analysis_tmpdir, target)
                if os.path.exists(portable_analysis_file):
                    self._analysis_tools.localize(portable_analysis_file,
                                                  analysis_file)
                if os.path.exists(analysis_file):
                    analyses_to_merge.append(analysis_file)

            # Merge them into the global analysis.
            if analyses_to_merge:
                with temporary_dir() as tmpdir:
                    if sources_to_strip:
                        throwaway = os.path.join(tmpdir, 'throwaway')
                        trimmed_analysis = os.path.join(tmpdir, 'trimmed')
                        self._analysis_tools.split_to_paths(
                            self._analysis_file,
                            [(sources_to_strip, throwaway)], trimmed_analysis)
                    else:
                        trimmed_analysis = self._analysis_file
                    if os.path.exists(trimmed_analysis):
                        analyses_to_merge.append(trimmed_analysis)
                    tmp_analysis = os.path.join(tmpdir, 'analysis')
                    with self.context.new_workunit(name='merge_analysis'):
                        self._analysis_tools.merge_from_paths(
                            analyses_to_merge, tmp_analysis)

                    sources_by_cached_target = self._sources_for_targets(
                        cached_targets)

                    # Record the cached target -> sources mapping for future use.
                    for target, sources in sources_by_cached_target.items():
                        self._record_sources_by_target(target, sources)

                    # Everything's good so move the merged analysis to its final location.
                    if os.path.exists(tmp_analysis):
                        self.move(tmp_analysis, self._analysis_file)

        self._ensure_analysis_tmpdir()
        return self.do_check_artifact_cache(
            vts, post_process_cached_vts=post_process_cached_vts)

    def _write_to_artifact_cache(self, analysis_file, vts, sources_by_target):
        vt_by_target = dict([(vt.target, vt) for vt in vts.versioned_targets])

        split_analysis_files = [
            JvmCompile._analysis_for_target(self._analysis_tmpdir, t)
            for t in vts.targets
        ]
        portable_split_analysis_files = [
            JvmCompile._portable_analysis_for_target(self._analysis_tmpdir, t)
            for t in vts.targets
        ]

        # Set up args for splitting the analysis into per-target files.
        splits = zip([sources_by_target.get(t, []) for t in vts.targets],
                     split_analysis_files)
        splits_args_tuples = [(analysis_file, splits)]

        # Set up args for rebasing the splits.
        relativize_args_tuples = zip(split_analysis_files,
                                     portable_split_analysis_files)

        # Set up args for artifact cache updating.
        vts_artifactfiles_pairs = []
        classes_by_source = self._compute_classes_by_source(analysis_file)
        for target, sources in sources_by_target.items():
            artifacts = []
            for source in sources:
                artifacts.extend(classes_by_source.get(source, []))
            vt = vt_by_target.get(target)
            if vt is not None:
                # NOTE: analysis_file doesn't exist yet.
                vts_artifactfiles_pairs.append((vt, artifacts + [
                    JvmCompile._portable_analysis_for_target(
                        self._analysis_tmpdir, target)
                ]))

        update_artifact_cache_work = self.get_update_artifact_cache_work(
            vts_artifactfiles_pairs)
        if update_artifact_cache_work:
            work_chain = [
                Work(self._analysis_tools.split_to_paths, splits_args_tuples,
                     'split'),
                Work(self._analysis_tools.relativize, relativize_args_tuples,
                     'relativize'), update_artifact_cache_work
            ]
            self.context.submit_background_work_chain(
                work_chain, parent_workunit_name='cache')

    def _compute_classes_by_source(self, analysis_file=None):
        """Compute src->classes.

    Srcs are relative to buildroot. Classes are absolute paths.
    """
        if analysis_file is None:
            analysis_file = self._analysis_file

        if not os.path.exists(analysis_file):
            return {}
        buildroot = get_buildroot()
        products = self._analysis_parser.parse_products_from_path(
            analysis_file)
        classes_by_src = {}
        for src, classes in products.items():
            relsrc = os.path.relpath(src, buildroot)
            classes_by_src[relsrc] = classes
        return classes_by_src

    def _compute_deleted_sources(self):
        """Computes the list of sources present in the last analysis that have since been deleted.

    This is a global list. We have no way of associating them to individual targets.
    Paths are relative to buildroot.
    """
        with self.context.new_workunit('find-deleted-sources'):
            if os.path.exists(self._analysis_file):
                products = self._analysis_parser.parse_products_from_path(
                    self._analysis_file)
                buildroot = get_buildroot()
                old_srcs = products.keys()  # Absolute paths.
                return [
                    os.path.relpath(src, buildroot) for src in old_srcs
                    if not os.path.exists(src)
                ]
            else:
                return []

    def _get_previous_sources_by_target(self, target):
        """Returns the target's sources as recorded on the last successful build of target.

    Returns a list of absolute paths.
    """
        path = os.path.join(self._target_sources_dir, target.identifier)
        if os.path.exists(path):
            with open(path, 'r') as infile:
                return [s.rstrip() for s in infile.readlines()]
        else:
            return []

    def _record_sources_by_target(self, target, sources):
        # Record target -> source mapping for future use.
        with open(os.path.join(self._target_sources_dir, target.identifier),
                  'w') as outfile:
            for src in sources:
                outfile.write(os.path.join(get_buildroot(), src))
                outfile.write('\n')

    def _compute_current_sources_by_target(self, targets):
        """Returns map target -> list of sources (relative to buildroot)."""
        def calculate_sources(target):
            sources = [
                s for s in target.sources_relative_to_buildroot()
                if s.endswith(self._file_suffix)
            ]
            # TODO: Make this less hacky. Ideally target.java_sources will point to sources, not targets.
            if hasattr(target, 'java_sources') and target.java_sources:
                sources.extend(
                    self._resolve_target_sources(target.java_sources, '.java'))
            return sources

        return dict([(t, calculate_sources(t)) for t in targets])

    def _find_locally_changed_targets(self, sources_by_target):
        """Finds the targets whose sources have been modified locally.

    Returns a list of targets, or None if no SCM is available.
    """
        # Compute the src->targets mapping. There should only be one target per source,
        # but that's not yet a hard requirement, so the value is a list of targets.
        # TODO(benjy): Might this inverse mapping be needed elsewhere too?
        targets_by_source = defaultdict(list)
        for tgt, srcs in sources_by_target.items():
            for src in srcs:
                targets_by_source[src].append(tgt)

        ret = OrderedSet()
        scm = get_scm()
        if not scm:
            return None
        changed_files = scm.changed_files(include_untracked=True)
        for f in changed_files:
            ret.update(targets_by_source.get(f, []))
        return list(ret)

    def _resolve_target_sources(self, target_sources, extension=None):
        """Given a list of pants targets, extract their sources as a list.

    Filters against the extension if given and optionally returns the paths relative to the target
    base.
    """
        resolved_sources = []
        for target in target_sources:
            if target.has_sources():
                resolved_sources.extend(target.sources_relative_to_buildroot())
        return resolved_sources

    def _compute_classpath_elements_by_class(self, classpath):
        # Don't consider loose classes dirs in our classpath. Those will be considered
        # separately, by looking at products.
        def non_product(path):
            return not (path.startswith(self._pants_workdir)
                        and os.path.isdir(path))

        classpath_jars = filter(non_product, classpath)
        if self._class_to_jarfile is None:
            self._class_to_jarfile = {}
            for jarpath in self.find_all_bootstrap_jars() + classpath_jars:
                # Per the classloading spec, a 'jar' in this context can also be a .zip file.
                if os.path.isfile(jarpath) and ((jarpath.endswith('.jar')
                                                 or jarpath.endswith('.zip'))):
                    with open_zip(jarpath, 'r') as jar:
                        for cls in jar.namelist():
                            # First jar with a given class wins, just like when classloading.
                            if cls.endswith(
                                    b'.class'
                            ) and not cls in self._class_to_jarfile:
                                self._class_to_jarfile[cls] = jarpath
                elif os.path.isdir(jarpath):
                    for dirpath, _, filenames in os.walk(jarpath,
                                                         followlinks=True):
                        for f in filter(lambda x: x.endswith('.class'),
                                        filenames):
                            cls = os.path.relpath(os.path.join(dirpath, f),
                                                  jarpath)
                            if not cls in self._class_to_jarfile:
                                self._class_to_jarfile[cls] = jarpath
        return self._class_to_jarfile

    def find_all_bootstrap_jars(self):
        def get_path(key):
            return self.context.java_sysprops.get(key, '').split(':')

        def find_jars_in_dirs(dirs):
            ret = []
            for d in dirs:
                if os.path.isdir(d):
                    ret.extend(
                        filter(lambda s: s.endswith('.jar'), os.listdir(d)))
            return ret

        # Note: assumes HotSpot, or some JVM that supports sun.boot.class.path.
        # TODO: Support other JVMs? Not clear if there's a standard way to do so.
        # May include loose classes dirs.
        boot_classpath = get_path('sun.boot.class.path')

        # Note that per the specs, overrides and extensions must be in jars.
        # Loose class files will not be found by the JVM.
        override_jars = find_jars_in_dirs(get_path('java.endorsed.dirs'))
        extension_jars = find_jars_in_dirs(get_path('java.ext.dirs'))

        # Note that this order matters: it reflects the classloading order.
        bootstrap_jars = filter(
            os.path.isfile, override_jars + boot_classpath + extension_jars)
        return bootstrap_jars  # Technically, may include loose class dirs from boot_classpath.

    @property
    def _analysis_tools(self):
        if self._lazy_analysis_tools is None:
            self._lazy_analysis_tools = self.create_analysis_tools()
        return self._lazy_analysis_tools

    @property
    def _analysis_parser(self):
        return self._analysis_tools.parser

    def _sources_for_targets(self, targets):
        """Returns a map target->sources for the specified targets."""
        if self._sources_by_target is None:
            raise TaskError('self._sources_by_target not computed yet.')
        return dict((t, self._sources_by_target.get(t, [])) for t in targets)

    # Work in a tmpdir so we don't stomp the main analysis files on error.
    # The tmpdir is cleaned up in a shutdown hook, because background work
    # may need to access files we create there even after this method returns.
    def _ensure_analysis_tmpdir(self):
        # Do this lazily, so we don't trigger creation of a worker pool unless we need it.
        if not os.path.exists(self._analysis_tmpdir):
            os.makedirs(self._analysis_tmpdir)
            if self._delete_scratch:
                self.context.background_worker_pool().add_shutdown_hook(
                    lambda: safe_rmtree(self._analysis_tmpdir))

    def _create_empty_products(self):
        make_products = lambda: defaultdict(MultipleRootedProducts)
        if self.context.products.is_required_data('classes_by_source'):
            self.context.products.safe_create_data('classes_by_source',
                                                   make_products)
        if self.context.products.is_required_data('classes_by_target'):
            self.context.products.safe_create_data('classes_by_target',
                                                   make_products)
        if self.context.products.is_required_data('resources_by_target'):
            self.context.products.safe_create_data('resources_by_target',
                                                   make_products)

    def _register_products(self, targets, analysis_file):
        classes_by_source = self.context.products.get_data('classes_by_source')
        classes_by_target = self.context.products.get_data('classes_by_target')
        resources_by_target = self.context.products.get_data(
            'resources_by_target')

        if classes_by_source is not None or classes_by_target is not None:
            computed_classes_by_source = self._compute_classes_by_source(
                analysis_file)
            for target in targets:
                target_products = classes_by_target[
                    target] if classes_by_target is not None else None
                for source in self._sources_by_target.get(
                        target, []):  # Source is relative to buildroot.
                    classes = computed_classes_by_source.get(
                        source, [])  # Classes are absolute paths.
                    if classes_by_target is not None:
                        target_products.add_abs_paths(self._classes_dir,
                                                      classes)
                    if classes_by_source is not None:
                        classes_by_source[source].add_abs_paths(
                            self._classes_dir, classes)

        # TODO(pl): https://github.com/pantsbuild/pants/issues/206
        if resources_by_target is not None:
            for target in targets:
                target_resources = resources_by_target[target]
                for root, abs_paths in self.extra_products(target):
                    target_resources.add_abs_paths(root, abs_paths)
Exemplo n.º 11
0
    def __init__(self, context, workdir, minimum_version=None, jdk=False):
        # TODO(John Sirois): XXX plumb minimum_version via config or flags
        super(JvmCompile, self).__init__(context,
                                         workdir,
                                         minimum_version=minimum_version,
                                         jdk=jdk)
        concrete_class = type(self)
        config_section = concrete_class._config_section

        def get_lang_specific_option(opt):
            full_opt_name = self._language + '_' + opt
            return getattr(context.options, full_opt_name, None)

        # Global workdir.
        self._pants_workdir = context.config.getdefault('pants_workdir')

        # Various working directories.
        self._classes_dir = os.path.join(self.workdir, 'classes')
        self._resources_dir = os.path.join(self.workdir, 'resources')
        self._analysis_dir = os.path.join(self.workdir, 'analysis')
        self._target_sources_dir = os.path.join(self.workdir, 'target_sources')

        self._delete_scratch = get_lang_specific_option('delete_scratch')

        safe_mkdir(self._classes_dir)
        safe_mkdir(self._analysis_dir)
        safe_mkdir(self._target_sources_dir)

        self._analysis_file = os.path.join(self._analysis_dir,
                                           'global_analysis.valid')
        self._invalid_analysis_file = os.path.join(self._analysis_dir,
                                                   'global_analysis.invalid')

        # A temporary, but well-known, dir in which to munge analysis/dependency files in before
        # caching. It must be well-known so we know where to find the files when we retrieve them from
        # the cache.
        self._analysis_tmpdir = os.path.join(self._analysis_dir,
                                             'artifact_cache_tmpdir')

        # We can't create analysis tools until after construction.
        self._lazy_analysis_tools = None

        # Compiler options.
        self._args = context.config.getlist(config_section, 'args')
        if get_lang_specific_option('compile_warnings'):
            self._args.extend(
                context.config.getlist(config_section, 'warning_args'))
        else:
            self._args.extend(
                context.config.getlist(config_section, 'no_warning_args'))

        # The rough number of source files to build in each compiler pass.
        self._partition_size_hint = get_lang_specific_option(
            'partition_size_hint')
        if self._partition_size_hint == -1:
            self._partition_size_hint = context.config.getint(
                config_section, 'partition_size_hint', default=1000)

        # JVM options for running the compiler.
        self._jvm_options = context.config.getlist(config_section, 'jvm_args')

        # The ivy confs for which we're building.
        self._confs = context.config.getlist(config_section,
                                             'confs',
                                             default=['default'])

        # Runtime dependencies.
        runtime_deps = context.config.getlist(config_section,
                                              'runtime-deps',
                                              default=[])
        if runtime_deps:
            self._runtime_deps_key = self._language + '-runtime-deps'
            self.register_jvm_tool(self._runtime_deps_key, runtime_deps)
        else:
            self._runtime_deps_key = None

        # Set up dep checking if needed.
        def munge_flag(flag):
            return None if flag == 'off' else flag

        check_missing_deps = munge_flag(
            get_lang_specific_option('missing_deps'))
        check_missing_direct_deps = munge_flag(
            get_lang_specific_option('missing_direct_deps'))
        check_unnecessary_deps = munge_flag(
            get_lang_specific_option('unnecessary_deps'))

        if check_missing_deps or check_missing_direct_deps or check_unnecessary_deps:
            # Must init it here, so it can set requirements on the context.
            self._dep_analyzer = JvmDependencyAnalyzer(
                self.context, check_missing_deps, check_missing_direct_deps,
                check_unnecessary_deps)
        else:
            self._dep_analyzer = None

        # If non-zero, and we have fewer than this number of locally-changed targets,
        # then we partition them separately, to preserve stability in the face of repeated
        # compilations.
        self._locally_changed_targets_heuristic_limit = context.config.getint(
            config_section, 'locally_changed_targets_heuristic_limit', 0)

        self._class_to_jarfile = None  # Computed lazily as needed.

        self.context.products.require_data('exclusives_groups')
        self.setup_artifact_cache_from_config(config_section=config_section)

        # Sources (relative to buildroot) present in the last analysis that have since been deleted.
        # Populated in prepare_execute().
        self._deleted_sources = None

        # Map of target -> list of sources (relative to buildroot), for all targets in all chunks.
        # Populated in prepare_execute().
        self._sources_by_target = None
Exemplo n.º 12
0
    def __init__(self, *args, **kwargs):
        super(JvmCompile, self).__init__(*args, **kwargs)

        # Various working directories.
        self._classes_dir = os.path.join(self.workdir, 'classes')
        self._resources_dir = os.path.join(self.workdir, 'resources')
        self._analysis_dir = os.path.join(self.workdir, 'analysis')
        self._target_sources_dir = os.path.join(self.workdir, 'target_sources')

        self._delete_scratch = self.get_options().delete_scratch

        self._analysis_file = os.path.join(self._analysis_dir,
                                           'global_analysis.valid')
        self._invalid_analysis_file = os.path.join(self._analysis_dir,
                                                   'global_analysis.invalid')

        # A temporary, but well-known, dir in which to munge analysis/dependency files in before
        # caching. It must be well-known so we know where to find the files when we retrieve them from
        # the cache.
        self._analysis_tmpdir = os.path.join(self._analysis_dir,
                                             'artifact_cache_tmpdir')

        # We can't create analysis tools until after construction.
        self._lazy_analysis_tools = None

        # The rough number of source files to build in each compiler pass.
        self._partition_size_hint = self.get_options().partition_size_hint

        # JVM options for running the compiler.
        self._jvm_options = self.get_options().jvm_options

        # The ivy confs for which we're building.
        self._confs = self.get_options().confs

        self._args = list(self.get_options().args)
        if self.get_options().warnings:
            self._args.extend(self.get_options().warning_args)
        else:
            self._args.extend(self.get_options().no_warning_args)

        # Set up dep checking if needed.
        def munge_flag(flag):
            flag_value = getattr(self.get_options(), flag, None)
            return None if flag_value == 'off' else flag_value

        check_missing_deps = munge_flag('missing_deps')
        check_missing_direct_deps = munge_flag('missing_direct_deps')
        check_unnecessary_deps = munge_flag('unnecessary_deps')

        if check_missing_deps or check_missing_direct_deps or check_unnecessary_deps:
            target_whitelist = self.get_options().missing_deps_whitelist
            # Must init it here, so it can set requirements on the context.
            self._dep_analyzer = JvmDependencyAnalyzer(
                self.context, check_missing_deps, check_missing_direct_deps,
                check_unnecessary_deps, target_whitelist)
        else:
            self._dep_analyzer = None

        # If non-zero, and we have fewer than this number of locally-changed targets,
        # then we partition them separately, to preserve stability in the face of repeated
        # compilations.
        self._changed_targets_heuristic_limit = self.get_options(
        ).changed_targets_heuristic_limit

        self._upstream_class_to_path = None  # Computed lazily as needed.
        self.setup_artifact_cache()

        # Sources (relative to buildroot) present in the last analysis that have since been deleted.
        # Populated in prepare_execute().
        self._deleted_sources = None

        # Map of target -> list of sources (relative to buildroot), for all targets in all chunks.
        # Populated in prepare_execute().
        self._sources_by_target = None
Exemplo n.º 13
0
class JvmCompile(NailgunTaskBase, GroupMember):
  """A common framework for JVM compilation.

  To subclass for a specific JVM language, implement the static values and methods
  mentioned below under "Subclasses must implement".
  """

  @classmethod
  def register_options(cls, register):
    super(JvmCompile, cls).register_options(register)
    register('--partition-size-hint', type=int, default=sys.maxint, metavar='<# source files>',
             help='Roughly how many source files to attempt to compile together. Set to a large '
                  'number to compile all sources together. Set to 0 to compile target-by-target.')

    register('--jvm-options', type=Options.list,
             help='Run the compiler with these JVM options.')

    register('--args', action='append', default=list(cls.get_args_default(register.bootstrap)),
             help='Pass these args to the compiler.')

    register('--confs', type=Options.list, default=['default'],
             help='Compile for these Ivy confs.')

    register('--warnings', default=True, action='store_true',
             help='Compile with all configured warnings enabled.')

    register('--warning-args', action='append', default=list(cls.get_warning_args_default()),
             help='Extra compiler args to use when warnings are enabled.')

    register('--no-warning-args', action='append', default=list(cls.get_no_warning_args_default()),
             help='Extra compiler args to use when warnings are disabled.')

    register('--missing-deps', choices=['off', 'warn', 'fatal'], default='warn',
             help='Check for missing dependencies in {0} code. Reports actual dependencies A -> B '
                  'where there is no transitive BUILD file dependency path from A to B. If fatal, '
                  'missing deps are treated as a build error.'.format(cls._language))

    register('--missing-direct-deps', choices=['off', 'warn', 'fatal'], default='off',
             help='Check for missing direct dependencies in {0} code. Reports actual dependencies '
                  'A -> B where there is no direct BUILD file dependency path from A to B. This is '
                  'a very strict check; In practice it is common to rely on transitive, indirect '
                  'dependencies, e.g., due to type inference or when the main target in a BUILD '
                  'file is modified to depend on other targets in the same BUILD file, as an '
                  'implementation detail. However it may still be useful to use this on '
                  'occasion. '.format(cls._language))

    register('--missing-deps-whitelist', type=Options.list,
             help="Don't report these targets even if they have missing deps.")

    register('--unnecessary-deps', choices=['off', 'warn', 'fatal'], default='off',
             help='Check for declared dependencies in {0} code that are not needed. This is a very '
                  'strict check. For example, generated code will often legitimately have BUILD '
                  'dependencies that are unused in practice.'.format(cls._language))

    register('--changed-targets-heuristic-limit', type=int, default=0,
             help='If non-zero, and we have fewer than this number of locally-changed targets, '
                  'partition them separately, to preserve stability when compiling repeatedly.')

    register('--delete-scratch', default=True, action='store_true',
             help='Leave intermediate scratch files around, for debugging build problems.')

  @classmethod
  def product_types(cls):
    return ['classes_by_target', 'classes_by_source', 'resources_by_target']

  @classmethod
  def prepare(cls, options, round_manager):
    super(JvmCompile, cls).prepare(options, round_manager)

    # This task uses JvmDependencyAnalyzer as a helper, get its product needs
    JvmDependencyAnalyzer.prepare(options, round_manager)

    round_manager.require_data('compile_classpath')
    round_manager.require_data('ivy_cache_dir')
    round_manager.require_data('ivy_resolve_symlink_map')

    # Require codegen we care about
    # TODO(John Sirois): roll this up in Task - if the list of labels we care about for a target
    # predicate to filter the full build graph is exposed, the requirement can be made automatic
    # and in turn codegen tasks could denote the labels they produce automating wiring of the
    # produce side
    round_manager.require_data('java')
    round_manager.require_data('scala')

    # Allow the deferred_sources_mapping to take place first
    round_manager.require_data('deferred_sources')

  # Subclasses must implement.
  # --------------------------
  _language = None
  _file_suffix = None

  @classmethod
  def name(cls):
    return cls._language

  @classmethod
  def get_args_default(cls, bootstrap_option_values):
    """Override to set default for --args option.

    :param bootstrap_option_values: The values of the "bootstrap options" (e.g., pants_workdir).
                                    Implementations can use these when generating the default.
                                    See src/python/pants/options/options_bootstrapper.py for
                                    details.
    """
    return ()

  @classmethod
  def get_warning_args_default(cls):
    """Override to set default for --warning-args option."""
    return ()

  @classmethod
  def get_no_warning_args_default(cls):
    """Override to set default for --no-warning-args option."""
    return ()

  @property
  def config_section(self):
    return self.options_scope

  def select(self, target):
    return target.has_sources(self._file_suffix)

  def create_analysis_tools(self):
    """Returns an AnalysisTools implementation.

    Subclasses must implement.
    """
    raise NotImplementedError()

  def compile(self, args, classpath, sources, classes_output_dir, analysis_file):
    """Invoke the compiler.

    Must raise TaskError on compile failure.

    Subclasses must implement."""
    raise NotImplementedError()

  # Subclasses may override.
  # ------------------------
  def extra_compile_time_classpath_elements(self):
    """Extra classpath elements common to all compiler invocations.

    E.g., jars for compiler plugins.
    """
    return []

  def extra_products(self, target):
    """Any extra, out-of-band products created for a target.

    E.g., targets that produce scala compiler plugins produce an info file.
    Returns a list of pairs (root, [absolute paths of files under root]).
    """
    return []

  def post_process(self, relevant_targets):
    """Any extra post-execute work."""
    pass

  # Common code.
  # ------------
  @staticmethod
  def _analysis_for_target(analysis_dir, target):
    return os.path.join(analysis_dir, target.id + '.analysis')

  @staticmethod
  def _portable_analysis_for_target(analysis_dir, target):
    return JvmCompile._analysis_for_target(analysis_dir, target) + '.portable'

  def __init__(self, *args, **kwargs):
    super(JvmCompile, self).__init__(*args, **kwargs)

    # Various working directories.
    self._classes_dir = os.path.join(self.workdir, 'classes')
    self._resources_dir = os.path.join(self.workdir, 'resources')
    self._analysis_dir = os.path.join(self.workdir, 'analysis')
    self._target_sources_dir = os.path.join(self.workdir, 'target_sources')

    self._delete_scratch = self.get_options().delete_scratch

    self._analysis_file = os.path.join(self._analysis_dir, 'global_analysis.valid')
    self._invalid_analysis_file = os.path.join(self._analysis_dir, 'global_analysis.invalid')

    # A temporary, but well-known, dir in which to munge analysis/dependency files in before
    # caching. It must be well-known so we know where to find the files when we retrieve them from
    # the cache.
    self._analysis_tmpdir = os.path.join(self._analysis_dir, 'artifact_cache_tmpdir')

    # We can't create analysis tools until after construction.
    self._lazy_analysis_tools = None

    # The rough number of source files to build in each compiler pass.
    self._partition_size_hint = self.get_options().partition_size_hint

    # JVM options for running the compiler.
    self._jvm_options = self.get_options().jvm_options

    # The ivy confs for which we're building.
    self._confs = self.get_options().confs

    self._args = list(self.get_options().args)
    if self.get_options().warnings:
      self._args.extend(self.get_options().warning_args)
    else:
      self._args.extend(self.get_options().no_warning_args)

    # Set up dep checking if needed.
    def munge_flag(flag):
      flag_value = getattr(self.get_options(), flag, None)
      return None if flag_value == 'off' else flag_value

    check_missing_deps = munge_flag('missing_deps')
    check_missing_direct_deps = munge_flag('missing_direct_deps')
    check_unnecessary_deps = munge_flag('unnecessary_deps')

    if check_missing_deps or check_missing_direct_deps or check_unnecessary_deps:
      target_whitelist = self.get_options().missing_deps_whitelist
      # Must init it here, so it can set requirements on the context.
      self._dep_analyzer = JvmDependencyAnalyzer(self.context,
                                                 check_missing_deps,
                                                 check_missing_direct_deps,
                                                 check_unnecessary_deps,
                                                 target_whitelist)
    else:
      self._dep_analyzer = None

    # If non-zero, and we have fewer than this number of locally-changed targets,
    # then we partition them separately, to preserve stability in the face of repeated
    # compilations.
    self._changed_targets_heuristic_limit = self.get_options().changed_targets_heuristic_limit

    self._upstream_class_to_path = None  # Computed lazily as needed.
    self.setup_artifact_cache()

    # Sources (relative to buildroot) present in the last analysis that have since been deleted.
    # Populated in prepare_execute().
    self._deleted_sources = None

    # Map of target -> list of sources (relative to buildroot), for all targets in all chunks.
    # Populated in prepare_execute().
    self._sources_by_target = None

  def move(self, src, dst):
    if self._delete_scratch:
      shutil.move(src, dst)
    else:
      shutil.copy(src, dst)

  def _jvm_fingerprint_strategy(self):
    # Use a fingerprint strategy that allows us to also include java/scala versions.
    return JvmFingerprintStrategy(self.platform_version_info())

  def platform_version_info(self):
    """
    Provides extra platform information such as java version that will be used
    in the fingerprinter. This in turn ensures different platform versions create different
    cache artifacts.

    Sublclasses should override this and return a list of version info.
    """
    return None

  def pre_execute(self):
    # Only create these working dirs during execution phase, otherwise, they
    # would be wiped out by clean-all goal/task if it's specified.
    safe_mkdir(self._classes_dir)
    safe_mkdir(self._analysis_dir)
    safe_mkdir(self._target_sources_dir)

    # TODO(John Sirois): Ensuring requested product maps are available - if empty - should probably
    # be lifted to Task infra.

    # In case we have no relevant targets and return early create the requested product maps.
    self._create_empty_products()

  def prepare_execute(self, chunks):
    all_targets = list(itertools.chain(*chunks))

    # Update the classpath for us and for downstream tasks.
    compile_classpath = self.context.products.get_data('compile_classpath')
    for conf in self._confs:
      compile_classpath.update([(conf, self._classes_dir), (conf, self._resources_dir)])

    # Target -> sources (relative to buildroot).
    # TODO(benjy): Should sources_by_target be available in all Tasks?
    self._sources_by_target = self._compute_current_sources_by_target(all_targets)

    # Split the global analysis file into valid and invalid parts.
    cache_manager = self.create_cache_manager(invalidate_dependents=True,
                                              fingerprint_strategy=self._jvm_fingerprint_strategy())
    invalidation_check = cache_manager.check(all_targets)
    if invalidation_check.invalid_vts:
      # The analysis for invalid and deleted sources is no longer valid.
      invalid_targets = [vt.target for vt in invalidation_check.invalid_vts]
      invalid_sources_by_target = {}
      for tgt in invalid_targets:
        invalid_sources_by_target[tgt] = self._sources_by_target[tgt]
      invalid_sources = list(itertools.chain.from_iterable(invalid_sources_by_target.values()))
      self._deleted_sources = self._compute_deleted_sources()

      self._ensure_analysis_tmpdir()
      tmpdir = os.path.join(self._analysis_tmpdir, str(uuid.uuid4()))
      os.mkdir(tmpdir)
      valid_analysis_tmp = os.path.join(tmpdir, 'valid_analysis')
      newly_invalid_analysis_tmp = os.path.join(tmpdir, 'newly_invalid_analysis')
      invalid_analysis_tmp = os.path.join(tmpdir, 'invalid_analysis')
      if self._analysis_parser.is_nonempty_analysis(self._analysis_file):
        with self.context.new_workunit(name='prepare-analysis'):
          self._analysis_tools.split_to_paths(self._analysis_file,
              [(invalid_sources + self._deleted_sources, newly_invalid_analysis_tmp)],
              valid_analysis_tmp)
          if self._analysis_parser.is_nonempty_analysis(self._invalid_analysis_file):
            self._analysis_tools.merge_from_paths(
              [self._invalid_analysis_file, newly_invalid_analysis_tmp], invalid_analysis_tmp)
          else:
            invalid_analysis_tmp = newly_invalid_analysis_tmp

          # Now it's OK to overwrite the main analysis files with the new state.
          self.move(valid_analysis_tmp, self._analysis_file)
          self.move(invalid_analysis_tmp, self._invalid_analysis_file)
    else:
      self._deleted_sources = []

  # TODO(benjy): Break this monstrosity up? Previous attempts to do so
  #              turned out to be more trouble than it was worth.
  def execute_chunk(self, relevant_targets):
    # TODO(benjy): Add a pre-execute goal for injecting deps into targets, so e.g.,
    # we can inject a dep on the scala runtime library and still have it ivy-resolve.

    if not relevant_targets:
      return

    # Get the classpath generated by upstream JVM tasks and our own prepare_execute().
    compile_classpath = self.context.products.get_data('compile_classpath')

    # Add any extra compile-time-only classpath elements.
    # TODO(benjy): Model compile-time vs. runtime classpaths more explicitly.
    def extra_compile_classpath_iter():
      for conf in self._confs:
        for jar in self.extra_compile_time_classpath_elements():
           yield (conf, jar)
    compile_classpath = OrderedSet(list(extra_compile_classpath_iter()) + list(compile_classpath))

    # Target -> sources (relative to buildroot), for just this chunk's targets.
    sources_by_target = self._sources_for_targets(relevant_targets)

    # If needed, find targets that we've changed locally (as opposed to
    # changes synced in from the SCM).
    # TODO(benjy): Should locally_changed_targets be available in all Tasks?
    locally_changed_targets = None
    if self._changed_targets_heuristic_limit:
      locally_changed_targets = self._find_locally_changed_targets(sources_by_target)
      if (locally_changed_targets and
          len(locally_changed_targets) > self._changed_targets_heuristic_limit):
        locally_changed_targets = None

    # Invalidation check. Everything inside the with block must succeed for the
    # invalid targets to become valid.
    with self.invalidated(relevant_targets,
                          invalidate_dependents=True,
                          partition_size_hint=self._partition_size_hint,
                          locally_changed_targets=locally_changed_targets,
                          fingerprint_strategy=self._jvm_fingerprint_strategy(),
                          topological_order=True) as invalidation_check:
      if invalidation_check.invalid_vts:
        # Find the invalid sources for this chunk.
        invalid_targets = [vt.target for vt in invalidation_check.invalid_vts]
        invalid_sources_by_target = self._sources_for_targets(invalid_targets)

        tmpdir = os.path.join(self._analysis_tmpdir, str(uuid.uuid4()))
        os.mkdir(tmpdir)

        # Register products for all the valid targets.
        # We register as we go, so dependency checking code can use this data.
        valid_targets = list(set(relevant_targets) - set(invalid_targets))
        self._register_products(valid_targets, self._analysis_file)

        # Figure out the sources and analysis belonging to each partition.
        partitions = []  # Each element is a triple (vts, sources_by_target, analysis).
        for vts in invalidation_check.invalid_vts_partitioned:
          partition_tmpdir = os.path.join(tmpdir, Target.maybe_readable_identify(vts.targets))
          os.mkdir(partition_tmpdir)
          sources = list(itertools.chain.from_iterable(
              [invalid_sources_by_target.get(t, []) for t in vts.targets]))
          de_duped_sources = list(OrderedSet(sources))
          if len(sources) != len(de_duped_sources):
            counts = [(src, len(list(srcs))) for src, srcs in itertools.groupby(sorted(sources))]
            self.context.log.warn(
                'De-duped the following sources:\n\t%s' %
                '\n\t'.join(sorted('%d %s' % (cnt, src) for src, cnt in counts if cnt > 1)))
          analysis_file = os.path.join(partition_tmpdir, 'analysis')
          partitions.append((vts, de_duped_sources, analysis_file))

        # Split per-partition files out of the global invalid analysis.
        if self._analysis_parser.is_nonempty_analysis(self._invalid_analysis_file) and partitions:
          with self.context.new_workunit(name='partition-analysis'):
            splits = [(x[1], x[2]) for x in partitions]
            # We have to pass the analysis for any deleted files through zinc, to give it
            # a chance to delete the relevant class files.
            if splits:
              splits[0] = (splits[0][0] + self._deleted_sources, splits[0][1])
            self._analysis_tools.split_to_paths(self._invalid_analysis_file, splits)

        # Now compile partitions one by one.
        for partition_index, partition in enumerate(partitions):
          (vts, sources, analysis_file) = partition
          cp_entries = [entry for conf, entry in compile_classpath if conf in self._confs]

          progress_message = '{} of {}'.format(partition_index + 1, len(partitions))
          self._process_target_partition(partition, cp_entries, progress_message)

          # No exception was thrown, therefore the compile succeded and analysis_file is now valid.
          if os.path.exists(analysis_file):  # The compilation created an analysis.
            # Merge the newly-valid analysis with our global valid analysis.
            new_valid_analysis = analysis_file + '.valid.new'
            if self._analysis_parser.is_nonempty_analysis(self._analysis_file):
              with self.context.new_workunit(name='update-upstream-analysis'):
                self._analysis_tools.merge_from_paths([self._analysis_file, analysis_file],
                                                      new_valid_analysis)
            else:  # We need to keep analysis_file around. Background tasks may need it.
              shutil.copy(analysis_file, new_valid_analysis)

            # Move the merged valid analysis to its proper location.
            # We do this before checking for missing dependencies, so that we can still
            # enjoy an incremental compile after fixing missing deps.
            self.move(new_valid_analysis, self._analysis_file)

            # Update the products with the latest classes. Must happen before the
            # missing dependencies check.
            self._register_products(vts.targets, analysis_file)
            if self._dep_analyzer:
              # Check for missing dependencies.
              actual_deps = self._analysis_parser.parse_deps_from_path(analysis_file,
                  lambda: self._compute_classpath_elements_by_class(cp_entries))
              with self.context.new_workunit(name='find-missing-dependencies'):
                self._dep_analyzer.check(sources, actual_deps, self.ivy_cache_dir)

            # Kick off the background artifact cache write.
            if self.artifact_cache_writes_enabled():
              self._write_to_artifact_cache(analysis_file, vts, invalid_sources_by_target)

          if self._analysis_parser.is_nonempty_analysis(self._invalid_analysis_file):
            with self.context.new_workunit(name='trim-downstream-analysis'):
              # Trim out the newly-valid sources from our global invalid analysis.
              new_invalid_analysis = analysis_file + '.invalid.new'
              discarded_invalid_analysis = analysis_file + '.invalid.discard'
              self._analysis_tools.split_to_paths(self._invalid_analysis_file,
                [(sources, discarded_invalid_analysis)], new_invalid_analysis)
              self.move(new_invalid_analysis, self._invalid_analysis_file)

          # Record the built target -> sources mapping for future use.
          for target in vts.targets:
            self._record_sources_by_target(target, sources_by_target.get(target, []))

          # Now that all the analysis accounting is complete, and we have no missing deps,
          # we can safely mark the targets as valid.
          vts.update()
      else:
        # Nothing to build. Register products for all the targets in one go.
        self._register_products(relevant_targets, self._analysis_file)

    self.post_process(relevant_targets)

  def _process_target_partition(self, partition, classpath, progress_message):
    """Needs invoking only on invalid targets.

    partition - a triple (vts, sources_by_target, analysis_file).
    classpath - a list of classpath entries.

    May be invoked concurrently on independent target sets.

    Postcondition: The individual targets in vts are up-to-date, as if each were
                   compiled individually.
    """
    (vts, sources, analysis_file) = partition

    if not sources:
      self.context.log.warn('Skipping %s compile for targets with no sources:\n  %s'
                            % (self._language, vts.targets))
    else:
      # Do some reporting.
      self.context.log.info(
        'Compiling a partition containing ',
        items_to_report_element(sources, 'source'),
        ' in ',
        items_to_report_element([t.address.reference() for t in vts.targets], 'target'),
        ' (partition ',
        progress_message,
        ').')
      with self.context.new_workunit('compile'):
        # The compiler may delete classfiles, then later exit on a compilation error. Then if the
        # change triggering the error is reverted, we won't rebuild to restore the missing
        # classfiles. So we force-invalidate here, to be on the safe side.
        vts.force_invalidate()
        self.compile(self._args, classpath, sources, self._classes_dir, analysis_file)

  def check_artifact_cache(self, vts):
    # Special handling for scala analysis files. Class files are retrieved directly into their
    # final locations in the global classes dir.

    def post_process_cached_vts(cached_vts):
      # Get all the targets whose artifacts we found in the cache.
      cached_targets = []
      for vt in cached_vts:
        for target in vt.targets:
          cached_targets.append(target)

      # The current global analysis may contain old data for modified targets for
      # which we got cache hits. We need to strip out this old analysis, to ensure
      # that the new data incoming from the cache doesn't collide with it during the merge.
      sources_to_strip = []
      if os.path.exists(self._analysis_file):
        for target in cached_targets:
          sources_to_strip.extend(self._get_previous_sources_by_target(target))

      # Localize the cached analyses.
      analyses_to_merge = []
      for target in cached_targets:
        analysis_file = JvmCompile._analysis_for_target(self._analysis_tmpdir, target)
        portable_analysis_file = JvmCompile._portable_analysis_for_target(self._analysis_tmpdir,
                                                                          target)
        if os.path.exists(portable_analysis_file):
          self._analysis_tools.localize(portable_analysis_file, analysis_file)
        if os.path.exists(analysis_file):
          analyses_to_merge.append(analysis_file)

      # Merge them into the global analysis.
      if analyses_to_merge:
        with temporary_dir() as tmpdir:
          if sources_to_strip:
            throwaway = os.path.join(tmpdir, 'throwaway')
            trimmed_analysis = os.path.join(tmpdir, 'trimmed')
            self._analysis_tools.split_to_paths(self._analysis_file,
                                            [(sources_to_strip, throwaway)],
                                            trimmed_analysis)
          else:
            trimmed_analysis = self._analysis_file
          if os.path.exists(trimmed_analysis):
            analyses_to_merge.append(trimmed_analysis)
          tmp_analysis = os.path.join(tmpdir, 'analysis')
          with self.context.new_workunit(name='merge_analysis'):
            self._analysis_tools.merge_from_paths(analyses_to_merge, tmp_analysis)

          sources_by_cached_target = self._sources_for_targets(cached_targets)

          # Record the cached target -> sources mapping for future use.
          for target, sources in sources_by_cached_target.items():
            self._record_sources_by_target(target, sources)

          # Everything's good so move the merged analysis to its final location.
          if os.path.exists(tmp_analysis):
            self.move(tmp_analysis, self._analysis_file)

    self._ensure_analysis_tmpdir()
    return self.do_check_artifact_cache(vts, post_process_cached_vts=post_process_cached_vts)

  def _write_to_artifact_cache(self, analysis_file, vts, sources_by_target):
    vt_by_target = dict([(vt.target, vt) for vt in vts.versioned_targets])

    vts_targets = [t for t in vts.targets if not t.has_label('no_cache')]

    split_analysis_files = [
        JvmCompile._analysis_for_target(self._analysis_tmpdir, t) for t in vts_targets]
    portable_split_analysis_files = [
        JvmCompile._portable_analysis_for_target(self._analysis_tmpdir, t) for t in vts_targets]

    # Set up args for splitting the analysis into per-target files.
    splits = zip([sources_by_target.get(t, []) for t in vts_targets], split_analysis_files)
    splits_args_tuples = [(analysis_file, splits)]

    # Set up args for rebasing the splits.
    relativize_args_tuples = zip(split_analysis_files, portable_split_analysis_files)

    # Set up args for artifact cache updating.
    vts_artifactfiles_pairs = []
    classes_by_source = self._compute_classes_by_source(analysis_file)
    resources_by_target = self.context.products.get_data('resources_by_target')
    for target, sources in sources_by_target.items():
      if target.has_label('no_cache'):
        continue
      artifacts = []
      if resources_by_target is not None:
        for _, paths in resources_by_target[target].abs_paths():
          artifacts.extend(paths)
      for source in sources:
        classes = classes_by_source.get(source, [])
        artifacts.extend(classes)

      vt = vt_by_target.get(target)
      if vt is not None:
        # NOTE: analysis_file doesn't exist yet.
        vts_artifactfiles_pairs.append(
            (vt,
             artifacts + [JvmCompile._portable_analysis_for_target(self._analysis_tmpdir, target)]))

    update_artifact_cache_work = self.get_update_artifact_cache_work(vts_artifactfiles_pairs)
    if update_artifact_cache_work:
      work_chain = [
        Work(self._analysis_tools.split_to_paths, splits_args_tuples, 'split'),
        Work(self._analysis_tools.relativize, relativize_args_tuples, 'relativize'),
        update_artifact_cache_work
      ]
      self.context.submit_background_work_chain(work_chain, parent_workunit_name='cache')

  def _compute_classes_by_source(self, analysis_file=None):
    """Compute src->classes.

    Srcs are relative to buildroot. Classes are absolute paths.
    """
    if analysis_file is None:
      analysis_file = self._analysis_file

    if not os.path.exists(analysis_file):
      return {}
    buildroot = get_buildroot()
    products = self._analysis_parser.parse_products_from_path(analysis_file)
    classes_by_src = {}
    for src, classes in products.items():
      relsrc = os.path.relpath(src, buildroot)
      classes_by_src[relsrc] = classes
    return classes_by_src

  def _compute_deleted_sources(self):
    """Computes the list of sources present in the last analysis that have since been deleted.

    This is a global list. We have no way of associating them to individual targets.
    Paths are relative to buildroot.
    """
    with self.context.new_workunit('find-deleted-sources'):
      if os.path.exists(self._analysis_file):
        products = self._analysis_parser.parse_products_from_path(self._analysis_file)
        buildroot = get_buildroot()
        old_srcs = products.keys()  # Absolute paths.
        return [os.path.relpath(src, buildroot) for src in old_srcs if not os.path.exists(src)]
      else:
        return []

  def _get_previous_sources_by_target(self, target):
    """Returns the target's sources as recorded on the last successful build of target.

    Returns a list of absolute paths.
    """
    path = os.path.join(self._target_sources_dir, target.identifier)
    if os.path.exists(path):
      with open(path, 'r') as infile:
        return [s.rstrip() for s in infile.readlines()]
    else:
      return []

  def _record_sources_by_target(self, target, sources):
    # Record target -> source mapping for future use.
    with open(os.path.join(self._target_sources_dir, target.identifier), 'w') as outfile:
      for src in sources:
        outfile.write(os.path.join(get_buildroot(), src))
        outfile.write('\n')

  def _compute_current_sources_by_target(self, targets):
    """Returns map target -> list of sources (relative to buildroot)."""
    def calculate_sources(target):
      sources = [s for s in target.sources_relative_to_buildroot() if s.endswith(self._file_suffix)]
      # TODO: Make this less hacky. Ideally target.java_sources will point to sources, not targets.
      if hasattr(target, 'java_sources') and target.java_sources:
        sources.extend(self._resolve_target_sources(target.java_sources, '.java'))
      return sources
    return dict([(t, calculate_sources(t)) for t in targets])

  def _find_locally_changed_targets(self, sources_by_target):
    """Finds the targets whose sources have been modified locally.

    Returns a list of targets, or None if no SCM is available.
    """
    # Compute the src->targets mapping. There should only be one target per source,
    # but that's not yet a hard requirement, so the value is a list of targets.
    # TODO(benjy): Might this inverse mapping be needed elsewhere too?
    targets_by_source = defaultdict(list)
    for tgt, srcs in sources_by_target.items():
      for src in srcs:
        targets_by_source[src].append(tgt)

    ret = OrderedSet()
    scm = get_scm()
    if not scm:
      return None
    changed_files = scm.changed_files(include_untracked=True, relative_to=get_buildroot())
    for f in changed_files:
      ret.update(targets_by_source.get(f, []))
    return list(ret)

  def _resolve_target_sources(self, target_sources, extension=None):
    """Given a list of pants targets, extract their sources as a list.

    Filters against the extension if given and optionally returns the paths relative to the target
    base.
    """
    resolved_sources = []
    for target in target_sources:
      if target.has_sources():
        resolved_sources.extend(target.sources_relative_to_buildroot())
    return resolved_sources

  def _compute_classpath_elements_by_class(self, classpath):
    # Don't consider loose classes dirs in our classes dir. Those will be considered
    # separately, by looking at products.
    def non_product(path):
      return path != self._classes_dir

    if self._upstream_class_to_path is None:
      self._upstream_class_to_path = {}
      classpath_entries = filter(non_product, classpath)
      for cp_entry in self.find_all_bootstrap_jars() + classpath_entries:
        # Per the classloading spec, a 'jar' in this context can also be a .zip file.
        if os.path.isfile(cp_entry) and ((cp_entry.endswith('.jar') or cp_entry.endswith('.zip'))):
          with open_zip64(cp_entry, 'r') as jar:
            for cls in jar.namelist():
              # First jar with a given class wins, just like when classloading.
              if cls.endswith(b'.class') and not cls in self._upstream_class_to_path:
                self._upstream_class_to_path[cls] = cp_entry
        elif os.path.isdir(cp_entry):
          for dirpath, _, filenames in safe_walk(cp_entry, followlinks=True):
            for f in filter(lambda x: x.endswith('.class'), filenames):
              cls = os.path.relpath(os.path.join(dirpath, f), cp_entry)
              if not cls in self._upstream_class_to_path:
                self._upstream_class_to_path[cls] = os.path.join(dirpath, f)
    return self._upstream_class_to_path

  def find_all_bootstrap_jars(self):
    def get_path(key):
      return self.context.java_sysprops.get(key, '').split(':')

    def find_jars_in_dirs(dirs):
      ret = []
      for d in dirs:
        if os.path.isdir(d):
          ret.extend(filter(lambda s: s.endswith('.jar'), os.listdir(d)))
      return ret

    # Note: assumes HotSpot, or some JVM that supports sun.boot.class.path.
    # TODO: Support other JVMs? Not clear if there's a standard way to do so.
    # May include loose classes dirs.
    boot_classpath = get_path('sun.boot.class.path')

    # Note that per the specs, overrides and extensions must be in jars.
    # Loose class files will not be found by the JVM.
    override_jars = find_jars_in_dirs(get_path('java.endorsed.dirs'))
    extension_jars = find_jars_in_dirs(get_path('java.ext.dirs'))

    # Note that this order matters: it reflects the classloading order.
    bootstrap_jars = filter(os.path.isfile, override_jars + boot_classpath + extension_jars)
    return bootstrap_jars  # Technically, may include loose class dirs from boot_classpath.

  @property
  def _analysis_tools(self):
    if self._lazy_analysis_tools is None:
      self._lazy_analysis_tools = self.create_analysis_tools()
    return self._lazy_analysis_tools

  @property
  def _analysis_parser(self):
    return self._analysis_tools.parser

  @property
  def ivy_cache_dir(self):
    ret = self.context.products.get_data('ivy_cache_dir')
    if ret is None:
      raise TaskError('ivy_cache_dir product accessed before it was created.')
    return ret

  def _sources_for_targets(self, targets):
    """Returns a map target->sources for the specified targets."""
    if self._sources_by_target is None:
      raise TaskError('self._sources_by_target not computed yet.')
    return dict((t, self._sources_by_target.get(t, [])) for t in targets)

  # Work in a tmpdir so we don't stomp the main analysis files on error.
  # The tmpdir is cleaned up in a shutdown hook, because background work
  # may need to access files we create there even after this method returns.
  def _ensure_analysis_tmpdir(self):
    # Do this lazily, so we don't trigger creation of a worker pool unless we need it.
    if not os.path.exists(self._analysis_tmpdir):
      os.makedirs(self._analysis_tmpdir)
      if self._delete_scratch:
        self.context.background_worker_pool().add_shutdown_hook(
            lambda: safe_rmtree(self._analysis_tmpdir))

  def _create_empty_products(self):
    make_products = lambda: defaultdict(MultipleRootedProducts)
    if self.context.products.is_required_data('classes_by_source'):
      self.context.products.safe_create_data('classes_by_source', make_products)

    # Whether or not anything else requires resources_by_target, this task
    # uses it internally.
    self.context.products.safe_create_data('resources_by_target', make_products)

    # JvmDependencyAnalyzer uses classes_by_target within this run
    self.context.products.safe_create_data('classes_by_target', make_products)

  def _resources_by_class_file(self, class_file_name, resource_mapping):
    assert class_file_name.endswith(".class")
    assert class_file_name.startswith(self.workdir)
    class_file_name = class_file_name[len(self._classes_dir) + 1:-len(".class")]
    class_name = class_file_name.replace("/", ".")
    return resource_mapping.get(class_name, [])

  def _register_products(self, targets, analysis_file):
    classes_by_source = self.context.products.get_data('classes_by_source')
    classes_by_target = self.context.products.get_data('classes_by_target')
    resources_by_target = self.context.products.get_data('resources_by_target')

    if classes_by_source is not None or classes_by_target is not None:
      computed_classes_by_source = self._compute_classes_by_source(analysis_file)
      resource_mapping = ResourceMapping(self._classes_dir)
      for target in targets:
        target_products = classes_by_target[target] if classes_by_target is not None else None
        for source in self._sources_by_target.get(target, []):  # Source is relative to buildroot.
          classes = computed_classes_by_source.get(source, [])  # Classes are absolute paths.
          for cls in classes:
            resources = self._resources_by_class_file(cls, resource_mapping)
            resources_by_target[target].add_abs_paths(self._classes_dir, resources)

          if classes_by_target is not None:
            target_products.add_abs_paths(self._classes_dir, classes)
          if classes_by_source is not None:
            classes_by_source[source].add_abs_paths(self._classes_dir, classes)

    # TODO(pl): https://github.com/pantsbuild/pants/issues/206
    if resources_by_target is not None:
      for target in targets:
        target_resources = resources_by_target[target]
        for root, abs_paths in self.extra_products(target):
          target_resources.add_abs_paths(root, abs_paths)
Exemplo n.º 14
0
class JvmCompileGlobalStrategy(JvmCompileStrategy):
  """A strategy for JVM compilation that uses a global classpath and analysis."""

  @classmethod
  def register_options(cls, register, language, supports_concurrent_execution):
    register('--missing-deps', choices=['off', 'warn', 'fatal'], default='warn',
             help='Check for missing dependencies in {0} code. Reports actual dependencies A -> B '
                  'where there is no transitive BUILD file dependency path from A to B. If fatal, '
                  'missing deps are treated as a build error.'.format(language))

    register('--missing-direct-deps', choices=['off', 'warn', 'fatal'], default='off',
             help='Check for missing direct dependencies in {0} code. Reports actual dependencies '
                  'A -> B where there is no direct BUILD file dependency path from A to B. This is '
                  'a very strict check; In practice it is common to rely on transitive, indirect '
                  'dependencies, e.g., due to type inference or when the main target in a BUILD '
                  'file is modified to depend on other targets in the same BUILD file, as an '
                  'implementation detail. However it may still be useful to use this on '
                  'occasion. '.format(language))

    register('--missing-deps-whitelist', type=Options.list,
             help="Don't report these targets even if they have missing deps.")

    register('--unnecessary-deps', choices=['off', 'warn', 'fatal'], default='off',
             help='Check for declared dependencies in {0} code that are not needed. This is a very '
                  'strict check. For example, generated code will often legitimately have BUILD '
                  'dependencies that are unused in practice.'.format(language))

    register('--changed-targets-heuristic-limit', type=int, default=0,
             help='If non-zero, and we have fewer than this number of locally-changed targets, '
                  'partition them separately, to preserve stability when compiling repeatedly.')

  def __init__(self, context, options, workdir, analysis_tools, language, sources_predicate):
    super(JvmCompileGlobalStrategy, self).__init__(context, options, workdir, analysis_tools,
                                                   language, sources_predicate)

    # Various working directories.
    # NB: These are grandfathered in with non-strategy-specific names, but to prevent
    # collisions within the buildcache, strategies should use strategy-specific subdirectories.
    self._analysis_dir = os.path.join(workdir, 'analysis')
    self._classes_dir = os.path.join(workdir, 'classes')

    self._analysis_file = os.path.join(self._analysis_dir, 'global_analysis.valid')
    self._invalid_analysis_file = os.path.join(self._analysis_dir, 'global_analysis.invalid')

    self._target_sources_dir = os.path.join(workdir, 'target_sources')

    # The rough number of source files to build in each compiler pass.
    self._partition_size_hint = options.partition_size_hint

    # Set up dep checking if needed.
    def munge_flag(flag):
      flag_value = getattr(options, flag, None)
      return None if flag_value == 'off' else flag_value

    check_missing_deps = munge_flag('missing_deps')
    check_missing_direct_deps = munge_flag('missing_direct_deps')
    check_unnecessary_deps = munge_flag('unnecessary_deps')

    if check_missing_deps or check_missing_direct_deps or check_unnecessary_deps:
      target_whitelist = options.missing_deps_whitelist
      # Must init it here, so it can set requirements on the context.
      self._dep_analyzer = JvmDependencyAnalyzer(self.context,
                                                 check_missing_deps,
                                                 check_missing_direct_deps,
                                                 check_unnecessary_deps,
                                                 target_whitelist)
    else:
      self._dep_analyzer = None

    # Computed lazily as needed.
    self._upstream_class_to_path = None

    # If non-zero, and we have fewer than this number of locally-changed targets,
    # then we partition them separately, to preserve stability in the face of repeated
    # compilations.
    self._changed_targets_heuristic_limit = options.changed_targets_heuristic_limit

    # Sources (relative to buildroot) present in the last analysis that have since been deleted.
    # Populated in prepare_compile().
    self._deleted_sources = None

  def name(self):
    return 'global'

  def compile_context(self, target):
    """Returns the default/stable compile context for the given target.

    Temporary compile contexts are private to the strategy.
    """
    return self.CompileContext(target,
                               self._analysis_file,
                               self._classes_dir,
                               self._sources_for_target(target))

  def move(self, src, dst):
    if self.delete_scratch:
      shutil.move(src, dst)
    else:
      shutil.copy(src, dst)

  def pre_compile(self):
    super(JvmCompileGlobalStrategy, self).pre_compile()

    # Only create these working dirs during execution phase, otherwise, they
    # would be wiped out by clean-all goal/task if it's specified.
    safe_mkdir(self._target_sources_dir)
    safe_mkdir(self._analysis_dir)
    safe_mkdir(self._classes_dir)

    # Look for invalid analysis files.
    for f in (self._invalid_analysis_file, self._analysis_file):
      self.validate_analysis(f)

  def prepare_compile(self, cache_manager, all_targets, relevant_targets):
    super(JvmCompileGlobalStrategy, self).prepare_compile(cache_manager, all_targets, relevant_targets)

    # Update the classpath for us and for downstream tasks.
    compile_classpaths = self.context.products.get_data('compile_classpath')
    for conf in self._confs:
      compile_classpaths.add_for_targets(all_targets, [(conf, self._classes_dir)])

    # Split the global analysis file into valid and invalid parts.
    invalidation_check = cache_manager.check(relevant_targets)
    if invalidation_check.invalid_vts:
      # The analysis for invalid and deleted sources is no longer valid.
      invalid_targets = [vt.target for vt in invalidation_check.invalid_vts]
      invalid_sources_by_target = {}
      for tgt in invalid_targets:
        invalid_sources_by_target[tgt] = self._sources_for_target(tgt)
      invalid_sources = list(itertools.chain.from_iterable(invalid_sources_by_target.values()))
      self._deleted_sources = self._compute_deleted_sources()

      tmpdir = os.path.join(self.analysis_tmpdir, str(uuid.uuid4()))
      os.mkdir(tmpdir)
      valid_analysis_tmp = os.path.join(tmpdir, 'valid_analysis')
      newly_invalid_analysis_tmp = os.path.join(tmpdir, 'newly_invalid_analysis')
      invalid_analysis_tmp = os.path.join(tmpdir, 'invalid_analysis')
      if self._analysis_parser.is_nonempty_analysis(self._analysis_file):
        with self.context.new_workunit(name='prepare-analysis'):
          self._analysis_tools.split_to_paths(self._analysis_file,
              [(invalid_sources + self._deleted_sources, newly_invalid_analysis_tmp)],
              valid_analysis_tmp)
          if self._analysis_parser.is_nonempty_analysis(self._invalid_analysis_file):
            self._analysis_tools.merge_from_paths(
              [self._invalid_analysis_file, newly_invalid_analysis_tmp], invalid_analysis_tmp)
          else:
            invalid_analysis_tmp = newly_invalid_analysis_tmp

          # Now it's OK to overwrite the main analysis files with the new state.
          self.move(valid_analysis_tmp, self._analysis_file)
          self.move(invalid_analysis_tmp, self._invalid_analysis_file)
    else:
      self._deleted_sources = []

  def invalidation_hints(self, relevant_targets):
    # If needed, find targets that we've changed locally (as opposed to
    # changes synced in from the SCM).
    # TODO(benjy): Should locally_changed_targets be available in all Tasks?
    locally_changed_targets = None
    if self._changed_targets_heuristic_limit:
      locally_changed_targets = self._find_locally_changed_targets(relevant_targets)
      if (locally_changed_targets and
          len(locally_changed_targets) > self._changed_targets_heuristic_limit):
        locally_changed_targets = None

    return (self._partition_size_hint, locally_changed_targets)

  def compile_chunk(self,
                    invalidation_check,
                    all_targets,
                    relevant_targets,
                    invalid_targets,
                    extra_compile_time_classpath_elements,
                    compile_vts,
                    register_vts,
                    update_artifact_cache_vts_work):
    """Executes compilations for the invalid targets contained in a single chunk.

    Has the side effects of populating:
    # valid/invalid analysis files
    # classes_by_source product
    # classes_by_target product
    # resources_by_target product
    """
    assert invalid_targets, "compile_chunk should only be invoked if there are invalid targets."

    extra_classpath_tuples = self._compute_extra_classpath(extra_compile_time_classpath_elements)

    # Get the classpath generated by upstream JVM tasks and our own prepare_compile().
    # NB: The global strategy uses the aggregated classpath (for all targets) to compile each
    # chunk, which avoids needing to introduce compile-time dependencies between annotation
    # processors and the classes they annotate.
    compile_classpath = ClasspathUtil.compute_classpath(all_targets, self.context.products.get_data(
      'compile_classpath'), extra_classpath_tuples, self._confs)

    # Find the invalid sources for this chunk.
    invalid_sources_by_target = {t: self._sources_for_target(t) for t in invalid_targets}

    tmpdir = os.path.join(self.analysis_tmpdir, str(uuid.uuid4()))
    os.mkdir(tmpdir)

    # Figure out the sources and analysis belonging to each partition.
    partitions = []  # Each element is a triple (vts, sources_by_target, analysis).
    for vts in invalidation_check.invalid_vts_partitioned:
      partition_tmpdir = os.path.join(tmpdir, Target.maybe_readable_identify(vts.targets))
      os.mkdir(partition_tmpdir)
      sources = list(itertools.chain.from_iterable(
          [invalid_sources_by_target.get(t, []) for t in vts.targets]))
      de_duped_sources = list(OrderedSet(sources))
      if len(sources) != len(de_duped_sources):
        counts = [(src, len(list(srcs))) for src, srcs in itertools.groupby(sorted(sources))]
        self.context.log.warn(
            'De-duped the following sources:\n\t{}'
            .format('\n\t'.join(sorted('{} {}'.format(cnt, src) for src, cnt in counts if cnt > 1))))
      analysis_file = os.path.join(partition_tmpdir, 'analysis')
      partitions.append((vts, de_duped_sources, analysis_file))

    # Split per-partition files out of the global invalid analysis.
    if self._analysis_parser.is_nonempty_analysis(self._invalid_analysis_file) and partitions:
      with self.context.new_workunit(name='partition-analysis'):
        splits = [(x[1], x[2]) for x in partitions]
        # We have to pass the analysis for any deleted files through zinc, to give it
        # a chance to delete the relevant class files.
        if splits:
          splits[0] = (splits[0][0] + self._deleted_sources, splits[0][1])
        self._analysis_tools.split_to_paths(self._invalid_analysis_file, splits)

    # Now compile partitions one by one.
    for partition_index, partition in enumerate(partitions):
      (vts, sources, analysis_file) = partition

      progress_message = 'partition {} of {}'.format(partition_index + 1, len(partitions))
      # We have to treat the global output dir as an upstream element, so compilers can
      # find valid analysis for previous partitions. We use the global valid analysis
      # for the upstream.
      upstream_analysis = ({self._classes_dir: self._analysis_file}
                           if os.path.exists(self._analysis_file) else {})
      compile_vts(vts,
                  sources,
                  analysis_file,
                  upstream_analysis,
                  compile_classpath,
                  self._classes_dir,
                  None,
                  progress_message)

      # No exception was thrown, therefore the compile succeeded and analysis_file is now valid.
      if os.path.exists(analysis_file):  # The compilation created an analysis.
        # Merge the newly-valid analysis with our global valid analysis.
        new_valid_analysis = analysis_file + '.valid.new'
        if self._analysis_parser.is_nonempty_analysis(self._analysis_file):
          with self.context.new_workunit(name='update-upstream-analysis'):
            self._analysis_tools.merge_from_paths([self._analysis_file, analysis_file],
                                                  new_valid_analysis)
        else:  # We need to keep analysis_file around. Background tasks may need it.
          shutil.copy(analysis_file, new_valid_analysis)

        # Move the merged valid analysis to its proper location.
        # We do this before checking for missing dependencies, so that we can still
        # enjoy an incremental compile after fixing missing deps.
        self.move(new_valid_analysis, self._analysis_file)

        # Update the products with the latest classes. Must happen before the
        # missing dependencies check.
        register_vts([self.compile_context(t) for t in vts.targets])
        if self._dep_analyzer:
          # Check for missing dependencies.
          actual_deps = self._analysis_parser.parse_deps_from_path(analysis_file,
              lambda: self._compute_classpath_elements_by_class(compile_classpath), self._classes_dir)
          with self.context.new_workunit(name='find-missing-dependencies'):
            self._dep_analyzer.check(sources, actual_deps)

        # Kick off the background artifact cache write.
        if update_artifact_cache_vts_work:
          self._write_to_artifact_cache(analysis_file,
                                        vts,
                                        update_artifact_cache_vts_work)

      if self._analysis_parser.is_nonempty_analysis(self._invalid_analysis_file):
        with self.context.new_workunit(name='trim-downstream-analysis'):
          # Trim out the newly-valid sources from our global invalid analysis.
          new_invalid_analysis = analysis_file + '.invalid.new'
          discarded_invalid_analysis = analysis_file + '.invalid.discard'
          self._analysis_tools.split_to_paths(self._invalid_analysis_file,
            [(sources, discarded_invalid_analysis)], new_invalid_analysis)
          self.move(new_invalid_analysis, self._invalid_analysis_file)

      # Record the built target -> sources mapping for future use.
      for target, sources in self._sources_for_targets(vts.targets).items():
        self._record_previous_sources_by_target(target, sources)

      # Now that all the analysis accounting is complete, and we have no missing deps,
      # we can safely mark the targets as valid.
      vts.update()

  def compute_resource_mapping(self, compile_contexts):
    return ResourceMapping(self._classes_dir)

  def compute_classes_by_source(self, compile_contexts):
    if not compile_contexts:
      return {}

    # This implementation requires that all contexts use the same analysis file and global classes.
    analysis_file = None
    for compile_context in compile_contexts:
      if compile_context.classes_dir != self._classes_dir:
        raise TaskError('Unrecognized classes directory for the global strategy: {}'.format(
            compile_context.classes_dir))
      if not analysis_file:
        analysis_file = compile_context.analysis_file
      else:
        if compile_context.analysis_file != analysis_file:
          raise TaskError('Inconsistent analysis file for the global strategy: {} vs {}'.format(
              compile_context.analysis_file, analysis_file))

    classes_by_src_by_context = defaultdict(dict)
    if os.path.exists(analysis_file):
      # Parse the global analysis once.
      buildroot = get_buildroot()
      products = self._analysis_parser.parse_products_from_path(analysis_file,
                                                                self._classes_dir)

      # Then iterate over contexts (targets), and add the classes for their sources.
      for compile_context in compile_contexts:
        classes_by_src = classes_by_src_by_context[compile_context]
        for source in compile_context.sources:
          absolute_source = os.path.join(buildroot, source)
          classes_by_src[source] = products.get(absolute_source, [])
    return classes_by_src_by_context

  def post_process_cached_vts(self, cached_vts):
    """Special post processing for global scala analysis files.

    Class files are retrieved directly into their final locations in the global classes dir.
    """

    # Get all the targets whose artifacts we found in the cache.
    cached_targets = []
    for vt in cached_vts:
      for target in vt.targets:
        cached_targets.append(target)

    # The current global analysis may contain old data for modified targets for
    # which we got cache hits. We need to strip out this old analysis, to ensure
    # that the new data incoming from the cache doesn't collide with it during the merge.
    sources_to_strip = []
    if os.path.exists(self._analysis_file):
      for target in cached_targets:
        sources_to_strip.extend(self._get_previous_sources_by_target(target))

    # Localize the cached analyses.
    analyses_to_merge = []
    for target in cached_targets:
      analysis_file = JvmCompileStrategy._analysis_for_target(self.analysis_tmpdir, target)
      portable_analysis_file = JvmCompileStrategy._portable_analysis_for_target(
          self.analysis_tmpdir, target)
      if os.path.exists(portable_analysis_file):
        self._analysis_tools.localize(portable_analysis_file, analysis_file)
      if os.path.exists(analysis_file):
        analyses_to_merge.append(analysis_file)

    # Merge them into the global analysis.
    if analyses_to_merge:
      with temporary_dir() as tmpdir:
        if sources_to_strip:
          throwaway = os.path.join(tmpdir, 'throwaway')
          trimmed_analysis = os.path.join(tmpdir, 'trimmed')
          self._analysis_tools.split_to_paths(self._analysis_file,
                                          [(sources_to_strip, throwaway)],
                                          trimmed_analysis)
        else:
          trimmed_analysis = self._analysis_file
        if os.path.exists(trimmed_analysis):
          analyses_to_merge.append(trimmed_analysis)
        tmp_analysis = os.path.join(tmpdir, 'analysis')
        with self.context.new_workunit(name='merge_analysis'):
          self._analysis_tools.merge_from_paths(analyses_to_merge, tmp_analysis)

        sources_by_cached_target = self._sources_for_targets(cached_targets)

        # Record the cached target -> sources mapping for future use.
        for target, sources in sources_by_cached_target.items():
          self._record_previous_sources_by_target(target, sources)

        # Everything's good so move the merged analysis to its final location.
        if os.path.exists(tmp_analysis):
          self.move(tmp_analysis, self._analysis_file)

  def _write_to_artifact_cache(self, analysis_file, vts, get_update_artifact_cache_work):
    vt_by_target = dict([(vt.target, vt) for vt in vts.versioned_targets])

    vts_targets = [t for t in vts.targets if not t.has_label('no_cache')]

    # Determine locations for analysis files that will be split in the background.
    split_analysis_files = [
        JvmCompileStrategy._analysis_for_target(self.analysis_tmpdir, t) for t in vts_targets]
    portable_split_analysis_files = [
        JvmCompileStrategy._portable_analysis_for_target(self.analysis_tmpdir, t) for t in vts_targets]

    # Set up args for splitting the analysis into per-target files.
    splits = zip([self._sources_for_target(t) for t in vts_targets], split_analysis_files)
    splits_args_tuples = [(analysis_file, splits)]

    # Set up args for rebasing the splits.
    relativize_args_tuples = zip(split_analysis_files, portable_split_analysis_files)

    # Compute the classes and resources for each vts.
    compile_contexts = [self.compile_context(t) for t in vts_targets]
    vts_artifactfiles_pairs = []
    classes_by_source_by_context = self.compute_classes_by_source(compile_contexts)
    resources_by_target = self.context.products.get_data('resources_by_target')
    for compile_context in compile_contexts:
      target = compile_context.target
      if target.has_label('no_cache'):
        continue
      artifacts = []
      if resources_by_target is not None:
        for _, paths in resources_by_target[target].abs_paths():
          artifacts.extend(paths)
      classes_by_source = classes_by_source_by_context[compile_context]
      for source in compile_context.sources:
        classes = classes_by_source.get(source, [])
        artifacts.extend(classes)

      vt = vt_by_target.get(target)
      if vt is not None:
        # NOTE: analysis_file doesn't exist yet.
        vts_artifactfiles_pairs.append(
            (vt, artifacts + [JvmCompileStrategy._portable_analysis_for_target(
                self.analysis_tmpdir, target)]))

    update_artifact_cache_work = get_update_artifact_cache_work(vts_artifactfiles_pairs)
    if update_artifact_cache_work:
      work_chain = [
        Work(self._analysis_tools.split_to_paths, splits_args_tuples, 'split'),
        Work(self._analysis_tools.relativize, relativize_args_tuples, 'relativize'),
        update_artifact_cache_work
      ]
      self.context.submit_background_work_chain(work_chain, parent_workunit_name='cache')

  def _get_previous_sources_by_target(self, target):
    """Returns the target's sources as recorded on the last successful build of target.

    Returns a list of absolute paths.
    """
    path = os.path.join(self._target_sources_dir, target.identifier)
    if os.path.exists(path):
      with open(path, 'r') as infile:
        return [s.rstrip() for s in infile.readlines()]
    else:
      return []

  def _record_previous_sources_by_target(self, target, sources):
    # Record target -> source mapping for future use.
    with open(os.path.join(self._target_sources_dir, target.identifier), 'w') as outfile:
      for src in sources:
        outfile.write(os.path.join(get_buildroot(), src))
        outfile.write('\n')

  def _compute_deleted_sources(self):
    """Computes the list of sources present in the last analysis that have since been deleted.

    This is a global list. We have no way of associating them to individual targets.
    Paths are relative to buildroot.
    """
    with self.context.new_workunit('find-deleted-sources'):
      if os.path.exists(self._analysis_file):
        products = self._analysis_parser.parse_products_from_path(self._analysis_file,
                                                                  self._classes_dir)
        buildroot = get_buildroot()
        old_srcs = products.keys()  # Absolute paths.
        return [os.path.relpath(src, buildroot) for src in old_srcs if not os.path.exists(src)]
      else:
        return []

  def _find_locally_changed_targets(self, relevant_targets):
    """Finds the targets whose sources have been modified locally.

    Returns a list of targets, or None if no SCM is available.
    """
    # Compute the src->targets mapping. There should only be one target per source,
    # but that's not yet a hard requirement, so the value is a list of targets.
    # TODO(benjy): Might this inverse mapping be needed elsewhere too?
    targets_by_source = defaultdict(list)
    for tgt, srcs in self._sources_for_targets(relevant_targets).items():
      for src in srcs:
        targets_by_source[src].append(tgt)

    ret = OrderedSet()
    scm = get_scm()
    if not scm:
      return None
    changed_files = scm.changed_files(include_untracked=True, relative_to=get_buildroot())
    for f in changed_files:
      ret.update(targets_by_source.get(f, []))
    return list(ret)

  def _compute_classpath_elements_by_class(self, classpath):
    # Don't consider loose classes dirs in our classes dir. Those will be considered
    # separately, by looking at products.
    def non_product(path):
      return path != self._classes_dir

    if self._upstream_class_to_path is None:
      self._upstream_class_to_path = {}
      classpath_entries = filter(non_product, classpath)
      for cp_entry in self._find_all_bootstrap_jars() + classpath_entries:
        # Per the classloading spec, a 'jar' in this context can also be a .zip file.
        if os.path.isfile(cp_entry) and (cp_entry.endswith('.jar') or cp_entry.endswith('.zip')):
          with open_zip(cp_entry, 'r') as jar:
            for cls in jar.namelist():
              # First jar with a given class wins, just like when classloading.
              if cls.endswith(b'.class') and not cls in self._upstream_class_to_path:
                self._upstream_class_to_path[cls] = cp_entry
        elif os.path.isdir(cp_entry):
          for dirpath, _, filenames in safe_walk(cp_entry, followlinks=True):
            for f in filter(lambda x: x.endswith('.class'), filenames):
              cls = os.path.relpath(os.path.join(dirpath, f), cp_entry)
              if not cls in self._upstream_class_to_path:
                self._upstream_class_to_path[cls] = os.path.join(dirpath, f)
    return self._upstream_class_to_path

  def _find_all_bootstrap_jars(self):
    def get_path(key):
      return self.context.java_sysprops.get(key, '').split(':')

    def find_jars_in_dirs(dirs):
      ret = []
      for d in dirs:
        if os.path.isdir(d):
          ret.extend(filter(lambda s: s.endswith('.jar'), os.listdir(d)))
      return ret

    # Note: assumes HotSpot, or some JVM that supports sun.boot.class.path.
    # TODO: Support other JVMs? Not clear if there's a standard way to do so.
    # May include loose classes dirs.
    boot_classpath = get_path('sun.boot.class.path')

    # Note that per the specs, overrides and extensions must be in jars.
    # Loose class files will not be found by the JVM.
    override_jars = find_jars_in_dirs(get_path('java.endorsed.dirs'))
    extension_jars = find_jars_in_dirs(get_path('java.ext.dirs'))

    # Note that this order matters: it reflects the classloading order.
    bootstrap_jars = filter(os.path.isfile, override_jars + boot_classpath + extension_jars)
    return bootstrap_jars  # Technically, may include loose class dirs from boot_classpath.