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 NET_STATS_DATA_HANDLER_H 17 #define NET_STATS_DATA_HANDLER_H 18 19 #include "net_stats_info.h" 20 #include "net_stats_database_defines.h" 21 22 namespace OHOS { 23 namespace NetManagerStandard { 24 25 class NetStatsDataHandler { 26 public: 27 NetStatsDataHandler(); 28 ~NetStatsDataHandler() = default; 29 int32_t WriteStatsData(const std::vector<NetStatsInfo> &infos, const std::string &tableName); 30 int32_t WriteIptablesData(const NetStatsInfo &infos, const std::string &tableName); 31 int32_t ReadStatsData(std::vector<NetStatsInfo> &infos, uint64_t start, uint64_t end); 32 int32_t ReadStatsData(std::vector<NetStatsInfo> &infos, uint64_t uid, uint64_t start, uint64_t end); 33 int32_t ReadStatsData(std::vector<NetStatsInfo> &infos, const std::string &iface, uint64_t start, uint64_t end); 34 int32_t ReadStatsData(std::vector<NetStatsInfo> &infos, const std::string &iface, const uint32_t uid, 35 uint64_t start, uint64_t end); 36 int32_t ReadStatsDataByIdent(std::vector<NetStatsInfo> &infos, const std::string &ident, uint64_t start, 37 uint64_t end); 38 int32_t ReadStatsData(std::vector<NetStatsInfo> &infos, uint32_t uid, const std::string &ident, uint64_t start, 39 uint64_t end); 40 int32_t ReadStatsDataByIdentAndUserId(std::vector<NetStatsInfo> &infos, const std::string &ident, 41 const int32_t userId, uint64_t start, uint64_t end); 42 int32_t DeleteByUid(uint64_t uid); 43 int32_t DeleteSimStatsByUid(uint64_t uid); 44 int32_t DeleteByDate(const std::string &tableName, uint64_t start, uint64_t end); 45 int32_t UpdateStatsFlag(uint32_t uid, uint32_t flag); 46 int32_t UpdateSimStatsFlag(uint32_t uid, uint32_t flag); 47 int32_t UpdateStatsFlagByUserId(int32_t userId, uint32_t flag); 48 int32_t UpdateSimStatsFlagByUserId(int32_t userId, uint32_t flag); 49 int32_t UpdateStatsUserIdByUserId(int32_t userId, int32_t newUserId); 50 int32_t UpdateSimStatsUserIdByUserId(int32_t userId, int32_t newUserId); 51 int32_t UpdateSimDataFlag(uint32_t oldFlag, uint32_t newFlag); 52 int32_t ClearData(); 53 int32_t BackupNetStatsData(const std::string &sourceDb, const std::string &backupDb); 54 55 private: 56 bool isDisplayTrafficAncoList = false; 57 }; 58 } // namespace NetManagerStandard 59 } // namespace OHOS 60 #endif // NET_STATS_DATA_HANDLER_H 61