← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/revisiontree.py

  • Committer: John Arbash Meinel
  • Date: 2007-03-01 21:56:19 UTC
  • mto: (2255.7.84 dirstate)
  • mto: This revision was merged to the branch mainline in revision 2322.
  • Revision ID: john@arbash-meinel.com-20070301215619-wpt6kz8yem3ypu1b
Update to dirstate locking.
Move all of WT4.lock_* functions locally, so that they can
properly interact and cleanup around when we lock/unlock the
dirstate file.
Change all Lock objects to be non-blocking. So that if someone
grabs a lock on the DirState we find out immediately, rather
than blocking.
Change WT4.unlock() so that if the dirstate is dirty, it will
save the contents even if it only has a read lock.
It does this by trying to take a write lock, if it fails
we just ignore it. If it succeeds, then we can flush to disk.
This is more important now that DirState tracks file changes.
It allows 'bzr status' to update the cached stat and sha values.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/revisiontree.py
 
1
# Copyright (C) 2005, 2007 Canonical Ltd
 
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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
"""RevisionTree - a Tree implementation backed by repository data for a revision."""
 
18
 
 
19
from cStringIO import StringIO
 
20
 
 
21
from bzrlib import (
 
22
    osutils,
 
23
    revision,
 
24
    )
 
25
from bzrlib.tree import Tree
 
26
 
 
27
 
 
28
class RevisionTree(Tree):
 
29
    """Tree viewing a previous revision.
 
30
 
 
31
    File text can be retrieved from the text store.
 
32
    """
 
33
    
 
34
    def __init__(self, branch, inv, revision_id):
 
35
        # for compatability the 'branch' parameter has not been renamed to 
 
36
        # repository at this point. However, we should change RevisionTree's
 
37
        # construction to always be via Repository and not via direct 
 
38
        # construction - this will mean that we can change the constructor
 
39
        # with much less chance of breaking client code.
 
40
        self._repository = branch
 
41
        self._weave_store = branch.weave_store
 
42
        self._inventory = inv
 
43
        self._revision_id = osutils.safe_revision_id(revision_id)
 
44
 
 
45
    def get_parent_ids(self):
 
46
        """See Tree.get_parent_ids.
 
47
 
 
48
        A RevisionTree's parents match the revision graph.
 
49
        """
 
50
        if self._revision_id in (None, revision.NULL_REVISION):
 
51
            parent_ids = []
 
52
        else:
 
53
            parent_ids = self._repository.get_revision(
 
54
                self._revision_id).parent_ids
 
55
        return parent_ids
 
56
        
 
57
    def get_revision_id(self):
 
58
        """Return the revision id associated with this tree."""
 
59
        return self._revision_id
 
60
 
 
61
    def get_weave(self, file_id):
 
62
        file_id = osutils.safe_file_id(file_id)
 
63
        return self._weave_store.get_weave(file_id,
 
64
                self._repository.get_transaction())
 
65
 
 
66
    def get_file_lines(self, file_id):
 
67
        file_id = osutils.safe_file_id(file_id)
 
68
        ie = self._inventory[file_id]
 
69
        weave = self.get_weave(file_id)
 
70
        return weave.get_lines(ie.revision)
 
71
 
 
72
    def get_file_text(self, file_id):
 
73
        file_id = osutils.safe_file_id(file_id)
 
74
        return ''.join(self.get_file_lines(file_id))
 
75
 
 
76
    def get_file(self, file_id):
 
77
        file_id = osutils.safe_file_id(file_id)
 
78
        return StringIO(self.get_file_text(file_id))
 
79
 
 
80
    def annotate_iter(self, file_id):
 
81
        """See Tree.annotate_iter"""
 
82
        file_id = osutils.safe_file_id(file_id)
 
83
        w = self.get_weave(file_id)
 
84
        return w.annotate_iter(self.inventory[file_id].revision)
 
85
 
 
86
    def get_file_size(self, file_id):
 
87
        file_id = osutils.safe_file_id(file_id)
 
88
        return self._inventory[file_id].text_size
 
89
 
 
90
    def get_file_sha1(self, file_id, path=None, stat_value=None):
 
91
        file_id = osutils.safe_file_id(file_id)
 
92
        ie = self._inventory[file_id]
 
93
        if ie.kind == "file":
 
94
            return ie.text_sha1
 
95
        return None
 
96
 
 
97
    def get_file_mtime(self, file_id, path=None):
 
98
        file_id = osutils.safe_file_id(file_id)
 
99
        ie = self._inventory[file_id]
 
100
        revision = self._repository.get_revision(ie.revision)
 
101
        return revision.timestamp
 
102
 
 
103
    def is_executable(self, file_id, path=None):
 
104
        file_id = osutils.safe_file_id(file_id)
 
105
        ie = self._inventory[file_id]
 
106
        if ie.kind != "file":
 
107
            return None
 
108
        return ie.executable
 
109
 
 
110
    def has_filename(self, filename):
 
111
        return bool(self.inventory.path2id(filename))
 
112
 
 
113
    def list_files(self, include_root=False):
 
114
        # The only files returned by this are those from the version
 
115
        entries = self.inventory.iter_entries()
 
116
        # skip the root for compatability with the current apis.
 
117
        if self.inventory.root is not None and not include_root:
 
118
            # skip the root for compatability with the current apis.
 
119
            entries.next()
 
120
        for path, entry in entries:
 
121
            yield path, 'V', entry.kind, entry.file_id, entry
 
122
 
 
123
    def get_symlink_target(self, file_id):
 
124
        file_id = osutils.safe_file_id(file_id)
 
125
        ie = self._inventory[file_id]
 
126
        return ie.symlink_target;
 
127
 
 
128
    def kind(self, file_id):
 
129
        file_id = osutils.safe_file_id(file_id)
 
130
        return self._inventory[file_id].kind
 
131
 
 
132
    def _comparison_data(self, entry, path):
 
133
        if entry is None:
 
134
            return None, False, None
 
135
        return entry.kind, entry.executable, None
 
136
 
 
137
    def _file_size(self, entry, stat_value):
 
138
        assert entry.text_size is not None
 
139
        return entry.text_size
 
140
 
 
141
    def lock_read(self):
 
142
        self._repository.lock_read()
 
143
 
 
144
    def __repr__(self):
 
145
        return '<%s instance at %x, rev_id=%r>' % (
 
146
            self.__class__.__name__, id(self), self._revision_id)
 
147
 
 
148
    def unlock(self):
 
149
        self._repository.unlock()
 
150
 
 
151
    def walkdirs(self, prefix=""):
 
152
        _directory = 'directory'
 
153
        inv = self.inventory
 
154
        top_id = inv.path2id(prefix)
 
155
        if top_id is None:
 
156
            pending = []
 
157
        else:
 
158
            pending = [(prefix, '', _directory, None, top_id, None)]
 
159
        while pending:
 
160
            dirblock = []
 
161
            currentdir = pending.pop()
 
162
            # 0 - relpath, 1- basename, 2- kind, 3- stat, id, v-kind
 
163
            if currentdir[0]:
 
164
                relroot = currentdir[0] + '/'
 
165
            else:
 
166
                relroot = ""
 
167
            # FIXME: stash the node in pending
 
168
            entry = inv[currentdir[4]]
 
169
            for name, child in entry.sorted_children():
 
170
                toppath = relroot + name
 
171
                dirblock.append((toppath, name, child.kind, None,
 
172
                    child.file_id, child.kind
 
173
                    ))
 
174
            yield (currentdir[0], entry.file_id), dirblock
 
175
            # push the user specified dirs from dirblock
 
176
            for dir in reversed(dirblock):
 
177
                if dir[2] == _directory:
 
178
                    pending.append(dir)