update ceph source to reef 18.1.2

[ceph.git] / ceph / src / rocksdb / monitoring / iostats_context_imp.h

diff --git a/ceph/src/rocksdb/monitoring/iostats_context_imp.h b/ceph/src/rocksdb/monitoring/iostats_context_imp.h
index 1a4d7bd0462e1f02076145ed1b5dbc8677e3c0fd..a0b4292dfd1d5142b5801992bf3715e26c080f47 100644 (file)
--- a/ceph/src/rocksdb/monitoring/iostats_context_imp.h
+++ b/ceph/src/rocksdb/monitoring/iostats_context_imp.h
@@ -1,45 +1,51 @@
 //  Copyright (c) 2011-present, Facebook, Inc.  All rights reserved.
-//  This source code is licensed under the BSD-style license found in the
-//  LICENSE file in the root directory of this source tree. An additional grant
-//  of patent rights can be found in the PATENTS file in the same directory.
+//  This source code is licensed under both the GPLv2 (found in the
+//  COPYING file in the root directory) and Apache 2.0 License
+//  (found in the LICENSE.Apache file in the root directory).
 //
 #pragma once
 #include "monitoring/perf_step_timer.h"
 #include "rocksdb/iostats_context.h"
 
-#ifndef IOS_CROSS_COMPILE
+#if !defined(NIOSTATS_CONTEXT)
+namespace ROCKSDB_NAMESPACE {
+extern thread_local IOStatsContext iostats_context;
+}  // namespace ROCKSDB_NAMESPACE
 
 // increment a specific counter by the specified value
-#define IOSTATS_ADD(metric, value)     \
-  (iostats_context.metric += value)
-
-// Increase metric value only when it is positive
-#define IOSTATS_ADD_IF_POSITIVE(metric, value)   \
-  if (value > 0) { IOSTATS_ADD(metric, value); }
+#define IOSTATS_ADD(metric, value)        \
+  if (!iostats_context.disable_iostats) { \
+    iostats_context.metric += value;      \
+  }
 
 // reset a specific counter to zero
-#define IOSTATS_RESET(metric)          \
-  (iostats_context.metric = 0)
+#define IOSTATS_RESET(metric) (iostats_context.metric = 0)
 
 // reset all counters to zero
-#define IOSTATS_RESET_ALL()                        \
-  (iostats_context.Reset())
+#define IOSTATS_RESET_ALL() (iostats_context.Reset())
 
-#define IOSTATS_SET_THREAD_POOL_ID(value)      \
+#define IOSTATS_SET_THREAD_POOL_ID(value) \
   (iostats_context.thread_pool_id = value)
 
-#define IOSTATS_THREAD_POOL_ID()               \
-  (iostats_context.thread_pool_id)
+#define IOSTATS_THREAD_POOL_ID() (iostats_context.thread_pool_id)
+
+#define IOSTATS(metric) (iostats_context.metric)
 
-#define IOSTATS(metric)                        \
-  (iostats_context.metric)
+// Declare and set start time of the timer
+#define IOSTATS_TIMER_GUARD(metric)                                     \
+  PerfStepTimer iostats_step_timer_##metric(&(iostats_context.metric)); \
+  iostats_step_timer_##metric.Start();
 
 // Declare and set start time of the timer
-#define IOSTATS_TIMER_GUARD(metric)                                       \
-  PerfStepTimer iostats_step_timer_ ## metric(&(iostats_context.metric));  \
-  iostats_step_timer_ ## metric.Start();
+#define IOSTATS_CPU_TIMER_GUARD(metric, clock)         \
+  PerfStepTimer iostats_step_timer_##metric(           \
+      &(iostats_context.metric), clock, true,          \
+      PerfLevel::kEnableTimeAndCPUTimeExceptForMutex); \
+  iostats_step_timer_##metric.Start();
+
+#define IOSTATS_SET_DISABLE(disable) (iostats_context.disable_iostats = disable)
 
-#else  // IOS_CROSS_COMPILE
+#else  // !NIOSTATS_CONTEXT
 
 #define IOSTATS_ADD(metric, value)
 #define IOSTATS_ADD_IF_POSITIVE(metric, value)
@@ -48,7 +54,9 @@
 #define IOSTATS_SET_THREAD_POOL_ID(value)
 #define IOSTATS_THREAD_POOL_ID()
 #define IOSTATS(metric) 0
+#define IOSTATS_SET_DISABLE(disable)
 
 #define IOSTATS_TIMER_GUARD(metric)
+#define IOSTATS_CPU_TIMER_GUARD(metric, clock) static_cast<void>(clock)
 
-#endif  // IOS_CROSS_COMPILE
+#endif  // !NIOSTATS_CONTEXT