def PerformStage(self):
        buildroot = self._build_root
        gs_context = gs.GSContext()
        cpv = portage_util.BestVisible(constants.CHROME_CP,
                                       buildroot=buildroot)
        version_number = cpv.version

        # We need the name of one board that has been setup in this
        # builder to find the Chrome ebuild. The chrome ebuild should be
        # the same for all the boards, so just use the first one.
        # If we don't have any boards, leave the called function to guess.
        board = self._boards[0] if self._boards else None
        arch_profiles = {}
        for arch in afdo.AFDO_ARCH_GENERATORS:
            afdo_file = afdo.GetLatestAFDOFile(cpv, arch, buildroot,
                                               gs_context)
            if not afdo_file:
                raise afdo.MissingAFDOData(
                    'Could not find appropriate AFDO profile')
            state = 'current' if version_number in afdo_file else 'previous'
            logging.info('Found %s %s AFDO profile %s', state, arch, afdo_file)
            arch_profiles[arch] = afdo_file

        # Now update the Chrome ebuild file with the AFDO profiles we found
        # for each architecture.
        afdo.UpdateChromeEbuildAFDOFile(board, arch_profiles)
    def PerformStage(self):
        """Collect a 'perf' profile and convert it into the AFDO format."""
        super(AFDODataGenerateStage, self).PerformStage()

        board = self._current_board
        if not afdo.CanGenerateAFDOData(board):
            logging.warning('Board %s cannot generate its own AFDO profile.',
                            board)
            return

        arch = self._GetCurrentArch()
        buildroot = self._build_root
        gs_context = gs.GSContext()
        cpv = portage_util.BestVisible(constants.CHROME_CP,
                                       buildroot=buildroot)
        afdo_file = None

        # Generation of AFDO could fail for different reasons.
        # We will ignore the failures and let the master PFQ builder try
        # to find an older AFDO profile.
        try:
            if afdo.WaitForAFDOPerfData(cpv, arch, buildroot, gs_context):
                afdo_file = afdo.GenerateAFDOData(cpv, arch, board, buildroot,
                                                  gs_context)
                assert afdo_file
                logging.info('Generated %s AFDO profile %s', arch, afdo_file)
            else:
                raise afdo.MissingAFDOData(
                    'Could not find current "perf" profile. '
                    'Master PFQ builder will try to use stale '
                    'AFDO profile.')
        # Will let system-exiting exceptions through.
        except Exception:
            logging.PrintBuildbotStepWarnings()
            logging.warning('AFDO profile generation failed with exception ',
                            exc_info=True)

            alert_msg = ('Please triage. This will become a fatal error.\n\n'
                         'arch=%s buildroot=%s\n\nURL=%s' %
                         (arch, buildroot, self._run.ConstructDashboardURL()))
            subject_msg = (
                'Failure in generation of AFDO Data for builder %s' %
                self._run.config.name)
            alerts.SendEmailLog(subject_msg,
                                afdo.AFDO_ALERT_RECIPIENTS,
                                server=alerts.SmtpServer(
                                    constants.GOLO_SMTP_SERVER),
                                message=alert_msg)
            # Re-raise whatever exception we got here. This stage will only
            # generate a warning but we want to make sure the warning is
            # generated.
            raise
    def PerformStage(self):
        buildroot = self._build_root
        gs_context = gs.GSContext()
        cpv = portage_util.PortageqBestVisible(constants.CHROME_CP,
                                               cwd=buildroot)

        # We need the name of one board that has been setup in this
        # builder to find the Chrome ebuild. The chrome ebuild should be
        # the same for all the boards, so just use the first one.
        # If we don't have any boards, leave the called function to guess.
        board = self._boards[0] if self._boards else None
        profiles = {}

        for source, getter in afdo.PROFILE_SOURCES.items():
            profile = getter(cpv, source, buildroot, gs_context)
            if not profile:
                raise afdo.MissingAFDOData(
                    'Could not find appropriate profile for %s' % source)
            logging.info('Found AFDO profile %s for %s', profile, source)
            profiles[source] = profile

        # Now update the Chrome ebuild file with the AFDO profiles we found
        # for each source.
        afdo.UpdateChromeEbuildAFDOFile(board, profiles)
    def PerformStage(self):
        """Collect a 'perf' profile and convert it into the AFDO format."""
        super(AFDODataGenerateStage, self).PerformStage()

        board = self._current_board
        if not afdo.CanGenerateAFDOData(board):
            logging.warning('Board %s cannot generate its own AFDO profile.',
                            board)
            return

        arch = self._GetCurrentArch()
        buildroot = self._build_root
        gs_context = gs.GSContext()
        cpv = portage_util.PortageqBestVisible(constants.CHROME_CP,
                                               cwd=buildroot)
        afdo_file = None

        # We have a mismatch between how we version the perf.data we collect and
        # how we version our AFDO profiles.
        #
        # This mismatch can cause us to generate garbage profiles, so we skip
        # profile updates for non-r1 revisions of Chrome.
        #
        # Going into more detail, a perf.data file looks like:
        # chromeos-chrome-amd64-68.0.3440.9.perf.data.bz2
        #
        # An AFDO profile looks like:
        # chromeos-chrome-amd64-68.0.3440.9_rc-r1.afdo.bz2
        #
        # And an unstripped Chrome looks like:
        # chromeos-chrome-amd64-68.0.3440.9_rc-r1.debug.bz2
        #
        # Notably, the perf.data is lacking the revision number of the Chrome it
        # was gathered on. This is problematic, since if there's a rev bump, we'll
        # end up using the perf.data collected on Chrome version $N-r1 with a
        # Chrome binary built from Chrome version $N-r2, which may have an entirely
        # different layout than Chrome version $N-r1.
        if cpv.rev != 'r1':
            logging.warning(
                'Non-r1 version of Chrome detected; skipping AFDO generation')
            return

        # Generation of AFDO could fail for different reasons.
        # We will ignore the failures and let the master PFQ builder try
        # to find an older AFDO profile.
        try:
            if afdo.WaitForAFDOPerfData(cpv, arch, buildroot, gs_context):
                afdo_file, uploaded_afdo = afdo.GenerateAFDOData(
                    cpv, arch, board, buildroot, gs_context)
                assert afdo_file
                logging.info('Generated %s AFDO profile %s', arch, afdo_file)

                # If there's no new profile, merging would only be redoing the last
                # merge and uploading nothing.
                if not uploaded_afdo:
                    logging.info('AFDO profile already existed in GS. Quit')
                    return

                merged_file, uploaded_merged = \
                    afdo.CreateAndUploadMergedAFDOProfile(gs_context, buildroot,
                                                          afdo_file)

                if merged_file is not None:
                    logging.info('Generated %s merged AFDO profile %s', arch,
                                 merged_file)

                # TODO(gbiv): once there's clarity that merged profiles are working
                # (e.g. a week goes by with Android/Linux mostly-happily using them),
                # we may want to turn them on for CrOS. Until then, `latest` is always
                # the raw AFDO file.
                if uploaded_merged and False:
                    newest_afdo_file = merged_file
                else:
                    newest_afdo_file = afdo_file

                afdo.UpdateLatestAFDOProfileInGS(cpv, arch, buildroot,
                                                 newest_afdo_file, gs_context)
                logging.info('Pointed newest profile at %s', newest_afdo_file)
            else:
                raise afdo.MissingAFDOData(
                    'Could not find current "perf" profile. '
                    'Master PFQ builder will try to use stale '
                    'AFDO profile.')
        # Will let system-exiting exceptions through.
        except Exception:
            logging.PrintBuildbotStepWarnings()
            logging.warning('AFDO profile generation failed with exception ',
                            exc_info=True)

            alert_msg = ('Please triage. This will become a fatal error.\n\n'
                         'arch=%s buildroot=%s\n\nURL=%s' %
                         (arch, buildroot, self._run.ConstructDashboardURL()))
            subject_msg = (
                'Failure in generation of AFDO Data for builder %s' %
                self._run.config.name)
            alerts.SendEmailLog(subject_msg,
                                afdo.AFDO_ALERT_RECIPIENTS,
                                server=alerts.SmtpServer(
                                    constants.GOLO_SMTP_SERVER),
                                message=alert_msg)
            # Re-raise whatever exception we got here. This stage will only
            # generate a warning but we want to make sure the warning is
            # generated.
            raise