← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/store/weave.py

  • Committer: Robert Collins
  • Date: 2005-10-07 06:13:14 UTC
  • mto: This revision was merged to the branch mainline in revision 1420.
  • Revision ID: robertc@robertcollins.net-20051007061313-9a835e979434e721
use transactions in the weave store interface, which enables caching for log

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/store/weave.py
41
41
 
42
42
    def __init__(self, transport):
43
43
        self._transport = transport
44
 
        self._cache = {}
45
 
        self.enable_cache = False
46
 
 
47
44
 
48
45
    def filename(self, file_id):
49
46
        """Return the path relative to the transport root."""
66
63
    def _put(self, file_id, f):
67
64
        return self._transport.put(self.filename(file_id), f)
68
65
 
69
 
    def get_weave(self, file_id):
70
 
        if self.enable_cache:
71
 
            if file_id in self._cache:
72
 
                mutter("cache hit in %s for %s", self, file_id)
73
 
                return self._cache[file_id]
 
66
    def get_weave(self, file_id, transaction):
 
67
        weave = transaction.map.find_weave(file_id)
 
68
        if weave:
 
69
            mutter("cache hit in %s for %s", self, file_id)
 
70
            return weave
74
71
        w = read_weave(self._get(file_id))
75
 
        if self.enable_cache:
76
 
            self._cache[file_id] = w
 
72
        transaction.map.add_weave(file_id, w)
 
73
        transaction.register_clean(w)
77
74
        return w
78
75
 
79
76
 
80
 
    def get_lines(self, file_id, rev_id):
 
77
    def get_lines(self, file_id, rev_id, transaction):
81
78
        """Return text from a particular version of a weave.
82
79
 
83
80
        Returned as a list of lines."""
84
 
        w = self.get_weave(file_id)
 
81
        w = self.get_weave(file_id, transaction)
85
82
        return w.get(w.lookup(rev_id))
86
83
    
87
 
 
88
 
    def get_weave_or_empty(self, file_id):
 
84
    def get_weave_or_empty(self, file_id, transaction):
89
85
        """Return a weave, or an empty one if it doesn't exist.""" 
90
86
        try:
91
 
            inf = self._get(file_id)
 
87
            return self.get_weave(file_id, transaction)
92
88
        except NoSuchFile:
93
 
            return Weave(weave_name=file_id)
94
 
        else:
95
 
            return read_weave(inf)
96
 
    
 
89
            weave = Weave(weave_name=file_id)
 
90
            transaction.map.add_weave(file_id, weave)
 
91
            transaction.register_clean(weave)
 
92
            return weave
97
93
 
98
 
    def put_weave(self, file_id, weave):
 
94
    def put_weave(self, file_id, weave, transaction):
99
95
        """Write back a modified weave"""
100
 
        if self.enable_cache:
101
 
            self._cache[file_id] = weave
102
 
 
 
96
        transaction.register_dirty(weave)
 
97
        # TODO FOR WRITE TRANSACTIONS: this should be done in a callback
 
98
        # from the transaction, when it decides to save.
103
99
        sio = StringIO()
104
100
        write_weave_v5(weave, sio)
105
101
        sio.seek(0)
107
103
        self._put(file_id, sio)
108
104
 
109
105
 
110
 
    def add_text(self, file_id, rev_id, new_lines, parents):
111
 
        w = self.get_weave_or_empty(file_id)
 
106
    def add_text(self, file_id, rev_id, new_lines, parents, transaction):
 
107
        w = self.get_weave_or_empty(file_id, transaction)
112
108
        parent_idxs = map(w.lookup, parents)
113
109
        w.add(rev_id, parent_idxs, new_lines)
114
 
        self.put_weave(file_id, w)
 
110
        self.put_weave(file_id, w, transaction)
115
111
        
116
 
    def add_identical_text(self, file_id, old_rev_id, new_rev_id, parents):
117
 
        w = self.get_weave_or_empty(file_id)
 
112
    def add_identical_text(self, file_id, old_rev_id, new_rev_id, parents,
 
113
                           transaction):
 
114
        w = self.get_weave_or_empty(file_id, transaction)
118
115
        parent_idxs = map(w.lookup, parents)
119
116
        w.add_identical(old_rev_id, new_rev_id, parent_idxs)
120
 
        self.put_weave(file_id, w)
 
117
        self.put_weave(file_id, w, transaction)
121
118
     
122
119
    def copy_multi(self, from_store, file_ids):
123
120
        assert isinstance(from_store, WeaveStore)