← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/hashcache.py

Make annotate cope better with revisions committed without a valid 
email address.  (Marien Zwart)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/hashcache.py
30
30
CACHE_HEADER = "### bzr hashcache v5\n"
31
31
 
32
32
import os, stat, time
 
33
import sha
33
34
 
34
35
from bzrlib.osutils import sha_file
35
36
from bzrlib.trace import mutter, warning
36
37
from bzrlib.atomicfile import AtomicFile
37
38
 
38
39
 
39
 
 
 
40
FP_MODE_COLUMN = 5
40
41
 
41
42
def _fingerprint(abspath):
42
43
    try:
51
52
    # we discard any high precision because it's not reliable; perhaps we
52
53
    # could do better on some systems?
53
54
    return (fs.st_size, long(fs.st_mtime),
54
 
            long(fs.st_ctime), fs.st_ino, fs.st_dev)
 
55
            long(fs.st_ctime), fs.st_ino, fs.st_dev, fs.st_mode)
55
56
 
56
57
 
57
58
class HashCache(object):
100
101
        self.update_count = 0
101
102
        self._cache = {}
102
103
 
103
 
 
104
104
    def cache_file_name(self):
 
105
        # FIXME: duplicate path logic here, this should be 
 
106
        # something like 'branch.controlfile'.
105
107
        return os.sep.join([self.basedir, '.bzr', 'stat-cache'])
106
108
 
107
 
 
108
 
 
109
 
 
110
109
    def clear(self):
111
110
        """Discard all cached information.
112
111
 
139
138
                del self._cache[path]
140
139
 
141
140
 
142
 
 
143
141
    def get_sha1(self, path):
144
142
        """Return the sha1 of a file.
145
143
        """
165
163
            return cache_sha1
166
164
        
167
165
        self.miss_count += 1
168
 
        digest = sha_file(file(abspath, 'rb', buffering=65000))
 
166
 
 
167
 
 
168
        mode = file_fp[FP_MODE_COLUMN]
 
169
        if stat.S_ISREG(mode):
 
170
            digest = sha_file(file(abspath, 'rb', buffering=65000))
 
171
        elif stat.S_ISLNK(mode):
 
172
            link_target = os.readlink(abspath)
 
173
            digest = sha.new(os.readlink(abspath)).hexdigest()
 
174
        else:
 
175
            raise BzrError("file %r: unknown file stat mode: %o"%(abspath,mode))
169
176
 
170
177
        now = int(time.time())
171
178
        if file_fp[1] >= now or file_fp[2] >= now:
181
188
            self.update_count += 1
182
189
            self.needs_write = True
183
190
            self._cache[path] = (digest, file_fp)
184
 
        
185
191
        return digest
186
192
        
187
 
 
188
 
 
189
 
 
190
193
    def write(self):
191
194
        """Write contents of cache to file."""
192
195
        outf = AtomicFile(self.cache_file_name(), 'wb')
207
210
        finally:
208
211
            if not outf.closed:
209
212
                outf.abort()
210
 
        
211
 
 
212
213
 
213
214
    def read(self):
214
215
        """Reinstate cache from file.
223
224
        try:
224
225
            inf = file(fn, 'rb', buffering=65000)
225
226
        except IOError, e:
226
 
            mutter("failed to open %s: %s" % (fn, e))
 
227
            mutter("failed to open %s: %s", fn, e)
227
228
            # better write it now so it is valid
228
229
            self.needs_write = True
229
230
            return
231
232
 
232
233
        hdr = inf.readline()
233
234
        if hdr != CACHE_HEADER:
234
 
            mutter('cache header marker not found at top of %s; discarding cache'
235
 
                   % fn)
 
235
            mutter('cache header marker not found at top of %s;'
 
236
                   ' discarding cache', fn)
236
237
            self.needs_write = True
237
238
            return
238
239
 
245
246
 
246
247
            pos += 3
247
248
            fields = l[pos:].split(' ')
248
 
            if len(fields) != 6:
 
249
            if len(fields) != 7:
249
250
                warning("bad line in hashcache: %r" % l)
250
251
                continue
251
252