Task manager: Get physical memory efficiently for all processes from SharedSampler

chrome/browser/task_manager/sampling/shared_sampler.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 CHROME_BROWSER_TASK_MANAGER_SAMPLING_SHARED_SAMPLER_H_
 #define CHROME_BROWSER_TASK_MANAGER_SAMPLING_SHARED_SAMPLER_H_
 
 #include <map>
 #include <memory>
 #include <utility>
@@ skipping 22 matching lines @@
 // some types like Idle Wakeups can only be collected this way.
 class SharedSampler : public base::RefCountedThreadSafe<SharedSampler> {
  public:
   explicit SharedSampler(
       const scoped_refptr<base::SequencedTaskRunner>& blocking_pool_runner);
 
   // Below are the types of callbacks that are invoked on the UI thread
   // when the refresh is done on the worker thread.
   // These callbacks are passed via RegisterCallbacks.
   using OnIdleWakeupsCallback = base::Callback<void(int)>;
+  using OnPhysicalMemoryCallback = base::Callback<void(int64_t)>;
 
   // Returns a combination of refresh flags supported by the shared sampler.
   int64_t GetSupportedFlags() const;
 
   // Registers task group specific callbacks.
   void RegisterCallbacks(base::ProcessId process_id,
-                         const OnIdleWakeupsCallback& on_idle_wakeups);
+                         const OnIdleWakeupsCallback& on_idle_wakeups,
+                         const OnPhysicalMemoryCallback& on_physical_memory);
 
   // Unregisters task group specific callbacks.
   void UnregisterCallbacks(base::ProcessId process_id);
 
   // Refreshes the expensive process' stats (for now only idle wakeups per
   // second) on the worker thread.
   void Refresh(base::ProcessId process_id, int64_t refresh_flags);
 
  private:
   friend class base::RefCountedThreadSafe<SharedSampler>;
   ~SharedSampler();
 
 #if defined(OS_WIN)
   // The UI-thread callbacks in TaskGroup registered with RegisterCallbacks and
   // to be called when refresh on the worker thread is done.
   struct Callbacks {
     Callbacks();
     Callbacks(Callbacks&& other);
     ~Callbacks();
 
     OnIdleWakeupsCallback on_idle_wakeups;
+    OnPhysicalMemoryCallback on_physical_memory;
 
    private:
     DISALLOW_COPY_AND_ASSIGN(Callbacks);
   };
 
   typedef std::map<base::ProcessId, Callbacks> CallbacksMap;
 
   // Contains all results of refresh for a single process.
   struct RefreshResult {
     base::ProcessId process_id;
     int idle_wakeups_per_second;
+    int64_t physical_bytes;
   };
 
   typedef std::vector<RefreshResult> RefreshResults;
 
   // Posted on the worker thread to do the actual refresh.
   std::unique_ptr<RefreshResults> RefreshOnWorkerThread();
 
   // Called on UI thread when the refresh is done.
   void OnRefreshDone(std::unique_ptr<RefreshResults> refresh_results);
 
@@ skipping 43 matching lines @@
   // To assert we're running on the correct thread.
   base::SequenceChecker worker_pool_sequenced_checker_;
 #endif  // defined(OS_WIN)
 
   DISALLOW_COPY_AND_ASSIGN(SharedSampler);
 };
 
 }  // namespace task_manager
 
 #endif  // CHROME_BROWSER_TASK_MANAGER_SAMPLING_SHARED_SAMPLER_H_