diff options
Diffstat (limited to 'project2/proj2_s4498062/dns/cache.py')
-rw-r--r-- | project2/proj2_s4498062/dns/cache.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/project2/proj2_s4498062/dns/cache.py b/project2/proj2_s4498062/dns/cache.py index 9cde66f..a8b62be 100644 --- a/project2/proj2_s4498062/dns/cache.py +++ b/project2/proj2_s4498062/dns/cache.py @@ -53,7 +53,7 @@ class RecordCache(object): FILE = '.dns.cache' - def __init__(self): + def __init__(self, ttl): """ Initialize the RecordCache Args: @@ -61,15 +61,16 @@ class RecordCache(object): """ self.records = [] self.read_cache_file() + self.ttl = ttl def __del__(self): self.write_cache_file() def remove_old(self): """Remove entries for which the TTL has expired""" - now = int(time.clock()) + now = int(time.time()) for record in reversed(self.records): - if record.ttl + record.timestamp < now: + if min(self.ttl, record.ttl) + record.timestamp < now: self.records.remove(record) def lookup(self, dname, type_, class_): |