Add capacity to semaphore so that initial size can be queried later.

[folly.git] / folly / fibers / Semaphore.h

diff --git a/folly/fibers/Semaphore.h b/folly/fibers/Semaphore.h
index 24e5707c67ea8222fe6cd405bb5ce4996c481c99..047abee8d888c076ecb261c43d43d427b0dab6e8 100644 (file)
--- a/folly/fibers/Semaphore.h
+++ b/folly/fibers/Semaphore.h
@@ -27,7 +27,8 @@ namespace fibers {
  */
 class Semaphore {
  public:
-  explicit Semaphore(size_t tokenCount) : tokens_(tokenCount) {}
+  explicit Semaphore(size_t tokenCount)
+      : capacity_(tokenCount), tokens_(capacity_) {}
 
   Semaphore(const Semaphore&) = delete;
   Semaphore(Semaphore&&) = delete;
@@ -44,10 +45,13 @@ class Semaphore {
    */
   void wait();
 
+  size_t getCapacity() const;
+
  private:
   bool waitSlow();
   bool signalSlow();
 
+  size_t capacity_;
   // Atomic counter
   std::atomic<int64_t> tokens_;
   folly::Synchronized<std::queue<folly::fibers::Baton*>> waitList_;