| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef CHROME_TEST_CHROMEDRIVER_NET_SYNC_WEBSOCKET_IMPL_H_ | 5 #ifndef CHROME_TEST_CHROMEDRIVER_NET_SYNC_WEBSOCKET_IMPL_H_ |
| 6 #define CHROME_TEST_CHROMEDRIVER_NET_SYNC_WEBSOCKET_IMPL_H_ | 6 #define CHROME_TEST_CHROMEDRIVER_NET_SYNC_WEBSOCKET_IMPL_H_ |
| 7 | 7 |
| 8 #include <list> | 8 #include <list> |
| 9 #include <string> | 9 #include <string> |
| 10 | 10 |
| (...skipping 14 matching lines...) Expand all Loading... |
| 25 | 25 |
| 26 namespace net { | 26 namespace net { |
| 27 class URLRequestContextGetter; | 27 class URLRequestContextGetter; |
| 28 } | 28 } |
| 29 | 29 |
| 30 class GURL; | 30 class GURL; |
| 31 | 31 |
| 32 class SyncWebSocketImpl : public SyncWebSocket { | 32 class SyncWebSocketImpl : public SyncWebSocket { |
| 33 public: | 33 public: |
| 34 explicit SyncWebSocketImpl(net::URLRequestContextGetter* context_getter); | 34 explicit SyncWebSocketImpl(net::URLRequestContextGetter* context_getter); |
| 35 virtual ~SyncWebSocketImpl(); | 35 ~SyncWebSocketImpl() override; |
| 36 | 36 |
| 37 // Overridden from SyncWebSocket: | 37 // Overridden from SyncWebSocket: |
| 38 virtual bool IsConnected() override; | 38 bool IsConnected() override; |
| 39 virtual bool Connect(const GURL& url) override; | 39 bool Connect(const GURL& url) override; |
| 40 virtual bool Send(const std::string& message) override; | 40 bool Send(const std::string& message) override; |
| 41 virtual StatusCode ReceiveNextMessage( | 41 StatusCode ReceiveNextMessage(std::string* message, |
| 42 std::string* message, | 42 const base::TimeDelta& timeout) override; |
| 43 const base::TimeDelta& timeout) override; | 43 bool HasNextMessage() override; |
| 44 virtual bool HasNextMessage() override; | |
| 45 | 44 |
| 46 private: | 45 private: |
| 47 struct CoreTraits; | 46 struct CoreTraits; |
| 48 class Core : public WebSocketListener, | 47 class Core : public WebSocketListener, |
| 49 public base::RefCountedThreadSafe<Core, CoreTraits> { | 48 public base::RefCountedThreadSafe<Core, CoreTraits> { |
| 50 public: | 49 public: |
| 51 explicit Core(net::URLRequestContextGetter* context_getter); | 50 explicit Core(net::URLRequestContextGetter* context_getter); |
| 52 | 51 |
| 53 bool IsConnected(); | 52 bool IsConnected(); |
| 54 bool Connect(const GURL& url); | 53 bool Connect(const GURL& url); |
| 55 bool Send(const std::string& message); | 54 bool Send(const std::string& message); |
| 56 SyncWebSocket::StatusCode ReceiveNextMessage( | 55 SyncWebSocket::StatusCode ReceiveNextMessage( |
| 57 std::string* message, | 56 std::string* message, |
| 58 const base::TimeDelta& timeout); | 57 const base::TimeDelta& timeout); |
| 59 bool HasNextMessage(); | 58 bool HasNextMessage(); |
| 60 | 59 |
| 61 // Overriden from WebSocketListener: | 60 // Overriden from WebSocketListener: |
| 62 virtual void OnMessageReceived(const std::string& message) override; | 61 void OnMessageReceived(const std::string& message) override; |
| 63 virtual void OnClose() override; | 62 void OnClose() override; |
| 64 | 63 |
| 65 private: | 64 private: |
| 66 friend class base::RefCountedThreadSafe<Core, CoreTraits>; | 65 friend class base::RefCountedThreadSafe<Core, CoreTraits>; |
| 67 friend class base::DeleteHelper<Core>; | 66 friend class base::DeleteHelper<Core>; |
| 68 friend struct CoreTraits; | 67 friend struct CoreTraits; |
| 69 | 68 |
| 70 virtual ~Core(); | 69 ~Core() override; |
| 71 | 70 |
| 72 void ConnectOnIO(const GURL& url, | 71 void ConnectOnIO(const GURL& url, |
| 73 bool* success, | 72 bool* success, |
| 74 base::WaitableEvent* event); | 73 base::WaitableEvent* event); |
| 75 void OnConnectCompletedOnIO(bool* connected, | 74 void OnConnectCompletedOnIO(bool* connected, |
| 76 base::WaitableEvent* event, | 75 base::WaitableEvent* event, |
| 77 int error); | 76 int error); |
| 78 void SendOnIO(const std::string& message, | 77 void SendOnIO(const std::string& message, |
| 79 bool* result, | 78 bool* result, |
| 80 base::WaitableEvent* event); | 79 base::WaitableEvent* event); |
| (...skipping 22 matching lines...) Expand all Loading... |
| 103 scoped_refptr<Core> core_; | 102 scoped_refptr<Core> core_; |
| 104 }; | 103 }; |
| 105 | 104 |
| 106 struct SyncWebSocketImpl::CoreTraits { | 105 struct SyncWebSocketImpl::CoreTraits { |
| 107 static void Destruct(const SyncWebSocketImpl::Core* core) { | 106 static void Destruct(const SyncWebSocketImpl::Core* core) { |
| 108 core->OnDestruct(); | 107 core->OnDestruct(); |
| 109 } | 108 } |
| 110 }; | 109 }; |
| 111 | 110 |
| 112 #endif // CHROME_TEST_CHROMEDRIVER_NET_SYNC_WEBSOCKET_IMPL_H_ | 111 #endif // CHROME_TEST_CHROMEDRIVER_NET_SYNC_WEBSOCKET_IMPL_H_ |
| OLD | NEW |