db.h

/* Copyright (c) 2021 Xie Meiyi(xiemeiyi@hust.edu.cn) and OceanBase and/or its affiliates. All rights reserved.
miniob is licensed under Mulan PSL v2.
You can use this software according to the terms and conditions of the Mulan PSL v2.
You may obtain a copy of Mulan PSL v2 at:
         http://license.coscl.org.cn/MulanPSL2
THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
See the Mulan PSL v2 for more details. */
 
//
// Created by Meiyi & Longda & Wangyunlai on 2021/5/12.
//
 
#pragma once
 
#include "common/sys/rc.h"
#include "common/lang/vector.h"
#include "common/lang/string.h"
#include "common/lang/unordered_map.h"
#include "common/lang/memory.h"
#include "common/lang/span.h"
#include "sql/parser/parse_defs.h"
#include "storage/buffer/disk_buffer_pool.h"
#include "storage/clog/disk_log_handler.h"
#include "storage/buffer/double_write_buffer.h"
#include "oblsm/include/ob_lsm.h"
 
class Table;
class LogHandler;
class BufferPoolManager;
class TrxKit;
 
class Db
{
public:
  Db() = default;
  ~Db();
 
  RC init(const char *name, const char *dbpath, const char *trx_kit_name, const char *log_handler_name,
      const char *storage_engine = "heap");
 
  RC create_table(const char *table_name, span<const AttrInfoSqlNode> attributes, const vector<string> &primary_keys,
      const StorageFormat storage_format = StorageFormat::ROW_FORMAT);
 
  Table *find_table(const char *table_name) const;
  Table *find_table(int32_t table_id) const;
 
  const char *name() const;
 
  void all_tables(vector<string> &table_names) const;
 
  RC sync();
 
  LogHandler &log_handler();
 
  BufferPoolManager &buffer_pool_manager();
 
  TrxKit &trx_kit();
 
  string path() const { return path_; }
 
  oceanbase::ObLsm *lsm() { return lsm_; }
 
private:
  RC open_all_tables();
  RC recover();
 
  RC init_meta();
  RC flush_meta();
 
  RC init_dblwr_buffer();
 
  StorageEngine get_storage_engine()
  {
    StorageEngine engine = StorageEngine::UNKNOWN_ENGINE;
    if (storage_engine_.length() == 0) {
      engine = StorageEngine::HEAP;
    } else if (0 == strcasecmp(storage_engine_.c_str(), "heap")) {
      engine = StorageEngine::HEAP;
    } else if (0 == strcasecmp(storage_engine_.c_str(), "lsm")) {
      engine = StorageEngine::LSM;
    } else {
      engine = StorageEngine::UNKNOWN_ENGINE;
    }
    return engine;
  }
 
private:
  string                         name_;                 
  string                         path_;                 
  unordered_map<string, Table *> opened_tables_;        
  unique_ptr<BufferPoolManager>  buffer_pool_manager_;  
  unique_ptr<LogHandler>         log_handler_;          
  unique_ptr<TrxKit>             trx_kit_;              
  oceanbase::ObLsm              *lsm_;                  
 
  int32_t next_table_id_ = 0;
 
  LSN    check_point_lsn_ = 0;  
  string storage_engine_;
};