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 #include "firmware_task_operator.h"
17
18 #include <memory>
19
20 #include "network_type.h"
21 #include "order.h"
22 #include "time_utils.h"
23 #include "update_define.h"
24 #include "upgrade_status.h"
25
26 namespace OHOS {
27 namespace UpdateEngine {
QueryTask(FirmwareTask & task)28 void FirmwareTaskOperator::QueryTask(FirmwareTask &task)
29 {
30 std::vector<FirmwareTask> taskList;
31 QueryAll(taskList);
32 if (taskList.empty()) {
33 FIRMWARE_LOGI("FirmwareTaskOperator QueryTask no task");
34 task.isExistTask = false;
35 return;
36 }
37 task = taskList[0];
38 task.isExistTask = true;
39 }
40
UpdateProgressByTaskId(const std::string & taskId,UpgradeStatus status,int32_t progress)41 bool FirmwareTaskOperator::UpdateProgressByTaskId(const std::string &taskId, UpgradeStatus status, int32_t progress)
42 {
43 NativeRdb::ValuesBucket values;
44 values.PutInt(COLUMN_TASK_STATUS, CAST_INT(status));
45 values.PutInt(COLUMN_TASK_PROGRESS, progress);
46 return UpdateByTaskId(taskId, values);
47 }
48
UpdateErrMsgByTaskId(const std::string & taskId,int errorCode,const std::string & errorMsg)49 bool FirmwareTaskOperator::UpdateErrMsgByTaskId(const std::string &taskId, int errorCode, const std::string &errorMsg)
50 {
51 NativeRdb::ValuesBucket values;
52 values.PutInt(COLUMN_TASK_ERROR_CODE, errorCode);
53 values.PutString(COLUMN_TASK_ERROR_MSG, errorMsg);
54 return UpdateByTaskId(taskId, values);
55 }
56
UpdateDownloadTaskIdByTaskId(const std::string & taskId,const std::string & downloadTaskId)57 bool FirmwareTaskOperator::UpdateDownloadTaskIdByTaskId(const std::string &taskId, const std::string &downloadTaskId)
58 {
59 NativeRdb::ValuesBucket values;
60 values.PutString(COLUMN_TASK_DOWNLOAD_TASK_ID, downloadTaskId);
61 return UpdateByTaskId(taskId, values);
62 }
63
UpdateCombinationTypeByTaskId(const std::string & taskId,const CombinationType & combinationType)64 bool FirmwareTaskOperator::UpdateCombinationTypeByTaskId(const std::string &taskId,
65 const CombinationType &combinationType)
66 {
67 NativeRdb::ValuesBucket values;
68 values.PutInt(COLUMN_TASK_COMBINATION_TYPE, CAST_INT(combinationType));
69 return UpdateByTaskId(taskId, values);
70 }
71
UpdateDownloadModeByTaskId(const std::string & taskId,DownloadMode downloadMode)72 bool FirmwareTaskOperator::UpdateDownloadModeByTaskId(const std::string &taskId, DownloadMode downloadMode)
73 {
74 NativeRdb::ValuesBucket values;
75 values.PutInt(COLUMN_TASK_DOWNLOAD_MODE, CAST_INT(downloadMode));
76 return UpdateByTaskId(taskId, values);
77 }
78
UpdateDownloadAllowNetworkByTaskId(const std::string & taskId,NetType downloadAllowNetwork)79 bool FirmwareTaskOperator::UpdateDownloadAllowNetworkByTaskId(const std::string &taskId, NetType downloadAllowNetwork)
80 {
81 NativeRdb::ValuesBucket values;
82 values.PutInt(COLUMN_TASK_DOWNLOAD_ALLOW_NETWORK, CAST_INT(downloadAllowNetwork));
83 return UpdateByTaskId(taskId, values);
84 }
85
UpdateDownloadOrderByTaskId(const std::string & taskId,Order downloadOrder)86 bool FirmwareTaskOperator::UpdateDownloadOrderByTaskId(const std::string &taskId, Order downloadOrder)
87 {
88 NativeRdb::ValuesBucket values;
89 values.PutInt(COLUMN_TASK_DOWNLOAD_ORDER, CAST_INT(downloadOrder));
90 return UpdateByTaskId(taskId, values);
91 }
92
UpdateDownloadOptionByTaskId(const std::string & taskId,DownloadMode downloadMode,NetType downloadAllowNetwork,Order downloadOrder)93 bool FirmwareTaskOperator::UpdateDownloadOptionByTaskId(
94 const std::string &taskId, DownloadMode downloadMode, NetType downloadAllowNetwork, Order downloadOrder)
95 {
96 NativeRdb::ValuesBucket values;
97 values.PutInt(COLUMN_TASK_DOWNLOAD_MODE, CAST_INT(downloadMode));
98 values.PutInt(COLUMN_TASK_DOWNLOAD_ALLOW_NETWORK, CAST_INT(downloadAllowNetwork));
99 values.PutInt(COLUMN_TASK_DOWNLOAD_ORDER, CAST_INT(downloadOrder));
100 return UpdateByTaskId(taskId, values);
101 }
102
UpdateUpgradeModeByTaskId(const std::string & taskId,UpgradeMode upgradeMode)103 bool FirmwareTaskOperator::UpdateUpgradeModeByTaskId(const std::string &taskId, UpgradeMode upgradeMode)
104 {
105 NativeRdb::ValuesBucket values;
106 values.PutInt(COLUMN_TASK_UPGRADE_MODE, CAST_INT(upgradeMode));
107 return UpdateByTaskId(taskId, values);
108 }
109
UpdateUpgradeOrderByTaskId(const std::string & taskId,Order upgradeOrder)110 bool FirmwareTaskOperator::UpdateUpgradeOrderByTaskId(const std::string &taskId, Order upgradeOrder)
111 {
112 NativeRdb::ValuesBucket values;
113 values.PutInt(COLUMN_TASK_UPGRADE_ORDER, CAST_INT(upgradeOrder));
114 return UpdateByTaskId(taskId, values);
115 }
116
UpdateByTaskId(const std::string & taskId,const NativeRdb::ValuesBucket & values)117 bool FirmwareTaskOperator::UpdateByTaskId(const std::string &taskId, const NativeRdb::ValuesBucket &values)
118 {
119 OHOS::NativeRdb::RdbPredicates predicates(GetTableName());
120 predicates.EqualTo(COLUMN_TASK_TASK_ID, taskId);
121 return TableBaseOperator::Update(values, predicates);
122 }
123 } // namespace UpdateEngine
124 } // namespace OHOS
125