← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/lockdir.py

  • Committer: Danny van Heumen
  • Date: 2010-03-09 16:38:10 UTC
  • mto: (4634.139.5 2.0)
  • mto: This revision was merged to the branch mainline in revision 5160.
  • Revision ID: danny@dannyvanheumen.nl-20100309163810-ujn8hcx08f75nlf1
Refined test to make use of locking hooks and also validate if lock is truly a checkout-lock.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/lockdir.py
1
 
# Copyright (C) 2006-2010 Canonical Ltd
 
1
# Copyright (C) 2006, 2007, 2008, 2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
110
110
    debug,
111
111
    errors,
112
112
    lock,
113
 
    osutils,
114
113
    )
115
114
import bzrlib.config
116
 
from bzrlib.decorators import only_raises
117
115
from bzrlib.errors import (
118
116
        DirectoryNotEmpty,
119
117
        FileExists,
151
149
# files/dirs created.
152
150
 
153
151
 
154
 
_DEFAULT_TIMEOUT_SECONDS = 30
 
152
_DEFAULT_TIMEOUT_SECONDS = 300
155
153
_DEFAULT_POLL_SECONDS = 1.0
156
154
 
157
155
 
243
241
        # have a similar bug allowing someone to think they got the lock
244
242
        # when it's already held.
245
243
        #
246
 
        # See <https://bugs.launchpad.net/bzr/+bug/498378> for one case.
 
244
        # See <https://bugs.edge.launchpad.net/bzr/+bug/498378> for one case.
247
245
        #
248
246
        # Strictly the check is unnecessary and a waste of time for most
249
247
        # people, but probably worth trapping if something is wrong.
296
294
                                            info_bytes)
297
295
        return tmpname
298
296
 
299
 
    @only_raises(LockNotHeld, LockBroken)
300
297
    def unlock(self):
301
298
        """Release a held lock
302
299
        """
425
422
 
426
423
        peek() reads the info file of the lock holder, if any.
427
424
        """
428
 
        return self._parse_info(self.transport.get_bytes(path))
 
425
        return self._parse_info(self.transport.get(path))
429
426
 
430
427
    def peek(self):
431
428
        """Check if the lock is held by anyone.
447
444
        # XXX: is creating this here inefficient?
448
445
        config = bzrlib.config.GlobalConfig()
449
446
        try:
 
447
            user = config.user_email()
 
448
        except errors.NoEmailInUsername:
450
449
            user = config.username()
451
 
        except errors.NoWhoami:
452
 
            user = osutils.getuser_unicode()
453
450
        s = rio.Stanza(hostname=get_host_name(),
454
451
                   pid=str(os.getpid()),
455
452
                   start_time=str(int(time.time())),
458
455
                   )
459
456
        return s.to_string()
460
457
 
461
 
    def _parse_info(self, info_bytes):
462
 
        stanza = rio.read_stanza(osutils.split_lines(info_bytes))
 
458
    def _parse_info(self, info_file):
 
459
        stanza = rio.read_stanza(info_file.readlines())
463
460
        if stanza is None:
464
461
            # see bug 185013; we fairly often end up with the info file being
465
462
            # empty after an interruption; we could log a message here but
539
536
                if deadline_str is None:
540
537
                    deadline_str = time.strftime('%H:%M:%S',
541
538
                                                 time.localtime(deadline))
542
 
                # As local lock urls are correct we display them.
543
 
                # We avoid displaying remote lock urls.
544
539
                lock_url = self.transport.abspath(self.path)
545
 
                if lock_url.startswith('file://'):
546
 
                    lock_url = lock_url.split('.bzr/')[0]
547
 
                else:
548
 
                    lock_url = ''
549
 
                user, hostname, pid, time_ago = formatted_info
550
 
                msg = ('%s lock %s '        # lock_url
551
 
                    'held by '              # start
552
 
                    '%s\n'                  # user
553
 
                    'at %s '                # hostname
554
 
                    '[process #%s], '       # pid
555
 
                    'acquired %s.')         # time ago
556
 
                msg_args = [start, lock_url, user, hostname, pid, time_ago]
557
 
                if timeout > 0:
558
 
                    msg += ('\nWill continue to try until %s, unless '
559
 
                        'you press Ctrl-C.')
560
 
                    msg_args.append(deadline_str)
561
 
                msg += '\nSee "bzr help break-lock" for more.'
562
 
                self._report_function(msg, *msg_args)
 
540
                self._report_function('%s %s\n'
 
541
                                      '%s\n' # held by
 
542
                                      '%s\n' # locked ... ago
 
543
                                      'Will continue to try until %s, unless '
 
544
                                      'you press Ctrl-C\n'
 
545
                                      'If you\'re sure that it\'s not being '
 
546
                                      'modified, use bzr break-lock %s',
 
547
                                      start,
 
548
                                      formatted_info[0],
 
549
                                      formatted_info[1],
 
550
                                      formatted_info[2],
 
551
                                      deadline_str,
 
552
                                      lock_url)
 
553
 
563
554
            if (max_attempts is not None) and (attempt_count >= max_attempts):
564
555
                self._trace("exceeded %d attempts")
565
556
                raise LockContention(self)
567
558
                self._trace("waiting %ss", poll)
568
559
                time.sleep(poll)
569
560
            else:
570
 
                # As timeout is always 0 for remote locks
571
 
                # this block is applicable only for local
572
 
                # lock contention
573
561
                self._trace("timeout after waiting %ss", timeout)
574
 
                raise LockContention('(local)', lock_url)
 
562
                raise LockContention(self)
575
563
 
576
564
    def leave_in_place(self):
577
565
        self._locked_via_token = True
622
610
 
623
611
    def _format_lock_info(self, info):
624
612
        """Turn the contents of peek() into something for the user"""
 
613
        lock_url = self.transport.abspath(self.path)
625
614
        start_time = info.get('start_time')
626
615
        if start_time is None:
627
616
            time_ago = '(unknown)'
628
617
        else:
629
618
            time_ago = format_delta(time.time() - int(info['start_time']))
630
 
        user = info.get('user', '<unknown>')
631
 
        hostname = info.get('hostname', '<unknown>')
632
 
        pid = info.get('pid', '<unknown>')
633
619
        return [
634
 
            user,
635
 
            hostname,
636
 
            pid,
637
 
            time_ago,
 
620
            'lock %s' % (lock_url,),
 
621
            'held by %s on host %s [process #%s]' %
 
622
                tuple([info.get(x, '<unknown>') for x in ['user', 'hostname', 'pid']]),
 
623
            'locked %s' % (time_ago,),
638
624
            ]
639
625
 
640
626
    def validate_token(self, token):