Implement UMA log throttling for cellular connections

components/metrics/data_use_tracker.h

+// Copyright 2016 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 COMPONENTS_METRICS_DATA_USE_TRACKER_H_
+#define COMPONENTS_METRICS_DATA_USE_TRACKER_H_
+
+#include "base/callback.h"
+#include "base/gtest_prod_util.h"
+#include "base/memory/weak_ptr.h"
+#include "components/prefs/pref_registry_simple.h"
+#include "components/prefs/pref_service.h"
+
+namespace metrics {
+
+typedef base::Callback<void(const std::string&, int)>
+    UpdateUsagePrefCallbackType;
+
+class DataUseTracker {

[Alexei Svitkine (slow), 2016/03/31 04:32:25]

Add a comment.

[gayane -on leave until 09-2017, 2016/03/31 17:54:58]

Done.

+ public:
+  // Returns a callback to data use pref updating function attached to
+  // |data_use_tracker| weak pointer. Should be called on UI thread.
+  static UpdateUsagePrefCallbackType GetForwardingCallback(
+      base::WeakPtr<DataUseTracker> data_use_tracker,
+      scoped_refptr<base::SequencedTaskRunner> ui_task_runner);
+
+  // Registers data use prefs using provided |registry|.
+  static void RegisterPrefs(PrefRegistrySimple* registry);
+
+  // Returns  whether a log with provided |log_bytes| can be uploaded according
+  // to data use ratio and UMA quota provided by finch experiment.
+  bool CanUploadUMALog(int log_bytes);
+
+  base::WeakPtr<DataUseTracker> GetWeakPtr();
+
+  DataUseTracker(PrefService* local_state);

[Alexei Svitkine (slow), 2016/03/31 04:32:25]

explicit

[gayane -on leave until 09-2017, 2016/03/31 17:54:58]

Done.

+  ~DataUseTracker();

[Alexei Svitkine (slow), 2016/03/31 04:32:25]

Put ctor and dtor at the top.

[gayane -on leave until 09-2017, 2016/03/31 17:54:58]

Done.

+
+ private:
+  FRIEND_TEST_ALL_PREFIXES(DataUseTrackerTest, CheckUpdateUsagePref);
+  FRIEND_TEST_ALL_PREFIXES(DataUseTrackerTest, CheckRemoveExpiredEntries);
+  FRIEND_TEST_ALL_PREFIXES(DataUseTrackerTest, CheckComputeTotalDataUse);
+  FRIEND_TEST_ALL_PREFIXES(DataUseTrackerTest, CheckCanUploadUMALog);
+
+  // Updates data usage prefs on UI thread according to what Prefservice
+  // expects.
+  void UpdateMetricsUsagePrefsOnUIThread(const std::string& service_name,
+                                         int message_size);
+
+  // Updates provided |pref_name| for a current date with the given message
+  // size.
+  void UpdateUsagePref(const std::string& pref_name, int message_size);
+
+  // Removes entries from the all data use  prefs.
+  void RemoveExpiredEntries();
+
+  // Removes entries from the given |pref_name| if they are more than 7 days
+  // old.
+  void RemoveExpiredEntriesForPref(const std::string& pref_name);
+
+  // Returns the current date as a string with a proper formatting.
+  std::string GetCurrentMeasurementDate();
+
+  // Computes data usage according to all the entries in the given dictionary
+  // pref.
+  int ComputeTotalDataUse(std::string pref_name);
+
+  bool GetUmaQuota(int* quota);
+  bool GetUmaRatio(double* ratio);
+
+  // Only for testing.
+  void SetMeasurementDateForTesting(const std::string& date);
+  void SetUmaQuotaForTesting(int uma_quota);
+  void SetUmaRatioForTesting(double uma_ratio);
+
+  PrefService* local_state_;
+
+  std::string date_for_testing_;
+  int uma_quota_for_testing_;
+  double uma_ratio_for_testing_;
+
+  base::WeakPtrFactory<DataUseTracker> weak_ptr_factory_;
+};
+
+}  // namespace metrics
+#endif  // COMPONENTS_METRICS_DATA_USE_TRACKER_H_