mercurial-scm/hg: hgext/blackbox.py annotate

annotate hgext/blackbox.py @ 34109:cf04db16f583

blackbox: do not cache file objects Having the blackbox file objects cached in `ui._bbfp` could in theory be troublesome if multiple processes (ex. chg servers) have file objects referring to a same file. (Although I spent some time and failed to build a convincing test case) This patch makes blackbox re-open the file every time to make the situation better. Ideally we also need proper locking. The caching logic traces back to the commit introducing blackbox (18242716a). That commit does not have details about why caching is necessary. Consider the fact that blackbox logs are not many, it seems fine to remove the fp cache to be more confident. Differential Revision: https://phab.mercurial-scm.org/D650

author	Jun Wu <quark@fb.com>
date	Wed, 06 Sep 2017 21:08:59 -0700
parents	50df6cf22717
children	029b33adbd17

rev	line source
18676 1506eb487ddd blackbox: fix copyright Bryan O'Sullivan <bryano@fb.com> parents: 18675 diff changeset	1 # blackbox.py - log repository events to a file for post-mortem debugging
18669 18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	2 #
18676 1506eb487ddd blackbox: fix copyright Bryan O'Sullivan <bryano@fb.com> parents: 18675 diff changeset	3 # Copyright 2010 Nicolas Dumazet
18669 18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	4 # Copyright 2013 Facebook, Inc.
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	5 #
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	6 # This software may be used and distributed according to the terms of the
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	7 # GNU General Public License version 2 or any later version.
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	8
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	9 """log repository events to a blackbox for debugging
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	10
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	11 Logs event information to .hg/blackbox.log to help debug and diagnose problems.
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	12 The events that get logged can be configured via the blackbox.track config key.
28246 b862e793ec10 blackbox: log dirty state timeless <timeless@mozdev.org> parents: 28245 diff changeset	13
19162 27013ace80eb blackbox: fix literal block syntax Takumi IINO <trot.thunder@gmail.com> parents: 19066 diff changeset	14 Examples::
18669 18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	15
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	16 [blackbox]
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	17 track = *
28303 ce24de063aa5 blackbox: rewrite dirty documentation noting it is expensive timeless <timeless@mozdev.org> parents: 28248 diff changeset	18 # dirty is EXPENSIVE (slow);
ce24de063aa5 blackbox: rewrite dirty documentation noting it is expensive timeless <timeless@mozdev.org> parents: 28248 diff changeset	19 # each log entry indicates `+` if the repository is dirty, like :hg:`id`.
28246 b862e793ec10 blackbox: log dirty state timeless <timeless@mozdev.org> parents: 28245 diff changeset	20 dirty = True
28305 f5ae291dfedf blackbox: optionally log event source timeless <timeless@mozdev.org> parents: 28304 diff changeset	21 # record the source of log messages
f5ae291dfedf blackbox: optionally log event source timeless <timeless@mozdev.org> parents: 28304 diff changeset	22 logsource = True
18669 18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	23
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	24 [blackbox]
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	25 track = command, commandfinish, commandexception, exthook, pythonhook
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	26
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	27 [blackbox]
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	28 track = incoming
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	29
19066 2cad301a7f06 blackbox: automatically rotate log files Bryan O'Sullivan <bryano@fb.com> parents: 19052 diff changeset	30 [blackbox]
2cad301a7f06 blackbox: automatically rotate log files Bryan O'Sullivan <bryano@fb.com> parents: 19052 diff changeset	31 # limit the size of a log file
2cad301a7f06 blackbox: automatically rotate log files Bryan O'Sullivan <bryano@fb.com> parents: 19052 diff changeset	32 maxsize = 1.5 MB
2cad301a7f06 blackbox: automatically rotate log files Bryan O'Sullivan <bryano@fb.com> parents: 19052 diff changeset	33 # rotate up to N log files when the current one gets too big
2cad301a7f06 blackbox: automatically rotate log files Bryan O'Sullivan <bryano@fb.com> parents: 19052 diff changeset	34 maxfiles = 3
2cad301a7f06 blackbox: automatically rotate log files Bryan O'Sullivan <bryano@fb.com> parents: 19052 diff changeset	35
18669 18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	36 """
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	37
28090 8113c88b8e6d blackbox: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 28027 diff changeset	38 from __future__ import absolute_import
8113c88b8e6d blackbox: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 28027 diff changeset	39
8113c88b8e6d blackbox: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 28027 diff changeset	40 import errno
8113c88b8e6d blackbox: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 28027 diff changeset	41 import re
8113c88b8e6d blackbox: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 28027 diff changeset	42
18669 18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	43 from mercurial.i18n import _
28245 caa2a0c6fbb7 blackbox: log working directory version timeless <timeless@mozdev.org> parents: 28244 diff changeset	44 from mercurial.node import hex
caa2a0c6fbb7 blackbox: log working directory version timeless <timeless@mozdev.org> parents: 28244 diff changeset	45
28090 8113c88b8e6d blackbox: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 28027 diff changeset	46 from mercurial import (
32337 46ba2cdda476 registrar: move cmdutil.command to registrar module (API) Yuya Nishihara <yuya@tcha.org> parents: 32155 diff changeset	47 registrar,
28248 851c41a21869 blackbox: properly replace ui class timeless <timeless@mozdev.org> parents: 28247 diff changeset	48 ui as uimod,
28090 8113c88b8e6d blackbox: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 28027 diff changeset	49 util,
8113c88b8e6d blackbox: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 28027 diff changeset	50 )
18669 18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	51
29841 d5883fd055c6 extensions: change magic "shipped with hg" string Augie Fackler <augie@google.com> parents: 28552 diff changeset	52 # Note for extension authors: ONLY specify testedwith = 'ships-with-hg-core' for
25186 80c5b2666a96 extensions: document that `testedwith = 'internal'` is special Augie Fackler <augie@google.com> parents: 23877 diff changeset	53 # extensions which SHIP WITH MERCURIAL. Non-mainline extensions should
80c5b2666a96 extensions: document that `testedwith = 'internal'` is special Augie Fackler <augie@google.com> parents: 23877 diff changeset	54 # be specifying the version(s) of Mercurial they are tested with, or
80c5b2666a96 extensions: document that `testedwith = 'internal'` is special Augie Fackler <augie@google.com> parents: 23877 diff changeset	55 # leave the attribute unspecified.
29841 d5883fd055c6 extensions: change magic "shipped with hg" string Augie Fackler <augie@google.com> parents: 28552 diff changeset	56 testedwith = 'ships-with-hg-core'
33134 7dc090faa8a4 blackbox: minor code reordering Pierre-Yves David <pierre-yves.david@octobus.net> parents: 32412 diff changeset	57
7dc090faa8a4 blackbox: minor code reordering Pierre-Yves David <pierre-yves.david@octobus.net> parents: 32412 diff changeset	58 cmdtable = {}
7dc090faa8a4 blackbox: minor code reordering Pierre-Yves David <pierre-yves.david@octobus.net> parents: 32412 diff changeset	59 command = registrar.command(cmdtable)
7dc090faa8a4 blackbox: minor code reordering Pierre-Yves David <pierre-yves.david@octobus.net> parents: 32412 diff changeset	60
33135 0a638f37f2d2 configitems: register 'blackbox.maxsize' as an example of 'configbytes' Pierre-Yves David <pierre-yves.david@octobus.net> parents: 33134 diff changeset	61 configtable = {}
0a638f37f2d2 configitems: register 'blackbox.maxsize' as an example of 'configbytes' Pierre-Yves David <pierre-yves.david@octobus.net> parents: 33134 diff changeset	62 configitem = registrar.configitem(configtable)
0a638f37f2d2 configitems: register 'blackbox.maxsize' as an example of 'configbytes' Pierre-Yves David <pierre-yves.david@octobus.net> parents: 33134 diff changeset	63
33189 1df74b71396d configitems: register the 'blackbox.dirty' config Pierre-Yves David <pierre-yves.david@octobus.net> parents: 33136 diff changeset	64 configitem('blackbox', 'dirty',
1df74b71396d configitems: register the 'blackbox.dirty' config Pierre-Yves David <pierre-yves.david@octobus.net> parents: 33136 diff changeset	65 default=False,
1df74b71396d configitems: register the 'blackbox.dirty' config Pierre-Yves David <pierre-yves.david@octobus.net> parents: 33136 diff changeset	66 )
33135 0a638f37f2d2 configitems: register 'blackbox.maxsize' as an example of 'configbytes' Pierre-Yves David <pierre-yves.david@octobus.net> parents: 33134 diff changeset	67 configitem('blackbox', 'maxsize',
33136 59c135bb31bc blackbox: use a human readable version of the default Pierre-Yves David <pierre-yves.david@octobus.net> parents: 33135 diff changeset	68 default='1 MB',
33135 0a638f37f2d2 configitems: register 'blackbox.maxsize' as an example of 'configbytes' Pierre-Yves David <pierre-yves.david@octobus.net> parents: 33134 diff changeset	69 )
33190 0ef40bb20264 configitems: register the 'blackbox.logsource' config Pierre-Yves David <pierre-yves.david@octobus.net> parents: 33189 diff changeset	70 configitem('blackbox', 'logsource',
0ef40bb20264 configitems: register the 'blackbox.logsource' config Pierre-Yves David <pierre-yves.david@octobus.net> parents: 33189 diff changeset	71 default=False,
0ef40bb20264 configitems: register the 'blackbox.logsource' config Pierre-Yves David <pierre-yves.david@octobus.net> parents: 33189 diff changeset	72 )
33135 0a638f37f2d2 configitems: register 'blackbox.maxsize' as an example of 'configbytes' Pierre-Yves David <pierre-yves.david@octobus.net> parents: 33134 diff changeset	73
28247 d2c0527af364 blackbox: store the blackbox ui object instead of the log file timeless <timeless@mozdev.org> parents: 28246 diff changeset	74 lastui = None
18669 18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	75
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	76 def wrapui(ui):
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	77 class blackboxui(ui.__class__):
28248 851c41a21869 blackbox: properly replace ui class timeless <timeless@mozdev.org> parents: 28247 diff changeset	78 def __init__(self, src=None):
851c41a21869 blackbox: properly replace ui class timeless <timeless@mozdev.org> parents: 28247 diff changeset	79 super(blackboxui, self).__init__(src)
851c41a21869 blackbox: properly replace ui class timeless <timeless@mozdev.org> parents: 28247 diff changeset	80 if src is None:
851c41a21869 blackbox: properly replace ui class timeless <timeless@mozdev.org> parents: 28247 diff changeset	81 self._partialinit()
851c41a21869 blackbox: properly replace ui class timeless <timeless@mozdev.org> parents: 28247 diff changeset	82 else:
28407 63da8bd0c65e blackbox: guard against recursion from dirty check timeless <timeless@mozdev.org> parents: 28305 diff changeset	83 self._bbinlog = False
28552 999e1acc61aa blackbox: do not assume self._bb{vfs,repo,fp} are set in blackboxui.__init__ Jun Wu <quark@fb.com> parents: 28407 diff changeset	84 self._bbrepo = getattr(src, '_bbrepo', None)
999e1acc61aa blackbox: do not assume self._bb{vfs,repo,fp} are set in blackboxui.__init__ Jun Wu <quark@fb.com> parents: 28407 diff changeset	85 self._bbvfs = getattr(src, '_bbvfs', None)
28248 851c41a21869 blackbox: properly replace ui class timeless <timeless@mozdev.org> parents: 28247 diff changeset	86
851c41a21869 blackbox: properly replace ui class timeless <timeless@mozdev.org> parents: 28247 diff changeset	87 def _partialinit(self):
851c41a21869 blackbox: properly replace ui class timeless <timeless@mozdev.org> parents: 28247 diff changeset	88 if util.safehasattr(self, '_bbvfs'):
851c41a21869 blackbox: properly replace ui class timeless <timeless@mozdev.org> parents: 28247 diff changeset	89 return
28407 63da8bd0c65e blackbox: guard against recursion from dirty check timeless <timeless@mozdev.org> parents: 28305 diff changeset	90 self._bbinlog = False
28248 851c41a21869 blackbox: properly replace ui class timeless <timeless@mozdev.org> parents: 28247 diff changeset	91 self._bbrepo = None
851c41a21869 blackbox: properly replace ui class timeless <timeless@mozdev.org> parents: 28247 diff changeset	92 self._bbvfs = None
851c41a21869 blackbox: properly replace ui class timeless <timeless@mozdev.org> parents: 28247 diff changeset	93
851c41a21869 blackbox: properly replace ui class timeless <timeless@mozdev.org> parents: 28247 diff changeset	94 def copy(self):
851c41a21869 blackbox: properly replace ui class timeless <timeless@mozdev.org> parents: 28247 diff changeset	95 self._partialinit()
851c41a21869 blackbox: properly replace ui class timeless <timeless@mozdev.org> parents: 28247 diff changeset	96 return self.__class__(self)
851c41a21869 blackbox: properly replace ui class timeless <timeless@mozdev.org> parents: 28247 diff changeset	97
18669 18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	98 @util.propertycache
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	99 def track(self):
19052 63a783d1ac85 blackbox: fix a case of name capture Bryan O'Sullivan <bryano@fb.com> parents: 18831 diff changeset	100 return self.configlist('blackbox', 'track', ['*'])
18669 18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	101
19066 2cad301a7f06 blackbox: automatically rotate log files Bryan O'Sullivan <bryano@fb.com> parents: 19052 diff changeset	102 def _openlogfile(self):
2cad301a7f06 blackbox: automatically rotate log files Bryan O'Sullivan <bryano@fb.com> parents: 19052 diff changeset	103 def rotate(oldpath, newpath):
2cad301a7f06 blackbox: automatically rotate log files Bryan O'Sullivan <bryano@fb.com> parents: 19052 diff changeset	104 try:
28026 a6db1163befa blackbox: refactor use of vfs as _bbvfs timeless <timeless@mozdev.org> parents: 28025 diff changeset	105 self._bbvfs.unlink(newpath)
25660 328739ea70c3 global: mass rewrite to use modern exception syntax Gregory Szorc <gregory.szorc@gmail.com> parents: 25186 diff changeset	106 except OSError as err:
19066 2cad301a7f06 blackbox: automatically rotate log files Bryan O'Sullivan <bryano@fb.com> parents: 19052 diff changeset	107 if err.errno != errno.ENOENT:
2cad301a7f06 blackbox: automatically rotate log files Bryan O'Sullivan <bryano@fb.com> parents: 19052 diff changeset	108 self.debug("warning: cannot remove '%s': %s\n" %
2cad301a7f06 blackbox: automatically rotate log files Bryan O'Sullivan <bryano@fb.com> parents: 19052 diff changeset	109 (newpath, err.strerror))
2cad301a7f06 blackbox: automatically rotate log files Bryan O'Sullivan <bryano@fb.com> parents: 19052 diff changeset	110 try:
2cad301a7f06 blackbox: automatically rotate log files Bryan O'Sullivan <bryano@fb.com> parents: 19052 diff changeset	111 if newpath:
28026 a6db1163befa blackbox: refactor use of vfs as _bbvfs timeless <timeless@mozdev.org> parents: 28025 diff changeset	112 self._bbvfs.rename(oldpath, newpath)
25660 328739ea70c3 global: mass rewrite to use modern exception syntax Gregory Szorc <gregory.szorc@gmail.com> parents: 25186 diff changeset	113 except OSError as err:
19066 2cad301a7f06 blackbox: automatically rotate log files Bryan O'Sullivan <bryano@fb.com> parents: 19052 diff changeset	114 if err.errno != errno.ENOENT:
2cad301a7f06 blackbox: automatically rotate log files Bryan O'Sullivan <bryano@fb.com> parents: 19052 diff changeset	115 self.debug("warning: cannot rename '%s' to '%s': %s\n" %
2cad301a7f06 blackbox: automatically rotate log files Bryan O'Sullivan <bryano@fb.com> parents: 19052 diff changeset	116 (newpath, oldpath, err.strerror))
2cad301a7f06 blackbox: automatically rotate log files Bryan O'Sullivan <bryano@fb.com> parents: 19052 diff changeset	117
33135 0a638f37f2d2 configitems: register 'blackbox.maxsize' as an example of 'configbytes' Pierre-Yves David <pierre-yves.david@octobus.net> parents: 33134 diff changeset	118 maxsize = self.configbytes('blackbox', 'maxsize')
34107 4f60720cf0df blackbox: fix rotation with chg Jun Wu <quark@fb.com> parents: 33436 diff changeset	119 name = 'blackbox.log'
19066 2cad301a7f06 blackbox: automatically rotate log files Bryan O'Sullivan <bryano@fb.com> parents: 19052 diff changeset	120 if maxsize > 0:
34107 4f60720cf0df blackbox: fix rotation with chg Jun Wu <quark@fb.com> parents: 33436 diff changeset	121 try:
4f60720cf0df blackbox: fix rotation with chg Jun Wu <quark@fb.com> parents: 33436 diff changeset	122 st = self._bbvfs.stat(name)
4f60720cf0df blackbox: fix rotation with chg Jun Wu <quark@fb.com> parents: 33436 diff changeset	123 except OSError:
4f60720cf0df blackbox: fix rotation with chg Jun Wu <quark@fb.com> parents: 33436 diff changeset	124 pass
4f60720cf0df blackbox: fix rotation with chg Jun Wu <quark@fb.com> parents: 33436 diff changeset	125 else:
4f60720cf0df blackbox: fix rotation with chg Jun Wu <quark@fb.com> parents: 33436 diff changeset	126 if st.st_size >= maxsize:
4f60720cf0df blackbox: fix rotation with chg Jun Wu <quark@fb.com> parents: 33436 diff changeset	127 path = self._bbvfs.join(name)
4f60720cf0df blackbox: fix rotation with chg Jun Wu <quark@fb.com> parents: 33436 diff changeset	128 maxfiles = self.configint('blackbox', 'maxfiles', 7)
4f60720cf0df blackbox: fix rotation with chg Jun Wu <quark@fb.com> parents: 33436 diff changeset	129 for i in xrange(maxfiles - 1, 1, -1):
4f60720cf0df blackbox: fix rotation with chg Jun Wu <quark@fb.com> parents: 33436 diff changeset	130 rotate(oldpath='%s.%d' % (path, i - 1),
4f60720cf0df blackbox: fix rotation with chg Jun Wu <quark@fb.com> parents: 33436 diff changeset	131 newpath='%s.%d' % (path, i))
4f60720cf0df blackbox: fix rotation with chg Jun Wu <quark@fb.com> parents: 33436 diff changeset	132 rotate(oldpath=path,
4f60720cf0df blackbox: fix rotation with chg Jun Wu <quark@fb.com> parents: 33436 diff changeset	133 newpath=maxfiles > 0 and path + '.1')
4f60720cf0df blackbox: fix rotation with chg Jun Wu <quark@fb.com> parents: 33436 diff changeset	134 return self._bbvfs(name, 'a')
19066 2cad301a7f06 blackbox: automatically rotate log files Bryan O'Sullivan <bryano@fb.com> parents: 19052 diff changeset	135
18669 18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	136 def log(self, event, msg, *opts):
28247 d2c0527af364 blackbox: store the blackbox ui object instead of the log file timeless <timeless@mozdev.org> parents: 28246 diff changeset	137 global lastui
18669 18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	138 super(blackboxui, self).log(event, msg, *opts)
28248 851c41a21869 blackbox: properly replace ui class timeless <timeless@mozdev.org> parents: 28247 diff changeset	139 self._partialinit()
18669 18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	140
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	141 if not '*' in self.track and not event in self.track:
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	142 return
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	143
34109 cf04db16f583 blackbox: do not cache file objects Jun Wu <quark@fb.com> parents: 34108 diff changeset	144 if self._bbvfs:
28247 d2c0527af364 blackbox: store the blackbox ui object instead of the log file timeless <timeless@mozdev.org> parents: 28246 diff changeset	145 ui = self
18669 18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	146 else:
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	147 # certain ui instances exist outside the context of
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	148 # a repo, so just default to the last blackbox that
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	149 # was seen.
28247 d2c0527af364 blackbox: store the blackbox ui object instead of the log file timeless <timeless@mozdev.org> parents: 28246 diff changeset	150 ui = lastui
18669 18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	151
34109 cf04db16f583 blackbox: do not cache file objects Jun Wu <quark@fb.com> parents: 34108 diff changeset	152 if not ui:
28407 63da8bd0c65e blackbox: guard against recursion from dirty check timeless <timeless@mozdev.org> parents: 28305 diff changeset	153 return
63da8bd0c65e blackbox: guard against recursion from dirty check timeless <timeless@mozdev.org> parents: 28305 diff changeset	154 if not lastui or ui._bbrepo:
63da8bd0c65e blackbox: guard against recursion from dirty check timeless <timeless@mozdev.org> parents: 28305 diff changeset	155 lastui = ui
63da8bd0c65e blackbox: guard against recursion from dirty check timeless <timeless@mozdev.org> parents: 28305 diff changeset	156 if ui._bbinlog:
34109 cf04db16f583 blackbox: do not cache file objects Jun Wu <quark@fb.com> parents: 34108 diff changeset	157 # recursion and failure guard
28407 63da8bd0c65e blackbox: guard against recursion from dirty check timeless <timeless@mozdev.org> parents: 28305 diff changeset	158 return
63da8bd0c65e blackbox: guard against recursion from dirty check timeless <timeless@mozdev.org> parents: 28305 diff changeset	159 try:
63da8bd0c65e blackbox: guard against recursion from dirty check timeless <timeless@mozdev.org> parents: 28305 diff changeset	160 ui._bbinlog = True
32412 043948c84647 devel: update blackbox to use default-date Boris Feld <boris.feld@octobus.net> parents: 32337 diff changeset	161 default = self.configdate('devel', 'default-date')
043948c84647 devel: update blackbox to use default-date Boris Feld <boris.feld@octobus.net> parents: 32337 diff changeset	162 date = util.datestr(default, '%Y/%m/%d %H:%M:%S')
18787 f56278a0a0c5 blackbox: use util.getuser for portability Bryan O'Sullivan <bryano@fb.com> parents: 18786 diff changeset	163 user = util.getuser()
32155 055cca8e167b py3: use %d to format integers into bytestrings Pulkit Goyal <7895pulkit@gmail.com> parents: 29841 diff changeset	164 pid = '%d' % util.getpid()
18669 18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	165 formattedmsg = msg[0] % msg[1:]
28245 caa2a0c6fbb7 blackbox: log working directory version timeless <timeless@mozdev.org> parents: 28244 diff changeset	166 rev = '(unknown)'
28246 b862e793ec10 blackbox: log dirty state timeless <timeless@mozdev.org> parents: 28245 diff changeset	167 changed = ''
28248 851c41a21869 blackbox: properly replace ui class timeless <timeless@mozdev.org> parents: 28247 diff changeset	168 if ui._bbrepo:
28247 d2c0527af364 blackbox: store the blackbox ui object instead of the log file timeless <timeless@mozdev.org> parents: 28246 diff changeset	169 ctx = ui._bbrepo[None]
28304 6b38888ab033 blackbox: remove hexfn timeless <timeless@mozdev.org> parents: 28303 diff changeset	170 parents = ctx.parents()
6b38888ab033 blackbox: remove hexfn timeless <timeless@mozdev.org> parents: 28303 diff changeset	171 rev = ('+'.join([hex(p.node()) for p in parents]))
33361 66fe60d5f291 blackbox: simplify the dirty check Matt Harbison <matt_harbison@yahoo.com> parents: 33190 diff changeset	172 if (ui.configbool('blackbox', 'dirty') and
66fe60d5f291 blackbox: simplify the dirty check Matt Harbison <matt_harbison@yahoo.com> parents: 33190 diff changeset	173 ctx.dirty(missing=True, merge=False, branch=False)):
28304 6b38888ab033 blackbox: remove hexfn timeless <timeless@mozdev.org> parents: 28303 diff changeset	174 changed = '+'
33190 0ef40bb20264 configitems: register the 'blackbox.logsource' config Pierre-Yves David <pierre-yves.david@octobus.net> parents: 33189 diff changeset	175 if ui.configbool('blackbox', 'logsource'):
28305 f5ae291dfedf blackbox: optionally log event source timeless <timeless@mozdev.org> parents: 28304 diff changeset	176 src = ' [%s]' % event
f5ae291dfedf blackbox: optionally log event source timeless <timeless@mozdev.org> parents: 28304 diff changeset	177 else:
f5ae291dfedf blackbox: optionally log event source timeless <timeless@mozdev.org> parents: 28304 diff changeset	178 src = ''
18786 ed39a8f94e95 blackbox: prevent failed I/O from causing hg to abort Bryan O'Sullivan <bryano@fb.com> parents: 18676 diff changeset	179 try:
34108 50df6cf22717 blackbox: inline _bbwrite Jun Wu <quark@fb.com> parents: 34107 diff changeset	180 fmt = '%s %s @%s%s (%s)%s> %s'
50df6cf22717 blackbox: inline _bbwrite Jun Wu <quark@fb.com> parents: 34107 diff changeset	181 args = (date, user, rev, changed, pid, src, formattedmsg)
34109 cf04db16f583 blackbox: do not cache file objects Jun Wu <quark@fb.com> parents: 34108 diff changeset	182 with ui._openlogfile() as fp:
cf04db16f583 blackbox: do not cache file objects Jun Wu <quark@fb.com> parents: 34108 diff changeset	183 fp.write(fmt % args)
cf04db16f583 blackbox: do not cache file objects Jun Wu <quark@fb.com> parents: 34108 diff changeset	184 except (IOError, OSError) as err:
18786 ed39a8f94e95 blackbox: prevent failed I/O from causing hg to abort Bryan O'Sullivan <bryano@fb.com> parents: 18676 diff changeset	185 self.debug('warning: cannot write to blackbox.log: %s\n' %
ed39a8f94e95 blackbox: prevent failed I/O from causing hg to abort Bryan O'Sullivan <bryano@fb.com> parents: 18676 diff changeset	186 err.strerror)
34109 cf04db16f583 blackbox: do not cache file objects Jun Wu <quark@fb.com> parents: 34108 diff changeset	187 # do not restore _bbinlog intentionally to avoid failed
cf04db16f583 blackbox: do not cache file objects Jun Wu <quark@fb.com> parents: 34108 diff changeset	188 # logging again
cf04db16f583 blackbox: do not cache file objects Jun Wu <quark@fb.com> parents: 34108 diff changeset	189 else:
cf04db16f583 blackbox: do not cache file objects Jun Wu <quark@fb.com> parents: 34108 diff changeset	190 ui._bbinlog = False
28407 63da8bd0c65e blackbox: guard against recursion from dirty check timeless <timeless@mozdev.org> parents: 28305 diff changeset	191 finally:
34109 cf04db16f583 blackbox: do not cache file objects Jun Wu <quark@fb.com> parents: 34108 diff changeset	192 pass
18669 18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	193
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	194 def setrepo(self, repo):
28407 63da8bd0c65e blackbox: guard against recursion from dirty check timeless <timeless@mozdev.org> parents: 28305 diff changeset	195 self._bbinlog = False
28248 851c41a21869 blackbox: properly replace ui class timeless <timeless@mozdev.org> parents: 28247 diff changeset	196 self._bbrepo = repo
28026 a6db1163befa blackbox: refactor use of vfs as _bbvfs timeless <timeless@mozdev.org> parents: 28025 diff changeset	197 self._bbvfs = repo.vfs
18669 18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	198
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	199 ui.__class__ = blackboxui
28248 851c41a21869 blackbox: properly replace ui class timeless <timeless@mozdev.org> parents: 28247 diff changeset	200 uimod.ui = blackboxui
18669 18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	201
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	202 def uisetup(ui):
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	203 wrapui(ui)
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	204
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	205 def reposetup(ui, repo):
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	206 # During 'hg pull' a httppeer repo is created to represent the remote repo.
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	207 # It doesn't have a .hg directory to put a blackbox in, so we don't do
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	208 # the blackbox setup for it.
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	209 if not repo.local():
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	210 return
18242716a014 blackbox: adds a blackbox extension Durham Goode <durham@fb.com> parents: diff changeset	211
19230 6d6f148cada9 blackbox: fix blackbox causing exceptions in tests Durham Goode <durham@fb.com> parents: 19162 diff changeset	212 if util.safehasattr(ui, 'setrepo'):
6d6f148cada9 blackbox: fix blackbox causing exceptions in tests Durham Goode <durham@fb.com> parents: 19162 diff changeset	213 ui.setrepo(repo)
33436 9bb4decd43b0 repovfs: add a ward to check if locks are properly taken Boris Feld <boris.feld@octobus.net> parents: 33361 diff changeset	214 repo._wlockfreeprefix.add('blackbox.log')
18673 f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	215
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	216 @command('^blackbox',
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	217 [('l', 'limit', 10, _('the number of events to show')),
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	218 ],
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	219 _('hg blackbox [OPTION]...'))
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	220 def blackbox(ui, repo, revs, *opts):
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	221 '''view the recent repository events
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	222 '''
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	223
28026 a6db1163befa blackbox: refactor use of vfs as _bbvfs timeless <timeless@mozdev.org> parents: 28025 diff changeset	224 if not repo.vfs.exists('blackbox.log'):
18673 f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	225 return
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	226
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	227 limit = opts.get('limit')
28244 c17d7b1c40be blackbox: rename fp variable timeless <timeless@mozdev.org> parents: 28243 diff changeset	228 fp = repo.vfs('blackbox.log', 'r')
c17d7b1c40be blackbox: rename fp variable timeless <timeless@mozdev.org> parents: 28243 diff changeset	229 lines = fp.read().split('\n')
18673 f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	230
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	231 count = 0
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	232 output = []
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	233 for line in reversed(lines):
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	234 if count >= limit:
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	235 break
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	236
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	237 # count the commands by matching lines like: 2013/01/23 19:13:36 root>
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	238 if re.match('^\d{4}/\d{2}/\d{2} \d{2}:\d{2}:\d{2} .> .', line):
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	239 count += 1
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	240 output.append(line)
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	241
f27598902007 blackbox: adds a 'blackbox' command for viewing recent logs Durham Goode <durham@fb.com> parents: 18669 diff changeset	242 ui.status('\n'.join(reversed(output)))

Mercurial > public > mercurial-scm > hg

annotate hgext/blackbox.py @ 34109:cf04db16f583