| // Copyright (c) 2013 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 CONTENT_BROWSER_LOADER_WAKE_LOCK_RESOURCE_THROTTLE_H_ |
| #define CONTENT_BROWSER_LOADER_WAKE_LOCK_RESOURCE_THROTTLE_H_ |
| |
| #include <memory> |
| #include <string> |
| |
| #include "base/compiler_specific.h" |
| #include "base/macros.h" |
| #include "base/timer/timer.h" |
| #include "content/public/browser/resource_throttle.h" |
| #include "services/device/public/mojom/wake_lock.mojom.h" |
| |
| namespace content { |
| |
| // This ResourceThrottle holds wake lock until large upload request finishes. |
| class WakeLockResourceThrottle : public ResourceThrottle { |
| public: |
| WakeLockResourceThrottle(const std::string& host); |
| ~WakeLockResourceThrottle() override; |
| |
| // ResourceThrottle overrides: |
| void WillStartRequest(bool* defer) override; |
| void WillProcessResponse(bool* defer) override; |
| const char* GetNameForLogging() const override; |
| |
| private: |
| void RequestWakeLock(); |
| |
| const std::string host_; |
| base::OneShotTimer timer_; |
| |
| // Destruction of wake_lock_ will trigger |
| // WakeLock::OnConnectionError on the service side, so there is no |
| // need to call CancelWakeLock() in the destructor. |
| device::mojom::WakeLockPtr wake_lock_; |
| |
| DISALLOW_COPY_AND_ASSIGN(WakeLockResourceThrottle); |
| }; |
| |
| } // namespace content |
| |
| #endif // CONTENT_BROWSER_LOADER_WAKE_LOCK_RESOURCE_THROTTLE_H_ |