__init__.py
127 lines
| 5.0 KiB
| text/x-python
|
PythonLexer
|
r6239 | # __init__.py - inotify-based status acceleration for Linux | ||
# | ||||
# Copyright 2006, 2007, 2008 Bryan O'Sullivan <bos@serpentine.com> | ||||
# Copyright 2007, 2008 Brendan Cully <brendan@kublai.com> | ||||
# | ||||
# This software may be used and distributed according to the terms | ||||
# of the GNU General Public License, incorporated herein by reference. | ||||
'''inotify-based status acceleration for Linux systems | ||||
''' | ||||
# todo: socket permissions | ||||
|
r7225 | from mercurial.i18n import _ | ||
|
r6239 | from mercurial import cmdutil, util | ||
import client, errno, os, server, socket | ||||
from weakref import proxy | ||||
def serve(ui, repo, **opts): | ||||
'''start an inotify server for this repository''' | ||||
timeout = opts.get('timeout') | ||||
if timeout: | ||||
timeout = float(timeout) * 1e3 | ||||
class service: | ||||
def init(self): | ||||
|
r6995 | try: | ||
self.master = server.Master(ui, repo, timeout) | ||||
except server.AlreadyStartedException, inst: | ||||
raise util.Abort(str(inst)) | ||||
|
r6239 | |||
def run(self): | ||||
try: | ||||
self.master.run() | ||||
finally: | ||||
self.master.shutdown() | ||||
service = service() | ||||
cmdutil.service(opts, initfn=service.init, runfn=service.run) | ||||
def reposetup(ui, repo): | ||||
|
r7522 | if not hasattr(repo, 'dirstate'): | ||
|
r6239 | return | ||
# XXX: weakref until hg stops relying on __del__ | ||||
repo = proxy(repo) | ||||
class inotifydirstate(repo.dirstate.__class__): | ||||
# Set to True if we're the inotify server, so we don't attempt | ||||
# to recurse. | ||||
inotifyserver = False | ||||
|
r6753 | def status(self, match, ignored, clean, unknown=True): | ||
|
r6603 | files = match.files() | ||
|
r7393 | if '.' in files: | ||
|
r7434 | files = [] | ||
|
r6239 | try: | ||
|
r6753 | if not ignored and not self.inotifyserver: | ||
|
r6239 | result = client.query(ui, repo, files, match, False, | ||
|
r6753 | clean, unknown) | ||
|
r7219 | if ui.config('inotify', 'debug'): | ||
r2 = super(inotifydirstate, self).status( | ||||
match, False, clean, unknown) | ||||
for c,a,b in zip('LMARDUIC', result, r2): | ||||
for f in a: | ||||
if f not in b: | ||||
ui.warn('*** inotify: %s +%s\n' % (c, f)) | ||||
for f in b: | ||||
if f not in a: | ||||
|
r7304 | ui.warn('*** inotify: %s -%s\n' % (c, f)) | ||
|
r7219 | result = r2 | ||
|
r6239 | if result is not None: | ||
return result | ||||
|
r6997 | except (OSError, socket.error), err: | ||
|
r7452 | autostart = ui.configbool('inotify', 'autostart', True) | ||
|
r6239 | if err[0] == errno.ECONNREFUSED: | ||
ui.warn(_('(found dead inotify server socket; ' | ||||
'removing it)\n')) | ||||
os.unlink(repo.join('inotify.sock')) | ||||
|
r7452 | if err[0] in (errno.ECONNREFUSED, errno.ENOENT) and autostart: | ||
|
r6239 | query = None | ||
ui.debug(_('(starting inotify server)\n')) | ||||
try: | ||||
|
r7329 | try: | ||
server.start(ui, repo) | ||||
query = client.query | ||||
except server.AlreadyStartedException, inst: | ||||
# another process may have started its own | ||||
# inotify server while this one was starting. | ||||
ui.debug(str(inst)) | ||||
query = client.query | ||||
|
r6239 | except Exception, inst: | ||
ui.warn(_('could not start inotify server: ' | ||||
'%s\n') % inst) | ||||
if query: | ||||
try: | ||||
return query(ui, repo, files or [], match, | ||||
|
r6753 | ignored, clean, unknown) | ||
|
r6239 | except socket.error, err: | ||
ui.warn(_('could not talk to new inotify ' | ||||
|
r6996 | 'server: %s\n') % err[-1]) | ||
|
r7452 | elif err[0] in (errno.ECONNREFUSED, errno.ENOENT): | ||
# silently ignore normal errors if autostart is False | ||||
ui.debug(_('(inotify server not running)\n')) | ||||
|
r6996 | else: | ||
ui.warn(_('failed to contact inotify server: %s\n') | ||||
% err[-1]) | ||||
ui.print_exc() | ||||
# replace by old status function | ||||
self.status = super(inotifydirstate, self).status | ||||
|
r6239 | |||
return super(inotifydirstate, self).status( | ||||
|
r6753 | match, ignored, clean, unknown) | ||
|
r6239 | |||
repo.dirstate.__class__ = inotifydirstate | ||||
cmdtable = { | ||||
'^inserve': | ||||
(serve, | ||||
[('d', 'daemon', None, _('run server in background')), | ||||
('', 'daemon-pipefds', '', _('used internally by daemon mode')), | ||||
('t', 'idle-timeout', '', _('minutes to sit idle before exiting')), | ||||
('', 'pid-file', '', _('name of file to write process ID to'))], | ||||
_('hg inserve [OPT]...')), | ||||
} | ||||