unittest_bit_alloc race fix

Signed-off-by: Ramesh Chander <Ramesh.Chander@sandisk.com>
Reviewed-by: Sage Weil <sage@redhat.com>

diff --git a/src/os/bluestore/BitAllocator.cc b/src/os/bluestore/BitAllocator.cc
index 9f241dbedf62fa188e32174097627e06c40f3b6a..8edd47919d89cee1931354985bfa06395e2c96f5 100644 (file)
--- a/src/os/bluestore/BitAllocator.cc
+++ b/src/os/bluestore/BitAllocator.cc
@@ -827,6 +827,12 @@ int64_t BitMapAreaIN::get_used_blocks()
   return m_used_blocks;
 }
 
+int64_t BitMapAreaIN::get_used_blocks_adj()
+{
+  std::lock_guard<std::mutex> l(m_blocks_lock);
+  return m_used_blocks - m_reserved_blocks;
+}
+
 bool BitMapAreaIN::reserve_blocks(int64_t num)
 {
   bool res = false;

diff --git a/src/os/bluestore/BitAllocator.h b/src/os/bluestore/BitAllocator.h
index a61874ea6511e042690369d9edf4b5b80ca18af8..2ce6b3c70adadffc4890b7661883cf7b01b31105 100644 (file)
--- a/src/os/bluestore/BitAllocator.h
+++ b/src/os/bluestore/BitAllocator.h
@@ -416,6 +416,7 @@ public:
   virtual void unreserve(int64_t num_blocks, int64_t allocated);
   virtual int64_t get_reserved_blocks();
   virtual int64_t get_used_blocks();
+  virtual int64_t get_used_blocks_adj();
   virtual int64_t size() {
     return m_total_blocks;
   }
@@ -530,7 +531,7 @@ public:
     return m_total_blocks - m_extra_blocks;
   }
   int64_t get_used_blocks() {
-    return BitMapAreaIN::get_used_blocks() - (m_extra_blocks + m_reserved_blocks);
+    return (BitMapAreaIN::get_used_blocks_adj() - m_extra_blocks);
   }
 
   BitAllocatorStats *get_stats() {