Move SpdySessionRequestMap to SpdySessionPool

net/spdy/chromium/spdy_session_pool.h

 // Copyright (c) 2012 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.
 
 #ifndef NET_SPDY_CHROMIUM_SPDY_SESSION_POOL_H_
 #define NET_SPDY_CHROMIUM_SPDY_SESSION_POOL_H_
 
 #include <stddef.h>
 
 #include <map>
 #include <memory>
 #include <set>
 #include <vector>
 
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/weak_ptr.h"
 #include "net/base/host_port_pair.h"
 #include "net/base/ip_endpoint.h"
 #include "net/base/net_errors.h"
 #include "net/base/net_export.h"
 #include "net/base/network_change_notifier.h"
 #include "net/cert/cert_database.h"
+#include "net/http/http_stream_factory_impl_request.h"
 #include "net/proxy/proxy_config.h"
 #include "net/proxy/proxy_server.h"
 #include "net/spdy/chromium/server_push_delegate.h"
 #include "net/spdy/chromium/spdy_session_key.h"
 #include "net/spdy/core/spdy_protocol.h"
 #include "net/spdy/platform/api/spdy_string.h"
 #include "net/ssl/ssl_config_service.h"
 
 namespace base {
 namespace trace_event {
@@ skipping 125 matching lines @@
 
   // CertDatabase::Observer methods:
 
   // We perform the same flushing as described above when certificate database
   // is changed.
   void OnCertDBChanged() override;
 
   void DumpMemoryStats(base::trace_event::ProcessMemoryDump* pmd,
                        const SpdyString& parent_dump_absolute_name) const;
 
+  // Called when a SpdySession is ready. It will find appropriate Requests and
+  // fulfill them. |direct| indicates whether or not |spdy_session| uses a
+  // proxy.
+  void OnNewSpdySessionReady(const base::WeakPtr<SpdySession>& spdy_session,
+                             bool direct,
+                             const SSLConfig& used_ssl_config,
+                             const ProxyInfo& used_proxy_info,
+                             bool was_alpn_negotiated,
+                             NextProto negotiated_protocol,
+                             bool using_spdy,
+                             NetLogSource source_dependency);
+
+  // Adds |request| to |spdy_session_request_map_| under |spdy_session_key| Key.
+  // Sets |spdy_session_key| as |request|'s SpdySessionKey.
+  void AddRequestToSpdySessionRequestMap(
+      const SpdySessionKey& spdy_session_key,
+      HttpStreamFactoryImpl::Request* request);
+
+  // Removes |request| from |spdy_session_request_map_|. No-op if |request| does
+  // not have a SpdySessionKey.
+  void RemoveRequestFromSpdySessionRequestMap(
+      HttpStreamFactoryImpl::Request* request);
+
  private:
   friend class SpdySessionPoolPeer;  // For testing.
 
+  typedef std::set<HttpStreamFactoryImpl::Request*> RequestSet;
+  typedef std::map<SpdySessionKey, RequestSet> SpdySessionRequestMap;
   typedef std::set<SpdySession*> SessionSet;
   typedef std::vector<base::WeakPtr<SpdySession> > WeakSessionList;
   typedef std::map<SpdySessionKey, base::WeakPtr<SpdySession> >
       AvailableSessionMap;
   typedef std::map<IPEndPoint, SpdySessionKey> AliasMap;
   typedef std::map<GURL, WeakSessionList> UnclaimedPushedStreamMap;
 
   // Returns true iff |session| is in |available_sessions_|.
   bool IsSessionAvailable(const base::WeakPtr<SpdySession>& session) const;
 
@@ skipping 56 matching lines @@
   bool enable_ping_based_connection_checking_;
 
   size_t session_max_recv_window_size_;
 
   // Settings that are sent in the initial SETTINGS frame
   // (if |enable_sending_initial_data_| is true),
   // and also control SpdySession parameters like initial receive window size
   // and maximum HPACK dynamic table size.
   const SettingsMap initial_settings_;
 
+  SpdySessionRequestMap spdy_session_request_map_;
+
   TimeFunc time_func_;
   ServerPushDelegate* push_delegate_;
 
   // Determines if a proxy is a trusted SPDY proxy, which is allowed to push
   // resources from origins that are different from those of their associated
   // streams. May be nullptr.
   ProxyDelegate* proxy_delegate_;
 
   DISALLOW_COPY_AND_ASSIGN(SpdySessionPool);
 };
 
 }  // namespace net
 
 #endif  // NET_SPDY_CHROMIUM_SPDY_SESSION_POOL_H_