/*
 * Copyright (C) 2024 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.
 */

#include "ringtone_rdb_transaction.h"

#include "ringtone_log.h"

namespace OHOS::Media {
using namespace std;
constexpr int32_t E_HAS_DB_ERROR = -222;
constexpr int32_t E_OK = 0;
constexpr int RDB_TRANSACTION_WAIT_MS = 1000;
std::mutex RingtoneRdbTransaction::transactionMutex_;
std::condition_variable RingtoneRdbTransaction::transactionCV_;
std::atomic<bool> RingtoneRdbTransaction::isInTransaction_(false);
constexpr int32_t MAX_TRY_TIMES = 30;
constexpr int32_t TRANSACTION_WAIT_INTERVAL = 50; // in milliseconds.

RingtoneRdbTransaction::RingtoneRdbTransaction(
    const shared_ptr<OHOS::NativeRdb::RdbStore> &rdbStore) : rdbStore_(rdbStore) {}

RingtoneRdbTransaction::~RingtoneRdbTransaction()
{
    if (isStart && !isFinish) {
        TransactionRollback();
    }
}

int32_t RingtoneRdbTransaction::Start()
{
    if (isStart || isFinish) {
        return 0;
    }
    int32_t errCode = BeginTransaction();
    if (errCode == 0) {
        isStart = true;
    }
    return errCode;
}

void RingtoneRdbTransaction::Finish()
{
    if (!isStart) {
        return;
    }
    if (!isFinish) {
        int32_t ret = TransactionCommit();
        if (ret == 0) {
            isFinish = true;
        }
    }
}

int32_t RingtoneRdbTransaction::BeginTransaction()
{
    if (rdbStore_ == nullptr) {
        RINGTONE_ERR_LOG("Pointer rdbStore_ is nullptr. Maybe it didn't init successfully.");
        return E_HAS_DB_ERROR;
    }
    RINGTONE_INFO_LOG("Start transaction");

    unique_lock<mutex> cvLock(transactionMutex_);
    if (isInTransaction_.load()) {
        transactionCV_.wait_for(cvLock, chrono::milliseconds(RDB_TRANSACTION_WAIT_MS),
            [this] () { return !(this->isInTransaction_.load()); });
    }

    int curTryTime = 0;
    while (curTryTime < MAX_TRY_TIMES) {
        if (rdbStore_->IsInTransaction()) {
            this_thread::sleep_for(chrono::milliseconds(TRANSACTION_WAIT_INTERVAL));
            if (isInTransaction_.load() || rdbStore_->IsInTransaction()) {
                curTryTime++;
                RINGTONE_INFO_LOG("RdbStore is in transaction, try %{public}d times...", curTryTime);
                continue;
            }
        }

        int32_t errCode = rdbStore_->BeginTransaction();
        if (errCode == SQLITE3_DATABASE_LOCKER) {
            curTryTime++;
            RINGTONE_ERR_LOG("Sqlite database file is locked! try %{public}d times...", curTryTime);
            continue;
        } else if (errCode != NativeRdb::E_OK) {
            RINGTONE_ERR_LOG("Start Transaction failed, errCode=%{public}d", errCode);
            isInTransaction_.store(false);
            transactionCV_.notify_one();
            return E_HAS_DB_ERROR;
        } else {
            isInTransaction_.store(true);
            return E_OK;
        }
    }

    RINGTONE_ERR_LOG("RdbStore is still in transaction after try %{public}d times, abort.", MAX_TRY_TIMES);
    return E_HAS_DB_ERROR;
}

int32_t RingtoneRdbTransaction::TransactionCommit()
{
    if (rdbStore_ == nullptr) {
        return E_HAS_DB_ERROR;
    }
    RINGTONE_INFO_LOG("Try commit transaction");

    if (!(isInTransaction_.load()) || !(rdbStore_->IsInTransaction())) {
        RINGTONE_ERR_LOG("no transaction now");
        return E_HAS_DB_ERROR;
    }

    int32_t errCode = rdbStore_->Commit();
    isInTransaction_.store(false);
    transactionCV_.notify_all();
    if (errCode != NativeRdb::E_OK) {
        RINGTONE_ERR_LOG("commit failed, errCode=%{public}d", errCode);
        return E_HAS_DB_ERROR;
    }

    return E_OK;
}

int32_t RingtoneRdbTransaction::TransactionRollback()
{
    if (rdbStore_ == nullptr) {
        return E_HAS_DB_ERROR;
    }
    RINGTONE_INFO_LOG("Try rollback transaction");

    if (!(isInTransaction_.load()) || !(rdbStore_->IsInTransaction())) {
        RINGTONE_ERR_LOG("no transaction now");
        return E_HAS_DB_ERROR;
    }

    int32_t errCode = rdbStore_->RollBack();
    isInTransaction_.store(false);
    transactionCV_.notify_all();
    if (errCode != NativeRdb::E_OK) {
        RINGTONE_ERR_LOG("rollback failed, errCode=%{public}d", errCode);
        return E_HAS_DB_ERROR;
    }

    return E_OK;
}
} // namespace OHOS::Media