1 /*
2 * Copyright (C) 2024 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 #include <sys/stat.h>
17 #include "dfx_database_utils.h"
18
19 #include "dfx_utils.h"
20 #include "medialibrary_rdbstore.h"
21 #include "medialibrary_unistore_manager.h"
22 #include "media_log.h"
23 #include "media_column.h"
24 #include "medialibrary_errno.h"
25 #include "result_set_utils.h"
26 #include "photo_album_column.h"
27
28 namespace OHOS {
29 namespace Media {
30 const std::string RECORD_COUNT = "recordCount";
31 const std::string ABNORMAL_VALUE = "-1";
32
QueryFromPhotos(int32_t mediaType,bool isLocal)33 int32_t DfxDatabaseUtils::QueryFromPhotos(int32_t mediaType, bool isLocal)
34 {
35 NativeRdb::RdbPredicates predicates(PhotoColumn::PHOTOS_TABLE);
36 predicates.EqualTo(MediaColumn::MEDIA_TYPE, mediaType);
37 if (isLocal) {
38 predicates.IsNull(PhotoColumn::PHOTO_CLOUD_ID);
39 } else {
40 predicates.IsNotNull(PhotoColumn::PHOTO_CLOUD_ID);
41 }
42 std::vector<std::string> columns = { "count(1) AS count" };
43 std::string queryColumn = "count";
44 int32_t count;
45 int32_t errCode = QueryInt(predicates, columns, queryColumn, count);
46 if (errCode != E_OK) {
47 MEDIA_ERR_LOG("query local image fail: %{public}d", errCode);
48 }
49 return count;
50 }
51
QueryAlbumInfoBySubtype(int32_t albumSubtype)52 AlbumInfo DfxDatabaseUtils::QueryAlbumInfoBySubtype(int32_t albumSubtype)
53 {
54 AlbumInfo albumInfo;
55 NativeRdb::RdbPredicates predicates(PhotoAlbumColumns::TABLE);
56 predicates.EqualTo(PhotoAlbumColumns::ALBUM_SUBTYPE, albumSubtype);
57 std::vector<std::string> columns = { PhotoAlbumColumns::ALBUM_COUNT, PhotoAlbumColumns::ALBUM_IMAGE_COUNT,
58 PhotoAlbumColumns::ALBUM_VIDEO_COUNT, PhotoAlbumColumns::ALBUM_CLOUD_ID };
59 auto resultSet = MediaLibraryRdbStore::QueryWithFilter(predicates, columns);
60 if (resultSet == nullptr || resultSet->GoToFirstRow() != NativeRdb::E_OK) {
61 MEDIA_ERR_LOG("query album fail");
62 return albumInfo;
63 }
64 albumInfo.count = GetInt32Val(PhotoAlbumColumns::ALBUM_COUNT, resultSet);
65 albumInfo.imageCount = GetInt32Val(PhotoAlbumColumns::ALBUM_IMAGE_COUNT, resultSet);
66 albumInfo.videoCount = GetInt32Val(PhotoAlbumColumns::ALBUM_VIDEO_COUNT, resultSet);
67 albumInfo.isLocal = GetStringVal(PhotoAlbumColumns::ALBUM_CLOUD_ID, resultSet) == "" ? true : false;
68 return albumInfo;
69 }
70
QueryDirtyCloudPhoto()71 std::vector<PhotoInfo> DfxDatabaseUtils::QueryDirtyCloudPhoto()
72 {
73 vector<PhotoInfo> photoInfoList;
74 NativeRdb::RdbPredicates predicates(PhotoColumn::PHOTOS_TABLE);
75 predicates.IsNotNull(PhotoColumn::PHOTO_CLOUD_ID);
76 predicates.NotEqualTo(PhotoColumn::PHOTO_DIRTY, static_cast<int32_t> (DirtyType::TYPE_SYNCED));
77 predicates.Limit(DIRTY_PHOTO_COUNT);
78 std::vector<std::string> columns = { MediaColumn::MEDIA_FILE_PATH, PhotoColumn::PHOTO_DIRTY,
79 PhotoColumn::PHOTO_CLOUD_ID };
80 auto resultSet = MediaLibraryRdbStore::QueryWithFilter(predicates, columns);
81 if (resultSet == nullptr) {
82 MEDIA_ERR_LOG("resultSet is null");
83 return photoInfoList;
84 }
85 while (resultSet->GoToNextRow() == NativeRdb::E_OK) {
86 PhotoInfo photoInfo;
87 photoInfo.data = DfxUtils::GetSafePath(GetStringVal(MediaColumn::MEDIA_FILE_PATH, resultSet));
88 photoInfo.dirty = GetInt32Val(PhotoColumn::PHOTO_DIRTY, resultSet);
89 photoInfo.cloudVersion = GetInt32Val(PhotoColumn::PHOTO_CLOUD_ID, resultSet);
90 photoInfoList.push_back(photoInfo);
91 }
92 return photoInfoList;
93 }
94
ParseResultSet(const string & querySql,int32_t mediaTypePara,int32_t & photoInfoCount)95 static bool ParseResultSet(const string &querySql, int32_t mediaTypePara, int32_t &photoInfoCount)
96 {
97 auto rdbStore = MediaLibraryUnistoreManager::GetInstance().GetRdbStore();
98 if (rdbStore == nullptr) {
99 MEDIA_ERR_LOG("rdbStore is nullptr!");
100 return false;
101 }
102 auto resultSet = rdbStore->QuerySql(querySql);
103 if (resultSet == nullptr) {
104 MEDIA_ERR_LOG("resultSet is null");
105 return false;
106 }
107 while (resultSet->GoToNextRow() == NativeRdb::E_OK) {
108 if (mediaTypePara > 0) {
109 int32_t mediaType = GetInt32Val(MediaColumn::MEDIA_TYPE, resultSet);
110 if (mediaType == mediaTypePara) {
111 photoInfoCount = GetInt32Val(RECORD_COUNT, resultSet);
112 }
113 } else {
114 photoInfoCount = GetInt32Val(RECORD_COUNT, resultSet);
115 }
116 }
117 return true;
118 }
119
QueryPhotoRecordInfo(PhotoRecordInfo & photoRecordInfo)120 int32_t DfxDatabaseUtils::QueryPhotoRecordInfo(PhotoRecordInfo &photoRecordInfo)
121 {
122 const string filterCondition = MediaColumn::MEDIA_TIME_PENDING + " = 0 AND " +
123 PhotoColumn::PHOTO_SYNC_STATUS + " = " +
124 to_string(static_cast<int32_t>(SyncStatusType::TYPE_VISIBLE)) + " AND " +
125 PhotoColumn::PHOTO_CLEAN_FLAG + " = " +
126 to_string(static_cast<int32_t>(CleanType::TYPE_NOT_CLEAN));
127
128 const string imageAndVideoCountQuerySql = "SELECT " + MediaColumn::MEDIA_TYPE + ", COUNT(*) AS " + RECORD_COUNT +
129 " FROM " + PhotoColumn::PHOTOS_TABLE + " WHERE " + filterCondition + " GROUP BY " + MediaColumn::MEDIA_TYPE;
130
131 const string abnormalSizeCountQuerySql = "SELECT COUNT(*) AS " + RECORD_COUNT + " FROM " +
132 PhotoColumn::PHOTOS_TABLE + " WHERE " + MediaColumn::MEDIA_SIZE + " = " + ABNORMAL_VALUE +
133 " AND " + filterCondition;
134
135 const string abnormalWidthHeightQuerySql = "SELECT COUNT(*) AS " + RECORD_COUNT + " FROM " +
136 PhotoColumn::PHOTOS_TABLE + " WHERE (" + PhotoColumn::PHOTO_WIDTH + " = " + ABNORMAL_VALUE +
137 " OR " + PhotoColumn::PHOTO_HEIGHT + " = " + ABNORMAL_VALUE + ") AND " + filterCondition;
138
139 const string abnormalVideoDurationQuerySql = "SELECT COUNT(*) AS " + RECORD_COUNT + " FROM " +
140 PhotoColumn::PHOTOS_TABLE + " WHERE " + MediaColumn::MEDIA_DURATION + " = " + ABNORMAL_VALUE +
141 " AND " + MediaColumn::MEDIA_TYPE + " = " + std::to_string(MEDIA_TYPE_VIDEO) + " AND " + filterCondition;
142
143 const string totalAbnormalRecordSql = "SELECT COUNT(*) AS " + RECORD_COUNT + " FROM " +
144 PhotoColumn::PHOTOS_TABLE + " WHERE (" + MediaColumn::MEDIA_SIZE + " = 0 OR " +
145 MediaColumn::MEDIA_SIZE + " IS NULL OR " + MediaColumn::MEDIA_MIME_TYPE + " IS NULL OR " +
146 MediaColumn::MEDIA_MIME_TYPE + " = '' OR " + PhotoColumn::PHOTO_HEIGHT + " = 0 OR " +
147 PhotoColumn::PHOTO_HEIGHT + " IS NULL OR " + PhotoColumn::PHOTO_WIDTH + " = 0 OR " +
148 PhotoColumn::PHOTO_WIDTH + " IS NULL OR ((" + MediaColumn::MEDIA_DURATION + " IS NULL OR " +
149 MediaColumn::MEDIA_DURATION + " = 0 ) AND " + MediaColumn::MEDIA_TYPE + " = " +
150 std::to_string(MEDIA_TYPE_VIDEO) + " )) AND " + filterCondition;
151
152 bool ret = ParseResultSet(imageAndVideoCountQuerySql, MEDIA_TYPE_VIDEO, photoRecordInfo.videoCount);
153 ret = ParseResultSet(imageAndVideoCountQuerySql, MEDIA_TYPE_IMAGE, photoRecordInfo.imageCount) && ret;
154 ret = ParseResultSet(abnormalSizeCountQuerySql, 0, photoRecordInfo.abnormalSizeCount) && ret;
155 ret = ParseResultSet(abnormalWidthHeightQuerySql, 0, photoRecordInfo.abnormalWidthOrHeightCount) && ret;
156 ret = ParseResultSet(abnormalVideoDurationQuerySql, 0, photoRecordInfo.abnormalVideoDurationCount) && ret;
157 ret = ParseResultSet(totalAbnormalRecordSql, 0, photoRecordInfo.toBeUpdatedRecordCount) && ret;
158
159 string databaseDir = MEDIA_DB_DIR + "/rdb";
160 if (access(databaseDir.c_str(), E_OK) != 0) {
161 MEDIA_WARN_LOG("can not get rdb through sandbox");
162 return E_FAIL;
163 }
164 string dbPath = databaseDir.append("/").append(MEDIA_DATA_ABILITY_DB_NAME);
165
166 struct stat statInfo {};
167 if (stat(dbPath.c_str(), &statInfo) != 0) {
168 MEDIA_ERR_LOG("stat syscall err");
169 return E_FAIL;
170 }
171 photoRecordInfo.dbFileSize = statInfo.st_size;
172
173 return ret ? E_OK : E_FAIL;
174 }
175
QueryAnalysisVersion(const std::string & table,const std::string & column)176 int32_t DfxDatabaseUtils::QueryAnalysisVersion(const std::string &table, const std::string &column)
177 {
178 NativeRdb::RdbPredicates predicates(table);
179 string whereClause = "max(" + column + ") AS version";
180 std::vector<std::string> columns = { whereClause };
181 string version = "version";
182 double count;
183 int32_t errCode = QueryDouble(predicates, columns, version, count);
184 if (errCode != E_OK) {
185 MEDIA_ERR_LOG("query analysis version fail: %{public}d", errCode);
186 }
187 return static_cast<int32_t> (count);
188 }
189
QueryDbVersion()190 int32_t DfxDatabaseUtils::QueryDbVersion()
191 {
192 int64_t dbVersion = 0;
193 MediaLibraryRdbStore::QueryPragma("user_version", dbVersion);
194 return static_cast<int32_t> (dbVersion);
195 }
196
QueryInt(const NativeRdb::AbsRdbPredicates & predicates,const std::vector<std::string> & columns,const std::string & queryColumn,int32_t & value)197 int32_t DfxDatabaseUtils::QueryInt(const NativeRdb::AbsRdbPredicates &predicates,
198 const std::vector<std::string> &columns, const std::string &queryColumn, int32_t &value)
199 {
200 auto resultSet = MediaLibraryRdbStore::QueryWithFilter(predicates, columns);
201 if (resultSet == nullptr || resultSet->GoToFirstRow() != NativeRdb::E_OK) {
202 return E_DB_FAIL;
203 }
204 value = GetInt32Val(queryColumn, resultSet);
205 return E_OK;
206 }
207
QueryDouble(const NativeRdb::AbsRdbPredicates & predicates,const std::vector<std::string> & columns,const std::string & queryColumn,double & value)208 int32_t DfxDatabaseUtils::QueryDouble(const NativeRdb::AbsRdbPredicates &predicates,
209 const std::vector<std::string> &columns, const std::string &queryColumn, double &value)
210 {
211 auto resultSet = MediaLibraryRdbStore::QueryWithFilter(predicates, columns);
212 if (resultSet == nullptr || resultSet->GoToFirstRow() != NativeRdb::E_OK) {
213 return E_DB_FAIL;
214 }
215 value = GetDoubleVal(queryColumn, resultSet);
216 return E_OK;
217 }
218
QueryDownloadedAndGeneratedThumb(int32_t & downloadedThumb,int32_t & generatedThumb)219 int32_t DfxDatabaseUtils::QueryDownloadedAndGeneratedThumb(int32_t& downloadedThumb, int32_t& generatedThumb)
220 {
221 // cloud image that are all generated
222 NativeRdb::RdbPredicates generatePredicates(PhotoColumn::PHOTOS_TABLE);
223 std::vector<std::string> columns = { "count(1) AS count" };
224 std::string queryColumn = "count";
225 int32_t cloudImage = 2;
226 int32_t thumbGeneratedFinished = 2;
227 generatePredicates.GreaterThanOrEqualTo(PhotoColumn::PHOTO_POSITION, cloudImage)->And()
228 ->GreaterThanOrEqualTo(PhotoColumn::PHOTO_THUMBNAIL_READY, thumbGeneratedFinished);
229 int32_t errCode = QueryInt(generatePredicates, columns, queryColumn, generatedThumb);
230 if (errCode != E_OK) {
231 MEDIA_ERR_LOG("query generated image fail: %{public}d", errCode);
232 return errCode;
233 }
234
235 // cloud image that are downloaded
236 NativeRdb::RdbPredicates downloadPredicates(PhotoColumn::PHOTOS_TABLE);
237 downloadPredicates.GreaterThanOrEqualTo(PhotoColumn::PHOTO_POSITION, cloudImage)->And()
238 ->EqualTo(PhotoColumn::PHOTO_THUMB_STATUS, 0);
239 errCode = QueryInt(downloadPredicates, columns, queryColumn, downloadedThumb);
240 if (errCode != E_OK) {
241 MEDIA_ERR_LOG("query downloaded image fail: %{public}d", errCode);
242 return errCode;
243 }
244 return E_OK;
245 }
246
QueryTotalCloudThumb(int32_t & totalDownload)247 int32_t DfxDatabaseUtils::QueryTotalCloudThumb(int32_t& totalDownload)
248 {
249 int32_t cloudImage = 2;
250 NativeRdb::RdbPredicates predicates(PhotoColumn::PHOTOS_TABLE);
251 predicates.GreaterThanOrEqualTo(PhotoColumn::PHOTO_POSITION, cloudImage);
252 std::vector<std::string> columns = { "count(1) AS count" };
253 std::string queryColumn = "count";
254 int32_t errCode = QueryInt(predicates, columns, queryColumn, totalDownload);
255 if (errCode != E_OK) {
256 MEDIA_ERR_LOG("query total download image fail: %{public}d", errCode);
257 return errCode;
258 }
259 return E_OK;
260 }
261
262 } // namespace Media
263 } // namespace OHOS