1/* 2 * Copyright (c) 2022 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 TEST_WUKONG_DATA_SET_H 17#define TEST_WUKONG_DATA_SET_H 18 19#include <map> 20#include <memory> 21#include <string> 22#include <vector> 23 24#include "filter.h" 25#include "format.h" 26#include "statistics.h" 27#include "table.h" 28#include "wukong_define.h" 29namespace OHOS { 30namespace WuKong { 31class DataSet { 32public: 33 DataSet() = default; 34 ~DataSet() = default; 35 void FilterData(std::map<std::string, std::string> data); 36 void SetFilterType(std::string filterType); 37 void SetFilterStragety(std::shared_ptr<Filter> filter); 38 void StatisticsData(); 39 void SetStatisticsStragety(std::shared_ptr<Statistics> statistics); 40 void FormatData(std::string name, std::string &content); 41 void SetFormatStragety(std::shared_ptr<Format> format); 42private: 43 // filtered data vector 44 std::vector<std::map<std::string, std::string>> dataVector_; 45 std::shared_ptr<Filter> filter_ = nullptr; 46 std::shared_ptr<Format> format_; 47 std::shared_ptr<Statistics> statistics_ = nullptr; 48 // filtered data vector statistical processing tables 49 std::map<std::string, std::shared_ptr<Table>> tables_; 50 std::string filterType_; 51}; 52} // namespace WuKong 53} // namespace OHOS 54#endif 55