| // Copyright 2018 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_COMMON_HEAP_PROFILER_CONTROLLER_H_ |
| #define CHROME_COMMON_HEAP_PROFILER_CONTROLLER_H_ |
| |
| #include "base/memory/weak_ptr.h" |
| |
| namespace base { |
| class TaskRunner; |
| } // namespace base |
| |
| // HeapProfilerController controls collection of sampled heap allocation |
| // snapshots for the current process. |
| class HeapProfilerController { |
| public: |
| HeapProfilerController(); |
| ~HeapProfilerController(); |
| |
| // Starts periodic heap snapshot collection. |
| void StartIfEnabled(); |
| |
| void SetTaskRunnerForTest(base::TaskRunner* task_runner) { |
| task_runner_for_test_ = task_runner; |
| } |
| |
| private: |
| void ScheduleNextSnapshot(); |
| void TakeSnapshot(); |
| void RetrieveAndSendSnapshot(); |
| |
| bool started_ = false; |
| base::TaskRunner* task_runner_for_test_ = nullptr; |
| base::WeakPtrFactory<HeapProfilerController> weak_factory_; |
| |
| DISALLOW_COPY_AND_ASSIGN(HeapProfilerController); |
| }; |
| |
| #endif // CHROME_COMMON_HEAP_PROFILER_CONTROLLER_H_ |