1 files changed, 76 insertions, 0 deletions
diff --git a/net/spdy/spdy_session_pool.h b/net/spdy/spdy_session_pool.h
new file mode 100644
index 0000000..98d78dd
--- /dev/null
+++ b/net/spdy/spdy_session_pool.h
@@ -0,0 +1,76 @@
+// Copyright (c) 2009 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_SPDY_SESSION_POOL_H_
+#define NET_SPDY_SPDY_SESSION_POOL_H_
+
+#include <map>
+#include <list>
+#include <string>
+
+#include "base/ref_counted.h"
+#include "base/scoped_ptr.h"
+#include "net/base/host_resolver.h"
+
+namespace net {
+
+class ClientSocketHandle;
+class FlipSession;
+class HttpNetworkSession;
+
+// This is a very simple pool for open FlipSessions.
+// TODO(mbelshe): Make this production ready.
+class FlipSessionPool : public base::RefCounted<FlipSessionPool> {
+ public:
+  FlipSessionPool();
+
+  // Either returns an existing FlipSession or creates a new FlipSession for
+  // use.
+  scoped_refptr<FlipSession> Get(
+      const HostResolver::RequestInfo& info, HttpNetworkSession* session);
+
+  // Builds a FlipSession from an existing socket.  Users should try calling
+  // Get() first to use an existing FlipSession so we don't get multiple
+  // FlipSessions per domain.  Note that ownership of |connection| is
+  // transferred from the caller to the FlipSession.
+  scoped_refptr<FlipSession> GetFlipSessionFromSocket(
+      const HostResolver::RequestInfo& info,
+      HttpNetworkSession* session,
+      ClientSocketHandle* connection);
+
+  // TODO(willchan): Consider renaming to HasReusableSession, since perhaps we
+  // should be creating a new session.
+  bool HasSession(const HostResolver::RequestInfo& info) const;
+
+  // Close all Flip Sessions; used for debugging.
+  void CloseAllSessions();
+
+ private:
+  friend class base::RefCounted<FlipSessionPool>;
+  friend class FlipSession;  // Needed for Remove().
+  friend class FlipSessionPoolPeer;  // For testing.
+
+  typedef std::list<scoped_refptr<FlipSession> > FlipSessionList;
+  typedef std::map<std::string, FlipSessionList*> FlipSessionsMap;
+
+  virtual ~FlipSessionPool();
+
+  // Removes a FlipSession from the FlipSessionPool.
+  void Remove(const scoped_refptr<FlipSession>& session);
+
+  // Helper functions for manipulating the lists.
+  FlipSessionList* AddSessionList(const std::string& domain);
+  FlipSessionList* GetSessionList(const std::string& domain);
+  const FlipSessionList* GetSessionList(const std::string& domain) const;
+  void RemoveSessionList(const std::string& domain);
+
+  // This is our weak session pool - one session per domain.
+  FlipSessionsMap sessions_;
+
+  DISALLOW_COPY_AND_ASSIGN(FlipSessionPool);
+};
+
+}  // namespace net
+
+#endif  // NET_SPDY_SPDY_SESSION_POOL_H_