Adding last_read to RandomContentFile and first_write to FileVerifier, so we can measure latency.

This commit is contained in:
Wesley Spikes 2011-07-15 12:05:02 -07:00
parent 12716d854b
commit 6afc7fdcdb

View file

@ -15,6 +15,7 @@ class RandomContentFile(object):
self.hash = hashlib.md5() self.hash = hashlib.md5()
self.digest_size = self.hash.digest_size self.digest_size = self.hash.digest_size
self.digest = None self.digest = None
self.last_read = 0
def seek(self, offset): def seek(self, offset):
assert offset == 0 assert offset == 0
@ -59,6 +60,7 @@ class RandomContentFile(object):
size -= digest_count size -= digest_count
data = self.digest[:digest_count] data = self.digest[:digest_count]
r.append(data) r.append(data)
self.last_read = time.time()
return ''.join(r) return ''.join(r)
@ -67,8 +69,11 @@ class FileVerifier(object):
self.size = 0 self.size = 0
self.hash = hashlib.md5() self.hash = hashlib.md5()
self.buf = '' self.buf = ''
self.first_write = 0
def write(self, data): def write(self, data):
if self.size == 0:
self.first_write = time.time()
self.size += len(data) self.size += len(data)
self.buf += data self.buf += data
digsz = -1*self.hash.digest_size digsz = -1*self.hash.digest_size