Index: net/socket/client_socket_pool_base.h |
diff --git a/net/socket/client_socket_pool_base.h b/net/socket/client_socket_pool_base.h |
deleted file mode 100644 |
index a7dcbf5263d520be60b4ab291265d9479da10faa..0000000000000000000000000000000000000000 |
--- a/net/socket/client_socket_pool_base.h |
+++ /dev/null |
@@ -1,878 +0,0 @@ |
-// 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. |
-// |
-// A ClientSocketPoolBase is used to restrict the number of sockets open at |
-// a time. It also maintains a list of idle persistent sockets for reuse. |
-// Subclasses of ClientSocketPool should compose ClientSocketPoolBase to handle |
-// the core logic of (1) restricting the number of active (connected or |
-// connecting) sockets per "group" (generally speaking, the hostname), (2) |
-// maintaining a per-group list of idle, persistent sockets for reuse, and (3) |
-// limiting the total number of active sockets in the system. |
-// |
-// ClientSocketPoolBase abstracts socket connection details behind ConnectJob, |
-// ConnectJobFactory, and SocketParams. When a socket "slot" becomes available, |
-// the ClientSocketPoolBase will ask the ConnectJobFactory to create a |
-// ConnectJob with a SocketParams. Subclasses of ClientSocketPool should |
-// implement their socket specific connection by subclassing ConnectJob and |
-// implementing ConnectJob::ConnectInternal(). They can control the parameters |
-// passed to each new ConnectJob instance via their ConnectJobFactory subclass |
-// and templated SocketParams parameter. |
-// |
-#ifndef NET_SOCKET_CLIENT_SOCKET_POOL_BASE_H_ |
-#define NET_SOCKET_CLIENT_SOCKET_POOL_BASE_H_ |
- |
-#include <cstddef> |
-#include <deque> |
-#include <list> |
-#include <map> |
-#include <set> |
-#include <string> |
-#include <vector> |
- |
-#include "base/basictypes.h" |
-#include "base/memory/ref_counted.h" |
-#include "base/memory/scoped_ptr.h" |
-#include "base/memory/weak_ptr.h" |
-#include "base/time/time.h" |
-#include "base/timer/timer.h" |
-#include "net/base/address_list.h" |
-#include "net/base/completion_callback.h" |
-#include "net/base/load_states.h" |
-#include "net/base/load_timing_info.h" |
-#include "net/base/net_errors.h" |
-#include "net/base/net_export.h" |
-#include "net/base/net_log.h" |
-#include "net/base/network_change_notifier.h" |
-#include "net/base/priority_queue.h" |
-#include "net/base/request_priority.h" |
-#include "net/socket/client_socket_handle.h" |
-#include "net/socket/client_socket_pool.h" |
-#include "net/socket/stream_socket.h" |
- |
-namespace net { |
- |
-class ClientSocketHandle; |
- |
-// ConnectJob provides an abstract interface for "connecting" a socket. |
-// The connection may involve host resolution, tcp connection, ssl connection, |
-// etc. |
-class NET_EXPORT_PRIVATE ConnectJob { |
- public: |
- class NET_EXPORT_PRIVATE Delegate { |
- public: |
- Delegate() {} |
- virtual ~Delegate() {} |
- |
- // Alerts the delegate that the connection completed. |job| must |
- // be destroyed by the delegate. A scoped_ptr<> isn't used because |
- // the caller of this function doesn't own |job|. |
- virtual void OnConnectJobComplete(int result, |
- ConnectJob* job) = 0; |
- |
- private: |
- DISALLOW_COPY_AND_ASSIGN(Delegate); |
- }; |
- |
- // A |timeout_duration| of 0 corresponds to no timeout. |
- ConnectJob(const std::string& group_name, |
- base::TimeDelta timeout_duration, |
- RequestPriority priority, |
- Delegate* delegate, |
- const BoundNetLog& net_log); |
- virtual ~ConnectJob(); |
- |
- // Accessors |
- const std::string& group_name() const { return group_name_; } |
- const BoundNetLog& net_log() { return net_log_; } |
- |
- // Releases ownership of the underlying socket to the caller. |
- // Returns the released socket, or NULL if there was a connection |
- // error. |
- scoped_ptr<StreamSocket> PassSocket(); |
- |
- // Begins connecting the socket. Returns OK on success, ERR_IO_PENDING if it |
- // cannot complete synchronously without blocking, or another net error code |
- // on error. In asynchronous completion, the ConnectJob will notify |
- // |delegate_| via OnConnectJobComplete. In both asynchronous and synchronous |
- // completion, ReleaseSocket() can be called to acquire the connected socket |
- // if it succeeded. |
- int Connect(); |
- |
- virtual LoadState GetLoadState() const = 0; |
- |
- // If Connect returns an error (or OnConnectJobComplete reports an error |
- // result) this method will be called, allowing the pool to add |
- // additional error state to the ClientSocketHandle (post late-binding). |
- virtual void GetAdditionalErrorState(ClientSocketHandle* handle) {} |
- |
- const LoadTimingInfo::ConnectTiming& connect_timing() const { |
- return connect_timing_; |
- } |
- |
- const BoundNetLog& net_log() const { return net_log_; } |
- |
- protected: |
- RequestPriority priority() const { return priority_; } |
- void SetSocket(scoped_ptr<StreamSocket> socket); |
- StreamSocket* socket() { return socket_.get(); } |
- void NotifyDelegateOfCompletion(int rv); |
- void ResetTimer(base::TimeDelta remainingTime); |
- |
- // Connection establishment timing information. |
- LoadTimingInfo::ConnectTiming connect_timing_; |
- |
- private: |
- virtual int ConnectInternal() = 0; |
- |
- void LogConnectStart(); |
- void LogConnectCompletion(int net_error); |
- |
- // Alerts the delegate that the ConnectJob has timed out. |
- void OnTimeout(); |
- |
- const std::string group_name_; |
- const base::TimeDelta timeout_duration_; |
- // TODO(akalin): Support reprioritization. |
- const RequestPriority priority_; |
- // Timer to abort jobs that take too long. |
- base::OneShotTimer<ConnectJob> timer_; |
- Delegate* delegate_; |
- scoped_ptr<StreamSocket> socket_; |
- BoundNetLog net_log_; |
- // A ConnectJob is idle until Connect() has been called. |
- bool idle_; |
- |
- DISALLOW_COPY_AND_ASSIGN(ConnectJob); |
-}; |
- |
-namespace internal { |
- |
-// ClientSocketPoolBaseHelper is an internal class that implements almost all |
-// the functionality from ClientSocketPoolBase without using templates. |
-// ClientSocketPoolBase adds templated definitions built on top of |
-// ClientSocketPoolBaseHelper. This class is not for external use, please use |
-// ClientSocketPoolBase instead. |
-class NET_EXPORT_PRIVATE ClientSocketPoolBaseHelper |
- : public ConnectJob::Delegate, |
- public NetworkChangeNotifier::IPAddressObserver { |
- public: |
- typedef uint32 Flags; |
- |
- // Used to specify specific behavior for the ClientSocketPool. |
- enum Flag { |
- NORMAL = 0, // Normal behavior. |
- NO_IDLE_SOCKETS = 0x1, // Do not return an idle socket. Create a new one. |
- }; |
- |
- class NET_EXPORT_PRIVATE Request { |
- public: |
- Request(ClientSocketHandle* handle, |
- const CompletionCallback& callback, |
- RequestPriority priority, |
- bool ignore_limits, |
- Flags flags, |
- const BoundNetLog& net_log); |
- |
- virtual ~Request(); |
- |
- ClientSocketHandle* handle() const { return handle_; } |
- const CompletionCallback& callback() const { return callback_; } |
- RequestPriority priority() const { return priority_; } |
- bool ignore_limits() const { return ignore_limits_; } |
- Flags flags() const { return flags_; } |
- const BoundNetLog& net_log() const { return net_log_; } |
- |
- private: |
- ClientSocketHandle* const handle_; |
- const CompletionCallback callback_; |
- // TODO(akalin): Support reprioritization. |
- const RequestPriority priority_; |
- const bool ignore_limits_; |
- const Flags flags_; |
- const BoundNetLog net_log_; |
- |
- DISALLOW_COPY_AND_ASSIGN(Request); |
- }; |
- |
- class ConnectJobFactory { |
- public: |
- ConnectJobFactory() {} |
- virtual ~ConnectJobFactory() {} |
- |
- virtual scoped_ptr<ConnectJob> NewConnectJob( |
- const std::string& group_name, |
- const Request& request, |
- ConnectJob::Delegate* delegate) const = 0; |
- |
- virtual base::TimeDelta ConnectionTimeout() const = 0; |
- |
- private: |
- DISALLOW_COPY_AND_ASSIGN(ConnectJobFactory); |
- }; |
- |
- ClientSocketPoolBaseHelper( |
- HigherLayeredPool* pool, |
- int max_sockets, |
- int max_sockets_per_group, |
- base::TimeDelta unused_idle_socket_timeout, |
- base::TimeDelta used_idle_socket_timeout, |
- ConnectJobFactory* connect_job_factory); |
- |
- ~ClientSocketPoolBaseHelper() override; |
- |
- // Adds a lower layered pool to |this|, and adds |this| as a higher layered |
- // pool on top of |lower_pool|. |
- void AddLowerLayeredPool(LowerLayeredPool* lower_pool); |
- |
- // See LowerLayeredPool::IsStalled for documentation on this function. |
- bool IsStalled() const; |
- |
- // See LowerLayeredPool for documentation on these functions. It is expected |
- // in the destructor that no higher layer pools remain. |
- void AddHigherLayeredPool(HigherLayeredPool* higher_pool); |
- void RemoveHigherLayeredPool(HigherLayeredPool* higher_pool); |
- |
- // See ClientSocketPool::RequestSocket for documentation on this function. |
- int RequestSocket(const std::string& group_name, |
- scoped_ptr<const Request> request); |
- |
- // See ClientSocketPool::RequestSocket for documentation on this function. |
- void RequestSockets(const std::string& group_name, |
- const Request& request, |
- int num_sockets); |
- |
- // See ClientSocketPool::CancelRequest for documentation on this function. |
- void CancelRequest(const std::string& group_name, |
- ClientSocketHandle* handle); |
- |
- // See ClientSocketPool::ReleaseSocket for documentation on this function. |
- void ReleaseSocket(const std::string& group_name, |
- scoped_ptr<StreamSocket> socket, |
- int id); |
- |
- // See ClientSocketPool::FlushWithError for documentation on this function. |
- void FlushWithError(int error); |
- |
- // See ClientSocketPool::CloseIdleSockets for documentation on this function. |
- void CloseIdleSockets(); |
- |
- // See ClientSocketPool::IdleSocketCount() for documentation on this function. |
- int idle_socket_count() const { |
- return idle_socket_count_; |
- } |
- |
- // See ClientSocketPool::IdleSocketCountInGroup() for documentation on this |
- // function. |
- int IdleSocketCountInGroup(const std::string& group_name) const; |
- |
- // See ClientSocketPool::GetLoadState() for documentation on this function. |
- LoadState GetLoadState(const std::string& group_name, |
- const ClientSocketHandle* handle) const; |
- |
- base::TimeDelta ConnectRetryInterval() const { |
- // TODO(mbelshe): Make this tuned dynamically based on measured RTT. |
- // For now, just use the max retry interval. |
- return base::TimeDelta::FromMilliseconds( |
- ClientSocketPool::kMaxConnectRetryIntervalMs); |
- } |
- |
- int NumUnassignedConnectJobsInGroup(const std::string& group_name) const { |
- return group_map_.find(group_name)->second->unassigned_job_count(); |
- } |
- |
- int NumConnectJobsInGroup(const std::string& group_name) const { |
- return group_map_.find(group_name)->second->jobs().size(); |
- } |
- |
- int NumActiveSocketsInGroup(const std::string& group_name) const { |
- return group_map_.find(group_name)->second->active_socket_count(); |
- } |
- |
- bool HasGroup(const std::string& group_name) const; |
- |
- // Called to enable/disable cleaning up idle sockets. When enabled, |
- // idle sockets that have been around for longer than a period defined |
- // by kCleanupInterval are cleaned up using a timer. Otherwise they are |
- // closed next time client makes a request. This may reduce network |
- // activity and power consumption. |
- static bool cleanup_timer_enabled(); |
- static bool set_cleanup_timer_enabled(bool enabled); |
- |
- // Closes all idle sockets if |force| is true. Else, only closes idle |
- // sockets that timed out or can't be reused. Made public for testing. |
- void CleanupIdleSockets(bool force); |
- |
- // Closes one idle socket. Picks the first one encountered. |
- // TODO(willchan): Consider a better algorithm for doing this. Perhaps we |
- // should keep an ordered list of idle sockets, and close them in order. |
- // Requires maintaining more state. It's not clear if it's worth it since |
- // I'm not sure if we hit this situation often. |
- bool CloseOneIdleSocket(); |
- |
- // Checks higher layered pools to see if they can close an idle connection. |
- bool CloseOneIdleConnectionInHigherLayeredPool(); |
- |
- // See ClientSocketPool::GetInfoAsValue for documentation on this function. |
- base::DictionaryValue* GetInfoAsValue(const std::string& name, |
- const std::string& type) const; |
- |
- base::TimeDelta ConnectionTimeout() const { |
- return connect_job_factory_->ConnectionTimeout(); |
- } |
- |
- static bool connect_backup_jobs_enabled(); |
- static bool set_connect_backup_jobs_enabled(bool enabled); |
- |
- void EnableConnectBackupJobs(); |
- |
- // ConnectJob::Delegate methods: |
- void OnConnectJobComplete(int result, ConnectJob* job) override; |
- |
- // NetworkChangeNotifier::IPAddressObserver methods: |
- void OnIPAddressChanged() override; |
- |
- private: |
- friend class base::RefCounted<ClientSocketPoolBaseHelper>; |
- |
- // Entry for a persistent socket which became idle at time |start_time|. |
- struct IdleSocket { |
- IdleSocket() : socket(NULL) {} |
- |
- // An idle socket can't be used if it is disconnected or has been used |
- // before and has received data unexpectedly (hence no longer idle). The |
- // unread data would be mistaken for the beginning of the next response if |
- // we were to use the socket for a new request. |
- // |
- // Note that a socket that has never been used before (like a preconnected |
- // socket) may be used even with unread data. This may be, e.g., a SPDY |
- // SETTINGS frame. |
- bool IsUsable() const; |
- |
- // An idle socket should be removed if it can't be reused, or has been idle |
- // for too long. |now| is the current time value (TimeTicks::Now()). |
- // |timeout| is the length of time to wait before timing out an idle socket. |
- bool ShouldCleanup(base::TimeTicks now, base::TimeDelta timeout) const; |
- |
- StreamSocket* socket; |
- base::TimeTicks start_time; |
- }; |
- |
- typedef PriorityQueue<const Request*> RequestQueue; |
- typedef std::map<const ClientSocketHandle*, const Request*> RequestMap; |
- |
- // A Group is allocated per group_name when there are idle sockets or pending |
- // requests. Otherwise, the Group object is removed from the map. |
- // |active_socket_count| tracks the number of sockets held by clients. |
- class Group { |
- public: |
- Group(); |
- ~Group(); |
- |
- bool IsEmpty() const { |
- return active_socket_count_ == 0 && idle_sockets_.empty() && |
- jobs_.empty() && pending_requests_.empty(); |
- } |
- |
- bool HasAvailableSocketSlot(int max_sockets_per_group) const { |
- return NumActiveSocketSlots() < max_sockets_per_group; |
- } |
- |
- int NumActiveSocketSlots() const { |
- return active_socket_count_ + static_cast<int>(jobs_.size()) + |
- static_cast<int>(idle_sockets_.size()); |
- } |
- |
- bool IsStalledOnPoolMaxSockets(int max_sockets_per_group) const { |
- return HasAvailableSocketSlot(max_sockets_per_group) && |
- pending_requests_.size() > jobs_.size(); |
- } |
- |
- // Returns the priority of the top of the pending request queue |
- // (which may be less than the maximum priority over the entire |
- // queue, due to how we prioritize requests with |ignore_limits| |
- // set over others). |
- RequestPriority TopPendingPriority() const { |
- // NOTE: FirstMax().value()->priority() is not the same as |
- // FirstMax().priority()! |
- return pending_requests_.FirstMax().value()->priority(); |
- } |
- |
- // Set a timer to create a backup job if it takes too long to |
- // create one and if a timer isn't already running. |
- void StartBackupJobTimer(const std::string& group_name, |
- ClientSocketPoolBaseHelper* pool); |
- |
- bool BackupJobTimerIsRunning() const; |
- |
- // If there's a ConnectJob that's never been assigned to Request, |
- // decrements |unassigned_job_count_| and returns true. |
- // Otherwise, returns false. |
- bool TryToUseUnassignedConnectJob(); |
- |
- void AddJob(scoped_ptr<ConnectJob> job, bool is_preconnect); |
- // Remove |job| from this group, which must already own |job|. |
- void RemoveJob(ConnectJob* job); |
- void RemoveAllJobs(); |
- |
- bool has_pending_requests() const { |
- return !pending_requests_.empty(); |
- } |
- |
- size_t pending_request_count() const { |
- return pending_requests_.size(); |
- } |
- |
- // Gets (but does not remove) the next pending request. Returns |
- // NULL if there are no pending requests. |
- const Request* GetNextPendingRequest() const; |
- |
- // Returns true if there is a connect job for |handle|. |
- bool HasConnectJobForHandle(const ClientSocketHandle* handle) const; |
- |
- // Inserts the request into the queue based on priority |
- // order. Older requests are prioritized over requests of equal |
- // priority. |
- void InsertPendingRequest(scoped_ptr<const Request> request); |
- |
- // Gets and removes the next pending request. Returns NULL if |
- // there are no pending requests. |
- scoped_ptr<const Request> PopNextPendingRequest(); |
- |
- // Finds the pending request for |handle| and removes it. Returns |
- // the removed pending request, or NULL if there was none. |
- scoped_ptr<const Request> FindAndRemovePendingRequest( |
- ClientSocketHandle* handle); |
- |
- void IncrementActiveSocketCount() { active_socket_count_++; } |
- void DecrementActiveSocketCount() { active_socket_count_--; } |
- |
- int unassigned_job_count() const { return unassigned_job_count_; } |
- const std::set<ConnectJob*>& jobs() const { return jobs_; } |
- const std::list<IdleSocket>& idle_sockets() const { return idle_sockets_; } |
- int active_socket_count() const { return active_socket_count_; } |
- std::list<IdleSocket>* mutable_idle_sockets() { return &idle_sockets_; } |
- |
- private: |
- // Returns the iterator's pending request after removing it from |
- // the queue. |
- scoped_ptr<const Request> RemovePendingRequest( |
- const RequestQueue::Pointer& pointer); |
- |
- // Called when the backup socket timer fires. |
- void OnBackupJobTimerFired( |
- std::string group_name, |
- ClientSocketPoolBaseHelper* pool); |
- |
- // Checks that |unassigned_job_count_| does not execeed the number of |
- // ConnectJobs. |
- void SanityCheck(); |
- |
- // Total number of ConnectJobs that have never been assigned to a Request. |
- // Since jobs use late binding to requests, which ConnectJobs have or have |
- // not been assigned to a request are not tracked. This is incremented on |
- // preconnect and decremented when a preconnect is assigned, or when there |
- // are fewer than |unassigned_job_count_| ConnectJobs. Not incremented |
- // when a request is cancelled. |
- size_t unassigned_job_count_; |
- |
- std::list<IdleSocket> idle_sockets_; |
- std::set<ConnectJob*> jobs_; |
- RequestQueue pending_requests_; |
- int active_socket_count_; // number of active sockets used by clients |
- // A timer for when to start the backup job. |
- base::OneShotTimer<Group> backup_job_timer_; |
- }; |
- |
- typedef std::map<std::string, Group*> GroupMap; |
- |
- typedef std::set<ConnectJob*> ConnectJobSet; |
- |
- struct CallbackResultPair { |
- CallbackResultPair(); |
- CallbackResultPair(const CompletionCallback& callback_in, int result_in); |
- ~CallbackResultPair(); |
- |
- CompletionCallback callback; |
- int result; |
- }; |
- |
- typedef std::map<const ClientSocketHandle*, CallbackResultPair> |
- PendingCallbackMap; |
- |
- Group* GetOrCreateGroup(const std::string& group_name); |
- void RemoveGroup(const std::string& group_name); |
- void RemoveGroup(GroupMap::iterator it); |
- |
- // Called when the number of idle sockets changes. |
- void IncrementIdleCount(); |
- void DecrementIdleCount(); |
- |
- // Start cleanup timer for idle sockets. |
- void StartIdleSocketTimer(); |
- |
- // Scans the group map for groups which have an available socket slot and |
- // at least one pending request. Returns true if any groups are stalled, and |
- // if so (and if both |group| and |group_name| are not NULL), fills |group| |
- // and |group_name| with data of the stalled group having highest priority. |
- bool FindTopStalledGroup(Group** group, std::string* group_name) const; |
- |
- // Called when timer_ fires. This method scans the idle sockets removing |
- // sockets that timed out or can't be reused. |
- void OnCleanupTimerFired() { |
- CleanupIdleSockets(false); |
- } |
- |
- // Removes |job| from |group|, which must already own |job|. |
- void RemoveConnectJob(ConnectJob* job, Group* group); |
- |
- // Tries to see if we can handle any more requests for |group|. |
- void OnAvailableSocketSlot(const std::string& group_name, Group* group); |
- |
- // Process a pending socket request for a group. |
- void ProcessPendingRequest(const std::string& group_name, Group* group); |
- |
- // Assigns |socket| to |handle| and updates |group|'s counters appropriately. |
- void HandOutSocket(scoped_ptr<StreamSocket> socket, |
- ClientSocketHandle::SocketReuseType reuse_type, |
- const LoadTimingInfo::ConnectTiming& connect_timing, |
- ClientSocketHandle* handle, |
- base::TimeDelta time_idle, |
- Group* group, |
- const BoundNetLog& net_log); |
- |
- // Adds |socket| to the list of idle sockets for |group|. |
- void AddIdleSocket(scoped_ptr<StreamSocket> socket, Group* group); |
- |
- // Iterates through |group_map_|, canceling all ConnectJobs and deleting |
- // groups if they are no longer needed. |
- void CancelAllConnectJobs(); |
- |
- // Iterates through |group_map_|, posting |error| callbacks for all |
- // requests, and then deleting groups if they are no longer needed. |
- void CancelAllRequestsWithError(int error); |
- |
- // Returns true if we can't create any more sockets due to the total limit. |
- bool ReachedMaxSocketsLimit() const; |
- |
- // This is the internal implementation of RequestSocket(). It differs in that |
- // it does not handle logging into NetLog of the queueing status of |
- // |request|. |
- int RequestSocketInternal(const std::string& group_name, |
- const Request& request); |
- |
- // Assigns an idle socket for the group to the request. |
- // Returns |true| if an idle socket is available, false otherwise. |
- bool AssignIdleSocketToRequest(const Request& request, Group* group); |
- |
- static void LogBoundConnectJobToRequest( |
- const NetLog::Source& connect_job_source, const Request& request); |
- |
- // Same as CloseOneIdleSocket() except it won't close an idle socket in |
- // |group|. If |group| is NULL, it is ignored. Returns true if it closed a |
- // socket. |
- bool CloseOneIdleSocketExceptInGroup(const Group* group); |
- |
- // Checks if there are stalled socket groups that should be notified |
- // for possible wakeup. |
- void CheckForStalledSocketGroups(); |
- |
- // Posts a task to call InvokeUserCallback() on the next iteration through the |
- // current message loop. Inserts |callback| into |pending_callback_map_|, |
- // keyed by |handle|. |
- void InvokeUserCallbackLater( |
- ClientSocketHandle* handle, const CompletionCallback& callback, int rv); |
- |
- // Invokes the user callback for |handle|. By the time this task has run, |
- // it's possible that the request has been cancelled, so |handle| may not |
- // exist in |pending_callback_map_|. We look up the callback and result code |
- // in |pending_callback_map_|. |
- void InvokeUserCallback(ClientSocketHandle* handle); |
- |
- // Tries to close idle sockets in a higher level socket pool as long as this |
- // this pool is stalled. |
- void TryToCloseSocketsInLayeredPools(); |
- |
- GroupMap group_map_; |
- |
- // Map of the ClientSocketHandles for which we have a pending Task to invoke a |
- // callback. This is necessary since, before we invoke said callback, it's |
- // possible that the request is cancelled. |
- PendingCallbackMap pending_callback_map_; |
- |
- // Timer used to periodically prune idle sockets that timed out or can't be |
- // reused. |
- base::RepeatingTimer<ClientSocketPoolBaseHelper> timer_; |
- |
- // The total number of idle sockets in the system. |
- int idle_socket_count_; |
- |
- // Number of connecting sockets across all groups. |
- int connecting_socket_count_; |
- |
- // Number of connected sockets we handed out across all groups. |
- int handed_out_socket_count_; |
- |
- // The maximum total number of sockets. See ReachedMaxSocketsLimit. |
- const int max_sockets_; |
- |
- // The maximum number of sockets kept per group. |
- const int max_sockets_per_group_; |
- |
- // Whether to use timer to cleanup idle sockets. |
- bool use_cleanup_timer_; |
- |
- // The time to wait until closing idle sockets. |
- const base::TimeDelta unused_idle_socket_timeout_; |
- const base::TimeDelta used_idle_socket_timeout_; |
- |
- const scoped_ptr<ConnectJobFactory> connect_job_factory_; |
- |
- // TODO(vandebo) Remove when backup jobs move to TransportClientSocketPool |
- bool connect_backup_jobs_enabled_; |
- |
- // A unique id for the pool. It gets incremented every time we |
- // FlushWithError() the pool. This is so that when sockets get released back |
- // to the pool, we can make sure that they are discarded rather than reused. |
- int pool_generation_number_; |
- |
- // Used to add |this| as a higher layer pool on top of lower layer pools. May |
- // be NULL if no lower layer pools will be added. |
- HigherLayeredPool* pool_; |
- |
- // Pools that create connections through |this|. |this| will try to close |
- // their idle sockets when it stalls. Must be empty on destruction. |
- std::set<HigherLayeredPool*> higher_pools_; |
- |
- // Pools that this goes through. Typically there's only one, but not always. |
- // |this| will check if they're stalled when it has a new idle socket. |this| |
- // will remove itself from all lower layered pools on destruction. |
- std::set<LowerLayeredPool*> lower_pools_; |
- |
- base::WeakPtrFactory<ClientSocketPoolBaseHelper> weak_factory_; |
- |
- DISALLOW_COPY_AND_ASSIGN(ClientSocketPoolBaseHelper); |
-}; |
- |
-} // namespace internal |
- |
-template <typename SocketParams> |
-class ClientSocketPoolBase { |
- public: |
- class Request : public internal::ClientSocketPoolBaseHelper::Request { |
- public: |
- Request(ClientSocketHandle* handle, |
- const CompletionCallback& callback, |
- RequestPriority priority, |
- internal::ClientSocketPoolBaseHelper::Flags flags, |
- bool ignore_limits, |
- const scoped_refptr<SocketParams>& params, |
- const BoundNetLog& net_log) |
- : internal::ClientSocketPoolBaseHelper::Request( |
- handle, callback, priority, ignore_limits, flags, net_log), |
- params_(params) {} |
- |
- const scoped_refptr<SocketParams>& params() const { return params_; } |
- |
- private: |
- const scoped_refptr<SocketParams> params_; |
- }; |
- |
- class ConnectJobFactory { |
- public: |
- ConnectJobFactory() {} |
- virtual ~ConnectJobFactory() {} |
- |
- virtual scoped_ptr<ConnectJob> NewConnectJob( |
- const std::string& group_name, |
- const Request& request, |
- ConnectJob::Delegate* delegate) const = 0; |
- |
- virtual base::TimeDelta ConnectionTimeout() const = 0; |
- |
- private: |
- DISALLOW_COPY_AND_ASSIGN(ConnectJobFactory); |
- }; |
- |
- // |max_sockets| is the maximum number of sockets to be maintained by this |
- // ClientSocketPool. |max_sockets_per_group| specifies the maximum number of |
- // sockets a "group" can have. |unused_idle_socket_timeout| specifies how |
- // long to leave an unused idle socket open before closing it. |
- // |used_idle_socket_timeout| specifies how long to leave a previously used |
- // idle socket open before closing it. |
- ClientSocketPoolBase( |
- HigherLayeredPool* self, |
- int max_sockets, |
- int max_sockets_per_group, |
- ClientSocketPoolHistograms* histograms, |
- base::TimeDelta unused_idle_socket_timeout, |
- base::TimeDelta used_idle_socket_timeout, |
- ConnectJobFactory* connect_job_factory) |
- : histograms_(histograms), |
- helper_(self, max_sockets, max_sockets_per_group, |
- unused_idle_socket_timeout, used_idle_socket_timeout, |
- new ConnectJobFactoryAdaptor(connect_job_factory)) {} |
- |
- virtual ~ClientSocketPoolBase() {} |
- |
- // These member functions simply forward to ClientSocketPoolBaseHelper. |
- void AddLowerLayeredPool(LowerLayeredPool* lower_pool) { |
- helper_.AddLowerLayeredPool(lower_pool); |
- } |
- |
- void AddHigherLayeredPool(HigherLayeredPool* higher_pool) { |
- helper_.AddHigherLayeredPool(higher_pool); |
- } |
- |
- void RemoveHigherLayeredPool(HigherLayeredPool* higher_pool) { |
- helper_.RemoveHigherLayeredPool(higher_pool); |
- } |
- |
- // RequestSocket bundles up the parameters into a Request and then forwards to |
- // ClientSocketPoolBaseHelper::RequestSocket(). |
- int RequestSocket(const std::string& group_name, |
- const scoped_refptr<SocketParams>& params, |
- RequestPriority priority, |
- ClientSocketHandle* handle, |
- const CompletionCallback& callback, |
- const BoundNetLog& net_log) { |
- scoped_ptr<const Request> request( |
- new Request(handle, callback, priority, |
- internal::ClientSocketPoolBaseHelper::NORMAL, |
- params->ignore_limits(), |
- params, net_log)); |
- return helper_.RequestSocket(group_name, request.Pass()); |
- } |
- |
- // RequestSockets bundles up the parameters into a Request and then forwards |
- // to ClientSocketPoolBaseHelper::RequestSockets(). Note that it assigns the |
- // priority to DEFAULT_PRIORITY and specifies the NO_IDLE_SOCKETS flag. |
- void RequestSockets(const std::string& group_name, |
- const scoped_refptr<SocketParams>& params, |
- int num_sockets, |
- const BoundNetLog& net_log) { |
- const Request request(NULL /* no handle */, |
- CompletionCallback(), |
- DEFAULT_PRIORITY, |
- internal::ClientSocketPoolBaseHelper::NO_IDLE_SOCKETS, |
- params->ignore_limits(), |
- params, |
- net_log); |
- helper_.RequestSockets(group_name, request, num_sockets); |
- } |
- |
- void CancelRequest(const std::string& group_name, |
- ClientSocketHandle* handle) { |
- return helper_.CancelRequest(group_name, handle); |
- } |
- |
- void ReleaseSocket(const std::string& group_name, |
- scoped_ptr<StreamSocket> socket, |
- int id) { |
- return helper_.ReleaseSocket(group_name, socket.Pass(), id); |
- } |
- |
- void FlushWithError(int error) { helper_.FlushWithError(error); } |
- |
- bool IsStalled() const { return helper_.IsStalled(); } |
- |
- void CloseIdleSockets() { return helper_.CloseIdleSockets(); } |
- |
- int idle_socket_count() const { return helper_.idle_socket_count(); } |
- |
- int IdleSocketCountInGroup(const std::string& group_name) const { |
- return helper_.IdleSocketCountInGroup(group_name); |
- } |
- |
- LoadState GetLoadState(const std::string& group_name, |
- const ClientSocketHandle* handle) const { |
- return helper_.GetLoadState(group_name, handle); |
- } |
- |
- virtual void OnConnectJobComplete(int result, ConnectJob* job) { |
- return helper_.OnConnectJobComplete(result, job); |
- } |
- |
- int NumUnassignedConnectJobsInGroup(const std::string& group_name) const { |
- return helper_.NumUnassignedConnectJobsInGroup(group_name); |
- } |
- |
- int NumConnectJobsInGroup(const std::string& group_name) const { |
- return helper_.NumConnectJobsInGroup(group_name); |
- } |
- |
- int NumActiveSocketsInGroup(const std::string& group_name) const { |
- return helper_.NumActiveSocketsInGroup(group_name); |
- } |
- |
- bool HasGroup(const std::string& group_name) const { |
- return helper_.HasGroup(group_name); |
- } |
- |
- void CleanupIdleSockets(bool force) { |
- return helper_.CleanupIdleSockets(force); |
- } |
- |
- base::DictionaryValue* GetInfoAsValue(const std::string& name, |
- const std::string& type) const { |
- return helper_.GetInfoAsValue(name, type); |
- } |
- |
- base::TimeDelta ConnectionTimeout() const { |
- return helper_.ConnectionTimeout(); |
- } |
- |
- ClientSocketPoolHistograms* histograms() const { |
- return histograms_; |
- } |
- |
- void EnableConnectBackupJobs() { helper_.EnableConnectBackupJobs(); } |
- |
- bool CloseOneIdleSocket() { return helper_.CloseOneIdleSocket(); } |
- |
- bool CloseOneIdleConnectionInHigherLayeredPool() { |
- return helper_.CloseOneIdleConnectionInHigherLayeredPool(); |
- } |
- |
- private: |
- // This adaptor class exists to bridge the |
- // internal::ClientSocketPoolBaseHelper::ConnectJobFactory and |
- // ClientSocketPoolBase::ConnectJobFactory types, allowing clients to use the |
- // typesafe ClientSocketPoolBase::ConnectJobFactory, rather than having to |
- // static_cast themselves. |
- class ConnectJobFactoryAdaptor |
- : public internal::ClientSocketPoolBaseHelper::ConnectJobFactory { |
- public: |
- typedef typename ClientSocketPoolBase<SocketParams>::ConnectJobFactory |
- ConnectJobFactory; |
- |
- explicit ConnectJobFactoryAdaptor(ConnectJobFactory* connect_job_factory) |
- : connect_job_factory_(connect_job_factory) {} |
- virtual ~ConnectJobFactoryAdaptor() {} |
- |
- virtual scoped_ptr<ConnectJob> NewConnectJob( |
- const std::string& group_name, |
- const internal::ClientSocketPoolBaseHelper::Request& request, |
- ConnectJob::Delegate* delegate) const override { |
- const Request& casted_request = static_cast<const Request&>(request); |
- return connect_job_factory_->NewConnectJob( |
- group_name, casted_request, delegate); |
- } |
- |
- base::TimeDelta ConnectionTimeout() const override { |
- return connect_job_factory_->ConnectionTimeout(); |
- } |
- |
- const scoped_ptr<ConnectJobFactory> connect_job_factory_; |
- }; |
- |
- // Histograms for the pool |
- ClientSocketPoolHistograms* const histograms_; |
- internal::ClientSocketPoolBaseHelper helper_; |
- |
- DISALLOW_COPY_AND_ASSIGN(ClientSocketPoolBase); |
-}; |
- |
-} // namespace net |
- |
-#endif // NET_SOCKET_CLIENT_SOCKET_POOL_BASE_H_ |