U  W[@sVdZddlmZmZddlmZmZddlmZm Z ddl m Z Gdddej Z dS) z An in-memory caching resolver. )divisionabsolute_import)dnscommon)failurelog)deferc@sReZdZdZdZdddZddZdd Zd d Zdd d Z dddZ ddZ dS) CacheResolverz A resolver that serves records from a local, memory cache. @ivar _reactor: A provider of L{interfaces.IReactorTime}. NrcCsdtj|i|_||_i|_|dkr2ddlm}||_|r`| D]\}\}}| |||qDdS)Nr)reactor) r ResolverBase__init__cacheverbosecanceltwisted.internetr _reactoritems cacheResult)selfr rr querysecondspayloadr5/usr/lib/python3/dist-packages/twisted/names/cache.pyr s  zCacheResolver.__init__c Csb||_|j}|jD]B\}\}\}}}||}|||D]} | j|kr@|j|=qq@qdSN)__dict__rrr rttl) rstatenowkwhenansaddnsdiffZrecrrr __setstate__(s  zCacheResolver.__setstate__cCs(|jD] }|q |j|jSr)rvaluesclearr)rcrrr __getstate__4s  zCacheResolver.__getstate__c s |j}t|||}z|j|\}\}} } WnFtk rv|jdkr\tdt |t t t|YSX|jrtdt |||z8fdd|Dfdd| Dfdd| Df} Wn,tk rt t t|YSXt | SdS)NzCache miss for zCache hit for c s.g|]&}t|jj|j|j|j|jqSrrZRRHeadernametypeclsrr.0rr$rr Ksz)CacheResolver._lookup..c s.g|]&}t|jj|j|j|j|jqSrr+r/r2rrr3Msc s.g|]&}t|jj|j|j|j|jqSrr+r/r2rrr3Os)rrrZQueryr KeyErrorrrmsgreprrfailrFailure DomainError ValueErrorZsucceed) rr,r.r-timeoutrqr r!Zauthr"resultrr2r_lookup;s2     zCacheResolver._lookupcCsttt|Sr)rr7rr8rr9)rr,r;rrrlookupAllRecordsWszCacheResolver.lookupAllRecordscCs|jdkrtd||p$|j|f|j|<||jkrH|j|t|dt|dt|d}|r|dj}|D]}t ||j}q~nd}|j ||j ||j|<dS)a Cache a DNS entry. @param query: a L{dns.Query} instance. @param payload: a 3-tuple of lists of L{dns.RRHeader} records, the matching result of the query (answers, authority and additional). @param cacheTime: The time (seconds since epoch) at which the entry is considered to have been added to the cache. If L{None} is given, the current time is used. r*zAdding %r to cacherN) rrr5rrr rlistrminZ callLater clearEntry)rrrZ cacheTimesmr1rrrr[s  $ zCacheResolver.cacheResultcCs|j|=|j|=dSr)r r)rrrrrrC{szCacheResolver.clearEntry)NrN)N)N) __name__ __module__ __qualname____doc__r r r%r)r>r?rrCrrrrr s    r N)rIZ __future__rrZ twisted.namesrrZtwisted.pythonrrrrr r rrrrs