Adds custom ttl argument to HostCache::Set.

net/base/host_cache.cc

 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "net/base/host_cache.h"
 
 #include "base/logging.h"
 #include "net/base/net_errors.h"
 
 namespace net {
@@ skipping 35 matching lines @@
   Entry* entry = it->second.get();
   if (CanUseEntry(entry, now))
     return entry;
 
   return NULL;
 }
 
 HostCache::Entry* HostCache::Set(const Key& key,
                                  int error,
                                  const AddressList& addrlist,
+                                 base::TimeDelta ttl,
                                  base::TimeTicks now) {
   DCHECK(CalledOnValidThread());
   if (caching_is_disabled())
     return NULL;
 
-  base::TimeTicks expiration = now +
-      (error == OK ? success_entry_ttl_ : failure_entry_ttl_);
+  if (ttl == base::TimeDelta())
+    ttl = (error == OK ? success_entry_ttl_ : failure_entry_ttl_);
+
+  base::TimeTicks expiration = now + ttl;
 
   scoped_refptr<Entry>& entry = entries_[key];
   if (!entry) {
     // Entry didn't exist, creating one now.
     Entry* ptr = new Entry(error, addrlist, expiration);
     entry = ptr;
 
     // Compact the cache if we grew it beyond limit -- exclude |entry| from
     // being pruned though!
     if (entries_.size() > max_entries_)
       Compact(now, ptr);
     return ptr;
   } else {
     // Update an existing cache entry.
     entry->error = error;
     entry->addrlist = addrlist;
     entry->expiration = expiration;
     return entry.get();
   }
 }
 
+HostCache::Entry* HostCache::Set(const Key& key,
+                                 int error,
+                                 const AddressList& addrlist,
+                                 base::TimeTicks now) {
+  return Set(key, error, addrlist, base::TimeDelta(), now);
+}
+
 void HostCache::clear() {
   DCHECK(CalledOnValidThread());
   entries_.clear();
 }
 
 size_t HostCache::size() const {
   DCHECK(CalledOnValidThread());
   return entries_.size();
 }
 
@@ skipping 61 matching lines @@
     } else {
       ++it;
     }
   }
 
   if (entries_.size() > max_entries_)
     DLOG(WARNING) << "Still above max entries limit";
 }
 
 }  // namespace net