/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/memorybranch.py

  • Committer: Jelmer Vernooij
  • Date: 2020-07-28 02:11:05 UTC
  • mfrom: (7490.40.78 work)
  • mto: This revision was merged to the branch mainline in revision 7520.
  • Revision ID: jelmer@jelmer.uk-20200728021105-fzq7g6f8bl1g0aet
Merge lp:brz/3.1.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/memorybranch.py
 
1
# Copyright (C) 2020 Breezy Developers
 
2
#
 
3
# This program is free software; you can redistribute it and/or modify
 
4
# it under the terms of the GNU General Public License as published by
 
5
# the Free Software Foundation; either version 2 of the License, or
 
6
# (at your option) any later version.
 
7
#
 
8
# This program is distributed in the hope that it will be useful,
 
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
# GNU General Public License for more details.
 
12
#
 
13
# You should have received a copy of the GNU General Public License
 
14
# along with this program; if not, write to the Free Software
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
16
 
 
17
"""MemoryBranch object.
 
18
"""
 
19
 
 
20
from __future__ import absolute_import
 
21
 
 
22
from . import config as _mod_config, errors, osutils
 
23
from .branch import Branch
 
24
from .lock import LogicalLockResult, _RelockDebugMixin
 
25
from .revision import NULL_REVISION
 
26
from .tag import DisabledTags, MemoryTags
 
27
 
 
28
 
 
29
class MemoryBranch(Branch, _RelockDebugMixin):
 
30
 
 
31
    def __init__(self, repository, last_revision_info, tags=None):
 
32
        self.repository = repository
 
33
        self._last_revision_info = last_revision_info
 
34
        self._revision_history_cache = None
 
35
        if tags is not None:
 
36
            self.tags = MemoryTags(tags)
 
37
        else:
 
38
            self.tags = DisabledTags(self)
 
39
        self._partial_revision_history_cache = []
 
40
        self._last_revision_info_cache = None
 
41
        self._revision_id_to_revno_cache = None
 
42
        self._partial_revision_id_to_revno_cache = {}
 
43
        self.base = 'memory://' + osutils.rand_chars(10)
 
44
 
 
45
    def get_config(self):
 
46
        return _mod_config.Config()
 
47
 
 
48
    def lock_read(self):
 
49
        self.repository.lock_read()
 
50
        return LogicalLockResult(self.unlock)
 
51
 
 
52
    def lock_write(self, token=None):
 
53
        self.repository.lock_write()
 
54
        return BranchWriteLockResult(self.unlock, None)
 
55
 
 
56
    def unlock(self):
 
57
        self.repository.unlock()
 
58
 
 
59
    def last_revision_info(self):
 
60
        return self._last_revision_info
 
61
 
 
62
    def _gen_revision_history(self):
 
63
        """Generate the revision history from last revision
 
64
        """
 
65
        last_revno, last_revision = self.last_revision_info()
 
66
        with self.lock_read():
 
67
            self._extend_partial_history()
 
68
            return list(reversed(self._partial_revision_history_cache))
 
69
 
 
70
    def get_rev_id(self, revno, history=None):
 
71
        """Find the revision id of the specified revno."""
 
72
        with self.lock_read():
 
73
            if revno == 0:
 
74
                return NULL_REVISION
 
75
            last_revno, last_revid = self.last_revision_info()
 
76
            if revno == last_revno:
 
77
                return last_revid
 
78
            if last_revno is None:
 
79
                self._extend_partial_history()
 
80
                return self._partial_revision_history_cache[
 
81
                    len(self._partial_revision_history_cache) - revno]
 
82
            else:
 
83
                if revno <= 0 or revno > last_revno:
 
84
                    raise errors.NoSuchRevision(self, revno)
 
85
                distance_from_last = last_revno - revno
 
86
                if len(self._partial_revision_history_cache) <= \
 
87
                        distance_from_last:
 
88
                    self._extend_partial_history(distance_from_last)
 
89
                return self._partial_revision_history_cache[distance_from_last]