diff options
Diffstat (limited to 'storage/rocksdb/rdb_threads.cc')
-rw-r--r-- | storage/rocksdb/rdb_threads.cc | 83 |
1 files changed, 83 insertions, 0 deletions
diff --git a/storage/rocksdb/rdb_threads.cc b/storage/rocksdb/rdb_threads.cc new file mode 100644 index 00000000..06683d6e --- /dev/null +++ b/storage/rocksdb/rdb_threads.cc @@ -0,0 +1,83 @@ +/* + Portions Copyright (c) 2015-Present, Facebook, Inc. + Portions Copyright (c) 2012, Monty Program 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 */ + +#ifdef USE_PRAGMA_IMPLEMENTATION +#pragma implementation // gcc: Class implementation +#endif + +#include <my_global.h> + +/* The C++ file's header */ +#include "./rdb_threads.h" + +namespace myrocks { + +void *Rdb_thread::thread_func(void *const thread_ptr) { + DBUG_ASSERT(thread_ptr != nullptr); + Rdb_thread *const thread = static_cast<Rdb_thread *>(thread_ptr); + if (!thread->m_run_once.exchange(true)) { + thread->setname(); + thread->run(); + thread->uninit(); + } + return nullptr; +} + +void Rdb_thread::init( +#ifdef HAVE_PSI_INTERFACE + my_core::PSI_mutex_key stop_bg_psi_mutex_key, + my_core::PSI_cond_key stop_bg_psi_cond_key +#endif +) { + DBUG_ASSERT(!m_run_once); + mysql_mutex_init(stop_bg_psi_mutex_key, &m_signal_mutex, MY_MUTEX_INIT_FAST); + mysql_cond_init(stop_bg_psi_cond_key, &m_signal_cond, nullptr); +} + +void Rdb_thread::uninit() { + mysql_mutex_destroy(&m_signal_mutex); + mysql_cond_destroy(&m_signal_cond); +} + +int Rdb_thread::create_thread(const std::string &thread_name +#ifdef HAVE_PSI_INTERFACE + , + PSI_thread_key background_psi_thread_key +#endif +) { + // Make a copy of the name so we can return without worrying that the + // caller will free the memory + m_name = thread_name; + + return mysql_thread_create(background_psi_thread_key, &m_handle, nullptr, + thread_func, this); + +} + +void Rdb_thread::signal(const bool stop_thread) { + RDB_MUTEX_LOCK_CHECK(m_signal_mutex); + + if (stop_thread) { + m_stop = true; + } + + mysql_cond_signal(&m_signal_cond); + + RDB_MUTEX_UNLOCK_CHECK(m_signal_mutex); +} + +} // namespace myrocks |