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 #ifndef OMIT_MULTI_VER
17 #include "multi_ver_storage_engine.h"
18 
19 #include "multi_ver_storage_executor.h"
20 
21 #include "db_errno.h"
22 
23 namespace DistributedDB {
MultiVerStorageEngine()24 MultiVerStorageEngine::MultiVerStorageEngine()
25     : kvDB_(nullptr),
26       dataStorage_(nullptr),
27       commitStorage_(nullptr),
28       kvDataStorage_(nullptr)
29 {}
30 
~MultiVerStorageEngine()31 MultiVerStorageEngine::~MultiVerStorageEngine()
32 {
33     kvDB_ = nullptr;
34     dataStorage_ = nullptr;
35     commitStorage_ = nullptr;
36     kvDataStorage_ = nullptr;
37 }
38 
InitDatabases(IKvDB * kvDB,IKvDBMultiVerDataStorage * dataStorage,IKvDBCommitStorage * commitStorage,MultiVerKvDataStorage * kvDataStorage,const StorageEngineAttr & poolSize)39 int MultiVerStorageEngine::InitDatabases(IKvDB *kvDB, IKvDBMultiVerDataStorage *dataStorage,
40     IKvDBCommitStorage *commitStorage, MultiVerKvDataStorage *kvDataStorage, const StorageEngineAttr &poolSize)
41 {
42     if (StorageEngine::CheckEngineAttr(poolSize)) {
43         return -E_INVALID_ARGS;
44     }
45     if (kvDB == nullptr || dataStorage == nullptr ||
46         commitStorage == nullptr || kvDataStorage == nullptr) {
47         return -E_INVALID_DB;
48     }
49     engineAttr_ = poolSize;
50     kvDB_ = kvDB;
51     dataStorage_ = dataStorage;
52     commitStorage_ = commitStorage;
53     kvDataStorage_ = kvDataStorage;
54     return Init();
55 }
56 
CreateNewExecutor(bool isWrite,StorageExecutor * & handle)57 int MultiVerStorageEngine::CreateNewExecutor(bool isWrite, StorageExecutor *&handle)
58 {
59     handle = new (std::nothrow) MultiVerStorageExecutor(kvDB_, dataStorage_, commitStorage_, kvDataStorage_, isWrite);
60     if (handle == nullptr) {
61         return -E_OUT_OF_MEMORY;
62     }
63     return E_OK;
64 }
65 }
66 #endif