1 /* 2 * Copyright (c) 2021 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 HTRACE_CPU_DATA_PARSER_H 16 #define HTRACE_CPU_DATA_PARSER_H 17 #include <cstdint> 18 #include <map> 19 #include <string> 20 #include "cpu_plugin_result.pb.h" 21 #include "hilog_plugin_result.pb.h" 22 #include "htrace_plugin_time_parser.h" 23 #include "trace_data/trace_data_cache.h" 24 #include "trace_streamer_config.h" 25 #include "trace_streamer_filters.h" 26 27 namespace SysTuning { 28 namespace TraceStreamer { 29 class HtraceCpuDataParser : public HtracePluginTimeParser { 30 public: 31 HtraceCpuDataParser(TraceDataCache* dataCache, const TraceStreamerFilters* ctx); 32 ~HtraceCpuDataParser(); 33 void Parse(CpuData& tracePacket, uint64_t ts); 34 void Finish(); 35 enum TSCpuDataType { TSCpuDataType_Usage, TSCpuDataType_ThreadInfo, TSCpuDataType_Load }; 36 class TsCpuData { 37 public: TsCpuData()38 TsCpuData() 39 { 40 ts_ = 0; 41 cpuDataType_ = TSCpuDataType_Usage; 42 } SetCpuUsage(uint64_t ts,CpuUsageInfo & usage)43 void SetCpuUsage(uint64_t ts, CpuUsageInfo& usage) 44 { 45 ts_ = ts; 46 cpuUsageData_ = std::make_unique<CpuUsageInfo>(usage); 47 cpuDataType_ = TSCpuDataType_Usage; 48 } SetThreadInfo(uint64_t ts,ThreadInfo & threadInfo)49 void SetThreadInfo(uint64_t ts, ThreadInfo& threadInfo) 50 { 51 ts_ = ts; 52 threadDataData_ = std::make_unique<ThreadInfo>(threadInfo); 53 cpuDataType_ = TSCpuDataType_ThreadInfo; 54 } SetExtInfo(double totalLoad,double userLoad,double sysLoad,double process_num)55 void SetExtInfo(double totalLoad, double userLoad, double sysLoad, double process_num) 56 { 57 totalLoad_ = totalLoad; 58 userLoad_ = userLoad; 59 sysLoad_ = sysLoad; 60 process_num_ = process_num; 61 cpuDataType_ = TSCpuDataType_Load; 62 } 63 uint64_t ts_; 64 std::unique_ptr<CpuUsageInfo> cpuUsageData_; 65 std::unique_ptr<ThreadInfo> threadDataData_; 66 TSCpuDataType cpuDataType_; 67 double userLoad_ = 0; 68 double sysLoad_ = 0; 69 double process_num_ = 0; 70 double totalLoad_ = 0; 71 }; 72 std::vector<std::unique_ptr<TsCpuData>> cpuData_; 73 74 private: 75 std::string threadStateDesc_[THREAD_WAITING + 1] = {"undefined", "Running", "Sleep", "Sloped", "Watting"}; 76 uint64_t lastLineSeq_ = 0; 77 }; 78 } // namespace TraceStreamer 79 } // namespace SysTuning 80 81 #endif // HTRACE_CPU_DATA_PARSER_H 82