mariadb/storage/rocksdb/rdb_cf_manager.h
Sergei Petrunia 9ab0d7b4e9 Merge from MyRocks upstream:
Copy of
    commit dcd9379eb5707bc7514a2ff4d9127790356505cb
    Author: Manuel Ung <mung@fb.com>
    Date:   Fri Jun 14 10:38:17 2019 -0700

        Skip valgrind for rocksdb.force_shutdown

        Summary:
        This test does unclean shutdown, and leaks memory.

        Squash with: D15749084

        Reviewed By: hermanlee

        Differential Revision: D15828957

        fbshipit-source-id: 30541455d74
2019-06-16 00:28:33 +03:00

108 lines
3.4 KiB
C++

/*
Copyright (c) 2014, SkySQL Ab
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., 51 Franklin Street, Fifth Floor, Boston, MA 02111-1301 USA */
#pragma once
/* C++ system header files */
#include <map>
#include <string>
#include <vector>
/* MySQL header files */
#include "./sql_class.h"
/* RocksDB header files */
#include "rocksdb/db.h"
/* MyRocks header files */
#include "./rdb_cf_options.h"
namespace myrocks {
/*
We need a Column Family (CF) manager. Its functions:
- create column families (synchronized, don't create the same twice)
- keep count in each column family.
= the count is kept on-disk.
= there are no empty CFs. initially count=1.
= then, when doing DDL, we increase or decrease it.
(atomicity is maintained by being in the same WriteBatch with DDLs)
= if DROP discovers that now count=0, it removes the CF.
Current state is:
- CFs are created in a synchronized way. We can't remove them, yet.
*/
class Rdb_cf_manager {
std::map<std::string, rocksdb::ColumnFamilyHandle *> m_cf_name_map;
std::map<uint32_t, rocksdb::ColumnFamilyHandle *> m_cf_id_map;
mutable mysql_mutex_t m_mutex;
std::unique_ptr<Rdb_cf_options> m_cf_options = nullptr;
public:
Rdb_cf_manager(const Rdb_cf_manager &) = delete;
Rdb_cf_manager &operator=(const Rdb_cf_manager &) = delete;
Rdb_cf_manager() = default;
static bool is_cf_name_reverse(const char *const name);
/*
This is called right after the DB::Open() call. The parameters describe
column
families that are present in the database. The first CF is the default CF.
*/
void init(std::unique_ptr<Rdb_cf_options> &&cf_options,
std::vector<rocksdb::ColumnFamilyHandle *> *const handles);
void cleanup();
/*
Used by CREATE TABLE.
- cf_name=nullptr means use default column family
*/
rocksdb::ColumnFamilyHandle *get_or_create_cf(rocksdb::DB *const rdb,
const std::string &cf_name);
/* Used by table open */
rocksdb::ColumnFamilyHandle *get_cf(
const std::string &cf_name, const bool lock_held_by_caller = false) const;
/* Look up cf by id; used by datadic */
rocksdb::ColumnFamilyHandle *get_cf(const uint32_t id) const;
/* Used to iterate over column families for show status */
std::vector<std::string> get_cf_names(void) const;
/* Used to iterate over column families */
std::vector<rocksdb::ColumnFamilyHandle *> get_all_cf(void) const;
/* Used to delete cf by name */
int drop_cf(const std::string &cf_name);
void get_cf_options(const std::string &cf_name,
rocksdb::ColumnFamilyOptions *const opts)
MY_ATTRIBUTE((__nonnull__)) {
m_cf_options->get_cf_options(cf_name, opts);
}
void update_options_map(const std::string &cf_name,
const std::string &updated_options) {
m_cf_options->update(cf_name, updated_options);
}
};
} // namespace myrocks