~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/repofmt/weaverepo.py

Rather use assertRaises in test_changes_from_with_require_versioned

Show diffs side-by-side

added added

removed removed

Lines of Context:
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
 
"""Deprecated weave-based repository formats.
18
17
 
19
 
Weave based formats scaled linearly with history size and could not represent
20
 
ghosts.
21
 
"""
 
18
"""Old weave-based repository formats"""
22
19
 
23
20
from StringIO import StringIO
24
21
 
25
22
from bzrlib import (
26
23
    bzrdir,
27
 
    debug,
28
 
    errors,
29
24
    lockable_files,
30
25
    lockdir,
31
 
    osutils,
32
 
    revision as _mod_revision,
33
26
    weave,
34
27
    weavefile,
35
28
    xml5,
36
29
    )
37
30
from bzrlib.decorators import needs_read_lock, needs_write_lock
38
31
from bzrlib.repository import (
39
 
    CommitBuilder,
40
32
    MetaDirRepository,
41
33
    MetaDirRepositoryFormat,
42
34
    Repository,
66
58
                              prefixed=prefixed, compressed=compressed,
67
59
                              dir_mode=dir_mode,
68
60
                              file_mode=file_mode)
 
61
            #if self._transport.should_cache():
 
62
            #    cache_path = os.path.join(self.cache_root, name)
 
63
            #    os.mkdir(cache_path)
 
64
            #    store = bzrlib.store.CachedStore(store, cache_path)
69
65
            return store
70
66
 
71
67
        # not broken out yet because the controlweaves|inventory_store
77
73
            text_store = get_store('text-store')
78
74
        super(AllInOneRepository, self).__init__(_format, a_bzrdir, a_bzrdir._control_files, _revision_store, control_store, text_store)
79
75
 
80
 
    @needs_read_lock
81
 
    def _all_possible_ids(self):
82
 
        """Return all the possible revisions that we could find."""
83
 
        if 'evil' in debug.debug_flags:
84
 
            mutter_callsite(3, "_all_possible_ids scales with size of history.")
85
 
        return self.get_inventory_weave().versions()
86
 
 
87
 
    @needs_read_lock
88
 
    def _all_revision_ids(self):
89
 
        """Returns a list of all the revision ids in the repository. 
90
 
 
91
 
        These are in as much topological order as the underlying store can 
92
 
        present: for weaves ghosts may lead to a lack of correctness until
93
 
        the reweave updates the parents list.
94
 
        """
95
 
        if self._revision_store.text_store.listable():
96
 
            return self._revision_store.all_revision_ids(self.get_transaction())
97
 
        result = self._all_possible_ids()
98
 
        # TODO: jam 20070210 Ensure that _all_possible_ids returns non-unicode
99
 
        #       ids. (It should, since _revision_store's API should change to
100
 
        #       return utf8 revision_ids)
101
 
        return self._eliminate_revisions_not_present(result)
102
 
 
103
 
    def _check_revision_parents(self, revision, inventory):
104
 
        """Private to Repository and Fetch.
105
 
        
106
 
        This checks the parentage of revision in an inventory weave for 
107
 
        consistency and is only applicable to inventory-weave-for-ancestry
108
 
        using repository formats & fetchers.
109
 
        """
110
 
        weave_parents = inventory.get_parents(revision.revision_id)
111
 
        weave_names = inventory.versions()
112
 
        for parent_id in revision.parent_ids:
113
 
            if parent_id in weave_names:
114
 
                # this parent must not be a ghost.
115
 
                if not parent_id in weave_parents:
116
 
                    # but it is a ghost
117
 
                    raise errors.CorruptRepository(self)
118
 
 
119
76
    def get_commit_builder(self, branch, parents, config, timestamp=None,
120
77
                           timezone=None, committer=None, revprops=None,
121
78
                           revision_id=None):
122
79
        self._check_ascii_revisionid(revision_id, self.get_commit_builder)
123
 
        result = WeaveCommitBuilder(self, parents, config, timestamp, timezone,
124
 
                              committer, revprops, revision_id)
125
 
        self.start_write_group()
126
 
        return result
127
 
 
128
 
    @needs_read_lock
129
 
    def get_revisions(self, revision_ids):
130
 
        revs = self._get_revisions(revision_ids)
131
 
        # weave corruption can lead to absent revision markers that should be
132
 
        # present.
133
 
        # the following test is reasonably cheap (it needs a single weave read)
134
 
        # and the weave is cached in read transactions. In write transactions
135
 
        # it is not cached but typically we only read a small number of
136
 
        # revisions. For knits when they are introduced we will probably want
137
 
        # to ensure that caching write transactions are in use.
138
 
        inv = self.get_inventory_weave()
139
 
        for rev in revs:
140
 
            self._check_revision_parents(rev, inv)
141
 
        return revs
142
 
 
143
 
    @needs_read_lock
144
 
    def get_revision_graph(self, revision_id=None):
145
 
        """Return a dictionary containing the revision graph.
146
 
        
147
 
        :param revision_id: The revision_id to get a graph from. If None, then
148
 
        the entire revision graph is returned. This is a deprecated mode of
149
 
        operation and will be removed in the future.
150
 
        :return: a dictionary of revision_id->revision_parents_list.
151
 
        """
152
 
        if 'evil' in debug.debug_flags:
153
 
            mutter_callsite(2,
154
 
                "get_revision_graph scales with size of history.")
155
 
        # special case NULL_REVISION
156
 
        if revision_id == _mod_revision.NULL_REVISION:
157
 
            return {}
158
 
        a_weave = self.get_inventory_weave()
159
 
        all_revisions = self._eliminate_revisions_not_present(
160
 
                                a_weave.versions())
161
 
        entire_graph = dict([(node, tuple(a_weave.get_parents(node))) for 
162
 
                             node in all_revisions])
163
 
        if revision_id is None:
164
 
            return entire_graph
165
 
        elif revision_id not in entire_graph:
166
 
            raise errors.NoSuchRevision(self, revision_id)
167
 
        else:
168
 
            # add what can be reached from revision_id
169
 
            result = {}
170
 
            pending = set([revision_id])
171
 
            while len(pending) > 0:
172
 
                node = pending.pop()
173
 
                result[node] = entire_graph[node]
174
 
                for revision_id in result[node]:
175
 
                    if revision_id not in result:
176
 
                        pending.add(revision_id)
177
 
            return result
178
 
 
179
 
    def has_revisions(self, revision_ids):
180
 
        """See Repository.has_revisions()."""
181
 
        result = set()
182
 
        transaction = self.get_transaction()
183
 
        for revision_id in revision_ids:
184
 
            if self._revision_store.has_revision_id(revision_id, transaction):
185
 
                result.add(revision_id)
186
 
        return result
 
80
        return Repository.get_commit_builder(self, branch, parents, config,
 
81
            timestamp, timezone, committer, revprops, revision_id)
187
82
 
188
83
    @needs_read_lock
189
84
    def is_shared(self):
206
101
        """Returns the policy for making working trees on new branches."""
207
102
        return True
208
103
 
209
 
    def revision_graph_can_have_wrong_parents(self):
210
 
        # XXX: This is an old format that we don't support full checking on, so
211
 
        # just claim that checking for this inconsistency is not required.
212
 
        return False
213
 
 
214
104
 
215
105
class WeaveMetaDirRepository(MetaDirRepository):
216
106
    """A subclass of MetaDirRepository to set weave specific policy."""
217
107
 
218
108
    _serializer = xml5.serializer_v5
219
109
 
220
 
    @needs_read_lock
221
 
    def _all_possible_ids(self):
222
 
        """Return all the possible revisions that we could find."""
223
 
        if 'evil' in debug.debug_flags:
224
 
            mutter_callsite(3, "_all_possible_ids scales with size of history.")
225
 
        return self.get_inventory_weave().versions()
226
 
 
227
 
    @needs_read_lock
228
 
    def _all_revision_ids(self):
229
 
        """Returns a list of all the revision ids in the repository. 
230
 
 
231
 
        These are in as much topological order as the underlying store can 
232
 
        present: for weaves ghosts may lead to a lack of correctness until
233
 
        the reweave updates the parents list.
234
 
        """
235
 
        if self._revision_store.text_store.listable():
236
 
            return self._revision_store.all_revision_ids(self.get_transaction())
237
 
        result = self._all_possible_ids()
238
 
        # TODO: jam 20070210 Ensure that _all_possible_ids returns non-unicode
239
 
        #       ids. (It should, since _revision_store's API should change to
240
 
        #       return utf8 revision_ids)
241
 
        return self._eliminate_revisions_not_present(result)
242
 
 
243
 
    def _check_revision_parents(self, revision, inventory):
244
 
        """Private to Repository and Fetch.
245
 
        
246
 
        This checks the parentage of revision in an inventory weave for 
247
 
        consistency and is only applicable to inventory-weave-for-ancestry
248
 
        using repository formats & fetchers.
249
 
        """
250
 
        weave_parents = inventory.get_parents(revision.revision_id)
251
 
        weave_names = inventory.versions()
252
 
        for parent_id in revision.parent_ids:
253
 
            if parent_id in weave_names:
254
 
                # this parent must not be a ghost.
255
 
                if not parent_id in weave_parents:
256
 
                    # but it is a ghost
257
 
                    raise errors.CorruptRepository(self)
258
 
 
259
110
    def get_commit_builder(self, branch, parents, config, timestamp=None,
260
111
                           timezone=None, committer=None, revprops=None,
261
112
                           revision_id=None):
262
113
        self._check_ascii_revisionid(revision_id, self.get_commit_builder)
263
 
        result = WeaveCommitBuilder(self, parents, config, timestamp, timezone,
264
 
                              committer, revprops, revision_id)
265
 
        self.start_write_group()
266
 
        return result
267
 
 
268
 
    @needs_read_lock
269
 
    def get_revision(self, revision_id):
270
 
        """Return the Revision object for a named revision"""
271
 
        # TODO: jam 20070210 get_revision_reconcile should do this for us
272
 
        r = self.get_revision_reconcile(revision_id)
273
 
        # weave corruption can lead to absent revision markers that should be
274
 
        # present.
275
 
        # the following test is reasonably cheap (it needs a single weave read)
276
 
        # and the weave is cached in read transactions. In write transactions
277
 
        # it is not cached but typically we only read a small number of
278
 
        # revisions. For knits when they are introduced we will probably want
279
 
        # to ensure that caching write transactions are in use.
280
 
        inv = self.get_inventory_weave()
281
 
        self._check_revision_parents(r, inv)
282
 
        return r
283
 
 
284
 
    @needs_read_lock
285
 
    def get_revision_graph(self, revision_id=None):
286
 
        """Return a dictionary containing the revision graph.
287
 
        
288
 
        :param revision_id: The revision_id to get a graph from. If None, then
289
 
        the entire revision graph is returned. This is a deprecated mode of
290
 
        operation and will be removed in the future.
291
 
        :return: a dictionary of revision_id->revision_parents_list.
292
 
        """
293
 
        if 'evil' in debug.debug_flags:
294
 
            mutter_callsite(3,
295
 
                "get_revision_graph scales with size of history.")
296
 
        # special case NULL_REVISION
297
 
        if revision_id == _mod_revision.NULL_REVISION:
298
 
            return {}
299
 
        a_weave = self.get_inventory_weave()
300
 
        all_revisions = self._eliminate_revisions_not_present(
301
 
                                a_weave.versions())
302
 
        entire_graph = dict([(node, tuple(a_weave.get_parents(node))) for 
303
 
                             node in all_revisions])
304
 
        if revision_id is None:
305
 
            return entire_graph
306
 
        elif revision_id not in entire_graph:
307
 
            raise errors.NoSuchRevision(self, revision_id)
308
 
        else:
309
 
            # add what can be reached from revision_id
310
 
            result = {}
311
 
            pending = set([revision_id])
312
 
            while len(pending) > 0:
313
 
                node = pending.pop()
314
 
                result[node] = entire_graph[node]
315
 
                for revision_id in result[node]:
316
 
                    if revision_id not in result:
317
 
                        pending.add(revision_id)
318
 
            return result
319
 
 
320
 
    def has_revisions(self, revision_ids):
321
 
        """See Repository.has_revisions()."""
322
 
        result = set()
323
 
        transaction = self.get_transaction()
324
 
        for revision_id in revision_ids:
325
 
            if self._revision_store.has_revision_id(revision_id, transaction):
326
 
                result.add(revision_id)
327
 
        return result
328
 
 
329
 
    def revision_graph_can_have_wrong_parents(self):
330
 
        # XXX: This is an old format that we don't support full checking on, so
331
 
        # just claim that checking for this inconsistency is not required.
332
 
        return False
 
114
        return MetaDirRepository.get_commit_builder(self, branch, parents,
 
115
            config, timestamp, timezone, committer, revprops, revision_id)
333
116
 
334
117
 
335
118
class PreSplitOutRepositoryFormat(RepositoryFormat):
337
120
 
338
121
    rich_root_data = False
339
122
    supports_tree_reference = False
340
 
    supports_ghosts = False
341
123
 
342
124
    def initialize(self, a_bzrdir, shared=False, _internal=False):
343
 
        """Create a weave repository."""
 
125
        """Create a weave repository.
 
126
        
 
127
        TODO: when creating split out bzr branch formats, move this to a common
 
128
        base for Format5, Format6. or something like that.
 
129
        """
344
130
        if shared:
345
131
            raise errors.IncompatibleFormat(self, a_bzrdir._format)
346
132
 
521
307
        """See RepositoryFormat._get_text_store()."""
522
308
        return self._get_versioned_file_store('weaves', transport, control_files)
523
309
 
 
310
 
524
311
class RepositoryFormat7(MetaDirRepositoryFormat):
525
312
    """Bzr repository 7.
526
313
 
534
321
    """
535
322
 
536
323
    _versionedfile_class = weave.WeaveFile
537
 
    supports_ghosts = False
538
324
 
539
325
    def _get_control_store(self, repo_transport, control_files):
540
326
        """Return the control store for this repository."""
616
402
            text_store=text_store)
617
403
 
618
404
 
619
 
class WeaveCommitBuilder(CommitBuilder):
620
 
    """A builder for weave based repos that don't support ghosts."""
621
 
 
622
 
    def _add_text_to_weave(self, file_id, new_lines, parents, nostore_sha):
623
 
        versionedfile = self.repository.weave_store.get_weave_or_empty(
624
 
            file_id, self.repository.get_transaction())
625
 
        result = versionedfile.add_lines(
626
 
            self._new_revision_id, parents, new_lines,
627
 
            nostore_sha=nostore_sha)[0:2]
628
 
        versionedfile.clear_cache()
629
 
        return result
630
 
 
631
 
 
632
405
_legacy_formats = [RepositoryFormat4(),
633
406
                   RepositoryFormat5(),
634
407
                   RepositoryFormat6()]