1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
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 <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
|