| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 HEADLESS_PUBLIC_UTIL_DETERMINISTIC_HTTP_PROTOCOL_HANDLER_H_ | 5 #ifndef HEADLESS_PUBLIC_UTIL_DETERMINISTIC_HTTP_PROTOCOL_HANDLER_H_ |
| 6 #define HEADLESS_PUBLIC_UTIL_DETERMINISTIC_HTTP_PROTOCOL_HANDLER_H_ | 6 #define HEADLESS_PUBLIC_UTIL_DETERMINISTIC_HTTP_PROTOCOL_HANDLER_H_ |
| 7 | 7 |
| 8 #include <memory> | 8 #include <memory> |
| 9 | 9 |
| 10 #include "base/single_thread_task_runner.h" | 10 #include "base/single_thread_task_runner.h" |
| 11 #include "headless/public/headless_export.h" | 11 #include "headless/public/headless_export.h" |
| 12 #include "net/url_request/url_request_job_factory.h" | 12 #include "net/url_request/url_request_job_factory.h" |
| 13 | 13 |
| 14 namespace net { | 14 namespace net { |
| 15 class URLRequestContext; | 15 class URLRequestContext; |
| 16 class URLRequestJobFactory; | 16 class URLRequestJobFactory; |
| 17 } // namespace | 17 } // namespace |
| 18 | 18 |
| 19 namespace headless { | 19 namespace headless { |
| 20 class DeterministicDispatcher; | 20 class DeterministicDispatcher; |
| 21 class HeadlessBrowserContext; |
| 21 | 22 |
| 22 // A deterministic protocol handler. Requests made to this protocol handler | 23 // A deterministic protocol handler. Requests made to this protocol handler |
| 23 // will return in order of creation, regardless of what order the network | 24 // will return in order of creation, regardless of what order the network |
| 24 // returns them in. This helps remove one large source of network related | 25 // returns them in. This helps remove one large source of network related |
| 25 // non determinism at the cost of slower page loads. | 26 // non determinism at the cost of slower page loads. |
| 26 class HEADLESS_EXPORT DeterministicHttpProtocolHandler | 27 class HEADLESS_EXPORT DeterministicHttpProtocolHandler |
| 27 : public net::URLRequestJobFactory::ProtocolHandler { | 28 : public net::URLRequestJobFactory::ProtocolHandler { |
| 28 public: | 29 public: |
| 29 // Note |deterministic_dispatcher| is expected to be shared across a number of | 30 // Note |deterministic_dispatcher| is expected to be shared across a number of |
| 30 // protocol handlers, e.g. for http & https protocols. | 31 // protocol handlers, e.g. for http & https protocols. |
| 31 DeterministicHttpProtocolHandler( | 32 DeterministicHttpProtocolHandler( |
| 32 DeterministicDispatcher* deterministic_dispatcher, | 33 DeterministicDispatcher* deterministic_dispatcher, |
| 33 scoped_refptr<base::SingleThreadTaskRunner> io_task_runner); | 34 scoped_refptr<base::SingleThreadTaskRunner> io_task_runner); |
| 34 ~DeterministicHttpProtocolHandler() override; | 35 ~DeterministicHttpProtocolHandler() override; |
| 35 | 36 |
| 36 net::URLRequestJob* MaybeCreateJob( | 37 net::URLRequestJob* MaybeCreateJob( |
| 37 net::URLRequest* request, | 38 net::URLRequest* request, |
| 38 net::NetworkDelegate* network_delegate) const override; | 39 net::NetworkDelegate* network_delegate) const override; |
| 39 | 40 |
| 41 void SetHeadlessBrowserContext( |
| 42 HeadlessBrowserContext* headless_browser_context) { |
| 43 headless_browser_context_ = headless_browser_context; |
| 44 } |
| 45 |
| 40 private: | 46 private: |
| 41 class NopGenericURLRequestJobDelegate; | 47 class NopGenericURLRequestJobDelegate; |
| 42 | 48 |
| 43 DeterministicDispatcher* deterministic_dispatcher_; // NOT OWNED. | 49 DeterministicDispatcher* deterministic_dispatcher_; // NOT OWNED. |
| 50 HeadlessBrowserContext* headless_browser_context_; // NOT OWNED. |
| 44 scoped_refptr<base::SingleThreadTaskRunner> io_task_runner_; | 51 scoped_refptr<base::SingleThreadTaskRunner> io_task_runner_; |
| 45 std::unique_ptr<NopGenericURLRequestJobDelegate> nop_delegate_; | 52 std::unique_ptr<NopGenericURLRequestJobDelegate> nop_delegate_; |
| 46 | 53 |
| 47 // |url_request_context_| and |url_request_job_factory_| are lazily created on | 54 // |url_request_context_| and |url_request_job_factory_| are lazily created on |
| 48 // the IO thread. The URLRequestContext is setup to bypass any user-specified | 55 // the IO thread. The URLRequestContext is setup to bypass any user-specified |
| 49 // protocol handlers including this one. This is necessary to actually fetch | 56 // protocol handlers including this one. This is necessary to actually fetch |
| 50 // http resources. | 57 // http resources. |
| 51 mutable std::unique_ptr<net::URLRequestContext> url_request_context_; | 58 mutable std::unique_ptr<net::URLRequestContext> url_request_context_; |
| 52 mutable std::unique_ptr<net::URLRequestJobFactory> url_request_job_factory_; | 59 mutable std::unique_ptr<net::URLRequestJobFactory> url_request_job_factory_; |
| 53 | 60 |
| 54 DISALLOW_COPY_AND_ASSIGN(DeterministicHttpProtocolHandler); | 61 DISALLOW_COPY_AND_ASSIGN(DeterministicHttpProtocolHandler); |
| 55 }; | 62 }; |
| 56 | 63 |
| 57 } // namespace headless | 64 } // namespace headless |
| 58 | 65 |
| 59 #endif // HEADLESS_PUBLIC_UTIL_DETERMINISTIC_HTTP_PROTOCOL_HANDLER_H_ | 66 #endif // HEADLESS_PUBLIC_UTIL_DETERMINISTIC_HTTP_PROTOCOL_HANDLER_H_ |
| OLD | NEW |