~bzr-pqm/bzr/bzr.dev

1563.2.21 by Robert Collins
Smoke test for RevisionStore factories creating revision stores.
1
# Copyright (C) 2006 by 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 version 2 as published by
5
# the Free Software Foundation.
6
#
7
# This program is distributed in the hope that it will be useful,
8
# but WITHOUT ANY WARRANTY; without even the implied warranty of
9
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10
# GNU General Public License for more details.
11
#
12
# You should have received a copy of the GNU General Public License
13
# along with this program; if not, write to the Free Software
14
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
15
16
"""VersionedFile based revision store.
17
1563.2.22 by Robert Collins
Move responsibility for repository.has_revision into RevisionStore
18
This stores revisions as individual entries in a knit, and signatures in a 
19
parallel knit.
1563.2.21 by Robert Collins
Smoke test for RevisionStore factories creating revision stores.
20
"""
21
22
1563.2.23 by Robert Collins
Add add_revision and get_revision methods to RevisionStore
23
import bzrlib
24
import bzrlib.errors as errors
1628.1.4 by Robert Collins
Change knit format to use non-delta, non-annotated revisions and signatures.
25
from bzrlib.knit import KnitVersionedFile, KnitPlainFactory
1563.2.21 by Robert Collins
Smoke test for RevisionStore factories creating revision stores.
26
from bzrlib.store.revision import RevisionStore
1563.2.22 by Robert Collins
Move responsibility for repository.has_revision into RevisionStore
27
from bzrlib.store.versioned import VersionedFileStore
1563.2.21 by Robert Collins
Smoke test for RevisionStore factories creating revision stores.
28
from bzrlib.transport import get_transport
29
30
31
class KnitRevisionStoreFactory(object):
32
    """Factory to create a KnitRevisionStore for testing."""
33
34
    def create(self, url):
35
        """Create a revision store at url."""
36
        t = get_transport(url)
1563.2.31 by Robert Collins
Convert Knit repositories to use knits.
37
        t.mkdir('revision-store')
38
        versioned_file_store = VersionedFileStore(
39
            t.clone('revision-store'),
1563.2.34 by Robert Collins
Remove the commit and rollback transaction methods as misleading, and implement a WriteTransaction
40
            precious=True,
1628.1.4 by Robert Collins
Change knit format to use non-delta, non-annotated revisions and signatures.
41
            versionedfile_class=KnitVersionedFile,
42
            versionedfile_kwargs={'delta':False, 'factory':KnitPlainFactory()})
1563.2.22 by Robert Collins
Move responsibility for repository.has_revision into RevisionStore
43
        return KnitRevisionStore(versioned_file_store)
1563.2.21 by Robert Collins
Smoke test for RevisionStore factories creating revision stores.
44
45
    def __str__(self):
46
        return "KnitRevisionStore"
1563.2.22 by Robert Collins
Move responsibility for repository.has_revision into RevisionStore
47
48
49
class KnitRevisionStore(RevisionStore):
50
    """A RevisionStore layering on a VersionedFileStore."""
51
52
    def __init__(self, versioned_file_store):
53
        """Create a KnitRevisionStore object.
54
55
        :param versioned_file_store: the text store to use for storing 
56
                                     revisions and signatures.
57
        """
1563.2.23 by Robert Collins
Add add_revision and get_revision methods to RevisionStore
58
        super(KnitRevisionStore, self).__init__()
1563.2.22 by Robert Collins
Move responsibility for repository.has_revision into RevisionStore
59
        self.versioned_file_store = versioned_file_store
60
1563.2.23 by Robert Collins
Add add_revision and get_revision methods to RevisionStore
61
    def _add_revision(self, revision, revision_as_file, transaction):
62
        """Template method helper to store revision in this store."""
1563.2.31 by Robert Collins
Convert Knit repositories to use knits.
63
        # FIXME: make this ghost aware at the knit level
64
        rf = self.get_revision_file(transaction)
1594.2.9 by Robert Collins
Teach Knit repositories how to handle ghosts without corrupting at all.
65
        self.get_revision_file(transaction).add_lines_with_ghosts(
1563.2.23 by Robert Collins
Add add_revision and get_revision methods to RevisionStore
66
            revision.revision_id,
1594.2.9 by Robert Collins
Teach Knit repositories how to handle ghosts without corrupting at all.
67
            revision.parent_ids,
1666.1.6 by Robert Collins
Make knit the default format.
68
            bzrlib.osutils.split_lines(revision_as_file.read()))
1563.2.23 by Robert Collins
Add add_revision and get_revision methods to RevisionStore
69
1563.2.30 by Robert Collins
Remove all but fetch references to revision_store, making the repository references that are weave specific use the RevisionTextStore.text_store attribute.
70
    def add_revision_signature_text(self, revision_id, signature_text, transaction):
71
        """See RevisionStore.add_revision_signature_text()."""
1563.2.35 by Robert Collins
cleanup deprecation warnings and finish conversion so the inventory is knit based too.
72
        self.get_signature_file(transaction).add_lines(
1563.2.29 by Robert Collins
Remove all but fetch references to repository.revision_store.
73
            revision_id, [], bzrlib.osutils.split_lines(signature_text))
74
75
    def all_revision_ids(self, transaction):
76
        """See RevisionStore.all_revision_ids()."""
77
        rev_file = self.get_revision_file(transaction)
78
        return rev_file.get_ancestry(rev_file.versions())
1563.2.27 by Robert Collins
Add signature text addition to the revisionstore api.
79
1756.1.2 by Aaron Bentley
Show logs using get_revisions
80
    def get_revisions(self, revision_ids, transaction):
81
        """See RevisionStore.get_revisions()."""
82
        texts = self._get_serialized_revisions(revision_ids, transaction)
83
        revisions = []
1563.2.23 by Robert Collins
Add add_revision and get_revision methods to RevisionStore
84
        try:
1756.1.2 by Aaron Bentley
Show logs using get_revisions
85
            for text, revision_id in zip(texts, revision_ids):
86
                r = self._serializer.read_revision_from_string(text)
87
                assert r.revision_id == revision_id
88
                revisions.append(r)
1563.2.23 by Robert Collins
Add add_revision and get_revision methods to RevisionStore
89
        except SyntaxError, e:
90
            raise errors.BzrError('failed to unpack revision_xml',
91
                                   [revision_id,
92
                                   str(e)])
1756.1.2 by Aaron Bentley
Show logs using get_revisions
93
        return revisions 
94
95
    def _get_serialized_revisions(self, revision_ids, transaction):
96
        texts = []
97
        vf = self.get_revision_file(transaction)
98
        try:
1756.2.1 by Aaron Bentley
Implement get_texts
99
            return vf.get_texts(revision_ids)
100
        except (errors.RevisionNotPresent), e:
101
            raise errors.NoSuchRevision(self, e.revision_id)
1563.2.23 by Robert Collins
Add add_revision and get_revision methods to RevisionStore
102
103
    def _get_revision_xml(self, revision_id, transaction):
104
        try:
105
            return self.get_revision_file(transaction).get_text(revision_id)
106
        except (errors.RevisionNotPresent):
107
            raise errors.NoSuchRevision(self, revision_id)
108
1563.2.22 by Robert Collins
Move responsibility for repository.has_revision into RevisionStore
109
    def get_revision_file(self, transaction):
110
        """Get the revision versioned file object."""
1563.2.25 by Robert Collins
Merge in upstream.
111
        return self.versioned_file_store.get_weave_or_empty('revisions', transaction)
1563.2.22 by Robert Collins
Move responsibility for repository.has_revision into RevisionStore
112
1563.2.35 by Robert Collins
cleanup deprecation warnings and finish conversion so the inventory is knit based too.
113
    def get_signature_file(self, transaction):
1563.2.29 by Robert Collins
Remove all but fetch references to repository.revision_store.
114
        """Get the signature text versioned file object."""
115
        return self.versioned_file_store.get_weave_or_empty('signatures', transaction)
116
1563.2.30 by Robert Collins
Remove all but fetch references to revision_store, making the repository references that are weave specific use the RevisionTextStore.text_store attribute.
117
    def _get_signature_text(self, revision_id, transaction):
118
        """See RevisionStore._get_signature_text()."""
119
        try:
1563.2.35 by Robert Collins
cleanup deprecation warnings and finish conversion so the inventory is knit based too.
120
            return self.get_signature_file(transaction).get_text(revision_id)
1563.2.30 by Robert Collins
Remove all but fetch references to revision_store, making the repository references that are weave specific use the RevisionTextStore.text_store attribute.
121
        except errors.RevisionNotPresent:
122
            raise errors.NoSuchRevision(self, revision_id)
123
1563.2.22 by Robert Collins
Move responsibility for repository.has_revision into RevisionStore
124
    def has_revision_id(self, revision_id, transaction):
125
        """True if the store contains revision_id."""
126
        return (revision_id is None
127
                or self.get_revision_file(transaction).has_version(revision_id))
1563.2.28 by Robert Collins
Add total_size to the revision_store api.
128
        
1563.2.29 by Robert Collins
Remove all but fetch references to repository.revision_store.
129
    def _has_signature(self, revision_id, transaction):
130
        """See RevisionStore._has_signature()."""
1563.2.35 by Robert Collins
cleanup deprecation warnings and finish conversion so the inventory is knit based too.
131
        return self.get_signature_file(transaction).has_version(revision_id)
1563.2.29 by Robert Collins
Remove all but fetch references to repository.revision_store.
132
1563.2.28 by Robert Collins
Add total_size to the revision_store api.
133
    def total_size(self, transaction):
134
        """ See RevisionStore.total_size()."""
1773.4.2 by Martin Pool
Cleanup of imports; undeprecate all_revision_ids()
135
        return (len(self.all_revision_ids(transaction)),
1666.1.6 by Robert Collins
Make knit the default format.
136
            self.versioned_file_store.total_size()[1])