barrier.h 2.24 KB
Newer Older
1 2 3 4 5 6 7 8 9
// Copyright 2017 the V8 project authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef V8_HEAP_BARRIER_H_
#define V8_HEAP_BARRIER_H_

#include "src/base/platform/condition-variable.h"
#include "src/base/platform/mutex.h"
10
#include "src/base/platform/time.h"
11 12 13 14 15 16 17

namespace v8 {
namespace internal {

// Barrier that can be used once to synchronize a dynamic number of tasks
// working concurrently.
//
18 19 20 21
// The barrier takes a timeout which is used to avoid waiting for too long. If
// any of the users ever reach the timeout they will disable the barrier and
// signal others to fall through.
//
22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38
// Usage:
//   void RunConcurrently(OneShotBarrier* shared_barrier) {
//     shared_barrier->Start();
//     do {
//       {
//         /* process work and create new work */
//         barrier->NotifyAll();
//         /* process work and create new work */
//       }
//     } while(!shared_barrier->Wait());
//   }
//
// Note: If Start() is not called in time, e.g., because the first concurrent
// task is already done processing all work, then Done() will return true
// immediately.
class OneshotBarrier {
 public:
39
  explicit OneshotBarrier(base::TimeDelta timeout) : timeout_(timeout) {}
40 41

  void Start() {
42
    base::MutexGuard guard(&mutex_);
43 44 45 46
    tasks_++;
  }

  void NotifyAll() {
47
    base::MutexGuard guard(&mutex_);
48 49 50 51
    if (waiting_ > 0) condition_.NotifyAll();
  }

  bool Wait() {
52
    base::MutexGuard guard(&mutex_);
53 54 55 56 57 58 59 60 61
    if (done_) return true;

    DCHECK_LE(waiting_, tasks_);
    waiting_++;
    if (waiting_ == tasks_) {
      done_ = true;
      condition_.NotifyAll();
    } else {
      // Spurious wakeup is ok here.
62 63 64 65 66
      if (!condition_.WaitFor(&mutex_, timeout_)) {
        // If predefined timeout was reached, Stop waiting and signal being done
        // also to other tasks.
        done_ = true;
      }
67 68 69 70 71 72 73 74 75 76 77
    }
    waiting_--;
    return done_;
  }

  // Only valid to be called in a sequential setting.
  bool DoneForTesting() const { return done_; }

 private:
  base::ConditionVariable condition_;
  base::Mutex mutex_;
78 79 80 81
  base::TimeDelta timeout_;
  int tasks_ = 0;
  int waiting_ = 0;
  bool done_ = false;
82 83 84 85 86 87
};

}  // namespace internal
}  // namespace v8

#endif  // V8_HEAP_BARRIER_H_