1 /* 2 * Copyright (c) 2023 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 OHOS_DISTRIBUTED_DATA_DATAMGR_SERVICE_RDB_ASSET_LOADER_H 17 #define OHOS_DISTRIBUTED_DATA_DATAMGR_SERVICE_RDB_ASSET_LOADER_H 18 19 #include "cloud/asset_loader.h" 20 #include "cloud/cloud_store_types.h" 21 #include "cloud/iAssetLoader.h" 22 #include "error/general_error.h" 23 #include "snapshot/snapshot.h" 24 25 namespace OHOS::DistributedRdb { 26 class RdbAssetLoader : public DistributedDB::IAssetLoader { 27 public: 28 using Type = DistributedDB::Type; 29 using Asset = DistributedDB::Asset; 30 using DBStatus = DistributedDB::DBStatus; 31 using BindAssets = DistributedData::BindAssets; 32 using AssetsRecord = DistributedData::AssetRecord; 33 using AssetStatus = DistributedData::Asset::Status; 34 using GeneralError = DistributedData::GeneralError; 35 using VBucket = DistributedData::VBucket; 36 37 explicit RdbAssetLoader(std::shared_ptr<DistributedData::AssetLoader> cloudAssetLoader, BindAssets *bindAssets); 38 39 ~RdbAssetLoader() = default; 40 41 DBStatus Download(const std::string &tableName, const std::string &gid, const Type &prefix, 42 std::map<std::string, std::vector<Asset>> &assets) override; 43 44 void BatchDownload(const std::string &tableName, std::vector<AssetRecord> &downloadAssets) override; 45 46 DBStatus RemoveLocalAssets(const std::vector<Asset> &assets) override; 47 48 DBStatus CancelDownload() override; 49 50 private: 51 static std::vector<AssetRecord> Convert(std::vector<AssetsRecord> &&assetsRecords); 52 static std::vector<AssetsRecord> Convert(std::vector<AssetRecord> &&assetRecords); 53 static DBStatus ConvertStatus(AssetStatus error); 54 static void UpdateStatus(AssetRecord &assetRecord, const VBucket &assets); 55 56 void PostEvent(std::set<std::string> &skipAssets, std::map<std::string, DistributedData::Value> &assets, 57 DistributedData::AssetEvent eventId, std::set<std::string> &deleteAssets); 58 void PostEvent(std::set<std::string> &skipAssets, std::vector<AssetsRecord> &assetsRecords, 59 DistributedData::AssetEvent eventId, std::set<std::string> &deleteAssets); 60 void PostEvent(DistributedData::AssetEvent eventId, DistributedData::Assets &assets, 61 std::set<std::string> &skipAssets, std::set<std::string> &deleteAssets); 62 63 std::shared_ptr<DistributedData::AssetLoader> assetLoader_; 64 BindAssets *snapshots_; 65 }; 66 } // namespace OHOS::DistributedRdb 67 #endif // OHOS_DISTRIBUTED_DATA_DATAMGR_SERVICE_RDB_ASSET_LOADER_H