summaryrefslogtreecommitdiffstats
path: root/storage/rocksdb/rdb_io_watchdog.h
diff options
context:
space:
mode:
Diffstat (limited to 'storage/rocksdb/rdb_io_watchdog.h')
-rw-r--r--storage/rocksdb/rdb_io_watchdog.h119
1 files changed, 119 insertions, 0 deletions
diff --git a/storage/rocksdb/rdb_io_watchdog.h b/storage/rocksdb/rdb_io_watchdog.h
new file mode 100644
index 00000000..8ee5b1f6
--- /dev/null
+++ b/storage/rocksdb/rdb_io_watchdog.h
@@ -0,0 +1,119 @@
+/*
+ Copyright (c) 2017, 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., 51 Franklin Street, Fifth Floor, Boston, MA 02111-1301 USA */
+
+#pragma once
+
+/* C++ standard header files */
+#include <signal.h>
+#include <stdlib.h>
+#include <string.h>
+#include <time.h>
+#include <atomic>
+#include <string>
+#include <vector>
+
+/* MySQL header files */
+#include "./my_global.h"
+#include "./my_stacktrace.h"
+
+/* MyRocks header files */
+#include "./rdb_utils.h"
+
+namespace myrocks {
+
+// Rdb_io_watchdog does not support Windows ATM.
+#ifdef HAVE_TIMER_DELETE
+
+class Rdb_io_watchdog {
+ const int RDB_IO_WRITE_BUFFER_SIZE = 4096;
+ const char *const RDB_IO_DUMMY_FILE_NAME = "myrocks_io_watchdog_write_file";
+
+ private:
+ timer_t m_io_check_timer, m_io_check_watchdog_timer;
+ std::atomic<bool> m_io_in_progress;
+ std::vector<std::string> m_dirs_to_check;
+ uint32_t m_write_timeout;
+ mysql_mutex_t m_reset_mutex;
+ char *m_buf;
+
+ int check_write_access(const std::string &dirname) const;
+ void io_check_callback(union sigval timer_data);
+ void expire_io_callback(union sigval timer_data);
+
+ int stop_timers() {
+ int ret = 0;
+
+ if (m_io_check_watchdog_timer) {
+ ret = timer_delete(m_io_check_watchdog_timer);
+
+ if (!ret) {
+ m_io_check_watchdog_timer = nullptr;
+ }
+ }
+
+ if (m_io_check_timer && !ret) {
+ ret = timer_delete(m_io_check_timer);
+
+ if (!ret) {
+ m_io_check_timer = nullptr;
+ }
+ }
+
+ return ret;
+ }
+
+ static void io_check_callback_wrapper(union sigval timer_data) {
+ Rdb_io_watchdog *io_watchdog =
+ static_cast<Rdb_io_watchdog *>(timer_data.sival_ptr);
+ DBUG_ASSERT(io_watchdog != nullptr);
+
+ io_watchdog->io_check_callback(timer_data);
+ }
+
+ static void expire_io_callback_wrapper(union sigval timer_data) {
+ Rdb_io_watchdog *io_watchdog =
+ static_cast<Rdb_io_watchdog *>(timer_data.sival_ptr);
+ DBUG_ASSERT(io_watchdog != nullptr);
+
+ io_watchdog->expire_io_callback(timer_data);
+ }
+
+ public:
+ explicit Rdb_io_watchdog(std::vector<std::string> &&directories)
+ : m_io_check_timer(nullptr),
+ m_io_check_watchdog_timer(nullptr),
+ m_io_in_progress(false),
+ m_dirs_to_check(std::move(directories)),
+ m_buf(nullptr) {
+ DBUG_ASSERT(m_dirs_to_check.size() > 0);
+ mysql_mutex_init(0, &m_reset_mutex, MY_MUTEX_INIT_FAST);
+ }
+
+ ~Rdb_io_watchdog() {
+ // We're shutting down. Ignore errors possibly coming from timer deletion.
+ static_cast<void>(stop_timers());
+ mysql_mutex_destroy(&m_reset_mutex);
+ free(m_buf);
+ }
+
+ int reset_timeout(const uint32_t write_timeout);
+
+ Rdb_io_watchdog(const Rdb_io_watchdog &) = delete;
+ Rdb_io_watchdog &operator=(const Rdb_io_watchdog &) = delete;
+};
+
+#endif
+} // namespace myrocks