/*
 * 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.
 */

#ifndef OHOS_CLOUDISK_RDB_TRANSACTION_H
#define OHOS_CLOUDISK_RDB_TRANSACTION_H

#include <atomic>
#include <condition_variable>
#include <cstdint>
#include <mutex>

#include "rdb_store.h"

namespace OHOS {
namespace FileManagement {
namespace CloudDisk {
enum TRANS_DB_IDX {
    PHOTO_RDB_IDX = 0,
    CLOUDDISK_RDB_IDX = 1,
};
static constexpr uint32_t RDB_NUM = 2;
/**
 * This class is used for database transaction creation, commit, and rollback
 * The usage of class is as follows:
 *   1. initialize TransactionOperations object with a rdb instance
 *          (for example: TranscationOperations opt(rdb))
 *   2. After init opt, you need call Start() function to start transaction
 *          int32_t err = opt.Start();
 *          if err != E_OK, transaction init failed
 *   3. If you need to commit transaction, then use
 *          int32_t err = opt.Finish();
 *          if err != E_OK, transaction commit failed and auto rollback
 *   4. If TransactionOperations is destructed without successfully finish, it will be auto rollback
 */
class TransactionOperations {
public:
    TransactionOperations(const std::shared_ptr<OHOS::NativeRdb::RdbStore> &rdbStore);
    ~TransactionOperations();
    std::pair<int32_t, std::shared_ptr<NativeRdb::Transaction>> Start(
        NativeRdb::Transaction::TransactionType type = NativeRdb::Transaction::EXCLUSIVE);
    void Finish();

private:
    int32_t BeginTransaction(NativeRdb::Transaction::TransactionType type);
    int32_t TransactionCommit();
    int32_t TransactionRollback();

    std::shared_ptr<OHOS::NativeRdb::RdbStore> rdbStore_;
    std::shared_ptr<OHOS::NativeRdb::Transaction> transaction_;

    bool isStart = false;
    bool isFinish = false;
};
} // namespace CloudDisk
} // namespace FileManagement
} // namespace OHOS
#endif // OHOS_CLOUDDISK_RDB_TRANSACTION_H