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