1 /*
2 * Copyright (C) 2025 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 #define MLOG_TAG "PhotoStorageOperation"
16 #include "photo_storage_operation.h"
17
18 #include "media_log.h"
19 #include "media_file_utils.h"
20 #include "medialibrary_tracer.h"
21 #include "medialibrary_type_const.h"
22 #include "preferences.h"
23 #include "preferences_helper.h"
24
25 namespace OHOS::Media {
FindStorage(std::shared_ptr<MediaLibraryRdbStore> rdbStore)26 std::shared_ptr<NativeRdb::ResultSet> PhotoStorageOperation::FindStorage(std::shared_ptr<MediaLibraryRdbStore> rdbStore)
27 {
28 MediaLibraryTracer tracer;
29 tracer.Start("PhotoStorageOperation::FindStorage");
30 bool conn = rdbStore == nullptr;
31 CHECK_AND_RETURN_RET_LOG(!conn, nullptr, "RdbStore is null");
32
33 TotalThumbnailSizeResult totalThumbnailSizeResult = {};
34 GetTotalThumbnailSize(rdbStore, totalThumbnailSizeResult);
35 MEDIA_INFO_LOG("Thumbnail stats: total_size = %{public}" PRId64 " bytes, count = %{public}d",
36 totalThumbnailSizeResult.totalThumbnailSize, totalThumbnailSizeResult.thumbnailCount);
37
38 TotalEditdataSizeResult totalEditdataSizeRusult = {};
39 GetTotalEditdataSize(rdbStore, totalEditdataSizeRusult);
40 MEDIA_INFO_LOG("Editdata stats: total_size = %{public}" PRId64 " bytes, count = %{public}d",
41 totalEditdataSizeRusult.totalEditdataSize, totalEditdataSizeRusult.editdataCount);
42
43 int64_t cacheSize = this->GetCacheSize();
44 int64_t highlightSize = this->GetHighlightSizeFromPreferences();
45 MEDIA_INFO_LOG("Media_Storage: cacheSize = %{public}" PRId64 ", highlightSize = %{public}" PRId64 "",
46 cacheSize, highlightSize);
47
48 int64_t totalExtSize = cacheSize + highlightSize + totalThumbnailSizeResult.totalThumbnailSize +
49 totalEditdataSizeRusult.totalEditdataSize;
50 std::string sql = this->SQL_DB_STORAGE_QUERY;
51 std::vector<NativeRdb::ValueObject> params = {totalExtSize};
52 return rdbStore->QuerySql(sql, params);
53 }
54
GetCacheSize()55 int64_t PhotoStorageOperation::GetCacheSize()
56 {
57 MediaLibraryTracer tracer;
58 tracer.Start("PhotoStorageOperation::GetCacheSize");
59 size_t totalSize = 0;
60 MediaFileUtils::StatDirSize(MEDIA_CACHE_DIR, totalSize);
61 return static_cast<int64_t>(totalSize);
62 }
63
GetHighlightSize()64 int64_t PhotoStorageOperation::GetHighlightSize()
65 {
66 MediaLibraryTracer tracer;
67 tracer.Start("PhotoStorageOperation::GetHighlightSize");
68 size_t totalSize = 0;
69 auto start = std::chrono::system_clock::now();
70 MediaFileUtils::StatDirSize(MEDIA_HIGHLIGHT_DIR, totalSize);
71 auto end = std::chrono::system_clock::now();
72 MEDIA_INFO_LOG("Media_Storage: GetHighlightSize size = %{public}zu, cost time = %{public}lld ms",
73 totalSize, std::chrono::duration_cast<std::chrono::milliseconds>(end - start).count());
74 this->SaveHighlightSizeToPreferences(totalSize);
75 return static_cast<int64_t>(totalSize);
76 }
77
QueryHighlightDirectorySize(std::shared_ptr<MediaLibraryRdbStore> rdbStore)78 std::shared_ptr<NativeRdb::ResultSet> PhotoStorageOperation::QueryHighlightDirectorySize(
79 std::shared_ptr<MediaLibraryRdbStore> rdbStore)
80 {
81 int64_t highlightSize = GetHighlightSize();
82 std::string sql = "SELECT ? AS dir_size";
83 std::vector<NativeRdb::ValueObject> params = {highlightSize};
84 return rdbStore->QuerySql(sql, params);
85 }
86
GetHighlightSizeFromPreferences()87 int64_t PhotoStorageOperation::GetHighlightSizeFromPreferences()
88 {
89 int32_t errCode;
90 std::shared_ptr<NativePreferences::Preferences> prefs =
91 NativePreferences::PreferencesHelper::GetPreferences(HIGHLIGHT_CONFIG, errCode);
92 CHECK_AND_RETURN_RET_WARN_LOG(prefs != nullptr && errCode == NativePreferences::E_OK, 0,
93 "get highlight preferences error: %{public}d", errCode);
94 return prefs->GetLong(HIGHLIGHT_DIRECTORY_SIZE, 0);
95 }
96
SaveHighlightSizeToPreferences(int64_t size)97 void PhotoStorageOperation::SaveHighlightSizeToPreferences(int64_t size)
98 {
99 int32_t errCode;
100 std::shared_ptr<NativePreferences::Preferences> prefs =
101 NativePreferences::PreferencesHelper::GetPreferences(HIGHLIGHT_CONFIG, errCode);
102 CHECK_AND_RETURN_WARN_LOG(prefs != nullptr && errCode == NativePreferences::E_OK,
103 "get highlight preferences error: %{public}d", errCode);
104 int32_t output = prefs->PutLong(HIGHLIGHT_DIRECTORY_SIZE, size);
105 prefs->FlushSync();
106 MEDIA_INFO_LOG("SaveHighlightSizeToPreferences output is %{public}d", output);
107 }
108
GetTotalThumbnailSize(std::shared_ptr<MediaLibraryRdbStore> rdbStore,TotalThumbnailSizeResult & totalThumbnailSizeResult)109 void PhotoStorageOperation::GetTotalThumbnailSize(std::shared_ptr<MediaLibraryRdbStore> rdbStore,
110 TotalThumbnailSizeResult &totalThumbnailSizeResult)
111 {
112 CHECK_AND_RETURN_LOG(rdbStore != nullptr, "RdbStore is null");
113
114 std::string sql = "SELECT "
115 "SUM(thumbnail_size) AS total_thumbnail_size, "
116 "COUNT(thumbnail_size) AS thumbnail_count "
117 "FROM tab_photos_ext";
118
119 auto statsResult = rdbStore->QuerySql(sql);
120 if (statsResult && (statsResult->GoToFirstRow() == NativeRdb::E_OK)) {
121 statsResult->GetLong(0, totalThumbnailSizeResult.totalThumbnailSize);
122 statsResult->GetInt(1, totalThumbnailSizeResult.thumbnailCount);
123 }
124 }
125
GetTotalEditdataSize(std::shared_ptr<MediaLibraryRdbStore> rdbStore,TotalEditdataSizeResult & totalEditdataSizeResult)126 void PhotoStorageOperation::GetTotalEditdataSize(std::shared_ptr<MediaLibraryRdbStore> rdbStore,
127 TotalEditdataSizeResult &totalEditdataSizeResult)
128 {
129 CHECK_AND_RETURN_LOG(rdbStore != nullptr, "RdbStore is null");
130
131 std::string sql = "SELECT "
132 "SUM(editdata_size) AS total_editdata_size, "
133 "COUNT(editdata_size) AS editdata_count "
134 "FROM tab_photos_ext";
135
136 auto statsResult = rdbStore->QuerySql(sql);
137 if (statsResult && (statsResult->GoToFirstRow() == NativeRdb::E_OK)) {
138 statsResult->GetLong(0, totalEditdataSizeResult.totalEditdataSize);
139 statsResult->GetInt(1, totalEditdataSizeResult.editdataCount);
140 }
141 }
142
GetLocalPhotoSize(std::shared_ptr<MediaLibraryRdbStore> rdbStore,LocalPhotoSizeResult & localPhotoSizeResult,int64_t totalExtSize)143 void PhotoStorageOperation::GetLocalPhotoSize(std::shared_ptr<MediaLibraryRdbStore> rdbStore,
144 LocalPhotoSizeResult &localPhotoSizeResult, int64_t totalExtSize)
145 {
146 CHECK_AND_RETURN_LOG(rdbStore != nullptr, "RdbStore is null");
147
148 std::string sql = this->SQL_DB_STORAGE_QUERY;
149 std::vector<NativeRdb::ValueObject> params = {totalExtSize};
150 auto statsResult = rdbStore->QuerySql(sql, params);
151 int32_t mediaType = MediaType::MEDIA_TYPE_DEFAULT;
152 while (statsResult && (statsResult->GoToNextRow() == NativeRdb::E_OK)) {
153 statsResult->GetInt(0, mediaType);
154 if (mediaType == MediaType::MEDIA_TYPE_IMAGE) {
155 statsResult->GetLong(1, localPhotoSizeResult.localImageSize);
156 } else if (mediaType == MediaType::MEDIA_TYPE_VIDEO) {
157 statsResult->GetLong(1, localPhotoSizeResult.localVideoSize);
158 }
159 }
160 statsResult->Close();
161 }
162 } // namespace OHOS::Media