HttpServerProperties - Implement MRU for AlternateProtocolMap. Persist

net/http/http_server_properties_impl.cc

Index: net/http/http_server_properties_impl.cc
diff --git a/net/http/http_server_properties_impl.cc b/net/http/http_server_properties_impl.cc
index de94dfed7e38e082f1a3fcc41e03c32e70ec3311..20caf91c7a5a21e0bca0fbd23ed6377bfc7ed828 100644
--- a/net/http/http_server_properties_impl.cc
+++ b/net/http/http_server_properties_impl.cc
@@ -19,7 +19,8 @@ namespace net {
 static const int kDefaultNumHostsToRemember = 200;
 
 HttpServerPropertiesImpl::HttpServerPropertiesImpl()
-    : pipeline_capability_map_(
+    : alternate_protocol_map_(AlternateProtocolMap::NO_AUTO_EVICT),

[Ryan Hamilton, 2014/03/05 17:51:22]

With NO_AUTO_EVICT, this means the in-memory map will be unbounded, right?  (I
think you mentioned that in your mail).  This seems fine for this CL, but just
confirming.

nit: Should we add a test of the MRU behavior of Has|GetAlternateProtocol?

[ramant (doing other things), 2014/03/05 22:43:22]

I went back and forth on the above. The current implementation has no bounds.
Thus made the MRUCache to be unbounded.

Tested the current code with a limit 10 and verified in net-internals that
MRUCache is evicting the right server.

A small change to set the limit: use a different const from
AlternateProtocolMap::NO_AUTO_EVICT. 
Thanks much for the above suggestion. Added the unit tests.

[Ryan Hamilton, 2014/03/05 23:10:20]

SGTM.  We can revisit this at any point.

+      pipeline_capability_map_(
         new CachedPipelineCapabilityMap(kDefaultNumHostsToRemember)),
       weak_ptr_factory_(this) {
   canoncial_suffixes_.push_back(".c.youtube.com");
@@ -44,15 +45,26 @@ void HttpServerPropertiesImpl::InitializeSpdyServers(
 
 void HttpServerPropertiesImpl::InitializeAlternateProtocolServers(
     AlternateProtocolMap* alternate_protocol_map) {
-  // First swap, and then add back all the ALTERNATE_PROTOCOL_BROKEN ones since
-  // those don't get persisted.
-  alternate_protocol_map_.swap(*alternate_protocol_map);
+  // Keep all the ALTERNATE_PROTOCOL_BROKEN ones since those don't
+  // get persisted.
+  for (AlternateProtocolMap::iterator it = alternate_protocol_map_.begin();
+       it != alternate_protocol_map_.end();) {
+    if (it->second.protocol != ALTERNATE_PROTOCOL_BROKEN) {
+      AlternateProtocolMap::iterator old_it = it;
+      ++it;
+      alternate_protocol_map_.Erase(old_it);
+    } else {
+      ++it;
+    }

[Ryan Hamilton, 2014/03/05 17:51:22]

alternatively:

AlternateProtocolMap::iterator old_it = it;
++it;
if (it->second.protocol != ALTERNATE_PROTOCOL_BROKEN) {
  alternate_protocol_map_.Erase(old_it);
}

It's shorter, but not sure if that's clearer.  What do you think?

[ramant (doing other things), 2014/03/05 22:43:22]

Done.

+  }

[Ryan Hamilton, 2014/03/05 17:51:22]

I take it there is no swap() method in MRUCache?

[ramant (doing other things), 2014/03/05 22:43:22]

Correct.

+
+  // Add the entries from persisted data.
   for (AlternateProtocolMap::const_iterator it =
-       alternate_protocol_map->begin();
+           alternate_protocol_map->begin();
        it != alternate_protocol_map->end(); ++it) {
-    if (it->second.protocol == ALTERNATE_PROTOCOL_BROKEN)
-      alternate_protocol_map_[it->first] = it->second;
+    alternate_protocol_map_.Put(it->first, it->second);
   }
+
   // Attempt to find canonical servers.
   int canonical_ports[] = { 80, 443 };
   for (size_t i = 0; i < canoncial_suffixes_.size(); ++i) {
@@ -61,8 +73,8 @@ void HttpServerPropertiesImpl::InitializeAlternateProtocolServers(
       HostPortPair canonical_host(canonical_suffix, canonical_ports[j]);
       // If we already have a valid canonical server, we're done.
       if (ContainsKey(canonical_host_to_origin_map_, canonical_host) &&
-          ContainsKey(alternate_protocol_map_,
-                      canonical_host_to_origin_map_[canonical_host])) {
+          (alternate_protocol_map_.Get(canonical_host_to_origin_map_[
+               canonical_host]) != alternate_protocol_map_.end())) {
         continue;
       }
       // Now attempt to find a server which matches this origin and set it as
@@ -147,7 +159,7 @@ base::WeakPtr<HttpServerProperties> HttpServerPropertiesImpl::GetWeakPtr() {
 void HttpServerPropertiesImpl::Clear() {
   DCHECK(CalledOnValidThread());
   spdy_servers_table_.clear();
-  alternate_protocol_map_.clear();
+  alternate_protocol_map_.Clear();
   spdy_settings_map_.clear();
   pipeline_capability_map_->Clear();
 }
@@ -185,8 +197,8 @@ void HttpServerPropertiesImpl::SetSupportsSpdy(
 }
 
 bool HttpServerPropertiesImpl::HasAlternateProtocol(
-    const HostPortPair& server) const {
-  if (ContainsKey(alternate_protocol_map_, server) ||
+    const HostPortPair& server) {
+  if (alternate_protocol_map_.Get(server) != alternate_protocol_map_.end() ||
       g_forced_alternate_protocol)
     return true;
 
@@ -195,19 +207,18 @@ bool HttpServerPropertiesImpl::HasAlternateProtocol(
 
 PortAlternateProtocolPair
 HttpServerPropertiesImpl::GetAlternateProtocol(
-    const HostPortPair& server) const {
+    const HostPortPair& server) {
   DCHECK(HasAlternateProtocol(server));
 
   // First check the map.
-  AlternateProtocolMap::const_iterator it =
-      alternate_protocol_map_.find(server);
+  AlternateProtocolMap::iterator it = alternate_protocol_map_.Get(server);
   if (it != alternate_protocol_map_.end())
     return it->second;
 
   // Next check the canonical host.
   CanonicalHostMap::const_iterator canonical_host = GetCanonicalHost(server);
   if (canonical_host != canonical_host_to_origin_map_.end())
-    return alternate_protocol_map_.find(canonical_host->second)->second;
+    return alternate_protocol_map_.Get(canonical_host->second)->second;
 
   // We must be forcing an alternate.
   DCHECK(g_forced_alternate_protocol);
@@ -247,7 +258,7 @@ void HttpServerPropertiesImpl::SetAlternateProtocol(
     }
   }
 
-  alternate_protocol_map_[server] = alternate;
+  alternate_protocol_map_.Put(server, alternate);
 
   // If this host ends with a canonical suffix, then set it as the
   // canonical host.
@@ -263,7 +274,14 @@ void HttpServerPropertiesImpl::SetAlternateProtocol(
 
 void HttpServerPropertiesImpl::SetBrokenAlternateProtocol(
     const HostPortPair& server) {
-  alternate_protocol_map_[server].protocol = ALTERNATE_PROTOCOL_BROKEN;
+  AlternateProtocolMap::iterator it = alternate_protocol_map_.Get(server);
+  if (it != alternate_protocol_map_.end()) {
+    it->second.protocol = ALTERNATE_PROTOCOL_BROKEN;
+    return;
+  }
+  PortAlternateProtocolPair alternate;
+  alternate.protocol = ALTERNATE_PROTOCOL_BROKEN;
+  alternate_protocol_map_.Put(server, alternate);
 }
 
 const AlternateProtocolMap&