comparison mercurial/ui.py @ 40524:25732c5678bc

ui: wrap whole _write() block with timeblockedsection

I think the cost of color labeling is negligible compared to the I/O
syscalls. Let's simply wrap the whole write() function so that we can
eliminate _write() and _write_err() in later changeset.

40523:0c7b2035a604

            self._writenobuf(self._write, *args, **opts)

    def _writenobuf(self, write, *args, **opts):
        self._progclear()
        msg = b''.join(args)
        try:
            if self._colormode == 'win32':
                # windows color printing is its own can of crab, defer to
                # the color module and that is it.
                color.win32print(self, write, msg, **opts)

                if self._colormode is not None:
                    label = opts.get(r'label', '')
                    msg = self.label(msg, label)
                write(msg)
        finally:
            pass

    def _write(self, data):
        # opencode timeblockedsection because this is a critical path
        starttime = util.timer()
        try:
            self.fout.write(data)
        except IOError as err:
            raise error.StdioError(err)
        finally:
            self._blockedtimes['stdio_blocked'] += \
                (util.timer() - starttime) * 1000

    def write_err(self, *args, **opts):
        if self._bufferstates and self._bufferstates[-1][0]:
            self.write(*args, **opts)
        else:
            self._writenobuf(self._write_err, *args, **opts)

    def _write_err(self, data):
        try:
            with self.timeblockedsection('stdio'):
                if not getattr(self.fout, 'closed', False):
                    self.fout.flush()
                self.ferr.write(data)
                # stderr may be buffered under win32 when redirected to files,
                # including stdout.

40524:25732c5678bc

            self._writenobuf(self._write, *args, **opts)

    def _writenobuf(self, write, *args, **opts):
        self._progclear()
        msg = b''.join(args)

        # opencode timeblockedsection because this is a critical path
        starttime = util.timer()
        try:
            if self._colormode == 'win32':
                # windows color printing is its own can of crab, defer to
                # the color module and that is it.
                color.win32print(self, write, msg, **opts)

                if self._colormode is not None:
                    label = opts.get(r'label', '')
                    msg = self.label(msg, label)
                write(msg)
        finally:
            self._blockedtimes['stdio_blocked'] += \
                (util.timer() - starttime) * 1000

    def _write(self, data):
        try:
            self.fout.write(data)
        except IOError as err:
            raise error.StdioError(err)

    def write_err(self, *args, **opts):
        if self._bufferstates and self._bufferstates[-1][0]:
            self.write(*args, **opts)
        else:
            self._writenobuf(self._write_err, *args, **opts)

    def _write_err(self, data):
        try:
            if True:
                if not getattr(self.fout, 'closed', False):
                    self.fout.flush()
                self.ferr.write(data)
                # stderr may be buffered under win32 when redirected to files,
                # including stdout.