1 /*
2  * Copyright (c) 2021 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 "single_ver_database_upgrader.h"
17 #include "db_errno.h"
18 #include "log_print.h"
19 #include "version.h"
20 
21 namespace DistributedDB {
Upgrade()22 int SingleVerDatabaseUpgrader::Upgrade()
23 {
24     int errCode = GetDatabaseVersion(dbVersion_);
25     if (errCode != E_OK) {
26         LOGE("[SingleUp][Upgrade] GetVersion error:%d.", errCode);
27         return errCode;
28     }
29     if (dbVersion_ > SINGLE_VER_STORE_VERSION_CURRENT) {
30         LOGE("[SingleUp][Upgrade] DbVersion=%d is newer.", dbVersion_);
31         return -E_VERSION_NOT_SUPPORT;
32     }
33     if (dbVersion_ != SINGLE_VER_STORE_VERSION_CURRENT) {
34         LOGI("[SingleUp][Upgrade] from %d to %d.", dbVersion_, SINGLE_VER_STORE_VERSION_CURRENT);
35     }
36     errCode = BeginUpgrade();
37     if (errCode != E_OK) {
38         LOGE("[SingleUp][Upgrade] Begin error:%d.", errCode);
39         return errCode;
40     }
41     errCode = ExecuteUpgrade();
42     if (errCode != E_OK) {
43         LOGE("[SingleUp][Upgrade] Execute error:%d.", errCode);
44         EndUpgrade(false);
45         return errCode;
46     }
47 
48     if (dbVersion_ < SINGLE_VER_STORE_VERSION_CURRENT) {
49         errCode = SetDatabaseVersion(SINGLE_VER_STORE_VERSION_CURRENT);
50         if (errCode != E_OK) {
51             LOGE("[SingleUp][Upgrade] SetVersion error:%d.", errCode);
52             EndUpgrade(false);
53             return errCode;
54         }
55     }
56 
57     errCode = EndUpgrade(true);
58     if (errCode != E_OK) {
59         LOGE("[SingleUp][Upgrade] End error:%d.", errCode);
60         return errCode;
61     }
62 
63     return E_OK;
64 }
65 
ExecuteUpgrade()66 int SingleVerDatabaseUpgrader::ExecuteUpgrade()
67 {
68     if (dbVersion_ <= SINGLE_VER_STORE_VERSION_CURRENT) {
69         return UpgradeFromDatabaseVersion(dbVersion_);
70     }
71     return E_OK;
72 }
73 } // namespace DistributedDB
74