From 06eaf7232e9a920468c0f8d74dcf2fe8b555501c Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Sat, 13 Apr 2024 14:24:36 +0200 Subject: Adding upstream version 1:10.11.6. Signed-off-by: Daniel Baumann --- storage/rocksdb/rdb_cf_manager.h | 108 +++++++++++++++++++++++++++++++++++++++ 1 file changed, 108 insertions(+) create mode 100644 storage/rocksdb/rdb_cf_manager.h (limited to 'storage/rocksdb/rdb_cf_manager.h') diff --git a/storage/rocksdb/rdb_cf_manager.h b/storage/rocksdb/rdb_cf_manager.h new file mode 100644 index 00000000..cf7b3d6c --- /dev/null +++ b/storage/rocksdb/rdb_cf_manager.h @@ -0,0 +1,108 @@ +/* + 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 +#include +#include + +/* 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 m_cf_name_map; + std::map m_cf_id_map; + + mutable mysql_mutex_t m_mutex; + + std::unique_ptr 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 &&cf_options, + std::vector *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 get_cf_names(void) const; + + /* Used to iterate over column families */ + std::vector 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 -- cgit v1.2.3