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 #define MLOG_TAG "PhotoDayMonthYearOperation"
17 
18 #include "photo_day_month_year_operation.h"
19 
20 #include "media_file_utils.h"
21 #include "media_log.h"
22 #include "result_set_utils.h"
23 
24 namespace OHOS {
25 namespace Media {
26 const int32_t UPDATE_BATCH_SIZE = 200;
27 
28 const std::string QUERY_NEED_UPDATE_FILE_IDS = ""
29     "SELECT file_id FROM Photos "
30     "WHERE"
31     "  date_added = 0"
32     "  OR date_taken = 0"
33     "  OR date_day <> strftime( '%Y%m%d', date_taken / 1000, 'unixepoch', 'localtime' );";
34 
35 const std::string UPDATE_DAY_MONTH_YEAR = ""
36     "UPDATE Photos "
37     "SET date_added ="
38     " CASE"
39     "  WHEN date_added <> 0 THEN"
40     "  date_added "
41     "  WHEN date_taken <> 0 THEN"
42     "  date_taken "
43     "  WHEN date_modified <> 0 THEN"
44     "  date_modified ELSE strftime( '%s', 'now' ) "
45     " END, "
46     "date_taken ="
47     " CASE"
48     "  WHEN date_taken <> 0 THEN"
49     "  date_taken "
50     "  WHEN date_added <> 0 THEN"
51     "  date_added "
52     "  WHEN date_modified <> 0 THEN"
53     "  date_modified ELSE strftime( '%s', 'now' ) "
54     " END, "
55     "date_day ="
56     " CASE"
57     "  WHEN date_taken <> 0 THEN"
58     "  strftime( '%Y%m%d', date_taken / 1000, 'unixepoch', 'localtime' ) "
59     "  WHEN date_added <> 0 THEN"
60     "  strftime( '%Y%m%d', date_added / 1000, 'unixepoch', 'localtime' ) "
61     "  WHEN date_modified <> 0 THEN"
62     "  strftime( '%Y%m%d', date_modified / 1000, 'unixepoch', 'localtime' ) "
63     "  ELSE strftime( '%Y%m%d', strftime( '%s', 'now' ) / 1000, 'unixepoch', 'localtime' ) "
64     " END, "
65     "date_month ="
66     " CASE"
67     "  WHEN date_taken <> 0 THEN"
68     "  strftime( '%Y%m', date_taken / 1000, 'unixepoch', 'localtime' ) "
69     "  WHEN date_added <> 0 THEN"
70     "  strftime( '%Y%m', date_added / 1000, 'unixepoch', 'localtime' ) "
71     "  WHEN date_modified <> 0 THEN"
72     "  strftime( '%Y%m', date_modified / 1000, 'unixepoch', 'localtime' ) "
73     "  ELSE strftime( '%Y%m', strftime( '%s', 'now' ) / 1000, 'unixepoch', 'localtime' ) "
74     " END, "
75     "date_year ="
76     " CASE"
77     "  WHEN date_taken <> 0 THEN"
78     "  strftime( '%Y', date_taken / 1000, 'unixepoch', 'localtime' ) "
79     "  WHEN date_added <> 0 THEN"
80     "  strftime( '%Y', date_added / 1000, 'unixepoch', 'localtime' ) "
81     "  WHEN date_modified <> 0 THEN"
82     "  strftime( '%Y', date_modified / 1000, 'unixepoch', 'localtime' ) "
83     "  ELSE strftime( '%Y', strftime( '%s', 'now' ) / 1000, 'unixepoch', 'localtime' ) "
84     " END, "
85     "dirty ="
86     " CASE"
87     "  WHEN dirty = 0 THEN"
88     "  2 ELSE dirty "
89     " END "
90     "WHERE"
91     "  ( date_added = 0 "
92     "  OR date_taken = 0 "
93     "  OR date_day <> strftime( '%Y%m%d', date_taken / 1000, 'unixepoch', 'localtime' ) )"
94     "  AND file_id IN ( ";
95 
UpdatePhotosDate(const std::shared_ptr<MediaLibraryRdbStore> rdbStore)96 int32_t PhotoDayMonthYearOperation::UpdatePhotosDate(const std::shared_ptr<MediaLibraryRdbStore> rdbStore)
97 {
98     MEDIA_INFO_LOG("update photos date start");
99     int64_t startTime = MediaFileUtils::UTCTimeMilliSeconds();
100     if (rdbStore == nullptr || !rdbStore->CheckRdbStore()) {
101         MEDIA_ERR_LOG("Pointer rdbStore_ is nullptr. Maybe it didn't init successfully.");
102         return NativeRdb::E_ERROR;
103     }
104 
105     auto resultSet = rdbStore->QueryByStep(QUERY_NEED_UPDATE_FILE_IDS);
106     CHECK_AND_RETURN_RET_LOG(resultSet != nullptr, NativeRdb::E_ERROR, "query photos by step failed");
107 
108     std::vector<std::string> needUpdateFileIds;
109     while (resultSet->GoToNextRow() == NativeRdb::E_OK) {
110         needUpdateFileIds.push_back(GetStringVal(PhotoColumn::MEDIA_ID, resultSet));
111     }
112     resultSet->Close();
113     auto needChangedSize = needUpdateFileIds.size();
114     CHECK_AND_RETURN_RET(needChangedSize > 0, NativeRdb::E_OK);
115 
116     int64_t totalChanged = 0;
117     for (size_t start = 0; start < needChangedSize; start += UPDATE_BATCH_SIZE) {
118         size_t end = std::min(start + UPDATE_BATCH_SIZE, needChangedSize);
119         std::stringstream updateSql;
120         updateSql << UPDATE_DAY_MONTH_YEAR;
121         for (size_t i = start; i < end; ++i) {
122             if (i != start) {
123                 updateSql << ", ";
124             }
125             updateSql << needUpdateFileIds[i];
126         }
127         updateSql << " );";
128         int64_t batchStart = MediaFileUtils::UTCTimeMilliSeconds();
129         int64_t changedRowCount = 0;
130         auto errCode = rdbStore->ExecuteForChangedRowCount(changedRowCount, updateSql.str());
131         if (errCode != NativeRdb::E_OK) {
132             MEDIA_ERR_LOG("update photos date failed, errCode: %{public}d, batchStart: %{public}" PRId64
133                 ", cost: %{public}" PRId64,
134                 errCode, batchStart, MediaFileUtils::UTCTimeMilliSeconds() - batchStart);
135         } else {
136             totalChanged += changedRowCount;
137             MEDIA_DEBUG_LOG("update photos date, batchStart: %{public}" PRId64 ", cost: %{public}" PRId64
138                 ", changedRowCount: %{public}" PRId64,
139                 batchStart, MediaFileUtils::UTCTimeMilliSeconds() - batchStart, changedRowCount);
140         }
141     }
142 
143     MEDIA_INFO_LOG("update photos date end, startTime: %{public}" PRId64 ", cost: %{public}" PRId64
144         ", needChangedSize: %{public}zu, totalChanged: %{public}" PRId64,
145         startTime, MediaFileUtils::UTCTimeMilliSeconds() - startTime, needChangedSize, totalChanged);
146     return NativeRdb::E_OK;
147 }
148 
UpdateIdxSchptDateDay(const std::shared_ptr<MediaLibraryRdbStore> rdbStore)149 int32_t PhotoDayMonthYearOperation::UpdateIdxSchptDateDay(const std::shared_ptr<MediaLibraryRdbStore> rdbStore)
150 {
151     MEDIA_INFO_LOG("update photo idx_schpt_date_day start");
152     int64_t startTime = MediaFileUtils::UTCTimeMilliSeconds();
153     if (rdbStore == nullptr || !rdbStore->CheckRdbStore()) {
154         MEDIA_ERR_LOG("Pointer rdbStore_ is nullptr. Maybe it didn't init successfully.");
155         return NativeRdb::E_ERROR;
156     }
157 
158     auto ret = rdbStore->ExecuteSql(PhotoColumn::DROP_SCHPT_DAY_INDEX);
159     CHECK_AND_RETURN_RET_LOG(ret == NativeRdb::E_OK, ret, "drop idx_schpt_date_day failed, ret=%{public}d", ret);
160 
161     ret = rdbStore->ExecuteSql(PhotoColumn::CREATE_SCHPT_DAY_INDEX);
162     CHECK_AND_RETURN_RET_LOG(ret == NativeRdb::E_OK, ret, "create idx_schpt_date_day failed, ret=%{public}d", ret);
163 
164     MEDIA_INFO_LOG("update photo idx_schpt_date_day end, startTime: %{public}" PRId64 ", cost: %{public}" PRId64,
165         startTime, (MediaFileUtils::UTCTimeMilliSeconds() - startTime));
166     return NativeRdb::E_OK;
167 }
168 
UpdatePhotosDateAndIdx(const std::shared_ptr<MediaLibraryRdbStore> rdbStore)169 int32_t PhotoDayMonthYearOperation::UpdatePhotosDateAndIdx(const std::shared_ptr<MediaLibraryRdbStore> rdbStore)
170 {
171     MEDIA_INFO_LOG("update phots date start");
172     int64_t startTime = MediaFileUtils::UTCTimeMilliSeconds();
173     if (rdbStore == nullptr || !rdbStore->CheckRdbStore()) {
174         MEDIA_ERR_LOG("Pointer rdbStore_ is nullptr. Maybe it didn't init successfully.");
175         return NativeRdb::E_ERROR;
176     }
177 
178     auto ret = UpdateIdxSchptDateDay(rdbStore);
179     CHECK_AND_RETURN_RET_LOG(ret == NativeRdb::E_OK, ret, "update idx_schpt_date_day failed, ret=%{public}d", ret);
180 
181     ret = UpdatePhotosDate(rdbStore);
182     CHECK_AND_RETURN_RET_LOG(ret == NativeRdb::E_OK, ret, "update day month year failed, ret=%{public}d", ret);
183 
184     MEDIA_INFO_LOG("update phots date end, startTime: %{public}" PRId64 ", cost: %{public}" PRId64, startTime,
185         (MediaFileUtils::UTCTimeMilliSeconds() - startTime));
186     return NativeRdb::E_OK;
187 }
188 
UpdatePhotosDate(NativeRdb::RdbStore & rdbStore)189 int32_t PhotoDayMonthYearOperation::UpdatePhotosDate(NativeRdb::RdbStore &rdbStore)
190 {
191     MEDIA_INFO_LOG("update photos date start");
192     int64_t startTime = MediaFileUtils::UTCTimeMilliSeconds();
193 
194     auto resultSet = rdbStore.QueryByStep(QUERY_NEED_UPDATE_FILE_IDS);
195     CHECK_AND_RETURN_RET_LOG(resultSet != nullptr, NativeRdb::E_ERROR, "query photos by step failed");
196 
197     std::vector<std::string> needUpdateFileIds;
198     while (resultSet->GoToNextRow() == NativeRdb::E_OK) {
199         needUpdateFileIds.push_back(GetStringVal(PhotoColumn::MEDIA_ID, resultSet));
200     }
201     resultSet->Close();
202     auto needChangedSize = needUpdateFileIds.size();
203     CHECK_AND_RETURN_RET(needChangedSize > 0, NativeRdb::E_OK);
204 
205     std::stringstream updateSql;
206     updateSql << UPDATE_DAY_MONTH_YEAR;
207     for (size_t i = 0; i < needChangedSize; ++i) {
208         if (i != 0) {
209             updateSql << ", ";
210         }
211         updateSql << needUpdateFileIds[i];
212     }
213     updateSql << " );";
214     int64_t changedRowCount = 0;
215     auto errCode = rdbStore.ExecuteForChangedRowCount(changedRowCount, updateSql.str());
216     if (errCode != NativeRdb::E_OK) {
217         MEDIA_ERR_LOG("update photos date failed, errCode: %{public}d, startTime: %{public}" PRId64
218             ", cost: %{public}" PRId64 ", needChangedSize: %{public}zu",
219             errCode, startTime, MediaFileUtils::UTCTimeMilliSeconds() - startTime, needChangedSize);
220         return errCode;
221     }
222     MEDIA_INFO_LOG("update photos date end, startTime: %{public}" PRId64 ", cost: %{public}" PRId64
223         ", needChangedSize: %{public}zu, changedRowCount: %{public}" PRId64,
224         startTime, MediaFileUtils::UTCTimeMilliSeconds() - startTime, needChangedSize, changedRowCount);
225     return NativeRdb::E_OK;
226 }
227 } // namespace Media
228 } // namespace OHOS
229