naiveproxy/src/net/nqe/connectivity_monitor.h
2021-09-02 00:21:20 +08:00

124 lines
4.9 KiB
C++

// Copyright 2020 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.
// Copyright 2020 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 NET_NQE_CONNECTIVITY_MONITOR_H_
#define NET_NQE_CONNECTIVITY_MONITOR_H_
#include <memory>
#include <set>
#include "base/callback.h"
#include "base/cancelable_callback.h"
#include "base/time/time.h"
#include "build/build_config.h"
#include "net/base/net_export.h"
#include "net/base/network_change_notifier.h"
#include "net/url_request/url_request.h"
#include "third_party/abseil-cpp/absl/types/optional.h"
namespace net {
#if defined(OS_ANDROID)
namespace android {
class NetworkActivationRequest;
}
#endif // defined(OS_ANDROID)
// ConnectivityMonitor is driven by NetworkQualityEstimator and is used to
// monitor progress of active URLRequests. If all active requests fail to make
// progress for a certain time interval, this will log accordingly and may
// report the problem to the operating system as a potential hint to fall back
// onto a more responsive network.
class NET_EXPORT_PRIVATE ConnectivityMonitor
: public NetworkChangeNotifier::NetworkObserver {
public:
// Constructs a new ConnectivityMonitor as below, but with builtin default
// TimeDelta values.
explicit ConnectivityMonitor();
// Constructs a new ConnectivityMonitor which assumes the current network has
// lost connectivity if it observes no request progress over a duration of at
// least `inactivity_threshold`. This observation will only occur at most once
// every `min_observation_interval`.
ConnectivityMonitor(base::TimeDelta inactivity_threshold,
base::TimeDelta min_failure_logging_interval);
ConnectivityMonitor(const ConnectivityMonitor&) = delete;
ConnectivityMonitor& operator=(const ConnectivityMonitor&) = delete;
~ConnectivityMonitor() override;
// Registers a new `request` to be tracked by the ConnectivityMonitor. Called
// just before the request's first header bytes hit the wire.
void TrackNewRequest(const URLRequest& request);
// Notifies the ConnectivityMonitor that progress has been made toward
// `request` completion. This means that some response bytes were received,
// and for a newly tracked request, the first call to this method signifies
// receipt of at least the first response header bytes.
void NotifyRequestProgress(const URLRequest& request);
// Indicates that `request` has been completed or is about to be destroyed,
// regardless of success or failure. If `request` was being tracked by this
// ConnectivityMonitor, it must no longer be tracked after this call.
void NotifyRequestCompleted(const URLRequest& request);
// Notifies the monitor of a change in the system's network configuration. As
// an example, this may be called when an Android device switches its default
// network from WiFi to mobile data.
void NotifyConnectionTypeChanged(NetworkChangeNotifier::ConnectionType type);
size_t num_active_requests_for_testing() const {
return active_requests_.size();
}
// Returns the amount of time since the ConnectivityMonitor first observed the
// current lapse in connectivity, if any.
absl::optional<base::TimeDelta> GetTimeSinceLastFailureForTesting();
// Registers a callback to hook into any time an activity deadline is reached.
void SetNextDeadlineCallbackForTesting(base::OnceClosure callback);
// Registers a callback to hook into the code path for OS reporting. Allows
// tests to effectively observe the OS reporting event.
void SetReportCallbackForTesting(base::OnceClosure callback);
private:
void ScheduleNextActivityDeadline(base::TimeDelta delay);
void OnActivityDeadlineExceeded();
void ReportConnectivityFailure();
void RequestMobileNetworkActivation();
// NetworkChangeNotifier::NetworkObserver:
void OnNetworkConnected(
NetworkChangeNotifier::NetworkHandle network) override;
void OnNetworkDisconnected(
NetworkChangeNotifier::NetworkHandle network) override;
void OnNetworkSoonToDisconnect(
NetworkChangeNotifier::NetworkHandle network) override;
void OnNetworkMadeDefault(
NetworkChangeNotifier::NetworkHandle network) override;
const base::TimeDelta inactivity_threshold_;
const base::TimeDelta min_failure_logging_interval_;
base::OnceClosure next_deadline_callback_for_testing_;
base::OnceClosure report_callback_for_testing_;
std::set<const URLRequest*> active_requests_;
base::CancelableOnceClosure next_activity_deadline_;
absl::optional<base::TimeTicks> time_last_failure_observed_;
NetworkChangeNotifier::ConnectionType current_connection_type_;
#if defined(OS_ANDROID)
std::unique_ptr<android::NetworkActivationRequest> mobile_network_request_;
#endif
};
} // namespace net
#endif // NET_NQE_CONNECTIVITY_MONITOR_H_