/*
   Copyright (c) 2016, Facebook, Inc.

   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; version 2 of the License.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */

#pragma once

/* C++ standard header files */
#include <condition_variable>
#include <mutex>
#include <queue>
#include <string>
#include <thread>
#include <vector>

/* RocksDB header files */
#include "rocksdb/db.h"
#include "rocksdb/sst_file_writer.h"

// define RDB_SST_INFO_USE_THREAD /* uncomment to use threads */

namespace myrocks {

class Rdb_sst_file {
  Rdb_sst_file(const Rdb_sst_file& p)= delete;
  Rdb_sst_file& operator=(const Rdb_sst_file& p)= delete;

  rocksdb::DB*                 m_db;
  rocksdb::ColumnFamilyHandle* m_cf;
  const rocksdb::DBOptions&    m_db_options;
  rocksdb::SstFileWriter*      m_sst_file_writer;
  std::string                  m_name;

 public:
  Rdb_sst_file(rocksdb::DB* db, rocksdb::ColumnFamilyHandle* cf,
               const rocksdb::DBOptions& db_options, const std::string& name);
  ~Rdb_sst_file();

  rocksdb::Status open();
  rocksdb::Status put(const rocksdb::Slice& key, const rocksdb::Slice& value);
  rocksdb::Status commit();
};

class Rdb_sst_info {
  Rdb_sst_info(const Rdb_sst_info& p)= delete;
  Rdb_sst_info& operator=(const Rdb_sst_info& p)= delete;

  rocksdb::DB*                 m_db;
  rocksdb::ColumnFamilyHandle* m_cf;
  const rocksdb::DBOptions&    m_db_options;
  uint64_t                     m_curr_size;
  uint64_t                     m_max_size;
  uint                         m_sst_count;
  std::string                  m_error_msg;
  std::string                  m_prefix;
  static std::string           m_suffix;
#if defined(RDB_SST_INFO_USE_THREAD)
  std::queue<Rdb_sst_file*>    m_queue;
  std::mutex                   m_mutex;
  std::condition_variable      m_cond;
  std::thread*                 m_thread;
  bool                         m_finished;
#endif
  Rdb_sst_file*                m_sst_file;

  int open_new_sst_file();
  void close_curr_sst_file();
  void set_error_msg(const std::string& msg);

#if defined(RDB_SST_INFO_USE_THREAD)
  void run_thread();

  static void thread_fcn(void* object);
#endif

 public:
  Rdb_sst_info(rocksdb::DB* db, const std::string& tablename,
               const std::string& indexname, rocksdb::ColumnFamilyHandle* cf,
               const rocksdb::DBOptions& db_options);
  ~Rdb_sst_info();

  int put(const rocksdb::Slice& key, const rocksdb::Slice& value);
  int commit();

  const std::string& error_message() const { return m_error_msg; }

  static void init(rocksdb::DB* db);
};

}  // namespace myrocks