1 /*
2 * Copyright (c) 2023-2024 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 #include "io_decorator.h"
17
18 namespace OHOS {
19 namespace HiviewDFX {
20 namespace UCollectUtil {
21 const std::string IO_COLLECTOR_NAME = "IoCollector";
22 StatInfoWrapper IoDecorator::statInfoWrapper_;
23
CollectProcessIo(int32_t pid)24 CollectResult<ProcessIo> IoDecorator::CollectProcessIo(int32_t pid)
25 {
26 auto task = [this, &pid] { return ioCollector_->CollectProcessIo(pid); };
27 return Invoke(task, statInfoWrapper_, IO_COLLECTOR_NAME + UC_SEPARATOR + __func__);
28 }
29
CollectRawDiskStats()30 CollectResult<std::string> IoDecorator::CollectRawDiskStats()
31 {
32 auto task = [this] { return ioCollector_->CollectRawDiskStats(); };
33 return Invoke(task, statInfoWrapper_, IO_COLLECTOR_NAME + UC_SEPARATOR + __func__);
34 }
35
CollectDiskStats(DiskStatsFilter filter,bool isUpdate)36 CollectResult<std::vector<DiskStats>> IoDecorator::CollectDiskStats(
37 DiskStatsFilter filter, bool isUpdate)
38 {
39 auto task = [this, &filter, &isUpdate] { return ioCollector_->CollectDiskStats(filter, isUpdate); };
40 return Invoke(task, statInfoWrapper_, IO_COLLECTOR_NAME + UC_SEPARATOR + __func__);
41 }
42
ExportDiskStats(DiskStatsFilter filter)43 CollectResult<std::string> IoDecorator::ExportDiskStats(DiskStatsFilter filter)
44 {
45 auto task = [this, &filter] { return ioCollector_->ExportDiskStats(filter); };
46 return Invoke(task, statInfoWrapper_, IO_COLLECTOR_NAME + UC_SEPARATOR + __func__);
47 }
48
CollectEMMCInfo()49 CollectResult<std::vector<EMMCInfo>> IoDecorator::CollectEMMCInfo()
50 {
51 auto task = [this] { return ioCollector_->CollectEMMCInfo(); };
52 return Invoke(task, statInfoWrapper_, IO_COLLECTOR_NAME + UC_SEPARATOR + __func__);
53 }
54
ExportEMMCInfo()55 CollectResult<std::string> IoDecorator::ExportEMMCInfo()
56 {
57 auto task = [this] { return ioCollector_->ExportEMMCInfo(); };
58 return Invoke(task, statInfoWrapper_, IO_COLLECTOR_NAME + UC_SEPARATOR + __func__);
59 }
60
CollectAllProcIoStats(bool isUpdate)61 CollectResult<std::vector<ProcessIoStats>> IoDecorator::CollectAllProcIoStats(bool isUpdate)
62 {
63 auto task = [this, &isUpdate] { return ioCollector_->CollectAllProcIoStats(isUpdate); };
64 return Invoke(task, statInfoWrapper_, IO_COLLECTOR_NAME + UC_SEPARATOR + __func__);
65 }
66
ExportAllProcIoStats()67 CollectResult<std::string> IoDecorator::ExportAllProcIoStats()
68 {
69 auto task = [this] { return ioCollector_->ExportAllProcIoStats(); };
70 return Invoke(task, statInfoWrapper_, IO_COLLECTOR_NAME + UC_SEPARATOR + __func__);
71 }
72
CollectSysIoStats()73 CollectResult<SysIoStats> IoDecorator::CollectSysIoStats()
74 {
75 auto task = [this] { return ioCollector_->CollectSysIoStats(); };
76 return Invoke(task, statInfoWrapper_, IO_COLLECTOR_NAME + UC_SEPARATOR + __func__);
77 }
78
ExportSysIoStats()79 CollectResult<std::string> IoDecorator::ExportSysIoStats()
80 {
81 auto task = [this] { return ioCollector_->ExportSysIoStats(); };
82 return Invoke(task, statInfoWrapper_, IO_COLLECTOR_NAME + UC_SEPARATOR + __func__);
83 }
84
SaveStatCommonInfo()85 void IoDecorator::SaveStatCommonInfo()
86 {
87 std::map<std::string, StatInfo> statInfo = statInfoWrapper_.GetStatInfo();
88 std::vector<std::string> formattedStatInfo;
89 for (const auto& record : statInfo) {
90 formattedStatInfo.push_back(record.second.ToString());
91 }
92 WriteLinesToFile(formattedStatInfo, false);
93 }
94
ResetStatInfo()95 void IoDecorator::ResetStatInfo()
96 {
97 statInfoWrapper_.ResetStatInfo();
98 }
99 } // namespace UCollectUtil
100 } // namespace HiviewDFX
101 } // namespace OHOS
102