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
17
from bzrlib.lazy_import import lazy_import
18
lazy_import(globals(), """
22
from bzrlib.store import revision
23
from bzrlib.store.revision.knit import KnitRevisionStore
17
25
from bzrlib import (
37
45
import bzrlib.revision as _mod_revision
38
46
from bzrlib.store.versioned import VersionedFileStore
39
from bzrlib.trace import mutter, note, warning
47
from bzrlib.trace import mutter, mutter_callsite
40
48
from bzrlib.util import bencode
134
142
operation and will be removed in the future.
135
143
:return: a dictionary of revision_id->revision_parents_list.
145
if 'evil' in debug.debug_flags:
147
"get_revision_graph scales with size of history.")
137
148
# special case NULL_REVISION
138
149
if revision_id == _mod_revision.NULL_REVISION:
140
151
a_weave = self._get_revision_vf()
141
152
if revision_id is None:
142
153
return a_weave.get_graph()
143
elif revision_id not in a_weave:
154
if revision_id not in a_weave:
144
155
raise errors.NoSuchRevision(self, revision_id)
146
157
# add what can be reached from revision_id
153
164
:param revision_ids: an iterable of revisions to graph or None for all.
154
165
:return: a Graph object with the graph reachable from revision_ids.
167
if 'evil' in debug.debug_flags:
169
"get_revision_graph_with_ghosts scales with size of history.")
156
170
result = deprecated_graph.Graph()
157
171
vf = self._get_revision_vf()
158
172
versions = set(vf.versions())
217
231
class KnitRepository3(KnitRepository):
233
# knit3 repositories need a RootCommitBuilder
219
234
_commit_builder_class = RootCommitBuilder
221
236
def __init__(self, _format, a_bzrdir, control_files, _revision_store,
277
# Set this attribute in derived classes to control the repository class
278
# created by open and initialize.
279
repository_class = None
262
281
def _get_control_store(self, repo_transport, control_files):
263
282
"""Return the control store for this repository."""
264
283
return VersionedFileStore(
272
291
def _get_revision_store(self, repo_transport, control_files):
273
292
"""See RepositoryFormat._get_revision_store()."""
274
from bzrlib.store.revision.knit import KnitRevisionStore
275
293
versioned_file_store = VersionedFileStore(
277
295
file_mode=control_files._file_mode,
309
327
mutter('creating repository in %s.', a_bzrdir.transport.base)
310
dirs = ['revision-store', 'knits']
312
330
utf8_files = [('format', self.get_format_string())]
345
363
text_store = self._get_text_store(repo_transport, control_files)
346
364
control_store = self._get_control_store(repo_transport, control_files)
347
365
_revision_store = self._get_revision_store(repo_transport, control_files)
348
return KnitRepository(_format=self,
366
return self.repository_class(_format=self,
349
367
a_bzrdir=a_bzrdir,
350
368
control_files=control_files,
351
369
_revision_store=_revision_store,
428
448
"""See RepositoryFormat.get_format_description()."""
429
449
return "Knit repository format 3"
431
def open(self, a_bzrdir, _found=False, _override_transport=None):
432
"""See RepositoryFormat.open().
434
:param _override_transport: INTERNAL USE ONLY. Allows opening the
435
repository at a slightly different url
436
than normal. I.e. during 'upgrade'.
439
format = RepositoryFormat.find_format(a_bzrdir)
440
assert format.__class__ == self.__class__
441
if _override_transport is not None:
442
repo_transport = _override_transport
444
repo_transport = a_bzrdir.get_repository_transport(None)
445
control_files = lockable_files.LockableFiles(repo_transport, 'lock',
447
text_store = self._get_text_store(repo_transport, control_files)
448
control_store = self._get_control_store(repo_transport, control_files)
449
_revision_store = self._get_revision_store(repo_transport, control_files)
450
return self.repository_class(_format=self,
452
control_files=control_files,
453
_revision_store=_revision_store,
454
control_store=control_store,
455
text_store=text_store)
458
452
def _get_stream_as_bytes(knit, required_versions):
459
453
"""Generate a serialised data stream.
476
470
for version, options, length, parents in data_list:
477
471
data.append([version, options, parents, callable(length)])
478
472
return bencode.bencode(data)