From b65426ac3a6cc91a4f11d138e2df19005fd1aefa Mon Sep 17 00:00:00 2001
From: cobalt-github-releaser-bot
 <95661244+cobalt-github-releaser-bot@users.noreply.github.com>
Date: Mon, 19 Aug 2024 11:14:17 -0700
Subject: [PATCH] Cherry pick PR #3967: Add setting getter for CPU interval
 definition. (#4015)

Refer to the original PR: https://github.com/youtube/cobalt/pull/3967

b/341774149

Co-authored-by: aee <aee@google.com>
---
 .../testdata/cpu_usage_tracker_test.html      | 42 +++++++++++++++++++
 cobalt/browser/cpu_usage_tracker.cc           |  6 +++
 cobalt/browser/cpu_usage_tracker.h            |  1 +
 cobalt/h5vcc/h5vcc_settings.cc                | 11 +++++
 cobalt/h5vcc/h5vcc_settings.h                 |  2 +
 cobalt/h5vcc/h5vcc_settings.idl               |  1 +
 6 files changed, 63 insertions(+)
diff --git a/cobalt/black_box_tests/testdata/cpu_usage_tracker_test.html b/cobalt/black_box_tests/testdata/cpu_usage_tracker_test.html
index 7053f0bbaf3a..e790e747db4b 100644
--- a/cobalt/black_box_tests/testdata/cpu_usage_tracker_test.html
+++ b/cobalt/black_box_tests/testdata/cpu_usage_tracker_test.html
@@ -20,6 +20,24 @@
 </head>
 <body>
   <script>
+const assertObjectsEqual = (expected, actual, msg='') => {
+  const expectedType = typeof(expected);
+  const actualType = typeof(actual);
+  assertEqual(expectedType, actualType, `types match, (${msg})`);
+
+  if (expectedType === 'object') {
+    const expectedKeys = new Set(Object.keys(expected));
+    const actualKeys = new Set(Object.keys(actual));
+    assertEqual(expectedKeys.size, actualKeys.size, `key count matches, (${msg})`);
+    expectedKeys.forEach(key => {
+      assertTrue(actualKeys.has(key), `key ${key} exists, (${msg})`);
+      assertObjectsEqual(expected[key], actual[key], `value for ${key} matches, (${msg})`);
+    });
+  } else {
+    assertEqual(expected, actual, `simple match, (${msg})`);
+  }
+};
+
 const delay = durationMs =>
     new Promise(resolve => {
       setTimeout(resolve, durationMs);
@@ -31,6 +49,16 @@
 // Set invalid config.
 window.h5vcc.settings.set('cpu_usage_tracker_intervals', '');
 delay(2500).then(() => {
+  const defaultSetting = JSON.parse(
+      window.h5vcc.settings.getPersistentSettingAsString(
+          'cpu_usage_tracker_intervals'));
+  assertObjectsEqual([
+    {type: 'per_thread', seconds: 2},
+    {type: 'per_thread', seconds: 30},
+    {type: 'total', seconds: 2},
+    {type: 'total', seconds: 30},
+  ], defaultSetting);
+
   const cvalKeys = window.h5vcc.cVal.keys().filter(k => k.startsWith('CPU.'));
   // Check for default CVals.
   assertIncludes('CPU.Total.Usage.IntervalSeconds.2', cvalKeys);
@@ -74,6 +102,20 @@
   window.h5vcc.settings.set('cpu_usage_tracker_intervals', JSON.stringify(newConfig));
   return delay(1500);
 }).then(() => {
+  const newSetting = JSON.parse(
+      window.h5vcc.settings.getPersistentSettingAsString(
+          'cpu_usage_tracker_intervals'));
+  assertObjectsEqual([
+    {type: 'total', seconds: 1},
+    {type: 'total', seconds: 3},
+    {type: 'total', seconds: 4},
+    {type: 'total', seconds: 120},
+    {type: 'per_thread', seconds: 1},
+    {type: 'per_thread', seconds: 3},
+    {type: 'per_thread', seconds: 5},
+    {type: 'per_thread', seconds: 60},
+  ], newSetting);
+
   const cvalKeys = window.h5vcc.cVal.keys().filter(k => k.startsWith('CPU.'));
   assertIncludes('CPU.Total.Usage.IntervalSeconds.1', cvalKeys);
   assertIncludes('CPU.Total.Usage.IntervalSeconds.3', cvalKeys);
diff --git a/cobalt/browser/cpu_usage_tracker.cc b/cobalt/browser/cpu_usage_tracker.cc
index 36ad4b85649d..d53e4cc6385f 100644
--- a/cobalt/browser/cpu_usage_tracker.cc
+++ b/cobalt/browser/cpu_usage_tracker.cc
@@ -300,6 +300,12 @@ void CpuUsageTracker::UpdateIntervalsEnabled(bool enabled) {
   }
 }
 
+base::Value CpuUsageTracker::GetIntervalsDefinition() {
+  base::Value intervals;
+  storage_->Get(kIntervals, &intervals);
+  return std::move(intervals);
+}
+
 void CpuUsageTracker::UpdateIntervalsDefinition(
     const base::Value& intervals_value) {
   if (!task_runner_->RunsTasksInCurrentSequence()) {
diff --git a/cobalt/browser/cpu_usage_tracker.h b/cobalt/browser/cpu_usage_tracker.h
index 85c03380fb22..8ccdb8f48e4d 100644
--- a/cobalt/browser/cpu_usage_tracker.h
+++ b/cobalt/browser/cpu_usage_tracker.h
@@ -40,6 +40,7 @@ class CpuUsageTracker : base::CurrentThread::DestructionObserver {
   static CpuUsageTracker* GetInstance();
 
   void Initialize(persistent_storage::PersistentSettings*);
+  base::Value GetIntervalsDefinition();
   void UpdateIntervalsDefinition(const base::Value&);
   void UpdateIntervalsEnabled(bool);
   void StartOneTimeTracking();
diff --git a/cobalt/h5vcc/h5vcc_settings.cc b/cobalt/h5vcc/h5vcc_settings.cc
index 0cb5ca7ef878..3c6062001737 100644
--- a/cobalt/h5vcc/h5vcc_settings.cc
+++ b/cobalt/h5vcc/h5vcc_settings.cc
@@ -185,5 +185,16 @@ int H5vccSettings::GetPersistentSettingAsInt(const std::string& key,
   return default_setting;
 }
 
+std::string H5vccSettings::GetPersistentSettingAsString(
+    const std::string& key) const {
+  if (key.compare("cpu_usage_tracker_intervals") == 0) {
+    base::Value value =
+        browser::CpuUsageTracker::GetInstance()->GetIntervalsDefinition();
+    absl::optional<std::string> json = base::WriteJson(value);
+    return json.value_or(std::string());
+  }
+  return std::string();
+}
+
 }  // namespace h5vcc
 }  // namespace cobalt
diff --git a/cobalt/h5vcc/h5vcc_settings.h b/cobalt/h5vcc/h5vcc_settings.h
index e1541ebd48b5..972f3a79ec13 100644
--- a/cobalt/h5vcc/h5vcc_settings.h
+++ b/cobalt/h5vcc/h5vcc_settings.h
@@ -63,6 +63,8 @@ class H5vccSettings : public script::Wrappable {
   int GetPersistentSettingAsInt(const std::string& key,
                                 int default_setting) const;
 
+  std::string GetPersistentSettingAsString(const std::string& key) const;
+
   DEFINE_WRAPPABLE_TYPE(H5vccSettings);
 
  private:
diff --git a/cobalt/h5vcc/h5vcc_settings.idl b/cobalt/h5vcc/h5vcc_settings.idl
index 703e483897c4..10223ffbc047 100644
--- a/cobalt/h5vcc/h5vcc_settings.idl
+++ b/cobalt/h5vcc/h5vcc_settings.idl
@@ -16,4 +16,5 @@ interface H5vccSettings {
   boolean set(DOMString name, (long or DOMString) value);
   void setPersistentSettingAsInt(DOMString name, long value);
   long getPersistentSettingAsInt(DOMString name, long default_setting);
+  DOMString getPersistentSettingAsString(DOMString name);
 };
