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_IO_COLLECTOR_IMPL_H
17 #define HIVIEW_FRAMEWORK_NATIVE_UNIFIED_COLLECTION_IO_COLLECTOR_IMPL_H
18 
19 #include <mutex>
20 
21 #include "io_collector.h"
22 
23 namespace OHOS {
24 namespace HiviewDFX {
25 namespace UCollectUtil {
26 class IoCollectorImpl : public IoCollector {
27 public:
28     IoCollectorImpl();
29     virtual ~IoCollectorImpl() = default;
30 
31 public:
32     virtual CollectResult<ProcessIo> CollectProcessIo(int32_t pid) override;
33     virtual CollectResult<std::string> CollectRawDiskStats() override;
34     virtual CollectResult<std::vector<DiskStats>> CollectDiskStats(
35         DiskStatsFilter filter = DefaultDiskStatsFilter, bool isUpdate = false) override;
36     virtual CollectResult<std::string> ExportDiskStats(DiskStatsFilter filter = DefaultDiskStatsFilter) override;
37     virtual CollectResult<std::vector<EMMCInfo>> CollectEMMCInfo() override;
38     virtual CollectResult<std::string> ExportEMMCInfo() override;
39     virtual CollectResult<std::vector<ProcessIoStats>> CollectAllProcIoStats(bool isUpdate = false) override;
40     virtual CollectResult<std::string> ExportAllProcIoStats() override;
41     virtual CollectResult<SysIoStats> CollectSysIoStats() override;
42     virtual CollectResult<std::string> ExportSysIoStats() override;
43 
44 private:
45     void InitDiskData();
46     void GetDiskStats(DiskStatsFilter filter, bool isUpdate, std::vector<DiskStats>& diskStats);
47     void CalculateDiskStats(uint64_t period, bool isUpdate);
48     void CalculateDeviceDiskStats(const DiskData& currData, const std::string& deviceName, uint64_t period);
49     void CalculateEMMCInfo(std::vector<EMMCInfo>& mmcInfos);
50     void ReadEMMCInfo(const std::string& path, std::vector<EMMCInfo>& mmcInfos);
51     std::string GetEMMCPath(const std::string& path);
52     void InitProcIoData();
53     void GetProcIoStats(std::vector<ProcessIoStats>& allProcIoStats, bool isUpdate);
54     void CalculateAllProcIoStats(uint64_t period, bool isUpdate);
55     void CalculateProcIoStats(const ProcessIo& currData, int32_t pid, uint64_t period);
56     bool ProcIoStatsFilter(const ProcessIoStats& stats);
57     int32_t GetProcStateInCollectionPeriod(int32_t pid);
58     std::string CreateExportFileName(const std::string& filePrefix);
59 
60 private:
61     std::mutex collectDiskMutex_;
62     std::mutex collectProcIoMutex_;
63     std::mutex exportFileMutex_;
64     uint64_t preCollectDiskTime_ = 0;
65     uint64_t preCollectProcIoTime_ = 0;
66     uint64_t currCollectProcIoTime_ = 0;
67     std::unordered_map<std::string, DiskStatsDevice> diskStatsMap_;
68     std::unordered_map<int32_t, ProcessIoStatsInfo> procIoStatsMap_;
69 };
70 } // namespace UCollectUtil
71 } // namespace HiviewDFX
72 } // namespace OHOS
73 #endif // HIVIEW_FRAMEWORK_NATIVE_UNIFIED_COLLECTION_IO_COLLECTOR_IMPL_H