1 /*
2  * Copyright (c) 2023 Huawei Device Co., Ltd.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at
6  *
7  *     http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15 
16 #ifndef HIVIEW_FRAMEWORK_NATIVE_UNIFIED_COLLECTION_MEM_PROFILER_DECORATOR_H
17 #define HIVIEW_FRAMEWORK_NATIVE_UNIFIED_COLLECTION_MEM_PROFILER_DECORATOR_H
18 
19 #include "mem_profiler_collector.h"
20 #include "decorator.h"
21 
22 namespace OHOS {
23 namespace HiviewDFX {
24 namespace UCollectUtil {
25 class MemProfilerDecorator : public MemProfilerCollector, public UCDecorator {
26 public:
MemProfilerDecorator(std::shared_ptr<MemProfilerCollector> collector)27     MemProfilerDecorator(std::shared_ptr<MemProfilerCollector> collector) : memProfilerCollector_(collector) {};
28     virtual ~MemProfilerDecorator() = default;
29     int Start(ProfilerType type, int pid, int duration, int sampleInterval) override;
30     int Stop(int pid) override;
31     int Stop(const std::string& processName) override;
32     int Start(int fd, ProfilerType type, int pid, int duration, int sampleInterval) override;
33     int StartPrintNmd(int fd, int pid, int type) override;
34     int Start(int fd, ProfilerType type, std::string processName, int duration, int sampleInterval,
35               bool startup = false) override;
36     int Prepare() override;
37 
38     void GenerateStatInfo(uint64_t startTime, const std::string& funcName, int result);
39     static void SaveStatCommonInfo();
40     static void ResetStatInfo();
41 
42 private:
43     std::shared_ptr<MemProfilerCollector> memProfilerCollector_;
44     static StatInfoWrapper statInfoWrapper_;
45 };
46 } // namespace UCollectUtil
47 } // namespace HiviewDFX
48 } // namespace OHOS
49 #endif // HIVIEW_FRAMEWORK_NATIVE_UNIFIED_COLLECTION_MEM_PROFILER_DECORATOR_H
50