mirror of
https://github.com/MariaDB/server.git
synced 2025-01-31 02:51:44 +01:00
ebfc4e6ad0
copy of
commit 86587affafe77ef555f7c3839839de44f0f203f3
Author: Tian Xia <tianx@fb.com>
Date: Tue Oct 4 10:01:52 2016 -0700
Allow filtering of show commands through admission control
101 lines
3.1 KiB
C++
101 lines
3.1 KiB
C++
/*
|
|
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
|