OpenHarmony-v3.2.1-Release/s

/*
 * Copyright (c) 2021 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#ifdef RELATIONAL_STORE
#include "relational_store_instance.h"

#include <thread>
#include <algorithm>

#include "db_common.h"
#include "db_errno.h"
#include "sqlite_relational_store.h"
#include "log_print.h"

namespace DistributedDB {
RelationalStoreInstance *RelationalStoreInstance::instance_ = nullptr;
std::mutex RelationalStoreInstance::instanceLock_;

static std::mutex storeLock_;
static std::map<std::string, IRelationalStore *> dbs_;

RelationalStoreInstance::RelationalStoreInstance()
{}

RelationalStoreInstance *RelationalStoreInstance::GetInstance()
{
    std::lock_guard<std::mutex> lockGuard(instanceLock_);
    if (instance_ == nullptr) {
        instance_ = new (std::nothrow) RelationalStoreInstance();
        if (instance_ == nullptr) {
            LOGE("failed to new RelationalStoreManager!");
            return nullptr;
        }
    }
    return instance_;
}

int RelationalStoreInstance::ReleaseDataBaseConnection(RelationalStoreConnection *connection)
{
    if (connection == nullptr) {
        return -E_INVALID_DB;
    }
    auto manager = RelationalStoreInstance::GetInstance();
    if (manager == nullptr) {
        return -E_OUT_OF_MEMORY;
    }
    std::string identifier = connection->GetIdentifier();
    manager->EnterDBOpenCloseProcess(identifier);
    int errCode = connection->Close();
    manager->ExitDBOpenCloseProcess(identifier);

    if (errCode != E_OK) {
        LOGE("Release db connection failed. %d", errCode);
    }
    return errCode;
}

int RelationalStoreInstance::CheckDatabaseFileStatus(const std::string &id)
{
    std::lock_guard<std::mutex> lockGuard(storeLock_);
    if (dbs_.count(id) != 0 && dbs_[id] != nullptr) {
        return -E_BUSY;
    }
    return E_OK;
}

static IRelationalStore *GetFromCache(const RelationalDBProperties &properties, int &errCode)
{
    errCode = E_OK;
    std::string identifier = properties.GetStringProp(RelationalDBProperties::IDENTIFIER_DATA, "");
    std::lock_guard<std::mutex> lockGuard(storeLock_);
    auto iter = dbs_.find(identifier);
    if (iter == dbs_.end()) {
        errCode = -E_NOT_FOUND;
        return nullptr;
    }

    auto *db = iter->second;
    if (db == nullptr) {
        LOGE("Store cache is nullptr, there may be a logic error");
        errCode = -E_INTERNAL_ERROR;
        return nullptr;
    }
    db->IncObjRef(db);
    return db;
}

// Save to IKvDB to the global map
void RelationalStoreInstance::RemoveKvDBFromCache(const RelationalDBProperties &properties)
{
    std::string identifier = properties.GetStringProp(RelationalDBProperties::IDENTIFIER_DATA, "");
    std::lock_guard<std::mutex> lockGuard(storeLock_);
    dbs_.erase(identifier);
}

void RelationalStoreInstance::SaveRelationalDBToCache(IRelationalStore *store, const RelationalDBProperties &properties)
{
    std::string identifier = properties.GetStringProp(RelationalDBProperties::IDENTIFIER_DATA, "");
    std::lock_guard<std::mutex> lockGuard(storeLock_);
    if (dbs_.count(identifier) == 0) {
        dbs_.insert(std::pair<std::string, IRelationalStore *>(identifier, store));
    }
}

IRelationalStore *RelationalStoreInstance::OpenDatabase(const RelationalDBProperties &properties, int &errCode)
{
    auto db = new (std::nothrow) SQLiteRelationalStore();
    if (db == nullptr) {
        errCode = -E_OUT_OF_MEMORY;
        LOGE("Failed to get relational store! err:%d", errCode);
        return nullptr;
    }

    db->OnClose([this, properties]() {
        LOGI("Remove from the cache");
        this->RemoveKvDBFromCache(properties);
    });

    errCode = db->Open(properties);
    if (errCode != E_OK) {
        LOGE("Failed to open db! err:%d", errCode);
        RefObject::KillAndDecObjRef(db);
        return nullptr;
    }
    db->WakeUpSyncer();

    SaveRelationalDBToCache(db, properties);
    return db;
}

IRelationalStore *RelationalStoreInstance::GetDataBase(const RelationalDBProperties &properties, int &errCode)
{
    auto *db = GetFromCache(properties, errCode);
    if (db != nullptr) {
        LOGD("Get db from cache.");
        return db;
    }

    // file lock
    RelationalStoreInstance *manager = RelationalStoreInstance::GetInstance();
    if (manager == nullptr) {
        errCode = -E_OUT_OF_MEMORY;
        return nullptr;
    }

    db = manager->OpenDatabase(properties, errCode);
    if (errCode != E_OK) {
        LOGE("Create data base failed, errCode = [%d]", errCode);
    }
    return db;
}

namespace {
int CheckCompatibility(const RelationalDBProperties &prop, const RelationalDBProperties &existedProp)
{
    std::string canonicalDir = prop.GetStringProp(DBProperties::DATA_DIR, "");
    if (canonicalDir.empty() || canonicalDir != existedProp.GetStringProp(DBProperties::DATA_DIR, "")) {
        LOGE("Failed to check store path, the input path does not match with cached store.");
        return -E_INVALID_ARGS;
    }
    if (prop.GetIntProp(RelationalDBProperties::DISTRIBUTED_TABLE_MODE, DistributedTableMode::SPLIT_BY_DEVICE) !=
        existedProp.GetIntProp(RelationalDBProperties::DISTRIBUTED_TABLE_MODE, DistributedTableMode::SPLIT_BY_DEVICE)) {
        LOGE("Failed to check table mode.");
        return -E_INVALID_ARGS;
    }

    if (prop.IsEncrypted() != existedProp.IsEncrypted()) {
        LOGE("Failed to check cipher args.");
        return -E_INVALID_PASSWD_OR_CORRUPTED_DB;
    }
    if (prop.IsEncrypted() &&
        (prop.GetPasswd() != existedProp.GetPasswd() || prop.GetIterTimes() != existedProp.GetIterTimes() ||
         !DBCommon::IsSameCipher(prop.GetCipherType(), existedProp.GetCipherType()))) {
        LOGE("Failed to check cipher args.");
        return -E_INVALID_PASSWD_OR_CORRUPTED_DB;
    }
    return E_OK;
}
}

RelationalStoreConnection *RelationalStoreInstance::GetDatabaseConnection(const RelationalDBProperties &properties,
    int &errCode)
{
    std::string identifier = properties.GetStringProp(DBProperties::IDENTIFIER_DATA, "");
    LOGD("Begin to get [%s] database connection.", STR_MASK(DBCommon::TransferStringToHex(identifier)));
    RelationalStoreInstance *manager = RelationalStoreInstance::GetInstance();
    if (manager == nullptr) {
        errCode = -E_OUT_OF_MEMORY;
        return nullptr;
    }
    manager->EnterDBOpenCloseProcess(properties.GetStringProp(DBProperties::IDENTIFIER_DATA, ""));
    RelationalStoreConnection *connection = nullptr;
    IRelationalStore *db = GetDataBase(properties, errCode);
    if (db == nullptr) {
        LOGE("Failed to open the db:%d", errCode);
        goto END;
    }

    errCode = CheckCompatibility(properties, db->GetProperties());
    if (errCode != E_OK) {
        goto END;
    }

    connection = db->GetDBConnection(errCode);
    if (connection == nullptr) { // not kill db, Other operations like import may be used concurrently
        LOGE("Failed to get the db connect for delegate:%d", errCode);
    }

END:
    RefObject::DecObjRef(db); // restore the reference increased by the cache.
    manager->ExitDBOpenCloseProcess(properties.GetStringProp(DBProperties::IDENTIFIER_DATA, ""));
    return connection;
}

void RelationalStoreInstance::EnterDBOpenCloseProcess(const std::string &identifier)
{
    std::unique_lock<std::mutex> lock(relationalDBOpenMutex_);
    relationalDBOpenCondition_.wait(lock, [this, &identifier]() {
        return this->relationalDBOpenSet_.count(identifier) == 0;
    });
    (void)relationalDBOpenSet_.insert(identifier);
}

void RelationalStoreInstance::ExitDBOpenCloseProcess(const std::string &identifier)
{
    std::unique_lock<std::mutex> lock(relationalDBOpenMutex_);
    (void)relationalDBOpenSet_.erase(identifier);
    relationalDBOpenCondition_.notify_all();
}

void RelationalStoreInstance::Dump(int fd)
{
    std::lock_guard<std::mutex> autoLock(storeLock_);
    for (const auto &entry : dbs_) {
        RefObject::IncObjRef(entry.second);
        entry.second->Dump(fd);
        RefObject::DecObjRef(entry.second);
    }
}
} // namespace DistributedDB
#endif