1
# Copyright (C) 2005, 2006, 2007 Canonical Ltd
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.
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.
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
30
from bzrlib.decorators import needs_read_lock, needs_write_lock
31
from bzrlib.repository import (
33
MetaDirRepositoryFormat,
37
import bzrlib.revision as _mod_revision
38
from bzrlib.store.versioned import VersionedFileStore
39
from bzrlib.trace import mutter, note, warning
42
class _KnitParentsProvider(object):
44
def __init__(self, knit):
48
return 'KnitParentsProvider(%r)' % self._knit
50
def get_parents(self, revision_ids):
52
for revision_id in revision_ids:
53
if revision_id == _mod_revision.NULL_REVISION:
57
parents = self._knit.get_parents_with_ghosts(revision_id)
58
except errors.RevisionNotPresent:
62
parents = [_mod_revision.NULL_REVISION]
63
parents_list.append(parents)
67
class KnitRepository(MetaDirRepository):
68
"""Knit format repository."""
70
_serializer = xml5.serializer_v5
72
def _warn_if_deprecated(self):
73
# This class isn't deprecated
76
def _inventory_add_lines(self, inv_vf, revid, parents, lines):
77
inv_vf.add_lines_with_ghosts(revid, parents, lines)
80
def _all_revision_ids(self):
81
"""See Repository.all_revision_ids()."""
82
# Knits get the revision graph from the index of the revision knit, so
83
# it's always possible even if they're on an unlistable transport.
84
return self._revision_store.all_revision_ids(self.get_transaction())
86
def fileid_involved_between_revs(self, from_revid, to_revid):
87
"""Find file_id(s) which are involved in the changes between revisions.
89
This determines the set of revisions which are involved, and then
90
finds all file ids affected by those revisions.
92
from_revid = osutils.safe_revision_id(from_revid)
93
to_revid = osutils.safe_revision_id(to_revid)
94
vf = self._get_revision_vf()
95
from_set = set(vf.get_ancestry(from_revid))
96
to_set = set(vf.get_ancestry(to_revid))
97
changed = to_set.difference(from_set)
98
return self._fileid_involved_by_set(changed)
100
def fileid_involved(self, last_revid=None):
101
"""Find all file_ids modified in the ancestry of last_revid.
103
:param last_revid: If None, last_revision() will be used.
106
changed = set(self.all_revision_ids())
108
changed = set(self.get_ancestry(last_revid))
111
return self._fileid_involved_by_set(changed)
114
def get_ancestry(self, revision_id, topo_sorted=True):
115
"""Return a list of revision-ids integrated by a revision.
117
This is topologically sorted, unless 'topo_sorted' is specified as
120
if _mod_revision.is_null(revision_id):
122
revision_id = osutils.safe_revision_id(revision_id)
123
vf = self._get_revision_vf()
125
return [None] + vf.get_ancestry(revision_id, topo_sorted)
126
except errors.RevisionNotPresent:
127
raise errors.NoSuchRevision(self, revision_id)
130
def get_revision(self, revision_id):
131
"""Return the Revision object for a named revision"""
132
revision_id = osutils.safe_revision_id(revision_id)
133
return self.get_revision_reconcile(revision_id)
136
def get_revision_graph(self, revision_id=None):
137
"""Return a dictionary containing the revision graph.
139
:param revision_id: The revision_id to get a graph from. If None, then
140
the entire revision graph is returned. This is a deprecated mode of
141
operation and will be removed in the future.
142
:return: a dictionary of revision_id->revision_parents_list.
144
# special case NULL_REVISION
145
if revision_id == _mod_revision.NULL_REVISION:
147
revision_id = osutils.safe_revision_id(revision_id)
148
a_weave = self._get_revision_vf()
149
entire_graph = a_weave.get_graph()
150
if revision_id is None:
151
return a_weave.get_graph()
152
elif revision_id not in a_weave:
153
raise errors.NoSuchRevision(self, revision_id)
155
# add what can be reached from revision_id
157
pending = set([revision_id])
158
while len(pending) > 0:
160
result[node] = a_weave.get_parents(node)
161
for revision_id in result[node]:
162
if revision_id not in result:
163
pending.add(revision_id)
167
def get_revision_graph_with_ghosts(self, revision_ids=None):
168
"""Return a graph of the revisions with ghosts marked as applicable.
170
:param revision_ids: an iterable of revisions to graph or None for all.
171
:return: a Graph object with the graph reachable from revision_ids.
173
result = deprecated_graph.Graph()
174
vf = self._get_revision_vf()
175
versions = set(vf.versions())
177
pending = set(self.all_revision_ids())
180
pending = set(osutils.safe_revision_id(r) for r in revision_ids)
181
# special case NULL_REVISION
182
if _mod_revision.NULL_REVISION in pending:
183
pending.remove(_mod_revision.NULL_REVISION)
184
required = set(pending)
187
revision_id = pending.pop()
188
if not revision_id in versions:
189
if revision_id in required:
190
raise errors.NoSuchRevision(self, revision_id)
192
result.add_ghost(revision_id)
193
# mark it as done so we don't try for it again.
194
done.add(revision_id)
196
parent_ids = vf.get_parents_with_ghosts(revision_id)
197
for parent_id in parent_ids:
198
# is this queued or done ?
199
if (parent_id not in pending and
200
parent_id not in done):
202
pending.add(parent_id)
203
result.add_node(revision_id, parent_ids)
204
done.add(revision_id)
207
def _get_revision_vf(self):
208
""":return: a versioned file containing the revisions."""
209
vf = self._revision_store.get_revision_file(self.get_transaction())
212
def _get_history_vf(self):
213
"""Get a versionedfile whose history graph reflects all revisions.
215
For knit repositories, this is the revision knit.
217
return self._get_revision_vf()
220
def reconcile(self, other=None, thorough=False):
221
"""Reconcile this repository."""
222
from bzrlib.reconcile import KnitReconciler
223
reconciler = KnitReconciler(self, thorough=thorough)
224
reconciler.reconcile()
227
def revision_parents(self, revision_id):
228
revision_id = osutils.safe_revision_id(revision_id)
229
return self._get_revision_vf().get_parents(revision_id)
231
def _make_parents_provider(self):
232
return _KnitParentsProvider(self._get_revision_vf())
235
class KnitRepository3(KnitRepository):
237
def __init__(self, _format, a_bzrdir, control_files, _revision_store,
238
control_store, text_store):
239
KnitRepository.__init__(self, _format, a_bzrdir, control_files,
240
_revision_store, control_store, text_store)
241
self._serializer = xml7.serializer_v7
243
def deserialise_inventory(self, revision_id, xml):
244
"""Transform the xml into an inventory object.
246
:param revision_id: The expected revision id of the inventory.
247
:param xml: A serialised inventory.
249
result = self._serializer.read_inventory_from_string(xml)
250
assert result.root.revision is not None
253
def serialise_inventory(self, inv):
254
"""Transform the inventory object into XML text.
256
:param revision_id: The expected revision id of the inventory.
257
:param xml: A serialised inventory.
259
assert inv.revision_id is not None
260
assert inv.root.revision is not None
261
return KnitRepository.serialise_inventory(self, inv)
263
def get_commit_builder(self, branch, parents, config, timestamp=None,
264
timezone=None, committer=None, revprops=None,
266
"""Obtain a CommitBuilder for this repository.
268
:param branch: Branch to commit to.
269
:param parents: Revision ids of the parents of the new revision.
270
:param config: Configuration to use.
271
:param timestamp: Optional timestamp recorded for commit.
272
:param timezone: Optional timezone for timestamp.
273
:param committer: Optional committer to set for commit.
274
:param revprops: Optional dictionary of revision properties.
275
:param revision_id: Optional revision id.
277
revision_id = osutils.safe_revision_id(revision_id)
278
result = RootCommitBuilder(self, parents, config, timestamp, timezone,
279
committer, revprops, revision_id)
280
self.start_write_group()
284
class RepositoryFormatKnit(MetaDirRepositoryFormat):
285
"""Bzr repository knit format (generalized).
287
This repository format has:
288
- knits for file texts and inventory
289
- hash subdirectory based stores.
290
- knits for revisions and signatures
291
- TextStores for revisions and signatures.
292
- a format marker of its own
293
- an optional 'shared-storage' flag
294
- an optional 'no-working-trees' flag
298
def _get_control_store(self, repo_transport, control_files):
299
"""Return the control store for this repository."""
300
return VersionedFileStore(
303
file_mode=control_files._file_mode,
304
versionedfile_class=knit.KnitVersionedFile,
305
versionedfile_kwargs={'factory':knit.KnitPlainFactory()},
308
def _get_revision_store(self, repo_transport, control_files):
309
"""See RepositoryFormat._get_revision_store()."""
310
from bzrlib.store.revision.knit import KnitRevisionStore
311
versioned_file_store = VersionedFileStore(
313
file_mode=control_files._file_mode,
316
versionedfile_class=knit.KnitVersionedFile,
317
versionedfile_kwargs={'delta':False,
318
'factory':knit.KnitPlainFactory(),
322
return KnitRevisionStore(versioned_file_store)
324
def _get_text_store(self, transport, control_files):
325
"""See RepositoryFormat._get_text_store()."""
326
return self._get_versioned_file_store('knits',
329
versionedfile_class=knit.KnitVersionedFile,
330
versionedfile_kwargs={
331
'create_parent_dir':True,
333
'dir_mode':control_files._dir_mode,
337
def initialize(self, a_bzrdir, shared=False):
338
"""Create a knit format 1 repository.
340
:param a_bzrdir: bzrdir to contain the new repository; must already
342
:param shared: If true the repository will be initialized as a shared
345
mutter('creating repository in %s.', a_bzrdir.transport.base)
346
dirs = ['revision-store', 'knits']
348
utf8_files = [('format', self.get_format_string())]
350
self._upload_blank_content(a_bzrdir, dirs, files, utf8_files, shared)
351
repo_transport = a_bzrdir.get_repository_transport(None)
352
control_files = lockable_files.LockableFiles(repo_transport,
353
'lock', lockdir.LockDir)
354
control_store = self._get_control_store(repo_transport, control_files)
355
transaction = transactions.WriteTransaction()
356
# trigger a write of the inventory store.
357
control_store.get_weave_or_empty('inventory', transaction)
358
_revision_store = self._get_revision_store(repo_transport, control_files)
359
# the revision id here is irrelevant: it will not be stored, and cannot
361
_revision_store.has_revision_id('A', transaction)
362
_revision_store.get_signature_file(transaction)
363
return self.open(a_bzrdir=a_bzrdir, _found=True)
365
def open(self, a_bzrdir, _found=False, _override_transport=None):
366
"""See RepositoryFormat.open().
368
:param _override_transport: INTERNAL USE ONLY. Allows opening the
369
repository at a slightly different url
370
than normal. I.e. during 'upgrade'.
373
format = RepositoryFormat.find_format(a_bzrdir)
374
assert format.__class__ == self.__class__
375
if _override_transport is not None:
376
repo_transport = _override_transport
378
repo_transport = a_bzrdir.get_repository_transport(None)
379
control_files = lockable_files.LockableFiles(repo_transport,
380
'lock', lockdir.LockDir)
381
text_store = self._get_text_store(repo_transport, control_files)
382
control_store = self._get_control_store(repo_transport, control_files)
383
_revision_store = self._get_revision_store(repo_transport, control_files)
384
return KnitRepository(_format=self,
386
control_files=control_files,
387
_revision_store=_revision_store,
388
control_store=control_store,
389
text_store=text_store)
392
class RepositoryFormatKnit1(RepositoryFormatKnit):
393
"""Bzr repository knit format 1.
395
This repository format has:
396
- knits for file texts and inventory
397
- hash subdirectory based stores.
398
- knits for revisions and signatures
399
- TextStores for revisions and signatures.
400
- a format marker of its own
401
- an optional 'shared-storage' flag
402
- an optional 'no-working-trees' flag
405
This format was introduced in bzr 0.8.
408
def __ne__(self, other):
409
return self.__class__ is not other.__class__
411
def get_format_string(self):
412
"""See RepositoryFormat.get_format_string()."""
413
return "Bazaar-NG Knit Repository Format 1"
415
def get_format_description(self):
416
"""See RepositoryFormat.get_format_description()."""
417
return "Knit repository format 1"
419
def check_conversion_target(self, target_format):
423
class RepositoryFormatKnit3(RepositoryFormatKnit):
424
"""Bzr repository knit format 2.
426
This repository format has:
427
- knits for file texts and inventory
428
- hash subdirectory based stores.
429
- knits for revisions and signatures
430
- TextStores for revisions and signatures.
431
- a format marker of its own
432
- an optional 'shared-storage' flag
433
- an optional 'no-working-trees' flag
435
- support for recording full info about the tree root
436
- support for recording tree-references
439
repository_class = KnitRepository3
440
rich_root_data = True
441
supports_tree_reference = True
443
def _get_matching_bzrdir(self):
444
return bzrdir.format_registry.make_bzrdir('dirstate-with-subtree')
446
def _ignore_setting_bzrdir(self, format):
449
_matchingbzrdir = property(_get_matching_bzrdir, _ignore_setting_bzrdir)
451
def check_conversion_target(self, target_format):
452
if not target_format.rich_root_data:
453
raise errors.BadConversionTarget(
454
'Does not support rich root data.', target_format)
455
if not getattr(target_format, 'supports_tree_reference', False):
456
raise errors.BadConversionTarget(
457
'Does not support nested trees', target_format)
459
def get_format_string(self):
460
"""See RepositoryFormat.get_format_string()."""
461
return "Bazaar Knit Repository Format 3 (bzr 0.15)\n"
463
def get_format_description(self):
464
"""See RepositoryFormat.get_format_description()."""
465
return "Knit repository format 3"
467
def open(self, a_bzrdir, _found=False, _override_transport=None):
468
"""See RepositoryFormat.open().
470
:param _override_transport: INTERNAL USE ONLY. Allows opening the
471
repository at a slightly different url
472
than normal. I.e. during 'upgrade'.
475
format = RepositoryFormat.find_format(a_bzrdir)
476
assert format.__class__ == self.__class__
477
if _override_transport is not None:
478
repo_transport = _override_transport
480
repo_transport = a_bzrdir.get_repository_transport(None)
481
control_files = lockable_files.LockableFiles(repo_transport, 'lock',
483
text_store = self._get_text_store(repo_transport, control_files)
484
control_store = self._get_control_store(repo_transport, control_files)
485
_revision_store = self._get_revision_store(repo_transport, control_files)
486
return self.repository_class(_format=self,
488
control_files=control_files,
489
_revision_store=_revision_store,
490
control_store=control_store,
491
text_store=text_store)
added
removed
bzrlib/repofmt/knitrepo.py
