OLD | NEW |
| (Empty) |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #ifndef NET_WEBSOCKETS_WEBSOCKET_THROTTLE_H_ | |
6 #define NET_WEBSOCKETS_WEBSOCKET_THROTTLE_H_ | |
7 | |
8 #include <deque> | |
9 #include <map> | |
10 #include <set> | |
11 #include <string> | |
12 | |
13 #include "net/base/ip_endpoint.h" | |
14 #include "net/base/net_export.h" | |
15 | |
16 template <typename T> struct DefaultSingletonTraits; | |
17 | |
18 namespace net { | |
19 | |
20 class SocketStream; | |
21 class WebSocketJob; | |
22 | |
23 // SocketStreamThrottle for WebSocket protocol. | |
24 // Implements the client-side requirements in the spec. | |
25 // http://tools.ietf.org/html/draft-hixie-thewebsocketprotocol | |
26 // 4.1 Handshake | |
27 // 1. If the user agent already has a Web Socket connection to the | |
28 // remote host (IP address) identified by /host/, even if known by | |
29 // another name, wait until that connection has been established or | |
30 // for that connection to have failed. | |
31 class NET_EXPORT_PRIVATE WebSocketThrottle { | |
32 public: | |
33 // Returns the singleton instance. | |
34 static WebSocketThrottle* GetInstance(); | |
35 | |
36 // Puts |job| in |queue_| and queues for the destination addresses | |
37 // of |job|. | |
38 // If other job is using the same destination address, set |job| waiting. | |
39 // | |
40 // Returns true if successful. If the number of pending jobs will exceed | |
41 // the limit, does nothing and returns false. | |
42 bool PutInQueue(WebSocketJob* job); | |
43 | |
44 // Removes |job| from |queue_| and queues for the destination addresses | |
45 // of |job|, and then wakes up jobs that can now resume establishing a | |
46 // connection. | |
47 void RemoveFromQueue(WebSocketJob* job); | |
48 | |
49 private: | |
50 typedef std::deque<WebSocketJob*> ConnectingQueue; | |
51 typedef std::map<IPEndPoint, ConnectingQueue> ConnectingAddressMap; | |
52 | |
53 WebSocketThrottle(); | |
54 ~WebSocketThrottle(); | |
55 friend struct DefaultSingletonTraits<WebSocketThrottle>; | |
56 | |
57 // Examines if any of the given jobs can resume establishing a connection. If | |
58 // for all per-address queues for each resolved addresses | |
59 // (job->address_list()) of a job, the job is at the front of the queues, the | |
60 // job can resume establishing a connection, so wakes up the job. | |
61 void WakeupSocketIfNecessary( | |
62 const std::set<WebSocketJob*>& wakeup_candidates); | |
63 | |
64 // Key: string of host's address. Value: queue of sockets for the address. | |
65 ConnectingAddressMap addr_map_; | |
66 | |
67 // Queue of sockets for websockets in opening state. | |
68 ConnectingQueue queue_; | |
69 | |
70 DISALLOW_COPY_AND_ASSIGN(WebSocketThrottle); | |
71 }; | |
72 | |
73 } // namespace net | |
74 | |
75 #endif // NET_WEBSOCKETS_WEBSOCKET_THROTTLE_H_ | |
OLD | NEW |