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 #ifndef INTERFACES_INNER_API_UNIFIED_COLLECTION_UTILITY_IO_COLLECTOR_H
16 #define INTERFACES_INNER_API_UNIFIED_COLLECTION_UTILITY_IO_COLLECTOR_H
17 #include <cinttypes>
18 #include <memory>
19 
20 #include "collect_result.h"
21 #include "resource/io.h"
22 
23 namespace OHOS {
24 namespace HiviewDFX {
25 namespace UCollectUtil {
26 using DiskStatsFilter = std::function<bool(const DiskStats &)>;
DefaultDiskStatsFilter(const DiskStats & stats)27 static bool DefaultDiskStatsFilter(const DiskStats &stats)
28 {
29     return (stats.operReadRate == 0 && stats.operWriteRate == 0);
30 }
31 
32 class IoCollector {
33 public:
34     IoCollector() = default;
35     virtual ~IoCollector() = default;
36 
37 public:
38     virtual CollectResult<ProcessIo> CollectProcessIo(int32_t pid) = 0;
39     virtual CollectResult<std::string> CollectRawDiskStats() = 0;
40     virtual CollectResult<std::vector<DiskStats>> CollectDiskStats(
41         DiskStatsFilter filter = DefaultDiskStatsFilter, bool isUpdate = false) = 0;
42     virtual CollectResult<std::string> ExportDiskStats(DiskStatsFilter filter = DefaultDiskStatsFilter) = 0;
43     virtual CollectResult<std::vector<EMMCInfo>> CollectEMMCInfo() = 0;
44     virtual CollectResult<std::string> ExportEMMCInfo() = 0;
45     virtual CollectResult<std::vector<ProcessIoStats>> CollectAllProcIoStats(bool isUpdate = false) = 0;
46     virtual CollectResult<std::string> ExportAllProcIoStats() = 0;
47     virtual CollectResult<SysIoStats> CollectSysIoStats() = 0;
48     virtual CollectResult<std::string> ExportSysIoStats() = 0;
49     static std::shared_ptr<IoCollector> Create();
50 }; // IoCollector
51 } // UCollectUtil
52 } // HiviewDFX
53 } // OHOS
54 #endif // INTERFACES_INNER_API_UNIFIED_COLLECTION_UTILITY_IO_COLLECTOR_H