1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
// Copyright 2018 The Chromium Authors
// 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_ARC_SHARED_SAMPLER_H_
#define CHROME_BROWSER_TASK_MANAGER_SAMPLING_ARC_SHARED_SAMPLER_H_
#include <stdint.h>
#include <map>
#include <optional>
#include "base/containers/flat_map.h"
#include "base/functional/callback.h"
#include "base/memory/weak_ptr.h"
#include "base/process/process_handle.h"
#include "base/time/time.h"
#include "chromeos/ash/experiences/arc/mojom/process.mojom.h"
namespace task_manager {
// Defines sampler that will retrieve memory footprint metrics for all arc
// processes at once. Created by TaskManagerImpl on the UI thread.
class ArcSharedSampler {
public:
ArcSharedSampler();
ArcSharedSampler(const ArcSharedSampler&) = delete;
ArcSharedSampler& operator=(const ArcSharedSampler&) = delete;
~ArcSharedSampler();
using MemoryFootprintBytes = uint64_t;
using OnSamplingCompleteCallback =
base::RepeatingCallback<void(std::optional<MemoryFootprintBytes>)>;
// Registers task group specific callback.
void RegisterCallback(base::ProcessId process_id,
OnSamplingCompleteCallback on_sampling_complete);
// Unregisters task group specific callbacks.
void UnregisterCallback(base::ProcessId process_id);
// Triggers a refresh of process stats.
void Refresh();
private:
using CallbacksMap =
base::flat_map<base::ProcessId, OnSamplingCompleteCallback>;
// Called when ArcProcessService returns memory dump.
void OnReceiveMemoryDump(
int dump_type,
std::vector<arc::mojom::ArcMemoryDumpPtr> process_dumps);
// Holds callbacks registered by TaskGroup objects.
CallbacksMap callbacks_;
// Keeps track of whether there is a pending request for memory footprint of
// app or system processes.
int pending_memory_dump_types_ = 0;
// The timestamp of when the last refresh call finished, for system and
// app processes.
base::Time last_system_refresh_;
base::Time last_app_refresh_;
base::WeakPtrFactory<ArcSharedSampler> weak_ptr_factory_{this};
};
} // namespace task_manager
#endif // CHROME_BROWSER_TASK_MANAGER_SAMPLING_ARC_SHARED_SAMPLER_H_
|
