Added missing vm/thread_phase.hpp.

rubinius · Aug 29, 2015 · 74b3f65 · 74b3f65
1 parent f1a3952
commit 74b3f65
Showing 1 changed file with 83 additions and 0 deletions.
diff --git a/vm/thread_phase.hpp b/vm/thread_phase.hpp
@@ -0,0 +1,83 @@
+#ifndef RBX_THREAD_PHASE_HPP
+#define RBX_THREAD_PHASE_HPP
+
+#include "defines.hpp"
+#include "vm.hpp"
+#include "state.hpp"
+#include "thread_nexus.hpp"
+
+namespace rubinius {
+  /**
+   * Instantiation of an instance of this class causes Ruby execution on all
+   * threads to be suspended. Upon destruction of the instance, Ruby execution
+   * is resumed.
+   */
+  class LockPhase {
+    State* state_;
+
+  public:
+    LockPhase(STATE)
+      : state_(state)
+    {
+      state->shared().thread_nexus()->lock_or_wait(state->vm());
+    }
+
+    ~LockPhase() {
+      state_->shared().thread_nexus()->unlock();
+    }
+  };
+
+  class ManagedPhase {
+    State* state_;
+
+  public:
+    ManagedPhase(STATE)
+      : state_(state)
+    {
+      state_->vm()->become_managed();
+    }
+
+    ~ManagedPhase() {
+      state_->vm()->become_unmanaged();
+    }
+
+  };
+
+  class UnmanagedPhase {
+    State* state_;
+
+  public:
+    UnmanagedPhase(STATE)
+      : state_(state)
+    {
+      state_->vm()->become_unmanaged();
+    }
+
+    ~UnmanagedPhase() {
+      state_->vm()->become_managed();
+    }
+  };
+
+  template <class T>
+  class LockUnmanaged : public utilities::thread::LockGuardTemplate<T> {
+    State* state_;
+
+  public:
+    LockUnmanaged(STATE, T& in_lock)
+      : utilities::thread::LockGuardTemplate<T>(in_lock, false)
+      , state_(state)
+    {
+      state_->vm()->become_unmanaged();
+      this->lock();
+      state_->vm()->become_managed();
+    }
+
+    ~LockUnmanaged() {
+      this->unlock();
+    }
+  };
+
+  typedef LockUnmanaged<utilities::thread::Mutex> MutexLockUnmanaged;
+}
+
+#endif